Salome HOME
Fix for the Feature #102: Display of Lambert coordinates.
[modules/hydro.git] / src / HYDROGUI / HYDROGUI_Module.h
1 // Copyright (C) 2007-2013  CEA/DEN, EDF R&D, OPEN CASCADE
2 //
3 // Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
5 //
6 // This library is free software; you can redistribute it and/or
7 // modify it under the terms of the GNU Lesser General Public
8 // License as published by the Free Software Foundation; either
9 // version 2.1 of the License.
10 //
11 // This library is distributed in the hope that it will be useful,
12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14 // Lesser General Public License for more details.
15 //
16 // You should have received a copy of the GNU Lesser General Public
17 // License along with this library; if not, write to the Free Software
18 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
19 //
20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
21 //
22
23 #ifndef HYDROGUI_MODULE_H
24 #define HYDROGUI_MODULE_H
25
26 #include <HYDROData_Entity.h>
27
28 #include <LightApp_Module.h>
29
30 #include <QEvent>
31
32 #include <vtkNew.h>
33
34 class QGraphicsSceneMouseEvent;
35
36 class GraphicsView_Viewer;
37 class OCCViewer_Viewer;
38 class SVTK_Viewer;
39 class vtkWorldPointPicker;
40
41 class SUIT_ViewWindow;
42 class SUIT_ViewManager;
43
44 class HYDROGUI_DataModel;
45 class HYDROGUI_Displayer;
46 class HYDROGUI_OCCDisplayer;
47 class HYDROGUI_VTKPrsDisplayer;
48 class HYDROGUI_AbstractDisplayer;
49 class HYDROGUI_Shape;
50 class HYDROGUI_VTKPrs;
51
52 /**\class HYDROGUI_Module
53  *\brief The class representing the HYDROGUI module
54  */
55 class HYDROGUI_Module : public LightApp_Module
56 {
57   Q_OBJECT
58
59   enum CustomEvent { NewViewEvent = QEvent::User + 100 };
60
61 public:
62   
63   enum ViewManagerRole
64   {
65     VMR_Unknown = 0,
66     VMR_General,
67     VMR_TransformImage,
68     VMR_ReferenceImage,
69     VMR_ObserveImage,
70     //VMR_PreviewZone, // totally unused, to delete
71     VMR_PreviewCaseZones // actually unused, to review
72   };
73   typedef QPair< SUIT_ViewManager*, ViewManagerRole > ViewManagerInfo;
74
75   typedef QMap        < int, ViewManagerInfo > ViewManagerMap;
76   typedef QMapIterator< int, ViewManagerInfo > ViewManagerMapIterator;
77
78   struct ObjectState
79   {
80     bool Visibility;
81     double Transparency;
82     double ZValue;
83     ObjectState() : Visibility( false ), Transparency( 1.0 ), ZValue( 0.0 ) {}
84   };
85   typedef QMap< QString, ObjectState > Entry2ObjectStateMap;
86   typedef QMap< int, Entry2ObjectStateMap > ViewId2Entry2ObjectStateMap;
87
88   typedef QList<HYDROGUI_Shape*> ListOfShapes;
89   typedef QMap<int,ListOfShapes> ViewId2ListOfShapes;
90
91   typedef QList<HYDROGUI_VTKPrs*> ListOfVTKPrs;
92   typedef QMap<int,ListOfVTKPrs> ViewId2ListOfVTKPrs;
93
94 public:
95   HYDROGUI_Module();
96   virtual ~HYDROGUI_Module();
97
98   int                             getStudyId() const;
99
100   virtual void                    initialize( CAM_Application* );
101   virtual void                    windows( QMap<int, int>& ) const;
102   virtual void                    viewManagers( QStringList& ) const;
103
104   virtual void                    contextMenuPopup( const QString&, QMenu*, QString& );
105
106   virtual void                    update( const int );
107   virtual void                    updateCommandsStatus();
108
109   virtual void                    selectionChanged();
110
111   SUIT_ViewManager*               getViewManager( const int theId ) const;
112   HYDROGUI_DataModel*             getDataModel() const;
113   HYDROGUI_Displayer*             getDisplayer() const;
114   HYDROGUI_OCCDisplayer*          getOCCDisplayer() const;
115   HYDROGUI_VTKPrsDisplayer*       getVTKDisplayer() const;
116
117   GraphicsView_Viewer*            getViewer( const int theId ) const;
118   OCCViewer_Viewer*               getOCCViewer( const int theId ) const;
119   SVTK_Viewer*                    getVTKViewer( const int theId ) const;
120
121   int                             getViewManagerId( SUIT_ViewManager* theViewManager );
122   ViewManagerRole                 getViewManagerRole( SUIT_ViewManager* theViewManager );
123   void                            setViewManagerRole( SUIT_ViewManager* theViewManager,
124                                                       const ViewManagerRole theRole );
125
126   bool                            isObjectVisible( const int theViewId,
127                                                    const Handle(HYDROData_Entity)& theObject ) const;
128   void                            setObjectVisible( const int theViewId,
129                                                     const Handle(HYDROData_Entity)& theObject,
130                                                     const bool theState );
131   /** 
132    * Set IsToUpdate flag for all presentations of the given object to recompute them during 
133    * the next viewer(s) updating.
134    * @param theObject the data model object
135    * @param theState  the flag to set for object's presentations, it's true by default.
136    */
137   void                            setIsToUpdate( const Handle(HYDROData_Entity)& theObject,
138                                                  const bool theState = true );
139   HYDROGUI_Shape*                 getObjectShape( const int                       theViewId,
140                                                   const Handle(HYDROData_Entity)& theObject ) const;
141   void                            setObjectShape( const int                       theViewId,
142                                                   const Handle(HYDROData_Entity)& theObject,
143                                                   HYDROGUI_Shape*                 theShape );
144   void                            removeViewShapes( const int                     theViewId );
145   void                            removeObjectShape( const int                       theViewId,
146                                                      const Handle(HYDROData_Entity)& theObject );
147
148   HYDROGUI_VTKPrs*                getObjectVTKPrs( const int                       theViewId,
149                                                    const Handle(HYDROData_Entity)& theObject ) const;
150   void                            setObjectVTKPrs( const int                       theViewId,
151                                                    const Handle(HYDROData_Entity)& theObject,
152                                                    HYDROGUI_VTKPrs*                theShape );
153   void                            removeViewVTKPrs( const int                      theViewId );
154   void                            removeObjectVTKPrs( const int                      theViewId,
155                                                      const Handle(HYDROData_Entity)& theObject );
156
157   /**
158    * Update global imposed range of Z values for the given VTK viewer id
159    */
160   void                            updateVTKZRange( const int theViewId, double theRange[] );
161
162   /**
163    * Returns true if the object with the given entry can be renamed.
164    * @param theEntry the object entry
165    */
166   virtual bool                    renameAllowed( const QString& theEntry ) const;
167   /**
168    * Returns true if the object with the given entry is renamed.
169    * @param theEntry the object entry
170    * @param theName the new name
171    */
172   virtual bool                    renameObject( const QString& theEntry, const QString& theName );
173
174 protected:
175   CAM_DataModel*                  createDataModel();
176
177 public slots:
178   virtual bool                    activateModule( SUIT_Study* );
179   virtual bool                    deactivateModule( SUIT_Study* );
180
181 protected:
182   virtual LightApp_Operation*     createOperation( const int ) const;
183   virtual bool                    reusableOperation( const int id );
184
185   virtual void                    customEvent( QEvent* );
186   virtual bool                    eventFilter( QObject*, QEvent* );
187
188 protected slots:
189   void                            onOperation();
190   void                            onDelete();
191
192
193   bool                            onUndo( int theNumActions );
194   bool                            onRedo( int theNumActions );
195
196   virtual void                    onViewManagerAdded( SUIT_ViewManager* );
197   virtual void                    onViewManagerRemoved( SUIT_ViewManager* );
198   virtual void                    onViewCreated( SUIT_ViewWindow* );
199
200   void                            onViewPortMouseEvent( QGraphicsSceneMouseEvent* );
201
202   void                            onMouseMove( SUIT_ViewWindow*, QMouseEvent* );
203
204 private:
205   void                            updateViewer( HYDROGUI_AbstractDisplayer* theDisplayer, 
206                                                 const bool theIsInit = false, 
207                                                 const bool theIsForced = false, 
208                                                 const bool theDoFitAll = false );
209
210   void                            createSelector( SUIT_ViewManager* viewMgr );
211
212   bool                            setUpdateEnabled( const bool theState );
213   bool                            isUpdateEnabled() const;
214
215   QStringList                     storeSelection() const;
216   void                            restoreSelection( const QStringList& theEntryList );
217
218 private:
219   void                            createActions();
220   void                            createMenus();
221   void                            createPopups();
222   void                            createToolbars();
223
224   void                            createUndoRedoActions();
225   void                            updateUndoRedoControls();
226
227 private:
228   QAction*                        createAction( const int theId, const QString& theSuffix,
229                                                 const QString& theImg = QString::null,
230                                                 const int theKey = 0, const bool isToggle = false,
231                                                 const QString& theSlot = QString::null );
232
233 private:
234   HYDROGUI_Displayer*             myDisplayer;
235   HYDROGUI_OCCDisplayer*          myOCCDisplayer;
236   HYDROGUI_VTKPrsDisplayer*       myVTKDisplayer;
237
238   ViewManagerMap                  myViewManagerMap;
239   ViewId2Entry2ObjectStateMap      myObjectStateMap;
240
241   ViewId2ListOfShapes             myShapesMap;
242   ViewId2ListOfVTKPrs             myVTKPrsMap;
243   vtkNew<vtkWorldPointPicker>     myPicker;
244
245   bool                            myIsUpdateEnabled;
246
247   QStringList                     myGeomObjectsToImport; ///< entries of GEOM objects to be imported
248 };
249
250 #endif