]> SALOME platform Git repositories - plugins/blsurfplugin.git/blob - src/GUI/BLSURFPluginGUI_HypothesisCreator.h
Salome HOME
Fixed compilation on Windows
[plugins/blsurfplugin.git] / src / GUI / BLSURFPluginGUI_HypothesisCreator.h
1 // Copyright (C) 2007-2023  CEA, EDF
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, or (at your option) any later version.
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 // ---
21 // File    : BLSURFPluginGUI_HypothesisCreator.h
22 // Authors : Francis KLOSS (OCC) & Patrick LAUG (INRIA) & Lioka RAZAFINDRAZAKA (CEA)
23 //           & Aurelien ALLEAUME (DISTENE)
24 //           Size maps developement: Nicolas GEIMER (OCC) & Gilles DAVID (EURIWARE)
25 // ---
26 //
27 #ifndef BLSURFPLUGINGUI_HypothesisCreator_H
28 #define BLSURFPLUGINGUI_HypothesisCreator_H
29
30 // akl: avoid compilation warning on Linux : "_POSIX_C_SOURCE" and "_XOPEN_SOURCE" are redefined
31 #ifdef _POSIX_C_SOURCE
32 #undef _POSIX_C_SOURCE
33 #endif
34
35 #ifdef _XOPEN_SOURCE
36 #undef _XOPEN_SOURCE
37 #endif
38
39 #ifdef WIN32
40   #if defined BLSURFPLUGIN_GUI_EXPORTS || defined BLSURFPluginGUI_EXPORTS
41     #define BLSURFPLUGIN_GUI_EXPORT __declspec( dllexport )
42   #else
43     #define BLSURFPLUGIN_GUI_EXPORT __declspec( dllimport )
44   #endif
45 #else
46   #define BLSURFPLUGIN_GUI_EXPORT
47 #endif
48
49 #ifdef WIN32
50 // E.A. : On windows with python 2.6, there is a conflict
51 // E.A. : between pymath.h and Standard_math.h which define
52 // E.A. : some same symbols : acosh, asinh, ...
53 #include <Standard_Real.hxx>
54 #include <pymath.h>
55 #endif
56
57 #include <Python.h>
58
59 #include <QItemDelegate>
60
61 #include <SMESHGUI_Hypotheses.h>
62 #include "StdMeshersGUI_ObjectReferenceParamWdg.h"
63 #include <SALOMEconfig.h>
64 #include <cstring>
65 #include <map>
66 #include <set>
67 #include <vector>
68 #include <TopAbs_ShapeEnum.hxx>
69 #include <GeomSelectionTools.h>
70 #include <GEOM_Client.hxx>
71 #include CORBA_SERVER_HEADER(BLSURFPlugin_Algorithm)
72
73 class QGroupBox;
74 class QComboBox;
75 class QCheckBox;
76 class QLineEdit;
77 class QRadioButton;
78 class QTableWidget;
79 class QTreeWidget;
80 class QModelIndex;
81 class QSpinBox;
82 class QDoubleSpinBox;
83 class QMenu;
84 class QAction;
85 class QTreeWidgetItem;
86 class QTableWidgetItem;
87 class QObject;
88 class QSplitter;
89 class QGridLayout;
90 class QVBoxLayout;
91 class QSpacerItem;
92
93 class SMESHGUI_SpinBox;
94 class SMESH_NumberFilter;
95 class LightApp_SelectionMgr;
96 class BLSURFPluginGUI_StdWidget;
97 class BLSURFPluginGUI_AdvWidget;
98 class StdMeshersGUI_SubShapeSelectorWdg;
99 // class DlgBlSurfHyp_Enforced;
100
101 // Name
102 typedef std::string TEnfName;
103 // Entry
104 typedef std::string TEntry;
105 // List of entries
106 typedef std::set<TEntry> TEntryList;
107 // Enforced vertex = 3 coordinates
108 typedef std::vector<double> TEnfVertexCoords;
109 // List of enforced vertices
110 typedef std::set< TEnfVertexCoords > TEnfVertexCoordsList;
111 // Enforced vertex
112 struct TEnfVertex{
113   TEnfName name;
114   TEntry geomEntry;
115   TEnfVertexCoords coords;
116   TEnfName grpName;
117 };
118 // Attractor
119 struct TAttractor{
120   std::string attEntry;
121   double      startSize;
122   double      infDist;
123   double      constDist;
124   TAttractor( const char* theAttEntry, double theStartSize, double theInfDist, double theConstDist)
125     : attEntry( theAttEntry ),
126       startSize( theStartSize ),
127       infDist( theInfDist ),
128       constDist( theConstDist )
129   {}
130   void SetToDelete() { startSize = -1; }
131   bool IsToDelete() const { return startSize < 0; }
132 };
133 typedef std::vector< TAttractor > TAttractorVec;
134
135 struct CompareEnfVertices
136 {
137   bool operator () (const TEnfVertex* e1, const TEnfVertex* e2) const {
138     if (e1 && e2) {
139       if (e1->coords.size() && e2->coords.size())
140         return (e1->coords < e2->coords);
141       else
142         return (e1->geomEntry < e2->geomEntry);
143     }
144     return false;
145   }
146 };
147
148 // List of enforced vertices
149 typedef std::set< TEnfVertex*, CompareEnfVertices > TEnfVertexList;
150
151 // Map Face Entry / List of enforced vertices
152 typedef std::map< TEntry, TEnfVertexList > TFaceEntryEnfVertexListMap;
153
154 // Map Face Entry / InternalEnforcedVertices
155 typedef std::map< TEntry, bool > TFaceEntryInternalVerticesMap;
156
157 // PreCad Face and Edge periodicity
158 typedef std::vector<std::string> TPreCadPeriodicity;
159 typedef std::vector< TPreCadPeriodicity > TPreCadPeriodicityVector;
160
161 typedef struct
162 {
163   int     myTopology, myVerbosity;
164   int     myPhysicalMesh, myGeometricMesh;
165   double  myPhySize, myMinSize, myMaxSize;
166   bool    myPhySizeRel, myMinSizeRel, myMaxSizeRel;
167   bool    myUseMinSize, myUseMaxSize, myUseGradation, myUseVolumeGradation;
168   double  myGradation, myVolumeGradation, myAngleMesh, myChordalError;
169   bool    myAnisotropic, myOptimiseTinyEdges, myRemoveTinyEdges, myForceBadElementRemoval, myCorrectSurfaceIntersection;
170   double  myAnisotropicRatio, myTinyEdgeLength, myTinyEdgeOptimisLength, myBadElementAspectRatio, myCorrectSurfaceIntersectionMaxCost;
171   bool    myOptimizeMesh, myQuadraticMesh;
172   bool    mySmpsurface,mySmpedge,mySmppoint,myEnforcedVertex,myInternalEnforcedVerticesAllFaces;
173   int     myElementType;
174   bool    myUseSurfaceProximity;
175   int     myNbSurfaceProximityLayers;
176   double  mySurfaceProximityRatio;
177   bool    myUseVolumeProximity;
178   int     myNbVolumeProximityLayers;
179   double  myVolumeProximityRatio;
180   // bool    myPreCADMergeEdges, myPreCADProcess3DTopology, myPreCADDiscardInput;
181 //   bool    myGMFFileMode;
182   std::string myGMFFileName, myInternalEnforcedVerticesAllFacesGroup;
183   TEnfVertexList enfVertexList;
184   TFaceEntryEnfVertexListMap faceEntryEnfVertexListMap;
185   /* TODO GROUPS
186   TGroupNameEnfVertexListMap groupNameEnfVertexListMap;
187   */
188   TPreCadPeriodicityVector preCadPeriodicityVector;
189   QStringList hyperpatches, hyperEntries;
190   QString myName;
191 } BlsurfHypothesisData;
192
193
194 /*!
195  * \brief Class for creation of MG-CADSurf hypotheses
196 */
197 class BLSURFPLUGIN_GUI_EXPORT BLSURFPluginGUI_HypothesisCreator : public SMESHGUI_GenericHypothesisCreator
198 {
199   Q_OBJECT
200
201 public:
202   BLSURFPluginGUI_HypothesisCreator( const QString& );
203   virtual ~BLSURFPluginGUI_HypothesisCreator();
204
205   virtual bool        checkParams(QString& msg) const;
206   virtual QString     helpPage() const;
207
208 protected:
209   virtual QFrame*     buildFrame    ();
210   virtual void        retrieveParams() const;
211   virtual QString     storeParams   () const;
212
213   virtual QString     caption() const;
214   virtual QPixmap     icon() const;
215   virtual QString     type() const;
216
217 protected slots:
218   void                onStateChange();
219   // Advanced tab
220   void                onAddOption();
221   void                onChangeOptionName( int, int );
222   // Sizemap tab
223   void                onMapGeomContentModified();
224   void                onSmpItemClicked( QTreeWidgetItem *, int );
225   void                onTabChanged(int);
226   void                onAttractorClicked(int);
227   void                onConstSizeClicked(int);
228   void                onAddMap();
229   void                onRemoveMap();
230   void                onModifyMap();
231   void                onSetSizeMap(QTreeWidgetItem *, int);
232   // Enforced vertices tab
233   QTreeWidgetItem*    addEnforcedFace(std::string theFaceEntry, std::string theFaceName);
234   void                addEnforcedVertex(double x=0, double y=0, double z=0, 
235                                         std::string vertexName = "",
236                                         std::string geomEntry = "",
237                                         std::string groupName = "");
238   void                onAddEnforcedVertices();
239   void                onRemoveEnforcedVertex();
240   void                synchronizeCoords();
241   void                updateEnforcedVertexValues(QTreeWidgetItem* , int );
242   void                onSelectEnforcedVertex();
243   void                clearEnforcedVertexWidgets();
244   void                onInternalVerticesClicked(int);
245   // Enforced mesh tab
246   void                onAddEnforcedMesh();
247   void                onRemoveEnforcedMesh();
248   void                onEnforcedMeshSelected();
249   void                onEnfMeshTableSelected();
250   // Periodicity tab
251   void                onPeriodicityByVerticesChecked(bool);
252   void                onAddPeriodicity();
253   void                onRemovePeriodicity();
254   void                onPeriodicityTreeClicked(QTreeWidgetItem*, int);
255   void                onPeriodicityContentModified();
256   // HyperPatch tab
257   void                onHyPatchFaceSelection(bool);
258   void                onHyPatchGroupSelection(bool);
259   void                onHyPatchSelectionChanged();
260   void                onHyPatchAdd();
261   void                onHyPatchRemove();
262
263 private:
264   bool                readParamsFromHypo( BlsurfHypothesisData& ) const;
265   QString             readParamsFromWidgets( BlsurfHypothesisData& ) const;
266   bool                storeParamsToHypo( const BlsurfHypothesisData& ) const;
267   bool                sizeMapsValidation();
268   bool                sizeMapValidationFromRow(int,bool displayError = true);
269   bool                sizeMapValidationFromEntry(QString,bool displayError = true);
270   GeomSelectionTools* getGeomSelectionTool() const;
271   GEOM::GEOM_Gen_var  getGeomEngine();
272   bool                insertElement( GEOM::GEOM_Object_var, bool modify = false );
273   bool                insertAttractor(GEOM::GEOM_Object_var, GEOM::GEOM_Object_var, bool modify = false);
274   int                 findRowFromEntry(QString entry);
275   CORBA::Object_var   entryToObject(QString entry);
276   static LightApp_SelectionMgr* selectionMgr();
277   void                avoidSimultaneousSelection(ListOfWidgets &myCustomWidgets) const;
278   void                AddPreCadSequenceToVector(BlsurfHypothesisData& h_data, BLSURFPlugin::TPeriodicityList_var preCadFacePeriodicityVector, bool onFace) const;
279   void                addHyPatchToTable(const QString& tags, const QString& entries);
280   bool                hasGeom() const;
281
282   void                addEnforcedMesh( SMESH::SMESH_IDSource_ptr mesh, const QString& groupName );
283
284 private:
285   
286   QTabWidget*             myTabWidget;
287   QWidget*                myStdGroup;
288   BLSURFPluginGUI_StdWidget* myStdWidget;
289   QLineEdit*              myName;
290
291   QWidget*                myAdvGroup;
292   BLSURFPluginGUI_AdvWidget* myAdvWidget;
293
294   // Sizemap widgets
295   QWidget                 *mySmpGroup;
296   QTreeWidget             *mySizeMapTable;
297   QPushButton             *addMapButton;
298   QPushButton             *removeMapButton;
299   QPushButton             *modifyMapButton;
300   QTabWidget              *smpTab; 
301   QWidget                 *myAttractorGroup;
302   QWidget                 *mySmpStdGroup;
303   QCheckBox               *myAttractorCheck;
304   QCheckBox               *myConstSizeCheck;
305   QGroupBox               *myDistanceGroup;
306 //   QGroupBox               *myParamsGroup;
307 //   QWidget                 *myParamsGroup;
308   SMESHGUI_SpinBox        *myAttSizeSpin;
309   SMESHGUI_SpinBox        *myAttDistSpin;
310   SMESHGUI_SpinBox        *myAttDistSpin2;
311   SMESHGUI_SpinBox        *mySmpSizeSpin;
312   QLabel                  *myAttDistLabel;
313   QLabel                  *myAttDistLabel2;
314   QLabel                  *myAttSizeLabel;
315   // Selection widgets for size maps
316   StdMeshersGUI_ObjectReferenceParamWdg *myGeomSelWdg1;
317   StdMeshersGUI_ObjectReferenceParamWdg *myGeomSelWdg2;
318   StdMeshersGUI_ObjectReferenceParamWdg *myAttSelWdg;
319   StdMeshersGUI_ObjectReferenceParamWdg *myDistSelWdg;
320   GEOM::GEOM_Object_var                  mySMapObject;
321   GEOM::GEOM_Object_var                  myAttObject;
322   GEOM::GEOM_Object_var                  myDistObject;
323   
324   
325   
326   
327   QWidget*            myEnfGroup;
328   StdMeshersGUI_ObjectReferenceParamWdg *myEnfFaceWdg;
329   GEOM::GEOM_Object_var myEnfFace;
330   StdMeshersGUI_ObjectReferenceParamWdg *myEnfVertexWdg;
331   GEOM::GEOM_Object_var myEnfVertex;
332
333   QTreeWidget*        myEnforcedTreeWidget;
334   SMESHGUI_SpinBox*   myXCoord;
335   SMESHGUI_SpinBox*   myYCoord;
336   SMESHGUI_SpinBox*   myZCoord;
337
338   QLineEdit*          myGroupName;
339
340   QPushButton*        addVertexButton;
341   QPushButton*        removeVertexButton;
342
343   QCheckBox           *myInternalEnforcedVerticesAllFaces;
344   QLineEdit           *myInternalEnforcedVerticesAllFacesGroup;
345
346
347   // Enforced meshes
348   QWidget*             myEnfMeshGroup;
349   StdMeshersGUI_ObjectReferenceParamWdg *myEnfMeshWdg;
350   QTableWidget*        myEnfMeshTableWdg;
351   QLineEdit*           myEnforcedGroupName;
352   QPushButton*         myAddEnfMeshButton;
353   QPushButton*         myRemoveEnfMeshButton;
354
355
356
357   // map =  entry , size map
358   QMap<QString, QString>          mySMPMap;           // Map <face entry, size>
359   QMap<QString, TAttractorVec >   myATTMap;           // Map <face entry, att. entry, etc>
360   QMap<QString, TopAbs_ShapeEnum> mySMPShapeTypeMap;
361   GeomSelectionTools*             GeomToolSelected;
362   LightApp_SelectionMgr*          aSel;
363
364   // Periodicity
365   QWidget*      myPeriodicityGroup;
366   QSplitter*    myPeriodicitySplitter;
367   QTreeWidget*  myPeriodicityTreeWidget;
368   QWidget*      myPeriodicityRightWidget;
369   QGridLayout*  myPeriodicityRightGridLayout;
370   QGroupBox*    myPeriodicityGroupBox1;
371   QGroupBox*    myPeriodicityGroupBox2;
372   QGridLayout* aPeriodicityLayout1;
373   QGridLayout*  myPeriodicityGroupBox1Layout;
374   QGridLayout*  myPeriodicityGroupBox2Layout;
375   QRadioButton* myPeriodicityOnFaceRadioButton;
376   QRadioButton* myPeriodicityOnEdgeRadioButton;
377   QLabel*       myPeriodicityMainSourceLabel;
378   QLabel*       myPeriodicityMainTargetLabel;
379   QLabel*       myPeriodicitySourceLabel;
380   QLabel*       myPeriodicityTargetLabel;
381   StdMeshersGUI_ObjectReferenceParamWdg* myPeriodicitySourceFaceWdg;
382   GEOM::GEOM_Object_var myPeriodicityFace;
383   StdMeshersGUI_ObjectReferenceParamWdg* myPeriodicityTargetFaceWdg;
384   GEOM::GEOM_Object_var myPeriodicityEdge;
385   QLabel*       myPeriodicityP1SourceLabel;
386   QLabel*       myPeriodicityP2SourceLabel;
387   QLabel*       myPeriodicityP3SourceLabel;
388   QLabel*       myPeriodicityP1TargetLabel;
389   QLabel*       myPeriodicityP2TargetLabel;
390   QLabel*       myPeriodicityP3TargetLabel;
391   StdMeshersGUI_ObjectReferenceParamWdg* myPeriodicityP1SourceWdg;
392   StdMeshersGUI_ObjectReferenceParamWdg* myPeriodicityP2SourceWdg;
393   StdMeshersGUI_ObjectReferenceParamWdg* myPeriodicityP3SourceWdg;
394   StdMeshersGUI_ObjectReferenceParamWdg* myPeriodicityP1TargetWdg;
395   StdMeshersGUI_ObjectReferenceParamWdg* myPeriodicityP2TargetWdg;
396   StdMeshersGUI_ObjectReferenceParamWdg* myPeriodicityP3TargetWdg;
397   ListOfWidgets myPeriodicitySelectionWidgets;
398   QPushButton*  myPeriodicityAddButton;
399   QPushButton*  myPeriodicityRemoveButton;
400   QSpacerItem*  myPeriodicityVerticalSpacer;
401
402   QTableWidget*                      myHyPatchTable;
403   StdMeshersGUI_SubShapeSelectorWdg* myHyPatchFaceSelector;
404   QLineEdit*                         myHyPatchTagsLE;
405   QPushButton*                       myHyPatchFaceSelBtn;
406   QPushButton*                       myHyPatchGroupSelBtn;
407
408   BLSURFPlugin::string_array_var myOptions, myPreCADOptions, myCustomOptions;
409
410   PyObject *          main_mod;
411   PyObject *          main_dict;
412 };
413
414
415 class EnforcedTreeWidgetDelegate : public QItemDelegate
416 {
417   Q_OBJECT
418
419 public:
420   EnforcedTreeWidgetDelegate(QObject *parent = 0);
421
422   QWidget *createEditor(QWidget *parent, const QStyleOptionViewItem & option,
423                         const QModelIndex &index) const;
424
425   void setEditorData(QWidget *editor, const QModelIndex &index) const;
426   void setModelData(QWidget *editor, QAbstractItemModel *model,
427                     const QModelIndex &index) const;
428
429   void updateEditorGeometry(QWidget *editor, const QStyleOptionViewItem &option,
430                             const QModelIndex &index) const;
431
432   bool vertexExists(QAbstractItemModel *model, const QModelIndex &index, QString value) const;
433 };
434
435 #endif // BLSURFPLUGINGUI_HypothesisCreator_H