src/HEXABLOCKGUI/HEXABLOCKGUI.hxx

   1 // Copyright (C) 2009-2013  CEA/DEN, EDF R&D
   2 //
   3 // This library is free software; you can redistribute it and/or
   4 // modify it under the terms of the GNU Lesser General Public
   5 // License as published by the Free Software Foundation; either
   6 // version 2.1 of the License.
   7 //
   8 // This library is distributed in the hope that it will be useful,
   9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11 // Lesser General Public License for more details.
  12 //
  13 // You should have received a copy of the GNU Lesser General Public
  14 // License along with this library; if not, write to the Free Software
  15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  16 //
  17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  18 //
  19
  20 #ifndef _HEXABLOCKGUI_HXX_
  21 #define _HEXABLOCKGUI_HXX_
  22
  23 #include <iostream>
  24 #include <map>
  25 #include <set>
  26
  27 #include <QTreeView>
  28 #include <QModelIndex>
  29 #include <QDockWidget>
  30
  31
  32 #include "klinkitemselectionmodel.hxx"
  33
  34 #include <OB_Browser.h>
  35 #include <SalomeApp_Module.h>
  36 #include <SalomeApp_DataObject.h>
  37
  38 #include <LightApp_SelectionMgr.h>
  39 #include <LightApp_DataModel.h>
  40 #include <SOCC_Prs.h>
  41 #include <SALOMEconfig.h>
  42
  43 #include "HEXABLOCKGUI_Resource.hxx"
  44 #include "HEXABLOCKGUI_OCCSelector.h"
  45 #include "hexa_base.hxx"
  46
  47
  48 // #include CORBA_CLIENT_HEADER(HEXABLOCK_Gen)  // Abu Juin 2013
  49
  50 namespace HEXABLOCK
  51 {
  52   namespace GUI
  53   {
  54     class VtkDocumentGraphicView;
  55     class OccGraphicView;
  56     class GraphicViewsHandler;
  57     class DocumentDelegate;
  58     class DocumentModel;
  59     class PatternDataModel;
  60     class PatternBuilderModel;
  61     class AssociationsModel;
  62     class GroupsModel;
  63     class MeshModel;
  64     class PatternDataSelectionModel;
  65     class PatternBuilderSelectionModel;
  66     class GroupsSelectionModel;
  67     class MeshSelectionModel;
  68
  69     class HexaBaseDialog;
  70     class VertexDialog;
  71     class EdgeDialog;
  72     class QuadDialog;
  73     class HexaDialog;
  74     class VectorDialog;
  75     class CylinderDialog;
  76     class PipeDialog;
  77     class MakeGridDialog;
  78     class MakeCylinderDialog;
  79     class MakePipeDialog;
  80     class MakeCylindersDialog;
  81     class MakePipesDialog;
  82     class RemoveHexaDialog;
  83     class PrismQuadDialog;
  84     class JoinQuadDialog;
  85     class MergeDialog;
  86     class DisconnectDialog;
  87     class CutEdgeDialog;
  88     class MakeTransformationDialog;
  89     class MakeSymmetryDialog;
  90     class PerformTransformationDialog;
  91     class PerformSymmetryDialog;
  92     typedef class MyBasicGUI_PointDlg VertexAssocDialog;
  93     class EdgeAssocDialog;
  94     class QuadAssocDialog;
  95     class GroupDialog;
  96     class LawDialog;
  97     class PropagationDialog;
  98     class ComputeMeshDialog;
  99     class ReplaceHexaDialog;
 100     class QuadRevolutionDialog;
 101     class MakeHemiSphereDialog;
 102     class ModelInfoDialog;
 103     class AddShapeDialog;
 104   }
 105 }
 106
 107 class HEXABLOCKGUI_Resource;//SUIT_ResourceMgr;
 108 class SalomeApp_Study;
 109 class SalomeApp_Application;
 110 class SUIT_ViewWindow;
 111 class SVTK_ViewWindow;
 112 class OCCViewer_ViewWindow;
 113 class LightApp_VTKSelector;
 114
 115
 116 class HEXABLOCKGUI : public SalomeApp_Module
 117 {
 118   Q_OBJECT
 119
 120   friend class HEXABLOCKGUI_Resource;
 121
 122 public:
 123   HEXABLOCKGUI();
 124   virtual ~HEXABLOCKGUI();
 125
 126   enum ViewType {
 127       VTK,
 128       OCC,
 129       UNKNOWN
 130   };
 131
 132   static SalomeApp_Study*         activeStudy();
 133
 134   // static HEXABLOCK_ORB::HEXABLOCK_Gen_ptr InitHEXABLOCKGen( SalomeApp_Application* );
 135   static LightApp_SelectionMgr*   selectionMgr();
 136   static SALOMEDS::Study_var  ClientStudyToStudy (_PTR(Study) theStudy);
 137   static ViewType getActiveViewType();
 138   static HEXABLOCKGUI*  getInstance();
 139
 140   //add an object in the study
 141   void addInStudy(QMap<QString, TopoDS_Shape>& topo_shapes,
 142                          QMap<QString, QString>& docShapesEntry,
 143                          QMap<QString, QString>& docShapesName);
 144
 145   static HEXABLOCK::GUI::VtkDocumentGraphicView* currentDocGView;
 146   static HEXABLOCK::GUI::OccGraphicView* currentOccGView;
 147   static bool assocInProgress;
 148   static SalomeApp_Application* myApplication;
 149
 150   virtual void initialize( CAM_Application* app);
 151
 152   virtual bool renameAllowed( const QString& ) const;
 153   virtual bool renameObject( const QString& entry, const QString& name);
 154   virtual void windows( QMap<int, int>& theMap) const;
 155   virtual QString  engineIOR() const;
 156
 157   virtual void viewManagers(QStringList& list) const;
 158   virtual void setResource(SUIT_ResourceMgr* r);
 159   virtual void createPreferences();
 160   virtual void preferencesChanged( const QString& sect, const QString& name );
 161   virtual void studyActivated();
 162
 163   ///Returns a new HEXA_NS::Document and it's entry in the study
 164   std::pair <QString, HEXA_NS::Document*> newHexaDocument();
 165
 166   //------------------------------------
 167   void createAndFillDockWidget();
 168   void createActions();
 169   void createMenus();
 170   void createTools();
 171
 172   void initialMenus();
 173   void showAllMenus();
 174   void updateSelectors();
 175
 176   void switchModel(HEXABLOCK::GUI::VtkDocumentGraphicView* dgview);
 177   void switchOnGraphicView(HEXABLOCK::GUI::VtkDocumentGraphicView* dgview);
 178   void switchOffGraphicView(HEXABLOCK::GUI::VtkDocumentGraphicView* dgview,
 179                               bool saveCurrentDlg = true);
 180   void showDockWidgets(bool isVisible);
 181
 182   HEXABLOCK::GUI::VtkDocumentGraphicView* getCurrentVtkGraphicView() const { return currentDocGView; }
 183   HEXABLOCK::GUI::VtkDocumentGraphicView* getOrCreateDocument(SalomeApp_DataObject* studyObject);
 184   HEXABLOCK::GUI::VtkDocumentGraphicView* getDocGViewByEntry(const QString& entry) const
 185   {
 186       return ( (docs.contains(entry) ? docs[entry] : NULL ) );
 187   }
 188   HEXABLOCK::GUI::DocumentModel* getCurrentModel();
 189   HEXABLOCK::GUI::HexaBaseDialog* getDlgBox(HEXABLOCK::GUI::VtkDocumentGraphicView* view);
 190   QTreeView* getPatternDataTreeView() const { return _patternDataTreeView; }
 191   QTreeView* getPatternGeomTreeView() const { return _patternGeomTreeView; }
 192
 193 public slots:
 194   bool deactivateModule( SUIT_Study* theStudy);
 195   bool activateModule( SUIT_Study* theStudy);
 196   void onObjectBrowserClick(const QModelIndex& index);
 197
 198   void showDocumentMenus(bool show);
 199   void showPatternMenus(bool show);
 200   void showAssociationMenus(bool show);
 201   void showGroupsMenus(bool show);
 202   void showMeshMenus(bool show);
 203
 204 protected slots:
 205   void onWindowActivated( SUIT_ViewWindow* svw);
 206   void onWindowClosed( SUIT_ViewWindow* svw);
 207
 208   virtual void                        onViewManagerAdded( SUIT_ViewManager* );
 209   virtual void                        onViewManagerRemoved( SUIT_ViewManager* );
 210
 211   void onSelectionChanged( const QItemSelection & selected, const QItemSelection & deselected );
 212   void showActor();
 213   void hideActor();
 214   void showOnlyActor();
 215
 216 protected:
 217   bool createSComponent();
 218   virtual bool eventFilter(QObject *obj, QEvent *event);
 219
 220   void _showDialogBox( HEXABLOCK::GUI::HexaBaseDialog* diag );
 221   HEXABLOCK::GUI::HexaBaseDialog* currentDialog;
 222
 223   HEXABLOCKGUI_Resource* _myresource;
 224   static int _oldStudyId;
 225   // static HEXABLOCK_ORB::HEXABLOCK_Gen_var _hexaEngine;  // Hexa6 TODo
 226 private slots:
 227
 228   void slot_modelChanged(const QModelIndex &topLeft, const QModelIndex  &bottomRight);
 229
 230   void treeContextMenu(const QPoint& aPosition);
 231
 232   void test_association();
 233   void test();
 234
 235   void newDocument();
 236   void loadDocument( const QString &path = QString() );
 237   void saveDocument();
 238
 239   // new
 240   void addVertex();
 241   void addEdge();
 242   void addQuad();
 243   void addHexa();
 244
 245   void addVector();
 246   void addCylinder();
 247   void addPipe();
 248   void makeGrid();
 249   void makeCylinder();
 250   void makePipe();
 251   void makeCylinders();
 252   void makePipes();
 253
 254   void makeHemiSphere();  // NEW HEXA3
 255
 256   void removeHexa();
 257   void prismQuad();
 258   void joinQuad();
 259   void merge();
 260   void disconnectElts();
 261   void cutEdge();
 262   void makeTransformation();
 263   void makeSymmetry();
 264   void performTransformation();
 265   void performSymmetry();
 266
 267   void replaceHexa();    // NEW HEXA3
 268   void quadRevolution(); // NEW HEXA3
 269
 270
 271   void assocVertex();
 272   void assocEdge();
 273   void assocQuad();
 274
 275   void addGroup();
 276   void removeGroup (); //CS_TODO
 277   void addLaw();
 278   void removeLaw();
 279   void setPropagation();
 280   void computeMesh();
 281
 282   //show the current model's information (nb vertex, edge, ... and more)
 283   void showModelInfo();
 284
 285   //associate a shape to a document
 286   void addShape();
 287
 288   //Clears associations of a single element (VERTEX, EDGE, QUAD, ...)
 289   void clearAssociations();
 290
 291   //Clears associations under a tree (VERTEX_TREE, EDGE_TREE, ...)
 292   void clearAllAssociations();
 293
 294   void showAssociations();
 295
 296 private:
 297
 298   QList<HEXABLOCKGUI_OCCSelector*>  myOCCSelectors;
 299   QList<LightApp_VTKSelector*> myVTKSelectors;
 300
 301
 302   QStringList getQuickDirList();
 303   void restoreGraphicViews();
 304   void clearDialogs();
 305
 306   SOCC_Prs* getOccPrs(HEXABLOCK::GUI::VtkDocumentGraphicView* docGView)
 307   {
 308       if (docGView == NULL)
 309           return NULL;
 310       if (!occPrs.contains(docGView))
 311           occPrs[docGView] = new SOCC_Prs();
 312       return occPrs[docGView];
 313   }
 314
 315   // -------------------------------------------------------------------------------------------------
 316   //          MainWindow presentation
 317   // -------------------------------------------------------------------------------------------------
 318   QDockWidget *_dwPattern;       // Hexablock model edition
 319   QDockWidget *_dwAssociation;   // Hexablock association edition
 320   QDockWidget *_dwGroups;        // Hexablock groups edition
 321   QDockWidget *_dwMesh;          // Hexablock meshing edtion
 322   QDockWidget *_dwObjectBrowser; // Salome study
 323   QDockWidget *_dwInputPanel;    // user Input
 324
 325
 326   // Dialog Box ( to keep persistent values )
 327   HEXABLOCK::GUI::VertexDialog*                 _vertexDiag;
 328   HEXABLOCK::GUI::EdgeDialog*                   _edgeDiag;
 329   HEXABLOCK::GUI::QuadDialog*                   _quadDiag;
 330   HEXABLOCK::GUI::HexaDialog*                   _hexaDiag;
 331   HEXABLOCK::GUI::VectorDialog*                 _vectorDiag;
 332   HEXABLOCK::GUI::CylinderDialog*               _cylinderDiag;
 333   HEXABLOCK::GUI::PipeDialog*                   _pipeDiag;
 334   HEXABLOCK::GUI::MakeGridDialog*               _makeGridDiag;
 335   HEXABLOCK::GUI::MakeCylinderDialog*           _makeCylinderDiag;
 336   HEXABLOCK::GUI::MakePipeDialog*               _makePipeDiag;
 337   HEXABLOCK::GUI::MakeCylindersDialog*          _makeCylindersDiag;
 338   HEXABLOCK::GUI::MakePipesDialog*              _makePipesDiag;
 339   HEXABLOCK::GUI::RemoveHexaDialog*             _removeHexaDiag;
 340   HEXABLOCK::GUI::PrismQuadDialog*              _prismQuadDiag;
 341   HEXABLOCK::GUI::JoinQuadDialog*               _joinQuadDiag;
 342   HEXABLOCK::GUI::MergeDialog*                  _mergeDiag;
 343   HEXABLOCK::GUI::DisconnectDialog*             _disconnectDiag;
 344   HEXABLOCK::GUI::CutEdgeDialog*                _cutEdgeDiag;
 345   HEXABLOCK::GUI::MakeTransformationDialog*     _makeTransformationDiag;
 346   HEXABLOCK::GUI::MakeSymmetryDialog*           _makeSymmetryDiag;
 347   HEXABLOCK::GUI::PerformTransformationDialog*  _performTransformationDiag;
 348   HEXABLOCK::GUI::PerformSymmetryDialog*        _performSymmetryDiag;
 349   HEXABLOCK::GUI::VertexAssocDialog*            _vertexAssocDiag;
 350   HEXABLOCK::GUI::EdgeAssocDialog*              _edgeAssocDiag;
 351   HEXABLOCK::GUI::QuadAssocDialog*              _quadAssocDiag;
 352   HEXABLOCK::GUI::GroupDialog*                  _groupDiag;
 353   HEXABLOCK::GUI::LawDialog*                    _lawDiag;
 354   HEXABLOCK::GUI::PropagationDialog*            _propagationDiag;
 355   HEXABLOCK::GUI::ComputeMeshDialog*            _computeMeshDiag;
 356   HEXABLOCK::GUI::ReplaceHexaDialog*            _replaceHexaDiag;
 357   HEXABLOCK::GUI::QuadRevolutionDialog*         _quadRevolutionDiag;
 358   HEXABLOCK::GUI::MakeHemiSphereDialog*         _makeHemiSphereDiag;
 359   HEXABLOCK::GUI::ModelInfoDialog*              _modelInfoDiag;
 360   HEXABLOCK::GUI::AddShapeDialog*               _addShapeDiag;
 361
 362
 363   std::set<HEXABLOCK::GUI::HexaBaseDialog*> currentModelDialogs;
 364   // Actions
 365   int _menuId;
 366   // Object Browser
 367   QAction *_newAct;
 368   QAction *_importAct;
 369   QAction *_saveAct;
 370   QAction *_testAct;
 371
 372   // Pattern Data
 373   QAction *_addVertex;
 374   QAction *_addEdge;
 375   QAction *_addQuad;
 376   QAction *_addHexa;
 377
 378   // Pattern Builder
 379   QAction *_addVector;
 380   QAction *_addCylinder;
 381   QAction *_addPipe;
 382   QAction *_makeGrid; //Cartesian, Cylindrical, Spherical
 383   QAction *_makeCylinder;
 384   QAction *_makePipe;
 385   QAction *_makeCylinders;
 386   QAction *_makePipes;
 387
 388   QAction *_makeHemiSphere; // NEW HEXA3
 389
 390
 391   // Pattern Data Edition
 392   QAction *_removeHexa;
 393   QAction *_prismQuad;
 394   QAction *_joinQuad;
 395   QAction *_merge;//   QAction *_mergeVertices; //   QAction *_mergeEdges;
 396   QAction *_disconnect;
 397   QAction *_cutEdge;
 398   QAction *_makeTransformation; //   QAction *_makeTranslation;
 399   QAction *_makeSymmetry;
 400   QAction *_performTransformation;
 401   QAction *_performSymmetry;
 402
 403   QAction *_replaceHexa;    // NEW HEXA3
 404   QAction *_quadRevolution; // NEW HEXA3
 405
 406   // Pattern Data Association
 407   QAction *_assocVertex;
 408   QAction *_assocEdge;
 409   QAction *_assocQuad;
 410
 411   // Group
 412   QAction *_addGroup;
 413   QAction *_removeGroup ; //CS_TODO
 414
 415   // Law
 416   QAction *_addLaw;
 417   QAction *_removeLaw;
 418
 419   // Propagation
 420   QAction *_setPropagation;
 421
 422   // Meshing
 423   QAction *_computeMesh;
 424
 425   QAction* _showModelInfoAct;
 426
 427   QAction* _addShapeAct;
 428
 429
 430   QAction* _sep1; // tmonin
 431   QAction* _sep2; // tmonin
 432   QAction* _sep3; // tmonin
 433   QAction* _sep4; // ...... tparti
 434
 435   // -------------------------------------------------------------------------------------------------
 436   //          Model/View implementation
 437   // -------------------------------------------------------------------------------------------------
 438
 439   HEXABLOCK::GUI::AssociationsModel   *_associationsModel;    // sub-part of DocumentModel
 440
 441   //      VIEW      VIEW      VIEW      VIEW      VIEW      VIEW      VIEW      VIEW      VIEW      VIEW
 442   QTreeView* _patternDataTreeView;    //  document's pattern : 1 ( only one view )
 443   QTreeView* _patternBuilderTreeView; //  document's pattern : 1 ( only one view )
 444   QTreeView* _patternGeomTreeView; // the geometries' tree view
 445   QTreeView* _associationTreeView;    //  document's association : 1 ( only one view )
 446   QTreeView* _groupsTreeView; //  document's groups
 447   QTreeView* _meshTreeView;   //  document's mesh property: 1 ( only one view )
 448
 449   //      DELEGATE      DELEGATE      DELEGATE      DELEGATE      DELEGATE      DELEGATE      DELEGATE
 450   HEXABLOCK::GUI::DocumentDelegate    *_treeViewDelegate;  // specific editor for each item of the tree
 451
 452   HEXABLOCK::GUI::GraphicViewsHandler*  graphicViewsHandler; //vtk views hanlder (create, close, ...)
 453
 454   //  SALOME   SALOME    SALOME   SALOME     SALOME     SALOME     SALOME     SALOME     SALOME     SALOME
 455   QMap<QString, HEXABLOCK::GUI::VtkDocumentGraphicView*>  docs; //  key = entry
 456   QMap<HEXABLOCK::GUI::VtkDocumentGraphicView*, HEXABLOCK::GUI::HexaBaseDialog*>  gViewDlgBox; //  key = entry
 457   QMap<HEXABLOCK::GUI::VtkDocumentGraphicView*, SOCC_Prs*> occPrs;
 458
 459   bool _isSaved;
 460   bool moduleActivatedOnce;
 461
 462   QDir* loadDocLastPath;
 463   QDir* saveDocLastPath;
 464
 465   void testDocument();
 466   void test_make_cart_grid();
 467   void test_make_elmts_transform();
 468
 469   QString       addDocInStudy (HEXA_NS::Document* doc);  // Hexa6
 470   HEXA_NS::Hex* hexa_root;                               // Hexa6
 471 };
 472
 473 #endif