Salome HOME
Merge remote-tracking branch 'remotes/origin/BR_2017' into HEAD
[modules/hydro.git] / src / HYDROGUI / HYDROGUI_Operations.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_Operations.h"
20
21 #include "HYDROGUI_CopyPasteOp.h"
22 #include "HYDROGUI_CalculationOp.h"
23 #include "HYDROGUI_ChannelOp.h"
24 #include "HYDROGUI_DataModel.h"
25 #include "HYDROGUI_CopyPastePositionOp.h"
26 #include "HYDROGUI_DeleteOp.h"
27 #include "HYDROGUI_DigueOp.h"
28 #include "HYDROGUI_ExportImageOp.h"
29 #include "HYDROGUI_ExportFileOp.h"
30 #include "HYDROGUI_ImportImageOp.h"
31 #include "HYDROGUI_ImportPolylineOp.h"
32 #include "HYDROGUI_ImportSinusXOp.h"
33 #include "HYDROGUI_ImportBathymetryOp.h"
34 #include "HYDROGUI_ImmersibleZoneOp.h"
35 #include "HYDROGUI_Module.h"
36 #include "HYDROGUI_ObserveImageOp.h"
37 #include "HYDROGUI_PolylineOp.h"
38 #include "HYDROGUI_Poly3DOp.h"
39 #include "HYDROGUI_ProfileOp.h"
40 #include "HYDROGUI_RemoveImageRefsOp.h"
41 #include "HYDROGUI_ShowHideOp.h"
42 #include "HYDROGUI_StreamOp.h"
43 #include "HYDROGUI_TwoImagesOp.h"
44 #include "HYDROGUI_UpdateFlags.h"
45 #include "HYDROGUI_UpdateObjectOp.h"
46 #include "HYDROGUI_VisualStateOp.h"
47 #include "HYDROGUI_ImmersibleZoneOp.h"
48 #include "HYDROGUI_ImportGeomObjectOp.h"
49 #include "HYDROGUI_ImportObstacleFromFileOp.h"
50 #include "HYDROGUI_TranslateObstacleOp.h"
51 #include "HYDROGUI_ExportCalculationOp.h"
52 #include "HYDROGUI_ImportProfilesOp.h"
53 #include "HYDROGUI_GeoreferencementOp.h"
54 #include "HYDROGUI_SetColorOp.h"
55 #include "HYDROGUI_BathymetryBoundsOp.h"
56 #include "HYDROGUI_Tool2.h"
57 #include "HYDROGUI_ZLevelsOp.h"
58 #include "HYDROGUI_LocalCSOp.h"
59 #include "HYDROGUI_RiverBottomOp.h"
60 #include "HYDROGUI_ProfileInterpolateOp.h"
61 #include "HYDROGUI_RecognizeContoursOp.h"
62 #include "HYDROGUI_SubmersibleOp.h"
63 #include "HYDROGUI_StricklerTableOp.h"
64 #include "HYDROGUI_DuplicateOp.h"
65 #include "HYDROGUI_LandCoverMapOp.h"
66 #include "HYDROGUI_PolylineExtractionOp.h"
67 #include "HYDROGUI_ExportSinusXOp.h"
68 #include "HYDROGUI_MergePolylinesOp.h"
69 #include "HYDROGUI_SplitPolylinesOp.h"
70 #include "HYDROGUI_LandCoverColoringOp.h"
71 #include "HYDROGUI_SetTransparencyOp.h"
72 #include "HYDROGUI_ImportLandCoverMapOp.h"
73 #include "HYDROGUI_BathymetrySelectionOp.h"
74 #include "HYDROGUI_BathymetryOp.h"
75 #include "HYDROGUI_RegenerateRegionColorsOp.h"
76 #include "HYDROGUI_PolylineStyleOp.h"
77 #include "HYDROGUI_ZoneSetColorOp.h"
78 #include <HYDROData_Document.h>
79 #include <HYDROData_Obstacle.h>
80 #include <HYDROData_SplitToZonesTool.h>
81 #include <HYDROData_Iterator.h>
82
83 #include <GeometryGUI.h>
84 #include <GeometryGUI_Operations.h>
85 #include <GEOMBase.h>
86
87 #include <SalomeApp_Study.h>
88
89 #include <LightApp_Application.h>
90
91 #include <OCCViewer_ViewModel.h>
92 #include <OCCViewer_ViewManager.h>
93 #include <OCCViewer_ViewWindow.h>
94 #include <OCCViewer_ViewFrame.h>
95
96 #include <QtxListAction.h>
97 #include <QtxActionToolMgr.h>
98
99 #include <SUIT_Desktop.h>
100 #include <SUIT_ResourceMgr.h>
101 #include <SUIT_Session.h>
102 #include <SUIT_MessageBox.h>
103
104 #include <QAction>
105 #include <QApplication>
106
107 QAction* HYDROGUI_Module::createAction( const int theId, const QString& theSuffix, const QString& theImg,
108                                         const int theKey, const bool isToggle, const QString& theSlot )
109 {
110   QString aSlot = theSlot;
111   if( aSlot.isEmpty() )
112     aSlot = SLOT( onOperation() );
113   SUIT_ResourceMgr* aResMgr = SUIT_Session::session()->resourceMgr();
114   std::string anImg = theImg.toStdString();
115   QPixmap aPixmap = theImg.isEmpty() ? QPixmap() : aResMgr->loadPixmap( "HYDRO", tr( anImg.c_str() ) );
116   std::string aMenu    = ( "MEN_" + theSuffix ).toStdString();
117   std::string aDesktop = ( "DSK_" + theSuffix ).toStdString();
118   std::string aToolbar = ( "STB_" + theSuffix ).toStdString();
119   std::string aSlotStr = aSlot.toStdString();
120   return LightApp_Module::createAction( theId, tr( aDesktop.c_str() ), aPixmap,
121     tr( aMenu.c_str() ), tr( aToolbar.c_str() ),
122                 theKey, application()->desktop(), isToggle, this, aSlotStr.c_str() );
123 }
124
125 void HYDROGUI_Module::createActions()
126 {
127   createAction( SaveVisualStateId, "SAVE_VISUAL_STATE" );
128   createAction( LoadVisualStateId, "LOAD_VISUAL_STATE" );
129
130   createAction( CopyId, "COPY", "", Qt::CTRL + Qt::Key_C );
131   createAction( PasteId, "PASTE", "", Qt::CTRL + Qt::Key_V );
132
133   createAction( UpdateObjectId, "UPDATE_OBJECT" );
134   createAction( ForcedUpdateObjectId, "FORCED_UPDATE_OBJECT" );
135
136   createAction( ImportImageId, "IMPORT_IMAGE", "IMPORT_IMAGE_ICO", Qt::CTRL + Qt::Key_I );
137   createAction( EditImportedImageId, "EDIT_IMPORTED_IMAGE", "EDIT_IMPORTED_IMAGE_ICO" );
138   createAction( ObserveImageId, "OBSERVE_IMAGE", "OBSERVE_IMAGE_ICO" );
139   createAction( ExportImageId, "EXPORT_IMAGE", "EXPORT_IMAGE_ICO" );
140   createAction( RemoveImageRefsId, "REMOVE_IMAGE_REFERENCE", "REMOVE_IMAGE_REFERENCE_ICO" );
141   
142   createAction( ImportPolylineId, "IMPORT_POLYLINE", "IMPORT_POLYLINE_ICO" );
143   createAction( ImportSinusXId, "IMPORT_SINUSX", "IMPORT_SINUSX_ICO" );
144   createAction( ExportSinusXId, "EXPORT_SINUSX", "EXPORT_SINUSX_ICO" );
145   createAction( ImportLandCoverMapId, "IMPORT_LANDCOVER_MAP", "IMPORT_LANDCOVER_MAP_ICO" );
146
147   createAction( CreatePolylineId, "CREATE_POLYLINE", "CREATE_POLYLINE_ICO" );
148   createAction( EditPolylineId, "EDIT_POLYLINE", "EDIT_POLYLINE_ICO" ); 
149
150   createAction( CreatePolyline3DId, "CREATE_POLYLINE_3D", "CREATE_POLYLINE_3D_ICO" );
151   createAction( EditPolyline3DId, "EDIT_POLYLINE_3D", "EDIT_POLYLINE_3D_ICO" ); 
152
153   createAction( CreateProfileId, "CREATE_PROFILE", "CREATE_PROFILE_ICO" );
154   createAction( ImportProfilesId, "IMPORT_PROFILES", "IMPORT_PROFILES_ICO" );
155   createAction( EditProfileId, "EDIT_PROFILE", "EDIT_PROFILE_ICO" ); 
156   createAction( AllGeoreferencementId, "GEOREFERENCEMENT", "GEOREFERENCEMENT_ICO" ); 
157   createAction( SelectedGeoreferencementId, "GEOREFERENCEMENT", "GEOREFERENCEMENT_ICO" ); 
158   
159   createAction( ImportBathymetryId, "IMPORT_BATHYMETRY", "IMPORT_BATHYMETRY_ICO", Qt::CTRL + Qt::Key_B );
160   createAction( EditImportedBathymetryId, "EDIT_IMPORTED_BATHYMETRY", "EDIT_IMPORTED_BATHYMETRY_ICO" );
161   createAction( BathymetryBoundsId, "BATHYMETRY_BOUNDS", "BATHYMETRY_BOUNDS_ICO" );
162   createAction( BathymetrySelectionId, "BATHYMETRY_SELECTION", "BATHYMETRY_SELECTION_ICO",
163     0, true, SLOT( onBathymetrySelection() ) );
164
165   createAction( BathymetryTextId, "BATHYMETRY_TEXT", "BATHYMETRY_TEXT_ICO",
166     0, true, SLOT( onBathymetryText() ) );
167
168   createAction( BathymetryRescaleSelectionId, "BATHYMETRY_RESCALE_SELECTION", "BATHYMETRY_RESCALE_SELECTION_ICO" );
169   createAction( BathymetryRescaleVisibleId, "BATHYMETRY_RESCALE_VISIBLE", "BATHYMETRY_RESCALE_VISIBLE_ICO" );
170   createAction( BathymetryRescaleUserId, "BATHYMETRY_RESCALE_USER", "BATHYMETRY_RESCALE_USER_ICO" );
171   createAction( BathymetryRescaleDefaultId, "BATHYMETRY_RESCALE_DEFAULT", "BATHYMETRY_RESCALE_DEFAULT_ICO" );
172
173   createAction( CreateImmersibleZoneId, "CREATE_IMMERSIBLE_ZONE", "CREATE_IMMERSIBLE_ZONE_ICO" );
174   createAction( EditImmersibleZoneId, "EDIT_IMMERSIBLE_ZONE", "EDIT_IMMERSIBLE_ZONE_ICO" );
175
176   createAction( CreateStreamId, "CREATE_STREAM", "CREATE_STREAM_ICO" );
177   createAction( EditStreamId, "EDIT_STREAM", "EDIT_STREAM_ICO" );
178
179   createAction( CreateChannelId, "CREATE_CHANNEL", "CREATE_CHANNEL_ICO" );
180   createAction( EditChannelId, "EDIT_CHANNEL", "EDIT_CHANNEL_ICO" );
181
182   createAction( CreateDigueId, "CREATE_DIGUE", "CREATE_DIGUE_ICO" );
183   createAction( EditDigueId, "EDIT_DIGUE", "EDIT_DIGUE_ICO" );
184
185   createAction( ImportStricklerTableFromFileId, "IMPORT_STRICKLER_TABLE", "IMPORT_STRICKLER_TABLE_ICO" );
186   createAction( ExportStricklerTableFromFileId, "EXPORT_STRICKLER_TABLE", "EXPORT_STRICKLER_TABLE_ICO" );
187   createAction( EditStricklerTableId, "EDIT_STRICKLER_TABLE", "EDIT_STRICKLER_TABLE_ICO" );
188   createAction( DuplicateStricklerTableId, "DUPLICATE_STRICKLER_TABLE", "DUPLICATE_STRICKLER_TABLE_ICO" );
189
190   createAction( CreateLandCoverMapId, "CREATE_LAND_COVER_MAP", "CREATE_LAND_COVER_MAP_ICO" );
191   
192   createAction( AddLandCoverId, "ADD_LAND_COVER", "ADD_LAND_COVER_ICO" );
193   createAction( RemoveLandCoverId, "REMOVE_LAND_COVER", "REMOVE_LAND_COVER_ICO" );
194   createAction( SplitLandCoverId, "SPLIT_LAND_COVER", "SPLIT_LAND_COVER_ICO" );
195   createAction( MergeLandCoverId, "MERGE_LAND_COVER", "MERGE_LAND_COVER_ICO" );
196   createAction( ChangeLandCoverTypeId, "CHANGE_LAND_COVER_TYPE", "CHANGE_LAND_COVER_TYPE_ICO" );
197
198   createAction( ImportObstacleFromFileId, "IMPORT_OBSTACLE_FROM_FILE", "IMPORT_OBSTACLE_FROM_FILE_ICO" );
199   createAction( ImportGeomObjectAsObstacleId, "IMPORT_GEOM_OBJECT_AS_OBSTACLE", "IMPORT_GEOM_OBJECT_ICO" );
200   createAction( ImportGeomObjectAsPolylineId, "IMPORT_GEOM_OBJECT_AS_POLYLINE", "IMPORT_GEOM_OBJECT_ICO" );
201   createAction( CreateBoxId, "CREATE_BOX", "CREATE_BOX_ICO" );
202   createAction( CreateCylinderId, "CREATE_CYLINDER", "CREATE_CYLINDER_ICO" );
203   createAction( TranslateObstacleId, "TRANSLATE_OBSTACLE" );
204
205   createAction( CreateCalculationId, "CREATE_CALCULATION", "CREATE_CALCULATION_ICO" );
206   createAction( EditCalculationId, "EDIT_CALCULATION", "EDIT_CALCULATION_ICO" );
207   createAction( ExportCalculationId, "EXPORT_CALCULATION", "EXPORT_CALCULATION_ICO" );
208
209   createAction( FuseImagesId, "FUSE_IMAGES", "FUSE_IMAGES_ICO" );
210   createAction( EditFusedImageId, "EDIT_FUSED_IMAGE", "EDIT_FUSED_IMAGE_ICO" );
211
212   createAction( CutImagesId, "CUT_IMAGES", "CUT_IMAGES_ICO" );
213   createAction( EditCutImageId, "EDIT_CUT_IMAGE", "EDIT_CUT_IMAGE_ICO" );
214
215   createAction( SplitImageId, "SPLIT_IMAGE", "SPLIT_IMAGE_ICO" );
216   createAction( EditSplitImageId, "EDIT_SPLIT_IMAGE", "EDIT_SPLIT_IMAGE_ICO" );
217
218   createAction( RecognizeContoursId, "RECOGNIZE_CONTOURS", "RECOGNIZE_CONTOURS_ICO" );
219
220   createAction( CopyViewerPositionId, "COPY_VIEWER_POSITION", "" );
221
222   createAction( DeleteId, "DELETE", "", Qt::Key_Delete, false,
223                 SLOT( onDelete() ) );
224
225   createAction( SetColorId, "COLOR" );
226   createAction( SetTransparencyId, "TRANSPARENCY" );
227   createAction( SetZLevelId, "ZLEVEL" );
228   createAction( EditLocalCSId, "EDIT_LOCAL_CS" );
229
230   createAction( ShowId, "SHOW" );
231   createAction( ShowOnlyId, "SHOW_ONLY" );
232   createAction( ShowAllId, "SHOW_ALL" );
233   createAction( HideId, "HIDE" );
234   createAction( HideAllId, "HIDE_ALL" );
235
236   createAction( RiverBottomId, "CREATE_STREAM_BOTTOM", "CREATE_STREAM_BOTTOM_ICO" );
237   createAction( RiverBottomContextId, "CREATE_STREAM_BOTTOM", "CREATE_STREAM_BOTTOM_ICO" );
238   createAction( ProfileInterpolateId, "PROFILE_INTERPOLATE", "PROFILE_INTERPOLATE_ICO" );
239
240   createAction( SubmersibleId, "SUBMERSIBLE", "SUBMERSIBLE_ICO" );
241   createAction( UnSubmersibleId, "UNSUBMERSIBLE", "HYDRO_UNSUBMERSIBLE16_ICO" );
242   createAction( ExportToShapeFileID, "EXPORT_TO_SHAPE_FILE", "EXPORT_TO_SHAPE_FILE_ICO" );
243
244   createAction( PolylineExtractionId, "POLYLINE_EXTRACTION" );
245   createAction( SplitPolylinesId, "SPLIT_POLYLINES", "SPLIT_POLYLINES_ICO" );
246   createAction( MergePolylinesId, "MERGE_POLYLINES", "MERGE_POLYLINES_ICO" );
247
248   createAction( LandCoverScalarMapModeOnId, "LC_SCALARMAP_COLORING_ON" );
249   createAction( LandCoverScalarMapModeOffId, "LC_SCALARMAP_COLORING_OFF" );
250
251   createAction( RegenerateRegionColorsId, "REGENERATE_REGION_COLORS" );
252   createAction( ZoneSetColorId, "ZONE_SET_COLOR" );
253   createAction( ShowHideArrows, "SHOW_HIDE_ARROWS" );
254 }
255
256 void HYDROGUI_Module::createMenus()
257 {
258   int aFileMenu = createMenu( tr( "MEN_DESK_FILE" ), -1, -1, 0 );
259   int aCustomPos = 5; // to insert custom actions after "Save as" and before "Preferences"
260   createMenu( SaveVisualStateId, aFileMenu, aCustomPos );
261   createMenu( separator(), aFileMenu, -1, aCustomPos );
262
263   int anEditMenu = createMenu( tr( "MEN_DESK_EDIT" ), -1, -1, 5 );
264   createMenu( UndoId, anEditMenu );
265   createMenu( RedoId, anEditMenu );
266   createMenu( separator(), anEditMenu );
267   createMenu( CopyId, anEditMenu );
268   createMenu( PasteId, anEditMenu );
269
270   int aHydroMenu = 6; // Edit menu id == 5, View menu id == 10
271   int aHydroId = createMenu( tr( "MEN_DESK_HYDRO" ), -1, -1, aHydroMenu );
272   createMenu( ImportSinusXId, aHydroId, -1, -1 );
273   createMenu( ExportSinusXId, aHydroId, -1, -1 );  
274   
275   createMenu( ImportBathymetryId, aHydroId, -1, -1 );
276   createMenu( ImportStricklerTableFromFileId, aHydroId, -1, -1 );
277   createMenu( CreatePolyline3DId, aHydroId, -1, -1 );
278   createMenu( CreateImmersibleZoneId, aHydroId, -1, -1 );
279   createMenu( CreateChannelId, aHydroId, -1, -1 );
280   createMenu( CreateDigueId, aHydroId, -1, -1 );
281
282   int aNewImageId = createMenu( tr( "MEN_DESK_IMAGE" ), aHydroId, -1 );
283   createMenu( ImportImageId, aNewImageId, -1, -1 );
284   createMenu( separator(), aNewImageId );
285   createMenu( FuseImagesId, aNewImageId, -1, -1 );
286   createMenu( CutImagesId, aNewImageId, -1, -1 );
287   createMenu( SplitImageId, aNewImageId, -1, -1 );
288
289   int aNewPolylineId = createMenu( tr( "MEN_DESK_POLYLINE" ), aHydroId, -1 );
290   createMenu( ImportPolylineId, aNewPolylineId, -1, -1 );
291   createMenu( CreatePolylineId, aNewPolylineId, -1, -1 );
292   createMenu( separator(), aNewPolylineId );
293   createMenu( SplitPolylinesId, aNewPolylineId, -1, -1 );
294   createMenu( MergePolylinesId, aNewPolylineId, -1, -1 );
295
296   int aNewProfileId = createMenu( tr( "MEN_DESK_PROFILE" ), aHydroId, -1 );
297   createMenu( CreateProfileId, aNewProfileId, -1, -1 );
298   createMenu( ImportProfilesId, aNewProfileId, -1, -1 );
299   createMenu( separator(), aNewProfileId );
300   createMenu( AllGeoreferencementId, aNewProfileId, -1, -1 );
301
302   int aStreamMenuId = createMenu( tr( "MEN_DESK_STREAM" ), aHydroId, -1 );
303   createMenu( CreateStreamId, aStreamMenuId, -1, -1 );
304   createMenu( separator(), aStreamMenuId );
305   createMenu( RiverBottomId, aStreamMenuId, -1, -1 );
306   createMenu( ProfileInterpolateId, aStreamMenuId, -1, -1 );
307
308   int anObstacleMenuId = createMenu( tr( "MEN_DESK_OBSTACLE" ), aHydroId, -1 );
309   createMenu( ImportObstacleFromFileId, anObstacleMenuId, -1, -1 );
310   createMenu( separator(), anObstacleMenuId );
311   createMenu( CreateBoxId, anObstacleMenuId, -1, -1 );
312   createMenu( CreateCylinderId, anObstacleMenuId, -1, -1 );
313
314   int aLandCoverMapMenuId = createMenu( tr( "MEN_DESK_LAND_COVER_MAP" ), aHydroId, -1 );
315   createMenu( ImportLandCoverMapId, aLandCoverMapMenuId, -1, -1 );
316   createMenu( CreateLandCoverMapId, aLandCoverMapMenuId, -1, -1 );
317   createMenu( separator(), aLandCoverMapMenuId );
318   createMenu( AddLandCoverId, aLandCoverMapMenuId, -1, -1 );
319   createMenu( RemoveLandCoverId, aLandCoverMapMenuId, -1, -1 );
320   createMenu( separator(), aLandCoverMapMenuId );
321   createMenu( SplitLandCoverId, aLandCoverMapMenuId, -1, -1 );
322   createMenu( MergeLandCoverId, aLandCoverMapMenuId, -1, -1 );
323   createMenu( ChangeLandCoverTypeId, aLandCoverMapMenuId, -1, -1 );
324   
325   createMenu( CreateCalculationId, aHydroId, -1, -1 );
326   createMenu( separator(), aHydroId );
327   createMenu( EditLocalCSId, aHydroId, -1, -1 );
328   createMenu( separator(), aHydroId );
329 }
330
331 void HYDROGUI_Module::createPopups()
332 {
333 }
334
335 void HYDROGUI_Module::createToolbars()
336 {
337   int aToolBar = createTool( tr( "MEN_DESK_HYDRO" ) );
338   createTool( UndoId, aToolBar );
339   createTool( RedoId, aToolBar );
340
341   createTool( separator(), aToolBar );
342   createTool( ImportImageId, aToolBar );
343   createTool( ImportPolylineId, aToolBar );
344   createTool( ImportSinusXId, aToolBar );
345   createTool( ExportSinusXId, aToolBar );
346   createTool( ImportLandCoverMapId, aToolBar );
347
348   createTool( ImportBathymetryId, aToolBar );
349   createTool( CreatePolylineId, aToolBar );
350   createTool( CreatePolyline3DId, aToolBar );
351
352   createTool( separator(), aToolBar );
353   createTool( CreateProfileId, aToolBar );
354   createTool( ImportProfilesId, aToolBar );
355   createTool( AllGeoreferencementId, aToolBar );
356
357   createTool( separator(), aToolBar );
358   createTool( CreateChannelId, aToolBar );
359   createTool( CreateDigueId, aToolBar );
360
361   createTool( separator(), aToolBar );
362   createTool( CreateImmersibleZoneId, aToolBar );
363   createTool( CreateStreamId, aToolBar );
364
365   createTool( separator(), aToolBar );
366   createTool( ImportObstacleFromFileId, aToolBar );
367   createTool( CreateBoxId, aToolBar );
368   createTool( CreateCylinderId, aToolBar );
369
370   createTool( separator(), aToolBar );
371   createTool( ImportStricklerTableFromFileId, aToolBar );
372
373   createTool( separator(), aToolBar );
374   createTool( CreateLandCoverMapId, aToolBar );
375   createTool( AddLandCoverId, aToolBar );
376   createTool( RemoveLandCoverId, aToolBar );
377   createTool( SplitLandCoverId, aToolBar );
378   createTool( MergeLandCoverId, aToolBar );
379   createTool( ChangeLandCoverTypeId, aToolBar );
380   enableLCMActions();
381
382   createTool( separator(), aToolBar );
383   createTool( CreateCalculationId, aToolBar );
384
385   createTool( separator(), aToolBar );
386   createTool( FuseImagesId, aToolBar );
387   createTool( CutImagesId, aToolBar );
388   createTool( SplitImageId, aToolBar );
389
390   createTool( separator(), aToolBar );
391   createTool( BathymetrySelectionId, aToolBar );
392   createTool( BathymetryTextId, aToolBar );
393   createTool( BathymetryRescaleSelectionId, aToolBar );
394   createTool( BathymetryRescaleVisibleId, aToolBar );
395   createTool( BathymetryRescaleUserId, aToolBar );
396   createTool( BathymetryRescaleDefaultId, aToolBar );
397 }
398
399 void HYDROGUI_Module::createUndoRedoActions()
400 {
401   SUIT_ResourceMgr* aResMgr = SUIT_Session::session()->resourceMgr();
402
403   QtxListAction* anEditUndo = new QtxListAction( tr( "MEN_UNDO" ),
404     aResMgr->loadPixmap( "HYDRO", tr( "UNDO_ICO" ) ), tr( "DSK_UNDO" ),
405     Qt::CTRL + Qt::Key_Z, application()->desktop() );
406     
407   QtxListAction* anEditRedo = new QtxListAction( tr( "MEN_REDO" ),
408     aResMgr->loadPixmap( "HYDRO", tr( "REDO_ICO" ) ), tr( "DSK_REDO" ),
409     Qt::CTRL + Qt::Key_Y, application()->desktop() );
410   
411   registerAction( UndoId, anEditUndo );
412   registerAction( RedoId, anEditRedo );
413
414   anEditUndo->setComment( tr( "STB_UNDO" ) );
415   anEditRedo->setComment( tr( "STB_REDO" ) );
416
417   connect( anEditUndo, SIGNAL( triggered( int ) ), this, SLOT( onUndo( int ) ) );
418   connect( anEditRedo, SIGNAL( triggered( int ) ), this, SLOT( onRedo( int ) ) );
419 }
420
421 void HYDROGUI_Module::updateUndoRedoControls()
422 {
423   HYDROGUI_DataModel* aModel = getDataModel();
424
425   QtxListAction* aUndoAction = (QtxListAction*)action( UndoId );
426   QtxListAction* aRedoAction = (QtxListAction*)action( RedoId );
427
428   bool aCanUndo = aModel->canUndo();
429   bool aCanRedo = aModel->canRedo();
430
431   if( aCanUndo )
432     aUndoAction->addNames( aModel->undoNames() );
433   aUndoAction->setEnabled( aCanUndo );
434
435   if( aCanRedo )
436     aRedoAction->addNames( aModel->redoNames() );
437   aRedoAction->setEnabled( aCanRedo );
438 }
439
440 void HYDROGUI_Module::enableLCMActions()
441 {
442   HYDROData_Iterator anIt( HYDROData_Document::Document( getStudyId() ), KIND_LAND_COVER_MAP );
443   bool anEnableTools = anIt.More();
444
445   QtxListAction* anAction;
446   anAction = (QtxListAction*)action( AddLandCoverId );
447   if ( anAction ) anAction->setEnabled( anEnableTools );
448   anAction = (QtxListAction*)action( RemoveLandCoverId );
449   if ( anAction ) anAction->setEnabled( anEnableTools );
450   anAction = (QtxListAction*)action( SplitLandCoverId );
451   if ( anAction ) anAction->setEnabled( anEnableTools );
452   anAction = (QtxListAction*)action( MergeLandCoverId );
453   if ( anAction ) anAction->setEnabled( anEnableTools );
454   anAction = (QtxListAction*)action( ChangeLandCoverTypeId );
455   if ( anAction ) anAction->setEnabled( anEnableTools );
456 }
457
458 void HYDROGUI_Module::resetViewState()
459 {
460   OCCViewer_ViewManager* mgr = dynamic_cast<OCCViewer_ViewManager*>(getApp()->viewManager( OCCViewer_Viewer::Type()));
461   if( mgr )
462   {
463     foreach( SUIT_ViewWindow* wnd, mgr->getViews() )
464     {
465       OCCViewer_ViewFrame* vf = dynamic_cast<OCCViewer_ViewFrame*>( wnd );
466       if( vf )
467       {
468         for( int i=OCCViewer_ViewFrame::MAIN_VIEW; i<=OCCViewer_ViewFrame::TOP_RIGHT; i++ )
469         {
470           OCCViewer_ViewWindow* iwnd = vf->getView(i);
471           if( iwnd )
472             iwnd->resetState();
473         }
474       }
475       else
476       {
477         OCCViewer_ViewWindow* ownd = dynamic_cast<OCCViewer_ViewWindow*>( wnd );
478         if( ownd )
479           ownd->resetState();
480       }
481     }
482   }
483 }
484
485 void HYDROGUI_Module::onOperation()
486 {
487   const QAction* anAction = dynamic_cast<const QAction*>( sender() );
488   int anId = actionId( anAction );
489   if( anId >= 0 )
490   {
491     resetViewState();
492     startOperation( anId );
493   }
494
495   if( anId==ShowHideArrows )
496   {
497     setArrowsVisible( !arrowsVisible() );
498   }
499 }
500
501 bool HYDROGUI_Module::arrowsVisible() const
502 {
503   int aType;
504   SUIT_ResourceMgr* resMgr = application()->resourceMgr();
505   if( resMgr )
506   {
507     resMgr->value( "polyline", "arrow_type", aType );
508     return aType>0;
509   }
510   else
511     return false;
512 }
513
514 void HYDROGUI_Module::setArrowsVisible( bool isVisible )
515 {
516   SUIT_ResourceMgr* resMgr = application()->resourceMgr();
517
518   int aType=-1;
519   if( !resMgr )
520     return;
521
522   if( isVisible )
523   {
524     if( resMgr->hasValue( "polyline", "prev_arrow_type" ) )
525       resMgr->value( "polyline", "prev_arrow_type", aType );
526
527     if( aType<=0 )
528       aType = 1;
529     resMgr->setValue( "polyline", "arrow_type", aType );
530   }
531   else
532   {
533     resMgr->value( "polyline", "arrow_type", aType );
534     resMgr->setValue( "polyline", "prev_arrow_type", aType );
535     resMgr->setValue( "polyline", "arrow_type", 0 );
536   }
537   preferencesChanged( "polyline", "arrow_type" );
538 }
539
540 void HYDROGUI_Module::onDelete()
541 {
542   SUIT_Operation* anOp = application()->activeStudy()->activeOperation();
543   HYDROGUI_PolylineOp* aPolylineOp = dynamic_cast<HYDROGUI_PolylineOp*>( anOp );
544   if ( aPolylineOp && aPolylineOp->deleteEnabled() )
545     aPolylineOp->deleteSelected();
546   else
547     startOperation( DeleteId );
548 }
549
550 bool HYDROGUI_Module::onUndo( int theNumActions )
551 {
552   QApplication::setOverrideCursor( Qt::WaitCursor );
553   bool anIsOk = true;
554   HYDROGUI_DataModel* aModel = getDataModel();
555   if( aModel )
556   {
557     while( theNumActions > 0 )
558     {
559       if( !aModel->undo() )
560       {
561         anIsOk = false;
562         break;
563       }
564       theNumActions--;
565     }
566     update( UF_All );
567   }
568   QApplication::restoreOverrideCursor();
569   return anIsOk;
570 }
571
572 bool HYDROGUI_Module::onRedo( int theNumActions )
573 {
574   QApplication::setOverrideCursor( Qt::WaitCursor );
575   bool anIsOk = true;
576   HYDROGUI_DataModel* aModel = getDataModel();
577   if( aModel )
578   {
579     while( theNumActions > 0 )
580     {
581       if( !aModel->redo() )
582       {
583         anIsOk = false;
584         break;
585       }
586       theNumActions--;
587     }
588     update( UF_All );
589   }
590   QApplication::restoreOverrideCursor();
591   return anIsOk;
592 }
593
594 LightApp_Operation* HYDROGUI_Module::createOperation( const int theId ) const
595 {
596   LightApp_Operation* anOp = 0;
597   HYDROGUI_Module* aModule = const_cast<HYDROGUI_Module*>( this );
598   switch( theId )
599   {
600   case SaveVisualStateId:
601   case LoadVisualStateId:
602     anOp = new HYDROGUI_VisualStateOp( aModule, theId == LoadVisualStateId );
603     break;
604   case CopyId:
605   case PasteId:
606     anOp = new HYDROGUI_CopyPasteOp( aModule, theId == PasteId );
607     break;
608   case ImportImageId:
609   case EditImportedImageId:
610     anOp = new HYDROGUI_ImportImageOp( aModule, theId == EditImportedImageId );
611     break;
612   case ImportPolylineId:
613     anOp = new HYDROGUI_ImportPolylineOp( aModule/*, theId == EditImportedPolylineId*/ );
614     break;
615   case ImportSinusXId:
616     anOp = new HYDROGUI_ImportSinusXOp( aModule );
617     break;
618   case ExportSinusXId:
619     anOp = new HYDROGUI_ExportSinusXOp( aModule );
620     break;
621   case ObserveImageId:
622     anOp = new HYDROGUI_ObserveImageOp( aModule );
623     break;
624   case ExportImageId:
625     anOp = new HYDROGUI_ExportImageOp( aModule );
626     break;
627   case UpdateObjectId:
628   case ForcedUpdateObjectId:
629     anOp = new HYDROGUI_UpdateObjectOp( aModule, theId == ForcedUpdateObjectId );
630     break;
631   case ExportToShapeFileID:
632     anOp = new HYDROGUI_ExportFileOp( aModule );
633     break;
634   case ImportLandCoverMapId:
635     anOp = new HYDROGUI_ImportLandCoverMapOp( aModule );
636     break;
637   case RemoveImageRefsId:
638     anOp = new HYDROGUI_RemoveImageRefsOp( aModule );
639     break;
640   case CreatePolyline3DId:
641   case EditPolyline3DId:
642     anOp = new HYDROGUI_Poly3DOp( aModule, theId == EditPolyline3DId );
643     break;
644   case CreatePolylineId:
645   case EditPolylineId:
646     anOp = new HYDROGUI_PolylineOp( aModule, theId == EditPolylineId );
647     break;
648   case CreateProfileId:
649   case EditProfileId:
650     anOp = new HYDROGUI_ProfileOp( aModule, theId == EditProfileId );
651     break;
652   case ProfileInterpolateId:
653     anOp = new HYDROGUI_ProfileInterpolateOp( aModule );
654     break;
655   case ImportProfilesId:
656     anOp = new HYDROGUI_ImportProfilesOp( aModule ) ;
657     break;
658   case AllGeoreferencementId:
659     anOp = new HYDROGUI_GeoreferencementOp( aModule, HYDROGUI_GeoreferencementOp::All ) ;
660     break;
661   case SelectedGeoreferencementId:
662     anOp = new HYDROGUI_GeoreferencementOp( aModule, HYDROGUI_GeoreferencementOp::Selected ) ;
663     break;
664   case ImportBathymetryId:
665   case EditImportedBathymetryId:
666     anOp = new HYDROGUI_ImportBathymetryOp( aModule, theId == EditImportedBathymetryId  );
667     break;
668   case BathymetryBoundsId:
669     anOp = new HYDROGUI_BathymetryBoundsOp( aModule );
670     break;
671   case BathymetrySelectionId:
672     anOp = new HYDROGUI_BathymetrySelectionOp( aModule );
673     break;
674   case BathymetryTextId:
675   case BathymetryRescaleSelectionId:
676   case BathymetryRescaleVisibleId:
677   case BathymetryRescaleUserId:
678   case BathymetryRescaleDefaultId:
679     anOp = new HYDROGUI_BathymetryOp( aModule, theId );
680     break;
681
682   case CreateImmersibleZoneId:
683   case EditImmersibleZoneId:
684     anOp = new HYDROGUI_ImmersibleZoneOp( aModule, theId == EditImmersibleZoneId );
685     break;
686   case CreateStreamId:
687   case EditStreamId:
688     anOp = new HYDROGUI_StreamOp( aModule, theId == EditStreamId );
689     break;
690   case CreateChannelId:
691   case EditChannelId:
692     anOp = new HYDROGUI_ChannelOp( aModule, theId == EditChannelId );
693     break;
694   case CreateDigueId:
695   case EditDigueId:
696     anOp = new HYDROGUI_DigueOp( aModule, theId == EditDigueId );
697     break;
698   case ImportStricklerTableFromFileId:
699   case ExportStricklerTableFromFileId:
700   case EditStricklerTableId:
701     anOp = new HYDROGUI_StricklerTableOp( aModule, theId );
702     break;
703   case CreateLandCoverMapId:
704   case AddLandCoverId:
705   case RemoveLandCoverId:
706   case SplitLandCoverId:
707   case MergeLandCoverId:
708   case ChangeLandCoverTypeId:
709     anOp = new HYDROGUI_LandCoverMapOp( aModule, theId );
710     break;
711   case DuplicateStricklerTableId:
712     anOp = new HYDROGUI_DuplicateOp( aModule );
713     break;
714   case CreateCalculationId:
715   case EditCalculationId:
716     anOp = new HYDROGUI_CalculationOp( aModule, theId == EditCalculationId );
717     break;
718   case ExportCalculationId:
719     anOp = new HYDROGUI_ExportCalculationOp( aModule );
720     break;
721   case FuseImagesId:
722   case EditFusedImageId:
723     anOp = new HYDROGUI_TwoImagesOp( aModule, HYDROGUI_TwoImagesOp::Fuse, theId == EditFusedImageId );
724     break;
725   case CutImagesId:
726   case EditCutImageId:
727     anOp = new HYDROGUI_TwoImagesOp( aModule, HYDROGUI_TwoImagesOp::Cut, theId == EditCutImageId );
728     break;
729   case SplitImageId:
730   case EditSplitImageId:
731     anOp = new HYDROGUI_TwoImagesOp( aModule, HYDROGUI_TwoImagesOp::Split, theId == EditSplitImageId );
732     break;
733   case ImportObstacleFromFileId:
734     anOp = new HYDROGUI_ImportObstacleFromFileOp( aModule );
735     break;
736   case ImportGeomObjectAsObstacleId:
737     anOp = new HYDROGUI_ImportGeomObjectOp( aModule, HYDROGUI_ImportGeomObjectOp::ImportSelectedAsObstacle );
738     break;
739   case ImportGeomObjectAsPolylineId:
740     anOp = new HYDROGUI_ImportGeomObjectOp( aModule, HYDROGUI_ImportGeomObjectOp::ImportSelectedAsPolyline );
741     break;
742   case CreateBoxId:
743     anOp = new HYDROGUI_ImportGeomObjectOp( aModule, 
744       HYDROGUI_ImportGeomObjectOp::ImportCreatedAsObstacle, GEOMOp::OpBox );
745     break;
746   case CreateCylinderId:
747     anOp = new HYDROGUI_ImportGeomObjectOp( aModule, 
748       HYDROGUI_ImportGeomObjectOp::ImportCreatedAsObstacle, GEOMOp::OpCylinder );
749     break;
750   case TranslateObstacleId:
751     anOp = new HYDROGUI_TranslateObstacleOp( aModule );
752     break;
753   case CopyViewerPositionId:
754     anOp = new HYDROGUI_CopyPastePositionOp( aModule, false );
755     break;
756   case DeleteId:
757     anOp = new HYDROGUI_DeleteOp( aModule );
758     break;
759   case SetColorId:
760     anOp = new HYDROGUI_SetColorOp( aModule );
761     break;
762   case SetTransparencyId:
763     anOp = new HYDROGUI_SetTransparencyOp( aModule );
764     break;
765   case SetZLevelId:
766     anOp = new HYDROGUI_ZLevelsOp( aModule );
767     break;
768   case EditLocalCSId:
769     anOp = new HYDROGUI_LocalCSOp( aModule );
770     break;
771   case RiverBottomId:
772   case RiverBottomContextId:
773     anOp = new HYDROGUI_RiverBottomOp( aModule );
774     break;
775   case RecognizeContoursId:
776     anOp = new HYDROGUI_RecognizeContoursOp( aModule );
777     break;
778   case ShowId:
779   case ShowOnlyId:
780   case ShowAllId:
781   case HideId:
782   case HideAllId:
783     anOp = new HYDROGUI_ShowHideOp( aModule, theId );
784     break;
785   case SubmersibleId:
786     anOp = new HYDROGUI_SubmersibleOp( aModule, true );
787     break;
788   case UnSubmersibleId:
789     anOp = new HYDROGUI_SubmersibleOp( aModule, false );
790     break;
791   case PolylineExtractionId:
792     anOp = new HYDROGUI_PolylineExtractionOp( aModule );
793     break;
794   case SplitPolylinesId:
795     anOp = new HYDROGUI_SplitPolylinesOp( aModule );
796     break;
797   case MergePolylinesId:
798     anOp = new HYDROGUI_MergePolylinesOp( aModule );
799     break;
800   case LandCoverScalarMapModeOnId:
801   case LandCoverScalarMapModeOffId:
802     anOp = new HYDROGUI_LandCoverColoringOp( aModule, theId );
803     break;
804   case RegenerateRegionColorsId:
805     anOp = new HYDROGUI_RegenerateRegionColorsOp( aModule );
806     break;
807   case ZoneSetColorId:
808     anOp = new HYDROGUI_ZoneSetColorOp( aModule );
809     break;
810   }
811
812   if( !anOp )
813     anOp = LightApp_Module::createOperation( theId );
814
815   return anOp;
816 }
817
818 bool HYDROGUI_Module::reusableOperation( const int id )
819 {
820   if ( id == ImportGeomObjectAsObstacleId ||
821        id == ImportGeomObjectAsPolylineId ) {
822     return false;
823   }
824
825   return LightApp_Module::reusableOperation( id );
826 }
827
828 /**
829  * Returns true if the object with the given entry can be renamed.
830  * @param theEntry the object entry
831  */
832 bool HYDROGUI_Module::renameAllowed( const QString& theEntry ) const
833 {
834   // Allow to rename all HYDRO objects
835   Handle(HYDROData_Entity) anEntity = getDataModel()->objectByEntry( theEntry );
836   return !anEntity.IsNull();
837 }
838 /**
839  * Returns true if the object with the given entry is renamed.
840  * @param theEntry the object entry
841  * @param theName the new name
842  */
843 bool HYDROGUI_Module::renameObject( const QString& theEntry, const QString& theName )
844 {
845   Handle(HYDROData_Entity) anEntity = getDataModel()->objectByEntry( theEntry );
846   bool aRes = false;
847   if ( !anEntity.IsNull() )
848   {
849     HYDROGUI_DataModel* aModel = getDataModel();
850     if( aModel )
851     {
852       if( anEntity->GetName() != theName )
853       {
854         // check that there are no other objects with the same name in the document
855         Handle(HYDROData_Entity) anObject = HYDROGUI_Tool::FindObjectByName( this, theName );
856         if ( anObject.IsNull() )
857         {
858           SUIT_Operation* anOp = application()->activeStudy()->activeOperation();
859           if ( anOp && anOp->inherits( "HYDROGUI_CalculationOp" ) )
860           {
861             anEntity->SetName( theName );
862             aRes = true;
863           }
864           else
865           {
866             aRes = aModel->rename( anEntity, theName );
867           }
868         }
869         else
870         {
871           // Inform the user that the name is already used
872           QString aTitle = QObject::tr( "INSUFFICIENT_INPUT_DATA" );
873           QString aMessage = QObject::tr( "OBJECT_EXISTS_IN_DOCUMENT" ).arg( theName );
874           SUIT_MessageBox::critical( getApp()->desktop(), aTitle, aMessage );
875         }
876       }
877     }
878   }
879   return aRes;
880 }
881
882 void HYDROGUI_Module::onBathymetrySelection()
883 {
884   QAction* a = qobject_cast<QAction*>( sender() );
885   if( !a )
886     return;
887
888   bool isChecked = a->isChecked();
889   if( isChecked )
890     startOperation( BathymetrySelectionId );
891   else
892   {
893     LightApp_Operation* op = operation( BathymetryTextId );
894     if( op )
895       op->abort();
896
897     op = operation( BathymetrySelectionId );
898     if( op )
899       op->abort();
900   }
901 }
902
903 void HYDROGUI_Module::onBathymetryText()
904 {
905   QAction* a = qobject_cast<QAction*>( sender() );
906   if( !a )
907     return;
908
909   bool isChecked = a->isChecked();
910   if( isChecked )
911     startOperation( BathymetryTextId );
912   else
913   {
914     HYDROGUI_BathymetryOp* op = dynamic_cast<HYDROGUI_BathymetryOp*>( operation( BathymetryTextId ) );
915     if( op )
916       op->commit();
917   }
918 }