]> SALOME platform Git repositories - modules/hydro.git/blob - src/HYDROGUI/HYDROGUI_LandCoverMapOp.cxx
Salome HOME
refs #669, #671: implementation of change land cover type operation, set default...
[modules/hydro.git] / src / HYDROGUI / HYDROGUI_LandCoverMapOp.cxx
1 // Copyright (C) 2014-2015  EDF-R&D
2 // This library is free software; you can redistribute it and/or
3 // modify it under the terms of the GNU Lesser General Public
4 // License as published by the Free Software Foundation; either
5 // version 2.1 of the License, or (at your option) any later version.
6 //
7 // This library is distributed in the hope that it will be useful,
8 // but WITHOUT ANY WARRANTY; without even the implied warranty of
9 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
10 // Lesser General Public License for more details.
11 //
12 // You should have received a copy of the GNU Lesser General Public
13 // License along with this library; if not, write to the Free Software
14 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
15 //
16 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
17 //
18
19 #include "HYDROGUI_LandCoverMapOp.h"
20
21 #include "HYDROGUI_Module.h"
22 #include "HYDROGUI_OCCDisplayer.h"
23 #include "HYDROGUI_Operations.h"
24 #include "HYDROGUI_LandCoverMapDlg.h"
25 #include "HYDROGUI_Tool.h"
26 #include "HYDROGUI_UpdateFlags.h"
27 #include "HYDROGUI_DataObject.h"
28 #include "HYDROGUI_ShapeLandCoverMap.h"
29
30 #include <HYDROData_Iterator.h>
31 #include <HYDROData_StricklerTable.h>
32 #include <HYDROData_PolylineXY.h>
33 #include <HYDROData_Object.h>
34
35 #include <OCCViewer_ViewManager.h>
36 #include <OCCViewer_ViewModel.h>
37
38 #include <LightApp_Application.h>
39
40 #include <TopoDS.hxx>
41 #include <TopoDS_Face.hxx>
42 #include <TopTools_ListOfShape.hxx>
43 #include <TopTools_ListIteratorOfListOfShape.hxx>
44 #include <AIS_Shape.hxx>
45
46 #include <QApplication>
47
48 HYDROGUI_LandCoverMapOp::HYDROGUI_LandCoverMapOp( HYDROGUI_Module* theModule, const int theOperationId )
49 : HYDROGUI_Operation( theModule ),
50   myOperationId( theOperationId ),
51   myPreviewPrs( 0 )
52 {
53   switch( myOperationId )
54   {
55     case CreateLandCoverMapId:
56       setName( tr( "CREATE_LAND_COVER_MAP" ) );
57       break;
58     case AddLandCoverId:
59       setName( tr( "ADD_LAND_COVER" ) );
60       break;
61     case RemoveLandCoverId:
62       setName( tr( "REMOVE_LAND_COVER" ) );
63       break;
64     case SplitLandCoverId:
65       setName( tr( "SPLIT_LAND_COVER" ) );
66       break;
67     case MergeLandCoverId:
68       setName( tr( "MERGE_LAND_COVER" ) );
69       break;
70     case ChangeLandCoverTypeId:
71       setName( tr( "CHANGE_LAND_COVER_TYPE" ) );
72       break;
73   }  
74 }
75
76 HYDROGUI_LandCoverMapOp::~HYDROGUI_LandCoverMapOp()
77 {
78   closePreview();
79 }
80
81 void HYDROGUI_LandCoverMapOp::startOperation()
82 {
83   HYDROGUI_Operation::startOperation();
84
85   HYDROGUI_LandCoverMapDlg* aPanel = ::qobject_cast<HYDROGUI_LandCoverMapDlg*>( inputPanel() );
86   if ( !aPanel )
87     return;
88
89   aPanel->blockSignals( true );
90
91   aPanel->reset();
92
93   // Set name of the created/edited land cover map object
94   QString anObjectName = HYDROGUI_Tool::GenerateObjectName( module(), tr( "DEFAULT_LAND_COVER_MAP_NAME" ) );
95   if ( myOperationId != CreateLandCoverMapId )
96   {
97     if ( isApplyAndClose() )
98       myEditedObject = Handle(HYDROData_LandCoverMap)::DownCast( HYDROGUI_Tool::GetSelectedObject( module() ) );
99     if ( !myEditedObject.IsNull() )
100     {
101       anObjectName = myEditedObject->GetName();
102
103       // Construct a list of names of all land cover map objects defined within the data model
104       QStringList aLandCoverMapNames;
105       HYDROData_Iterator anIterator( doc(), KIND_LAND_COVER_MAP );
106       for( ; anIterator.More(); anIterator.Next() )
107       {
108         Handle(HYDROData_LandCoverMap) aLandCoverObj =
109           Handle(HYDROData_LandCoverMap)::DownCast( anIterator.Current() );     
110         if ( !aLandCoverObj.IsNull() )
111           aLandCoverMapNames.append( aLandCoverObj->GetName() );
112       }
113
114       //aLandCoverMapNames.sort();
115       aPanel->setObjectNames( aLandCoverMapNames );
116     }
117   }  
118   aPanel->setObjectName( anObjectName );
119
120   closePreview();
121   QStringList aPolylineFaceNames;
122   onCreatePreview( aPolylineFaceNames );
123
124   aPanel->blockSignals( false );
125
126   module()->update( UF_OCCViewer | UF_FitAll );
127 }
128
129 void HYDROGUI_LandCoverMapOp::abortOperation()
130 {
131   closePreview();
132
133   HYDROGUI_Operation::abortOperation();
134
135   module()->update( UF_OCCViewer | UF_FitAll );
136 }
137
138 void HYDROGUI_LandCoverMapOp::commitOperation()
139 {
140   closePreview();
141
142   HYDROGUI_Operation::commitOperation();
143
144   module()->update( UF_OCCViewer | UF_FitAll );
145 }
146
147 HYDROGUI_InputPanel* HYDROGUI_LandCoverMapOp::createInputPanel() const
148 {
149   HYDROGUI_LandCoverMapDlg* aPanel = new HYDROGUI_LandCoverMapDlg( module(), getName(), myOperationId );
150   connect( aPanel, SIGNAL( landCoverMapChanged( const QString& ) ),
151            this, SLOT( onLandCoverMapChanged( const QString& ) ) );
152   connect( aPanel, SIGNAL( CreatePreview( const QStringList& ) ),
153            this,   SLOT( onCreatePreview( const QStringList& ) ) );
154   return aPanel;
155 }
156
157 bool HYDROGUI_LandCoverMapOp::processApply( int& theUpdateFlags,
158                                             QString& theErrorMsg,
159                                             QStringList& theBrowseObjectsEntries )
160 {
161   HYDROGUI_LandCoverMapDlg* aPanel = ::qobject_cast<HYDROGUI_LandCoverMapDlg*>( inputPanel() );
162   if ( !aPanel )
163     return false;
164
165   // Check name of the created/edited object
166   QString anObjectName = aPanel->getObjectName().simplified();
167   if ( anObjectName.isEmpty() )
168   {
169     theErrorMsg = tr( "INCORRECT_OBJECT_NAME" );
170     return false;
171   }
172
173   if ( myOperationId == CreateLandCoverMapId || ( !myEditedObject.IsNull() && myEditedObject->GetName() != anObjectName ) )
174   {
175     // check that there are no other objects with the same name in the document
176     Handle(HYDROData_Entity) anObject = HYDROGUI_Tool::FindObjectByName( module(), anObjectName );
177     if( !anObject.IsNull() )
178     {
179       theErrorMsg = tr( "OBJECT_EXISTS_IN_DOCUMENT" ).arg( anObjectName );
180       return false;
181     }
182   }
183
184   Handle(HYDROData_PolylineXY) aPolyline;
185   Handle(HYDROData_Object) aFace;
186
187   TopTools_ListOfShape aFacesSelectedInViewer;
188
189   // Get polyline/face selected in combo-box
190   if ( myOperationId == CreateLandCoverMapId || myOperationId == AddLandCoverId || myOperationId == SplitLandCoverId )
191   {
192     Handle(HYDROData_Entity) aPolylineFace = aPanel->getPolylineFace();
193     aPolyline = Handle(HYDROData_PolylineXY)::DownCast( aPolylineFace );
194     aFace = Handle(HYDROData_Object)::DownCast( aPolylineFace );
195     if ( aPolyline.IsNull() && aFace.IsNull() )
196     {
197       theErrorMsg = tr( "POLYLINE_FACE_NOT_DEFINED" );
198       return false;
199     }
200   }
201   // Get face(s) selected in the 3d viewer
202   else if ( myOperationId == RemoveLandCoverId || 
203             myOperationId == MergeLandCoverId || 
204             myOperationId == ChangeLandCoverTypeId )
205   {
206     if ( myPreviewPrs )
207     {
208       // Fill in aFacesSelectedInViewer list
209       Handle(AIS_InteractiveContext) aCtx;
210       getSelectedShapes( aFacesSelectedInViewer, aCtx );
211     }
212   }
213
214   // Get selected Strickler type
215   QString aSelectedStricklerType;
216   if ( myOperationId == CreateLandCoverMapId || 
217        myOperationId == AddLandCoverId || 
218        myOperationId == MergeLandCoverId ||
219        myOperationId == ChangeLandCoverTypeId )
220   {
221     aSelectedStricklerType = aPanel->getSelectedStricklerTypeName();
222     if ( aSelectedStricklerType.isEmpty() )
223     {
224       theErrorMsg = tr( "STRICKLER_TYPE_NOT_DEFINED" );
225       return false;
226     }
227   }
228
229   // Create / find the new / edited land cover map object
230   Handle(HYDROData_LandCoverMap) aLandCoverMapObj = myOperationId != CreateLandCoverMapId ? myEditedObject :
231     Handle(HYDROData_LandCoverMap)::DownCast( doc()->CreateObject( KIND_LAND_COVER_MAP ) );
232
233   // Set land cover map name
234   aLandCoverMapObj->SetName( anObjectName );
235   
236   // Add land cover to new / edited land cover map
237   if ( myOperationId == CreateLandCoverMapId || myOperationId == AddLandCoverId )
238   {
239     bool aLandCoverAdded = false;
240     if ( !aPolyline.IsNull() )
241       aLandCoverAdded = aLandCoverMapObj->Add( aPolyline, aSelectedStricklerType );
242     else if ( !aFace.IsNull() )
243       aLandCoverAdded = aLandCoverMapObj->Add( aFace, aSelectedStricklerType );
244     if ( !aLandCoverAdded )
245     {
246       theErrorMsg = tr( "LAND_COVER_NOT_ADDED" );
247       return false;
248     }
249   }
250
251   // Remove land cover from edited land cover map
252   if ( myOperationId == RemoveLandCoverId )
253   {
254     bool aLandCoverRemoved = false;
255     if ( !aFacesSelectedInViewer.IsEmpty() )
256     {
257       aLandCoverRemoved = aLandCoverMapObj->Remove( aFacesSelectedInViewer );
258       if ( !aLandCoverRemoved )
259       {
260         theErrorMsg = tr( "LAND_COVER_NOT_REMOVED" );
261         return false;
262       }
263     }
264   }
265
266   // Split land cover(s) inside edited land cover map
267   if ( myOperationId == SplitLandCoverId )
268   {
269     bool aLandCoverSplitted = false;
270     if ( !aPolyline.IsNull() )
271       aLandCoverSplitted = aLandCoverMapObj->Split( aPolyline );
272     else if ( !aFace.IsNull() )
273     {
274       // Get the complete boundary of the object face as the splitting polyline
275       QList<TopoDS_Shape> aBoundShapes;
276       QStringList aBoundNames;
277       aFace->GetBoundaries( aBoundShapes, aBoundNames );
278
279       for( int i=0, n=aBoundShapes.size(); i<n; i++ )
280       {
281         TopoDS_Shape aShape = aBoundShapes[i];
282         if( aShape.IsNull() )
283           continue;
284
285         bool aSplitResult = aLandCoverMapObj->Split( aShape );
286         aLandCoverSplitted = ( i==0 ? aSplitResult : aLandCoverSplitted && aSplitResult );
287       }
288     }
289     if ( !aLandCoverSplitted )
290     {
291       theErrorMsg = tr( "LAND_COVER_NOT_SPLITTED" );
292       return false;
293     }
294   }
295
296   // Merge land covers inside edited land cover map
297   if ( myOperationId == MergeLandCoverId )
298   {
299     bool aLandCoverMerged = false;
300     if ( !aFacesSelectedInViewer.IsEmpty() )
301     {
302       aLandCoverMerged = aLandCoverMapObj->Merge( aFacesSelectedInViewer, aSelectedStricklerType );
303       if ( !aLandCoverMerged )
304       {
305         theErrorMsg = tr( "LAND_COVER_NOT_MERGED" );
306         return false;
307       }
308     }
309   }
310
311   // Change Strickler type for land cover(s) inside edited land cover map
312   if ( myOperationId == ChangeLandCoverTypeId )
313   {
314     bool aLandCoverChangeType = false;
315     if ( !aFacesSelectedInViewer.IsEmpty() )
316     {
317       aLandCoverChangeType = aLandCoverMapObj->ChangeType( aFacesSelectedInViewer, aSelectedStricklerType );
318       if ( !aLandCoverChangeType )
319       {
320         theErrorMsg = tr( "LAND_COVER_TYPE_NOT_CHANGED" );
321         return false;
322       }
323     }
324   }
325     
326   // Update land cover map object and close preview
327   aLandCoverMapObj->Update();
328
329   closePreview();
330
331   // Publish the newly created land cover map in the Object Browser
332   module()->setObjectVisible( HYDROGUI_Tool::GetActiveOCCViewId( module() ), aLandCoverMapObj, true );
333   if ( myOperationId == CreateLandCoverMapId )
334   {
335     QString anEntry = HYDROGUI_DataObject::dataObjectEntry( aLandCoverMapObj );
336     theBrowseObjectsEntries.append( anEntry );
337   }
338
339   // Update presentation of land cover object in the 3d viewer
340   module()->setIsToUpdate( aLandCoverMapObj );
341   module()->getOCCDisplayer()->SetToUpdateColorScale();
342
343   theUpdateFlags = UF_Model | UF_OCCViewer | UF_OCC_Forced | UF_VTKViewer;
344
345   return true;
346 }
347
348 void HYDROGUI_LandCoverMapOp::onLandCoverMapChanged( const QString& theName )
349 {
350   // If the edited land cover map was changed in the combo-box, update myEditedObject
351   if ( myOperationId != CreateLandCoverMapId )
352   {
353     myEditedObject = Handle(HYDROData_LandCoverMap)::DownCast( HYDROGUI_Tool::FindObjectByName( module(), theName ) );
354     if ( !myEditedObject.IsNull() )
355     {
356       // Show preview of the newly selected land cover map
357       closePreview();
358       QStringList aPolylineFaceNames;
359       onCreatePreview( aPolylineFaceNames );
360     }
361   }
362 }
363
364 void HYDROGUI_LandCoverMapOp::onCreatePreview( const QStringList& thePolylineFaceNames )
365 {
366   HYDROGUI_LandCoverMapDlg* aPanel = ::qobject_cast<HYDROGUI_LandCoverMapDlg*>( inputPanel() );
367   if ( !aPanel )
368     return;
369
370   QApplication::setOverrideCursor( Qt::WaitCursor );  
371
372   LightApp_Application* anApp = module()->getApp();
373   if ( !getPreviewManager() )
374     setPreviewManager( ::qobject_cast<OCCViewer_ViewManager*>( 
375                        anApp->getViewManager( OCCViewer_Viewer::Type(), true ) ) );
376   OCCViewer_ViewManager* aViewManager = getPreviewManager();
377   if ( aViewManager && !myPreviewPrs )
378   {
379     if ( OCCViewer_Viewer* aViewer = aViewManager->getOCCViewer() )
380     {
381       Handle(AIS_InteractiveContext) aCtx = aViewer->getAISContext();
382       if ( !aCtx.IsNull() )
383       {
384         connect( aViewer, SIGNAL( selectionChanged() ), this, SLOT( onViewerSelectionChanged() ) );
385         myPreviewPrs = new HYDROGUI_ShapeLandCoverMap( module()->getOCCDisplayer(), aCtx, myEditedObject, getPreviewZLayer()/*, theIsScalarMapMode*/ );
386       }
387     }
388   }
389
390   if ( aViewManager && myPreviewPrs )
391   {
392     TopoDS_Shape aLandCoverMapShape = myEditedObject->GetShape();
393     if( !aLandCoverMapShape.IsNull() )
394     {
395       if ( myOperationId == RemoveLandCoverId ||
396            myOperationId == MergeLandCoverId ||
397            myOperationId == ChangeLandCoverTypeId )
398         myPreviewPrs->setSelectionMode( AIS_Shape::SelectionMode( TopAbs_FACE ) ); 
399       myPreviewPrs->setShape( aLandCoverMapShape );      
400     }
401   }
402   
403   module()->update( UF_OCCViewer | UF_FitAll );
404
405   QApplication::restoreOverrideCursor();  
406 }
407
408 void HYDROGUI_LandCoverMapOp::onViewerSelectionChanged()
409 {
410   HYDROGUI_LandCoverMapDlg* aPanel = ::qobject_cast<HYDROGUI_LandCoverMapDlg*>( inputPanel() );
411   if ( !aPanel )
412     return;
413
414   Handle(AIS_InteractiveContext) aCtx = getInteractiveContext();
415   if ( !aCtx.IsNull() )
416   {
417     int aNbSelected = aCtx->NbSelected();
418
419     if ( myOperationId == RemoveLandCoverId || myOperationId == ChangeLandCoverTypeId )
420       // Enable Apply, Apply and Close buttons only if at least one face (land cover) is selected in the 3d viewer
421       aPanel->setApplyEnabled( aNbSelected > 0 );
422     else if ( myOperationId == MergeLandCoverId )
423       // Enable Apply, Apply and Close buttons only if at least two faces (land covers) are selected in the 3d viewer
424       aPanel->setApplyEnabled( aNbSelected > 1 );
425
426     if ( myOperationId == MergeLandCoverId || myOperationId == ChangeLandCoverTypeId )
427     {
428       if ( aNbSelected == 1 && !myEditedObject.IsNull() )
429       {
430         TopTools_ListOfShape aFacesSelectedInViewer;
431         getSelectedShapes( aFacesSelectedInViewer, aCtx );
432         if ( aFacesSelectedInViewer.Extent() == 1 )
433         {
434           QString aType = myEditedObject->StricklerType( TopoDS::Face( aFacesSelectedInViewer.First() ) );
435           if ( !aType.isEmpty() )
436             aPanel->setSelectedStricklerTypeName( aType );
437         }
438       }
439     }
440   }
441 }
442
443 void HYDROGUI_LandCoverMapOp::closePreview()
444 {
445   if( myPreviewPrs )
446   {
447     delete myPreviewPrs;
448     myPreviewPrs = 0;
449   }
450
451   HYDROGUI_LandCoverMapDlg* aPanel = ::qobject_cast<HYDROGUI_LandCoverMapDlg*>( inputPanel() );
452   if ( !aPanel )
453     return;
454
455   if ( myOperationId == RemoveLandCoverId || myOperationId == MergeLandCoverId || myOperationId == ChangeLandCoverTypeId )
456     aPanel->setApplyEnabled( false );
457 }
458
459 Handle(AIS_InteractiveContext) HYDROGUI_LandCoverMapOp::getInteractiveContext()
460 {
461   OCCViewer_ViewManager* aViewManager = getPreviewManager();
462   Handle(AIS_InteractiveContext) aCtx = NULL;
463   if ( aViewManager ) {
464     if ( OCCViewer_Viewer* aViewer = aViewManager->getOCCViewer() ) {
465       aCtx = aViewer->getAISContext();
466     }
467   }
468   return aCtx;
469 }
470
471 void HYDROGUI_LandCoverMapOp::getSelectedShapes( TopTools_ListOfShape& theSelectedShapes,
472                                                  Handle(AIS_InteractiveContext)& theCtx )
473 {
474   if ( theCtx.IsNull() )
475     theCtx = getInteractiveContext();
476
477   if ( !theCtx.IsNull() && theCtx->NbSelected() > 0 )
478   {
479     for ( theCtx->InitSelected(); theCtx->MoreSelected(); theCtx->NextSelected() )
480     {
481       TopoDS_Shape aSelectedShape = theCtx->SelectedShape();
482       if ( aSelectedShape.IsNull() )
483         continue;
484
485       theSelectedShapes.Append( aSelectedShape );
486     }
487   }
488 }