Salome HOME
0021014: EDF 1583 SMESH: Improvement of the Python Dump for the creation of groups
[modules/smesh.git] / src / SMESH_I / SMESH_2smeshpy.hxx
1 // Copyright (C) 2007-2011  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 // File      : SMESH_smesh.hxx
24 // Created   : Fri Nov 18 12:05:18 2005
25 // Author    : Edward AGAPOV (eap)
26 //
27 #ifndef SMESH_smesh_HeaderFile
28 #define SMESH_smesh_HeaderFile
29
30 #include <Standard_DefineHandle.hxx>
31 #include <Standard_Type.hxx>
32 #include <Standard_Transient.hxx>
33 #include <TCollection_AsciiString.hxx>
34 #include <TColStd_SequenceOfAsciiString.hxx>
35 #include <TColStd_SequenceOfInteger.hxx>
36
37 #include <list>
38 #include <map>
39
40 // ===========================================================================================
41 /*!
42  * This file was created in order to respond to requirement of bug PAL10494:
43  * SMESH python dump uses idl interface.
44  *
45  * The creation reason is that smesh.py commands defining hypotheses encapsulate
46  * several SMESH engine method calls. As well, the dependencies between smesh.py
47  * classes differ from ones between corresponding SMESH IDL interfaces.
48  * 
49  * Everything here is for internal usage by SMESH_2smeshpy::ConvertScript()
50  * declared in SMESH_PythonDump.hxx
51  *
52  * See comments to _pyHypothesis class to know how to assure convertion of a new
53  * type of hypothesis
54  */
55 // ===========================================================================================
56
57 class Resource_DataMapOfAsciiStringAsciiString;
58
59 // ===========================================================================================
60 // =====================
61 //    INTERNAL STUFF
62 // =====================
63 // ===========================================================================================
64
65 class _pyCommand;
66 class _pyObject;
67 class _pyGen;
68 class _pyMesh;
69 class _pySubMesh;
70 class _pyHypothesis;
71 class _pyAlgorithm;
72
73 DEFINE_STANDARD_HANDLE (_pyCommand   ,Standard_Transient);
74 DEFINE_STANDARD_HANDLE (_pyObject    ,Standard_Transient);
75 DEFINE_STANDARD_HANDLE (_pyGen       ,_pyObject);
76 DEFINE_STANDARD_HANDLE (_pyMesh      ,_pyObject);
77 DEFINE_STANDARD_HANDLE (_pySubMesh   ,_pyObject);
78 DEFINE_STANDARD_HANDLE (_pyMeshEditor,_pyObject);
79 DEFINE_STANDARD_HANDLE (_pyHypothesis,_pyObject);
80 DEFINE_STANDARD_HANDLE (_pyAlgorithm ,_pyHypothesis);
81
82 typedef TCollection_AsciiString _pyID;
83
84 // ===========================================================
85 /*!
86  * \brief Class operating on a command string looking like
87  *        ResultValue = Object.Method( Arg1, Arg2,...)
88  */
89 // ===========================================================
90
91 class _pyCommand: public Standard_Transient
92 {
93   int                             myOrderNb;            //!< position within the script
94   TCollection_AsciiString         myString;             //!< command text
95   TCollection_AsciiString         myRes, myObj, myMeth; //!< found parts of command
96   TColStd_SequenceOfAsciiString   myArgs;               //!< found arguments
97   TColStd_SequenceOfInteger       myBegPos;             //!< where myRes, myObj, ... begin
98   std::list< Handle(_pyCommand) > myDependentCmds; //!< commands that sould follow me in the script
99
100   enum { UNKNOWN=-1, EMPTY=0, RESULT_IND, OBJECT_IND, METHOD_IND, ARG1_IND };
101   int GetBegPos( int thePartIndex );
102   void SetBegPos( int thePartIndex, int thePosition );
103   void SetPart( int thePartIndex, const TCollection_AsciiString& theNewPart,
104                 TCollection_AsciiString& theOldPart);
105   void FindAllArgs() { GetArg(1); }
106
107 public:
108   _pyCommand() {};
109   _pyCommand( const TCollection_AsciiString& theString, int theNb )
110     : myString( theString ), myOrderNb( theNb ) {};
111   TCollection_AsciiString & GetString() { return myString; }
112   int GetOrderNb() const { return myOrderNb; }
113   void SetOrderNb( int theNb ) { myOrderNb = theNb; }
114   int Length() { return myString.Length(); }
115   void Clear() { myString.Clear(); myBegPos.Clear(); myArgs.Clear(); }
116   bool IsEmpty() const { return myString.IsEmpty(); }
117   TCollection_AsciiString GetIndentation();
118   const TCollection_AsciiString & GetResultValue();
119   const int GetNbResultValues();
120   const TCollection_AsciiString & GetResultValue(int res);
121   const TCollection_AsciiString & GetObject();
122   const TCollection_AsciiString & GetMethod();
123   const TCollection_AsciiString & GetArg( int index );
124   int GetNbArgs() { FindAllArgs(); return myArgs.Length(); }
125   //Handle(TColStd_HSequenceOfAsciiString) GetArgs();
126   void SetResultValue( const TCollection_AsciiString& theResult )
127   { GetResultValue(); SetPart( RESULT_IND, theResult, myRes ); }
128   void SetObject(const TCollection_AsciiString& theObject)
129   { GetObject(); SetPart( OBJECT_IND, theObject, myObj ); }
130   void SetMethod(const TCollection_AsciiString& theMethod)
131   { GetMethod(); SetPart( METHOD_IND, theMethod, myMeth ); }
132   void SetArg( int index, const TCollection_AsciiString& theArg);
133   void RemoveArgs();
134   static bool SkipSpaces( const TCollection_AsciiString & theSring, int & thePos );
135   static TCollection_AsciiString GetWord( const TCollection_AsciiString & theSring,
136                                           int & theStartPos, const bool theForward,
137                                           const bool dotIsWord = false);
138   void AddDependantCmd( Handle(_pyCommand) cmd, bool prepend = false)
139   { if (prepend) myDependentCmds.push_front( cmd ); else myDependentCmds.push_back( cmd ); }
140   bool SetDependentCmdsAfter() const;
141
142   bool AddAccessorMethod( _pyID theObjectID, const char* theAcsMethod );
143
144   DEFINE_STANDARD_RTTI (_pyCommand)
145 };
146
147 // -------------------------------------------------------------------------------------
148 /*!
149  * \brief Root of all objects. It counts calls of Process()
150  */
151 // -------------------------------------------------------------------------------------
152
153 class _pyObject: public Standard_Transient
154 {
155   Handle(_pyCommand) myCreationCmd;
156   int                myNbCalls;
157 public:
158   _pyObject(const Handle(_pyCommand)& theCreationCmd)
159     : myCreationCmd(theCreationCmd), myNbCalls(0) {}
160   const _pyID& GetID() { return myCreationCmd->GetResultValue(); }
161   static _pyID FatherID(const _pyID & childID);
162   const Handle(_pyCommand)& GetCreationCmd() { return myCreationCmd; }
163   int GetNbCalls() const { return myNbCalls; }
164   void  SetCreationCmd( Handle(_pyCommand) cmd ) { myCreationCmd = cmd; }
165   int GetCommandNb() { return myCreationCmd->GetOrderNb(); }
166   virtual void Process(const Handle(_pyCommand) & theCommand) { myNbCalls++; }
167   virtual void Flush() = 0;
168   virtual const char* AccessorMethod() const;
169
170   DEFINE_STANDARD_RTTI (_pyObject)
171 };
172
173 // -------------------------------------------------------------------------------------
174 /*!
175  * \brief Class corresponding to SMESH_Gen. It holds info on existing
176  *        meshes and hypotheses
177  */
178 // -------------------------------------------------------------------------------------
179 class _pyGen: public _pyObject
180 {
181 public:
182   _pyGen(Resource_DataMapOfAsciiStringAsciiString& theEntry2AccessorMethod,
183          Resource_DataMapOfAsciiStringAsciiString& theObjectNames);
184   Handle(_pyCommand) AddCommand( const TCollection_AsciiString& theCommand );
185   void Process( const Handle(_pyCommand)& theCommand );
186   void Flush();
187   Handle(_pyHypothesis) FindHyp( const _pyID& theHypID );
188   Handle(_pyHypothesis) FindAlgo( const _pyID& theGeom, const _pyID& theMesh,
189                                   const Handle(_pyHypothesis)& theHypothesis);
190   Handle(_pySubMesh) FindSubMesh( const _pyID& theSubMeshID );
191   void ExchangeCommands( Handle(_pyCommand) theCmd1, Handle(_pyCommand) theCmd2 );
192   void SetCommandAfter( Handle(_pyCommand) theCmd, Handle(_pyCommand) theAfterCmd );
193   void SetCommandBefore( Handle(_pyCommand) theCmd, Handle(_pyCommand) theBeforeCmd );
194   Handle(_pyCommand)& GetLastCommand();
195   std::list< Handle(_pyCommand) >& GetCommands() { return myCommands; }
196   void SetAccessorMethod(const _pyID& theID, const char* theMethod );
197   bool AddMeshAccessorMethod( Handle(_pyCommand) theCmd ) const;
198   bool AddAlgoAccessorMethod( Handle(_pyCommand) theCmd ) const;
199   const char* AccessorMethod() const;
200   _pyID GenerateNewID( const _pyID& theID );
201   void AddObject( Handle(_pyObject)& theObj );
202
203 private:
204   void setNeighbourCommand( Handle(_pyCommand)& theCmd,
205                             Handle(_pyCommand)& theOtherCmd,
206                             const bool theIsAfter );
207   
208 private:
209   std::map< _pyID, Handle(_pyMesh) >       myMeshes;
210   std::map< _pyID, Handle(_pyMeshEditor) > myMeshEditors;
211   std::map< _pyID, Handle(_pyObject) >     myObjects;
212   std::list< Handle(_pyHypothesis) >       myHypos;
213   std::list< Handle(_pyCommand) >          myCommands;
214   int                                      myNbCommands;
215   Resource_DataMapOfAsciiStringAsciiString& myID2AccessorMethod;
216   Resource_DataMapOfAsciiStringAsciiString& myObjectNames;
217   Handle(_pyCommand)                       myLastCommand;
218
219   DEFINE_STANDARD_RTTI (_pyGen)
220 };
221
222 // -------------------------------------------------------------------------------------
223 /*!
224  * \brief Contains commands concerning mesh substructures
225  */
226 // -------------------------------------------------------------------------------------
227 #define _pyMesh_ACCESS_METHOD "GetMesh()"
228 class _pyMesh: public _pyObject
229 {
230   std::list< Handle(_pyHypothesis) > myHypos;
231   std::list< Handle(_pyCommand) > myAddHypCmds;
232   std::list< Handle(_pySubMesh) > mySubmeshes;
233   bool                            myHasEditor;
234 public:
235   _pyMesh(const Handle(_pyCommand) creationCmd);
236   _pyMesh(const Handle(_pyCommand) theCreationCmd, const TCollection_AsciiString & id);
237   const _pyID& GetGeom() { return GetCreationCmd()->GetArg(1); }
238   void Process( const Handle(_pyCommand)& theCommand);
239   void Flush();
240   const char* AccessorMethod() const { return _pyMesh_ACCESS_METHOD; }
241 private:
242   static bool NeedMeshAccess( const Handle(_pyCommand)& theCommand );
243   static void AddMeshAccess( const Handle(_pyCommand)& theCommand )
244   { theCommand->SetObject( theCommand->GetObject() + "." _pyMesh_ACCESS_METHOD ); }
245
246   DEFINE_STANDARD_RTTI (_pyMesh)
247 };
248 #undef _pyMesh_ACCESS_METHOD 
249
250 // -------------------------------------------------------------------------------------
251 /*!
252  * \brief MeshEditor convert its commands to ones of mesh
253  */
254 // -------------------------------------------------------------------------------------
255 class _pyMeshEditor: public _pyObject
256 {
257   _pyID myMesh;
258   TCollection_AsciiString myCreationCmdStr;
259 public:
260   _pyMeshEditor(const Handle(_pyCommand)& theCreationCmd);
261   void Process( const Handle(_pyCommand)& theCommand);
262   virtual void Flush() {}
263
264   DEFINE_STANDARD_RTTI (_pyMesh)
265 };
266
267 // -------------------------------------------------------------------------------------
268 /*!
269  * \brief Root class for hypothesis
270  *
271  * HOWTO assure convertion of a new type of hypothesis
272  * In _pyHypothesis::NewHypothesis():
273  * 1. add a case for the name of the new hypothesis
274  * 2. use SetConvMethodAndType() to set
275  *    . for algo: algorithm name and method of Mesh creating the algo
276  *    . for hypo: name of the algorithm and method creating the hypothesis
277  * 3. append to myArgMethods interface methods setting param values in the
278  *    order they are used when creation method is called. If arguments of
279  *    the creation method can't be easily got from calls of hypothesis methods, you are
280  *    to derive a specific class from _pyHypothesis that would redefine Process(),
281  *    see _pyComplexParamHypo for example
282  */
283 // -------------------------------------------------------------------------------------
284 class _pyHypothesis: public _pyObject
285 {
286 protected:
287   bool    myIsAlgo, myIsWrapped;
288   _pyID   myGeom,   myMesh;
289   // a hypothesis can be used and created by different algos by different methods
290   std::map<TCollection_AsciiString, TCollection_AsciiString > myType2CreationMethod;
291   //TCollection_AsciiString       myCreationMethod, myType;
292   TColStd_SequenceOfAsciiString myArgs;           // creation arguments
293   TColStd_SequenceOfAsciiString myArgMethods;     // hypo methods setting myArgs
294   TColStd_SequenceOfInteger     myNbArgsByMethod; // nb args set by each method
295   std::list<Handle(_pyCommand)>  myArgCommands;
296   std::list<Handle(_pyCommand)>  myUnknownCommands;
297 public:
298   _pyHypothesis(const Handle(_pyCommand)& theCreationCmd);
299   void SetConvMethodAndType(const char* creationMethod, const char* type)
300   { myType2CreationMethod[ (char*)type ] = (char*)creationMethod; }
301   void AddArgMethod(const char* method, const int nbArgs = 1)
302   { myArgMethods.Append( (char*)method ); myNbArgsByMethod.Append( nbArgs ); }
303   const TColStd_SequenceOfAsciiString& GetArgs() const { return myArgs; }
304   const std::list<Handle(_pyCommand)>& GetArgCommands() const { return myArgCommands; }
305   void ClearAllCommands();
306   virtual bool IsAlgo() const { return myIsAlgo; }
307   bool IsValid() const { return !myType2CreationMethod.empty(); }
308   bool IsWrapped() const { return myIsWrapped; }
309   const _pyID & GetGeom() const { return myGeom; }
310   void SetMesh( const _pyID& theMeshId) { if ( myMesh.IsEmpty() ) myMesh = theMeshId; }
311   const _pyID & GetMesh() const { return myMesh; }
312   const TCollection_AsciiString& GetAlgoType() const
313   { return myType2CreationMethod.begin()->first; }
314   const TCollection_AsciiString& GetAlgoCreationMethod() const
315   { return myType2CreationMethod.begin()->second; }
316   bool CanBeCreatedBy(const TCollection_AsciiString& algoType ) const
317   { return myType2CreationMethod.find( algoType ) != myType2CreationMethod.end(); }
318   const TCollection_AsciiString& GetCreationMethod(const TCollection_AsciiString& algoType) const
319   { return myType2CreationMethod.find( algoType )->second; }
320   virtual bool IsWrappable(const _pyID& theMesh) { return !myIsWrapped && myMesh == theMesh; }
321   virtual bool Addition2Creation( const Handle(_pyCommand)& theAdditionCmd,
322                                   const _pyID&              theMesh);
323   static Handle(_pyHypothesis) NewHypothesis( const Handle(_pyCommand)& theCreationCmd);
324   void Process( const Handle(_pyCommand)& theCommand);
325   void Flush();
326   virtual void Assign( const Handle(_pyHypothesis)& theOther,
327                        const _pyID&                 theMesh );
328
329   DEFINE_STANDARD_RTTI (_pyHypothesis)
330 };
331
332 // -------------------------------------------------------------------------------------
333 /*!
334  * \brief Class representing smesh.Mesh_Algorithm
335  */
336 // -------------------------------------------------------------------------------------
337 class _pyAlgorithm: public _pyHypothesis
338 {
339 public:
340   _pyAlgorithm(const Handle(_pyCommand)& theCreationCmd);
341   virtual bool Addition2Creation( const Handle(_pyCommand)& theAdditionCmd,
342                                   const _pyID&              theMesh);
343   const char* AccessorMethod() const { return "GetAlgorithm()"; }
344   virtual bool IsWrappable(const _pyID& theMesh) { return !myIsWrapped; }
345
346   DEFINE_STANDARD_RTTI (_pyAlgorithm)
347 };
348
349 // -------------------------------------------------------------------------------------
350 /*!
351  * \brief Class for hypotheses having several parameters modified by one method
352  */
353 // -------------------------------------------------------------------------------------
354 class _pyComplexParamHypo: public _pyHypothesis
355 {
356 public:
357   _pyComplexParamHypo(const Handle(_pyCommand)& theCreationCmd): _pyHypothesis(theCreationCmd) {}
358   void Process( const Handle(_pyCommand)& theCommand);
359   void Flush();
360
361   DEFINE_STANDARD_RTTI (_pyComplexParamHypo)
362 };
363 DEFINE_STANDARD_HANDLE (_pyComplexParamHypo, _pyHypothesis);
364
365 // -------------------------------------------------------------------------------------
366 /*!
367  * \brief Class for LayerDistribution hypothesis conversion
368  */
369 // -------------------------------------------------------------------------------------
370 class _pyLayerDistributionHypo: public _pyHypothesis
371 {
372   Handle(_pyHypothesis) my1dHyp;
373   TCollection_AsciiString myAlgoMethod;
374 public:
375   _pyLayerDistributionHypo(const Handle(_pyCommand)& theCreationCmd, const char* algoMethod):
376     _pyHypothesis(theCreationCmd), myAlgoMethod((char*)algoMethod) {}
377   void Process( const Handle(_pyCommand)& theCommand);
378   void Flush();
379   bool Addition2Creation( const Handle(_pyCommand)& theAdditionCmd,
380                           const _pyID&              theMesh);
381
382   DEFINE_STANDARD_RTTI (_pyLayerDistributionHypo)
383 };
384 DEFINE_STANDARD_HANDLE (_pyLayerDistributionHypo, _pyHypothesis);
385
386 // -------------------------------------------------------------------------------------
387 /*!
388  * \brief Class representing NumberOfSegments hypothesis
389  */
390 // -------------------------------------------------------------------------------------
391 class _pyNumberOfSegmentsHyp: public _pyHypothesis
392 {
393 public:
394   _pyNumberOfSegmentsHyp(const Handle(_pyCommand)& theCrCmd): _pyHypothesis(theCrCmd) {}
395   virtual bool Addition2Creation( const Handle(_pyCommand)& theAdditionCmd,
396                                   const _pyID&              theMesh);
397   void Flush();
398
399   DEFINE_STANDARD_RTTI (_pyNumberOfSegmentsHyp)
400 };
401 DEFINE_STANDARD_HANDLE (_pyNumberOfSegmentsHyp, _pyHypothesis);
402
403 // -------------------------------------------------------------------------------------
404 /*!
405  * \brief Class representing SegmentLengthAroundVertex hypothesis
406  */
407 // -------------------------------------------------------------------------------------
408 class _pySegmentLengthAroundVertexHyp: public _pyHypothesis
409 {
410 public:
411   _pySegmentLengthAroundVertexHyp(const Handle(_pyCommand)& theCrCmd): _pyHypothesis(theCrCmd) {}
412   virtual bool Addition2Creation( const Handle(_pyCommand)& theAdditionCmd,
413                                   const _pyID&              theMesh);
414   DEFINE_STANDARD_RTTI (_pySegmentLengthAroundVertexHyp)
415 };
416 DEFINE_STANDARD_HANDLE (_pySegmentLengthAroundVertexHyp, _pyHypothesis);
417
418 // -------------------------------------------------------------------------------------
419 /*!
420  * \brief SelfEraser erases creation command if no more it's commands invoked
421  */
422 // -------------------------------------------------------------------------------------
423 class _pySelfEraser: public _pyObject
424 {
425 public:
426   _pySelfEraser(const Handle(_pyCommand)& theCreationCmd):_pyObject(theCreationCmd) {}
427   virtual void Flush();
428
429   DEFINE_STANDARD_RTTI (_pySelfEraser)
430 };
431 DEFINE_STANDARD_HANDLE (_pySelfEraser, _pyObject);
432
433 // -------------------------------------------------------------------------------------
434 /*!
435  * \brief SubMesh creation can be moved to the end of engine commands
436  */
437 // -------------------------------------------------------------------------------------
438 class _pySubMesh:  public _pyObject
439 {
440 public:
441   _pySubMesh(const Handle(_pyCommand)& theCreationCmd):_pyObject(theCreationCmd) {}
442   void Process( const Handle(_pyCommand)& theCommand);
443   virtual void Flush();
444   void SetCreator( const Handle(_pyObject)& theCreator ) { myCreator = theCreator; }
445
446   DEFINE_STANDARD_RTTI (_pySubMesh)
447 private:
448   Handle(_pyObject) myCreator;
449 };
450 // -------------------------------------------------------------------------------------
451 /*!
452  * \brief To convert creation of a group by filter
453  */
454 // -------------------------------------------------------------------------------------
455 class _pyGroup:  public _pyObject
456 {
457 public:
458   _pyGroup(const Handle(_pyCommand)& theCreationCmd):_pyObject(theCreationCmd) {}
459   void Process( const Handle(_pyCommand)& theCommand);
460   virtual void Flush() {}
461
462   DEFINE_STANDARD_RTTI (_pyGroup)
463 };
464 DEFINE_STANDARD_HANDLE (_pyGroup, _pyObject);
465
466 // -------------------------------------------------------------------------------------
467 /*!
468  * \brief A filter
469  */
470 // -------------------------------------------------------------------------------------
471 class _pyFilter:  public _pyObject
472 {
473 public:
474   _pyFilter(const Handle(_pyCommand)& theCreationCmd):_pyObject(theCreationCmd) {}
475   void Process( const Handle(_pyCommand)& theCommand);
476   virtual void Flush() {}
477
478   DEFINE_STANDARD_RTTI (_pyFilter)
479 };
480 DEFINE_STANDARD_HANDLE (_pyFilter, _pyObject);
481
482 #endif