Salome HOME
Merge from OCC_development_generic_2006
[modules/smesh.git] / src / SMESH_I / SMESH_2smeshpy.cxx
1 // File      : SMESH_2smeshpy.cxx
2 // Created   : Fri Nov 18 13:20:10 2005
3 // Author    : Edward AGAPOV (eap)
4
5 #include "SMESH_2smeshpy.hxx"
6
7 #include "SMESH_Gen_i.hxx"
8 #include "utilities.h"
9 #include "SMESH_PythonDump.hxx"
10 #include "Resource_DataMapOfAsciiStringAsciiString.hxx"
11
12 IMPLEMENT_STANDARD_HANDLE (_pyObject          ,Standard_Transient);
13 IMPLEMENT_STANDARD_HANDLE (_pyCommand         ,Standard_Transient);
14 IMPLEMENT_STANDARD_HANDLE (_pyGen             ,_pyObject);
15 IMPLEMENT_STANDARD_HANDLE (_pyMesh            ,_pyObject);
16 IMPLEMENT_STANDARD_HANDLE (_pyHypothesis      ,_pyObject);
17 IMPLEMENT_STANDARD_HANDLE (_pyAlgorithm       ,_pyHypothesis);
18 IMPLEMENT_STANDARD_HANDLE (_pyComplexParamHypo,_pyHypothesis);
19 IMPLEMENT_STANDARD_HANDLE (_pyNumberOfSegmentsHyp,_pyHypothesis);
20
21 IMPLEMENT_STANDARD_RTTIEXT(_pyObject          ,Standard_Transient);
22 IMPLEMENT_STANDARD_RTTIEXT(_pyCommand         ,Standard_Transient);
23 IMPLEMENT_STANDARD_RTTIEXT(_pyGen             ,_pyObject);
24 IMPLEMENT_STANDARD_RTTIEXT(_pyMesh            ,_pyObject);
25 IMPLEMENT_STANDARD_RTTIEXT(_pyHypothesis      ,_pyObject);
26 IMPLEMENT_STANDARD_RTTIEXT(_pyAlgorithm       ,_pyHypothesis);
27 IMPLEMENT_STANDARD_RTTIEXT(_pyComplexParamHypo,_pyHypothesis);
28 IMPLEMENT_STANDARD_RTTIEXT(_pyNumberOfSegmentsHyp,_pyHypothesis);
29
30 using namespace std;
31 using SMESH::TPythonDump;
32
33 /*!
34  * \brief Container of commands into which the initial script is split.
35  *        It also contains data coresponding to SMESH_Gen contents
36  */
37 static Handle(_pyGen) theGen;
38
39 static TCollection_AsciiString theEmptyString;
40
41 //#define DUMP_CONVERSION
42
43 #if !defined(_DEBUG_) && defined(DUMP_CONVERSION)
44 #undef DUMP_CONVERSION
45 #endif
46
47 //================================================================================
48 /*!
49  * \brief Convert python script using commands of smesh.py
50   * \param theScript - Input script
51   * \retval TCollection_AsciiString - Convertion result
52  */
53 //================================================================================
54
55 TCollection_AsciiString
56 SMESH_2smeshpy::ConvertScript(const TCollection_AsciiString& theScript,
57                               Resource_DataMapOfAsciiStringAsciiString& theEntry2AccessorMethod)
58 {
59   theGen = new _pyGen( theEntry2AccessorMethod );
60
61   // split theScript into separate commands
62   int from = 1, end = theScript.Length(), to;
63   while ( from < end && ( to = theScript.Location( "\n", from, end )))
64   {
65     if ( to != from )
66       // cut out and store a command
67       theGen->AddCommand( theScript.SubString( from, to - 1 ));
68     from = to + 1;
69   }
70   // finish conversion
71   theGen->Flush();
72 #ifdef DUMP_CONVERSION
73   cout << endl << " ######## RESULT ######## " << endl<< endl;
74 #endif
75   // concat commands back into a script
76   TCollection_AsciiString aScript;
77   list< Handle(_pyCommand) >::iterator cmd = theGen->GetCommands().begin();
78   for ( ; cmd != theGen->GetCommands().end(); ++cmd ) {
79 #ifdef DUMP_CONVERSION
80     cout << "## COM " << (*cmd)->GetOrderNb() << ": "<< (*cmd)->GetString() << endl;
81 #endif
82     if ( !(*cmd)->IsEmpty() ) {
83       aScript += "\n";
84       aScript += (*cmd)->GetString();
85     }
86   }
87   aScript += "\n";
88
89   theGen.Nullify();
90
91   return aScript;
92 }
93
94 //================================================================================
95 /*!
96  * \brief _pyGen constructor
97  */
98 //================================================================================
99
100 _pyGen::_pyGen(Resource_DataMapOfAsciiStringAsciiString& theEntry2AccessorMethod)
101   : _pyObject( new _pyCommand( TPythonDump::SMESHGenName(), 0 )),
102     myID2AccessorMethod( theEntry2AccessorMethod )
103 {
104   myNbCommands = 0;
105   // make that GetID() to return TPythonDump::SMESHGenName()
106   GetCreationCmd()->GetString() += "=";
107 }
108
109 //================================================================================
110 /*!
111  * \brief Convert a command using a specific converter
112   * \param theCommand - the command to convert
113   * \retval bool - convertion result
114  */
115 //================================================================================
116
117 void _pyGen::AddCommand( const TCollection_AsciiString& theCommand)
118 {
119   // store theCommand in the sequence
120   myCommands.push_back( new _pyCommand( theCommand, ++myNbCommands ));
121
122   Handle(_pyCommand) aCommand = myCommands.back();
123 #ifdef DUMP_CONVERSION
124   cout << "## COM " << myNbCommands << ": "<< aCommand->GetString() << endl;
125 #endif
126
127   _pyID objID = aCommand->GetObject();
128
129   if ( objID.IsEmpty() )
130     return;
131
132   // SMESH_Gen method?
133   if ( objID == this->GetID() ) {
134     this->Process( aCommand );
135     return;
136   }
137   // SMESH_Mesh method?
138   map< _pyID, Handle(_pyMesh) >::iterator id_mesh = myMeshes.find( objID );
139   if ( id_mesh != myMeshes.end() ) {
140     id_mesh->second->Process( aCommand );
141     return;
142   }
143   // SMESH_Hypothesis method
144   list< Handle(_pyHypothesis) >::iterator hyp = myHypos.begin();
145   for ( ; hyp != myHypos.end(); ++hyp )
146     if ( !(*hyp)->IsAlgo() && objID == (*hyp)->GetID() )
147       (*hyp)->Process( aCommand );
148
149   // Mesh provides SMESH_IDSource interface used in SMESH_MeshEditor.
150   // Add access to wrapped mesh
151   if ( objID == TPythonDump::MeshEditorName() ) {
152     // in all SMESH_MeshEditor's commands, a SMESH_IDSource is the first arg
153     id_mesh = myMeshes.find( aCommand->GetArg( 1 ));
154     if ( id_mesh != myMeshes.end() )
155       aCommand->SetArg( 1 , aCommand->GetArg( 1 ) + ".GetMesh()" );
156   }
157 }
158
159 //================================================================================
160 /*!
161  * \brief Convert the command or remember it for later conversion 
162   * \param theCommand - The python command calling a method of SMESH_Gen
163  */
164 //================================================================================
165
166 void _pyGen::Process( const Handle(_pyCommand)& theCommand )
167 {
168   // there are methods to convert:
169   // CreateMesh( shape )
170   // CreateHypothesis( theHypType, theLibName )
171   // Compute( mesh, geom )
172
173   if ( theCommand->GetMethod() == "CreateMesh" )
174   {
175     Handle(_pyMesh) mesh = new _pyMesh( theCommand );
176     myMeshes.insert( make_pair( mesh->GetID(), mesh ));
177     return;
178   }
179
180   // CreateHypothesis()
181   if ( theCommand->GetMethod() == "CreateHypothesis" )
182   {
183     myHypos.push_back( _pyHypothesis::NewHypothesis( theCommand ));
184     return;
185   }
186
187   // smeshgen.Compute( mesh, geom ) --> mesh.Compute()
188   if ( theCommand->GetMethod() == "Compute" )
189   {
190     const _pyID& meshID = theCommand->GetArg( 1 );
191     map< _pyID, Handle(_pyMesh) >::iterator id_mesh = myMeshes.find( meshID );
192     if ( id_mesh != myMeshes.end() ) {
193       theCommand->SetObject( meshID );
194       theCommand->RemoveArgs();
195       id_mesh->second->Flush();
196       return;
197     }
198   }
199
200   // smeshgen.Method() --> smesh.smesh.Method()
201   theCommand->SetObject( SMESH_2smeshpy::GenName() );
202 }
203
204 //================================================================================
205 /*!
206  * \brief Convert the remembered commands
207  */
208 //================================================================================
209
210 void _pyGen::Flush()
211 {
212   map< _pyID, Handle(_pyMesh) >::iterator id_mesh = myMeshes.begin();
213   for ( ; id_mesh != myMeshes.end(); ++id_mesh )
214     if ( ! id_mesh->second.IsNull() )
215       id_mesh->second->Flush();
216
217   list< Handle(_pyHypothesis) >::iterator hyp = myHypos.begin();
218   for ( ; hyp != myHypos.end(); ++hyp )
219     if ( !hyp->IsNull() ) {
220       (*hyp)->Flush();
221       // smeshgen.CreateHypothesis() --> smesh.smesh.CreateHypothesis()
222       if ( !(*hyp)->GetCreationCmd()->IsEmpty() )
223         (*hyp)->GetCreationCmd()->SetObject( SMESH_2smeshpy::GenName() );
224     }
225 }
226
227 //================================================================================
228 /*!
229  * \brief Find hypothesis by ID (entry)
230   * \param theHypID - The hypothesis ID
231   * \retval Handle(_pyHypothesis) - The found hypothesis
232  */
233 //================================================================================
234
235 Handle(_pyHypothesis) _pyGen::FindHyp( const _pyID& theHypID )
236 {
237   list< Handle(_pyHypothesis) >::iterator hyp = myHypos.begin();
238   for ( ; hyp != myHypos.end(); ++hyp )
239     if ( !hyp->IsNull() && theHypID == (*hyp)->GetID() )
240       return *hyp;
241   return Handle(_pyHypothesis)();
242 }
243
244 //================================================================================
245 /*!
246  * \brief Find algorithm the created algorithm
247   * \param theGeom - The shape ID the algorithm was created on
248   * \param theMesh - The mesh ID that created the algorithm
249   * \param dim - The algo dimension
250   * \retval Handle(_pyHypothesis) - The found algo
251  */
252 //================================================================================
253
254 Handle(_pyHypothesis) _pyGen::FindAlgo( const _pyID& theGeom, const _pyID& theMesh,
255                                       const TCollection_AsciiString& theAlgoType )
256 {
257   list< Handle(_pyHypothesis) >::iterator hyp = myHypos.begin();
258   for ( ; hyp != myHypos.end(); ++hyp )
259     if ( !hyp->IsNull() &&
260          (*hyp)->IsAlgo() &&
261          (*hyp)->GetType() == theAlgoType &&
262          (*hyp)->GetGeom() == theGeom &&
263          (*hyp)->GetMesh() == theMesh )
264       return *hyp;
265   return 0;
266 }
267
268 //================================================================================
269 /*!
270  * \brief Change order of commands in the script
271   * \param theCmd1 - One command
272   * \param theCmd2 - Another command
273  */
274 //================================================================================
275
276 void _pyGen::ExchangeCommands( Handle(_pyCommand) theCmd1, Handle(_pyCommand) theCmd2 )
277 {
278   list< Handle(_pyCommand) >::iterator pos1, pos2;
279   pos1 = find( myCommands.begin(), myCommands.end(), theCmd1 );
280   pos2 = find( myCommands.begin(), myCommands.end(), theCmd2 );
281   myCommands.insert( pos1, theCmd2 );
282   myCommands.insert( pos2, theCmd1 );
283   myCommands.erase( pos1 );
284   myCommands.erase( pos2 );
285
286   int nb1 = theCmd1->GetOrderNb();
287   theCmd1->SetOrderNb( theCmd2->GetOrderNb() );
288   theCmd2->SetOrderNb( nb1 );
289 }
290 //================================================================================
291 /*!
292  * \brief Set one command after the other
293   * \param theCmd - Command to move
294   * \param theAfterCmd - Command ater which to insert the first one
295  */
296 //================================================================================
297
298 void _pyGen::SetCommandAfter( Handle(_pyCommand) theCmd, Handle(_pyCommand) theAfterCmd )
299 {
300   list< Handle(_pyCommand) >::iterator pos;
301   pos = find( myCommands.begin(), myCommands.end(), theCmd );
302   myCommands.erase( pos );
303   pos = find( myCommands.begin(), myCommands.end(), theAfterCmd );
304   myCommands.insert( ++pos, theCmd );
305
306   int i = 1;
307   for ( pos = myCommands.begin(); pos != myCommands.end(); ++pos)
308     (*pos)->SetOrderNb( i++ );
309 }
310
311 //================================================================================
312 /*!
313  * \brief Set method to access to object wrapped with python class
314   * \param theID - The wrapped object entry
315   * \param theMethod - The accessor method
316  */
317 //================================================================================
318
319 void _pyGen::SetAccessorMethod(const _pyID& theID, const char* theMethod )
320 {
321   myID2AccessorMethod.Bind( theID, (char*) theMethod );
322 }
323
324 //================================================================================
325 /*!
326  * \brief Find out type of geom group
327   * \param grpID - The geom group entry
328   * \retval int - The type
329  */
330 //================================================================================
331
332 static bool sameGroupType( const _pyID&                   grpID,
333                            const TCollection_AsciiString& theType)
334 {
335   // define group type as smesh.Mesh.Group() does
336   int type = -1;
337   SALOMEDS::Study_var study = SMESH_Gen_i::GetSMESHGen()->GetCurrentStudy();
338   SALOMEDS::SObject_var aSObj = study->FindObjectID( grpID.ToCString() );
339   if ( !aSObj->_is_nil() ) {
340     GEOM::GEOM_Object_var aGeomObj = GEOM::GEOM_Object::_narrow( aSObj->GetObject() );
341     if ( !aGeomObj->_is_nil() ) {
342       switch ( aGeomObj->GetShapeType() ) {
343       case GEOM::VERTEX: type = SMESH::NODE; break;
344       case GEOM::EDGE:   type = SMESH::EDGE; break;
345       case GEOM::FACE:   type = SMESH::FACE; break;
346       case GEOM::SOLID:
347       case GEOM::SHELL:  type = SMESH::VOLUME; break;
348       case GEOM::COMPOUND: {
349         GEOM::GEOM_Gen_var aGeomGen = SMESH_Gen_i::GetSMESHGen()->GetGeomEngine();
350         if ( !aGeomGen->_is_nil() ) {
351           GEOM::GEOM_IGroupOperations_var aGrpOp =
352             aGeomGen->GetIGroupOperations( study->StudyId() );
353           if ( !aGrpOp->_is_nil() ) {
354             switch ( aGrpOp->GetType( aGeomObj )) {
355             case TopAbs_VERTEX: type = SMESH::NODE; break;
356             case TopAbs_EDGE:   type = SMESH::EDGE; break;
357             case TopAbs_FACE:   type = SMESH::FACE; break;
358             case TopAbs_SOLID:  type = SMESH::VOLUME; break;
359             default:;
360             }
361           }
362         }
363       }
364       default:;
365       }
366     }
367   }
368   if ( type < 0 ) {
369     MESSAGE("Type of the group " << grpID << " not found");
370     return false;
371   }
372   if ( theType.IsIntegerValue() )
373     return type == theType.IntegerValue();
374
375   switch ( type ) {
376   case SMESH::NODE:   return theType.Location( "NODE", 1, theType.Length() );
377   case SMESH::EDGE:   return theType.Location( "EDGE", 1, theType.Length() );
378   case SMESH::FACE:   return theType.Location( "FACE", 1, theType.Length() );
379   case SMESH::VOLUME: return theType.Location( "VOLUME", 1, theType.Length() );
380   default:;
381   }
382   return false;
383 }
384
385 //================================================================================
386 /*!
387  * \brief 
388   * \param theCreationCmd - 
389  */
390 //================================================================================
391
392 _pyMesh::_pyMesh(const Handle(_pyCommand) theCreationCmd): _pyObject(theCreationCmd)
393 {
394   // convert my creation command
395   Handle(_pyCommand) creationCmd = GetCreationCmd();
396   creationCmd->SetObject( SMESH_2smeshpy::SmeshpyName() );
397   creationCmd->SetMethod( "Mesh" );
398
399   theGen->SetAccessorMethod( GetID(), "GetMesh()" );
400 }
401
402 //================================================================================
403 /*!
404             case brief:
405             default:
406   * \param theCommand - Engine method called for this mesh
407  */
408 //================================================================================
409
410 void _pyMesh::Process( const Handle(_pyCommand)& theCommand )
411 {
412   // smesh.py wraps the following methods:
413   //
414   // 1. GetSubMesh(geom, name) + AddHypothesis(geom, algo)
415   //     --> in Mesh_Algorithm.Create(mesh, geom, hypo, so)
416   // 2. AddHypothesis(geom, hyp)
417   //     --> in Mesh_Algorithm.Hypothesis(hyp, args, so)
418   // 3. CreateGroupFromGEOM(type, name, grp)
419   //     --> in Mesh.Group(grp, name="")
420   // 4. ExportToMED(f, opt, version)
421   //     --> in Mesh.ExportToMED( f, version, opt=0 )
422   // 5. ExportMED(f, opt)
423   //     --> in Mesh.ExportMED( f,opt=0 )
424   // 6. ExportDAT(f)
425   //     --> in Mesh.ExportDAT( f )
426   // 7. ExportUNV(f)
427   //     --> in Mesh.ExportUNV(f)
428   // 8. ExportSTL(f, ascii)
429   //     --> in Mesh.ExportSTL(f, ascii=1)
430
431   const TCollection_AsciiString method = theCommand->GetMethod();
432   if ( method == "GetSubMesh" ) {
433     mySubmeshes.push_back( theCommand );
434   }
435   else if ( method == "AddHypothesis" ) { // mesh.AddHypothesis(geom, HYPO )
436     myAddHypCmds.push_back( theCommand );
437     // set mesh to hypo
438     const _pyID& hypID = theCommand->GetArg( 2 );
439     Handle(_pyHypothesis) hyp = theGen->FindHyp( hypID );
440     if ( !hyp.IsNull() && hyp->GetMesh().IsEmpty() )
441       hyp->SetMesh( this->GetID() );
442   }
443   else if ( method == "CreateGroupFromGEOM" ) {// (type, name, grp)
444     _pyID grp = theCommand->GetArg( 3 );
445     if ( sameGroupType( grp, theCommand->GetArg( 1 )) ) { // --> Group(grp)
446       theCommand->SetMethod( "Group" );
447       theCommand->RemoveArgs();
448       theCommand->SetArg( 1, grp );
449     }
450     else {
451       AddMeshAccess( theCommand );
452     }
453   }
454   else if ( method == "ExportToMED" ) {//(f, opt, version)
455     // --> (f, version, opt)
456     _pyID opt = theCommand->GetArg( 2 );
457     _pyID ver = theCommand->GetArg( 3 );
458     theCommand->SetArg( 2, ver );
459     theCommand->SetArg( 3, opt );
460   }
461   else if ( method == "RemoveHypothesis" ) // (geom, hyp)
462   {
463     const _pyID & hypID = theCommand->GetArg( 2 );
464
465     // check if this mesh still has corresponding addition command
466     bool hasAddCmd = false;
467     list< Handle(_pyCommand) >::iterator cmd = myAddHypCmds.begin();
468     while ( cmd != myAddHypCmds.end() )
469     {
470       // AddHypothesis(geom, hyp)
471       if ( hypID == (*cmd)->GetArg( 2 )) { // erase both commands
472         theCommand->Clear();
473         (*cmd)->Clear();
474         cmd = myAddHypCmds.erase( cmd );
475         hasAddCmd = true;
476       }
477       else {
478         ++cmd;
479       }
480     }
481     if ( ! hasAddCmd ) {
482       // access to wrapped mesh
483       AddMeshAccess( theCommand );
484       // access to wrapped algo
485       Handle(_pyHypothesis) hyp = theGen->FindHyp( hypID );
486       if ( !hyp.IsNull() && hyp->IsAlgo() && hyp->IsWrapped() )
487         theCommand->SetArg( 2, theCommand->GetArg( 2 ) + ".GetAlgorithm()" );
488     }
489   }
490   else { // apply theCommand to the mesh wrapped by smeshpy mesh
491     AddMeshAccess( theCommand );
492   }
493 }
494
495 //================================================================================
496 /*!
497  * \brief Convert creation and addition of all algos and hypos
498  */
499 //================================================================================
500
501 void _pyMesh::Flush()
502 {
503   list < Handle(_pyCommand) >::iterator cmd, cmd2;
504   map< _pyID, Handle(_pyCommand) > algo2additionCmd;
505
506   // try to convert algo addition like this:
507   // mesh.AddHypothesis(geom, ALGO ) --> ALGO = mesh.Algo()
508   for ( cmd = myAddHypCmds.begin(); cmd != myAddHypCmds.end(); ++cmd )
509   {
510     Handle(_pyCommand) addCmd = *cmd;
511     const _pyID& algoID = addCmd->GetArg( 2 );
512     Handle(_pyHypothesis) algo = theGen->FindHyp( algoID );
513     if ( algo.IsNull() || !algo->IsAlgo() )
514       continue;
515     // try to convert
516     _pyID geom = addCmd->GetArg( 1 );
517     if ( algo->Addition2Creation( addCmd, this->GetID() )) // OK
518     {
519       algo2additionCmd[ algo->GetID() ] = addCmd;
520
521       if ( geom != GetGeom() ) // local algo
522       {
523         // mesh.AddHypothesis(geom, ALGO ) --> mesh.AlgoMethod(geom)
524         addCmd->SetArg( addCmd->GetNbArgs() + 1,
525                         TCollection_AsciiString( "geom=" ) + geom );
526         // sm = mesh.GetSubMesh(geom, name) --> sm = ALGO.GetSubMesh()
527         for ( cmd2 = mySubmeshes.begin(); cmd2 != mySubmeshes.end(); ++cmd2 ) {
528           Handle(_pyCommand) subCmd = *cmd2;
529           if ( geom == subCmd->GetArg( 1 )) {
530             subCmd->SetObject( algo->GetID() );
531             subCmd->RemoveArgs();
532             if ( addCmd->GetOrderNb() > subCmd->GetOrderNb() )
533               theGen->SetCommandAfter( subCmd, addCmd );
534           }
535         }
536       }
537     }
538     else // ALGO was already created
539     {
540       // mesh.AddHypothesis(geom, ALGO ) --> mesh.GetMesh().AddHypothesis(geom, ALGO )
541       AddMeshAccess( addCmd );
542       // mesh.GetMesh().AddHypothesis(geom, ALGO ) ->
543       // mesh.GetMesh().AddHypothesis(geom, ALGO.GetAlgorithm() )
544       addCmd->SetArg( 2, addCmd->GetArg( 2 ) + ".GetAlgorithm()" );
545     }
546   }
547
548   // try to convert hypo addition like this:
549   // mesh.AddHypothesis(geom, HYPO ) --> HYPO = algo.Hypo()
550   for ( cmd = myAddHypCmds.begin(); cmd != myAddHypCmds.end(); ++cmd )
551   {
552     Handle(_pyCommand) addCmd = *cmd;
553     const _pyID& hypID = addCmd->GetArg( 2 );
554     Handle(_pyHypothesis) hyp = theGen->FindHyp( hypID );
555     if ( hyp.IsNull() || hyp->IsAlgo() )
556       continue;
557     const _pyID& geom = addCmd->GetArg( 1 );
558     // find algo created on <geom> for this mesh
559     Handle(_pyHypothesis) algo = theGen->FindAlgo( geom, this->GetID(), hyp->GetType() );
560     //_pyID algoID = algo.IsNull() ? "" : algo->GetID();
561     if ( !algo.IsNull() && hyp->Addition2Creation( addCmd, this->GetID() )) // OK
562     {
563       addCmd->SetObject( algo->GetID() );
564       Handle(_pyCommand) algoCmd = algo2additionCmd[ algo->GetID() ];
565       if ( !algoCmd.IsNull() && algoCmd->GetOrderNb() > addCmd->GetOrderNb() )
566         // algo was created later than hyp
567         theGen->ExchangeCommands( algoCmd, addCmd );
568     }
569     else
570     {
571       AddMeshAccess( addCmd );
572     }
573   }
574
575   // sm = mesh.GetSubMesh(geom, name) --> sm = mesh.GetMesh().GetSubMesh(geom, name)
576   for ( cmd = mySubmeshes.begin(); cmd != mySubmeshes.end(); ++cmd ) {
577     Handle(_pyCommand) subCmd = *cmd;
578     if ( subCmd->GetNbArgs() > 0 )
579       AddMeshAccess( subCmd );
580   }
581   myAddHypCmds.clear();
582   mySubmeshes.clear();
583 }
584
585 //================================================================================
586 /*!
587  * \brief _pyHypothesis constructor
588   * \param theCreationCmd - 
589  */
590 //================================================================================
591
592 _pyHypothesis::_pyHypothesis(const Handle(_pyCommand)& theCreationCmd):
593   _pyObject( theCreationCmd )
594 {
595   myDim = myIsAlgo = /*myIsLocal = */myIsWrapped = myIsConverted = false;
596 }
597
598 //================================================================================
599 /*!
600  * \brief Creates algorithm or hypothesis
601   * \param theCreationCmd - The engine command creating a hypothesis
602   * \retval Handle(_pyHypothesis) - Result _pyHypothesis
603  */
604 //================================================================================
605
606 Handle(_pyHypothesis) _pyHypothesis::NewHypothesis( const Handle(_pyCommand)& theCreationCmd)
607 {
608   // theCreationCmd: CreateHypothesis( "theHypType", "theLibName" )
609   ASSERT (( theCreationCmd->GetMethod() == "CreateHypothesis"));
610
611   Handle(_pyHypothesis) hyp, algo;
612
613   // "theHypType"
614   const TCollection_AsciiString & hypTypeWithQuotes = theCreationCmd->GetArg( 1 );
615   if ( hypTypeWithQuotes.IsEmpty() )
616     return hyp;
617   // theHypType
618   TCollection_AsciiString  hypType =
619     hypTypeWithQuotes.SubString( 2, hypTypeWithQuotes.Length() - 1 );
620
621   algo = new _pyAlgorithm( theCreationCmd );
622   hyp  = new _pyHypothesis( theCreationCmd );
623
624   // 1D Regular_1D ----------
625   if ( hypType == "Regular_1D" ) {
626     algo->myDim = 1;
627     algo->myCreationMethod = "Segment";
628   }
629   else if ( hypType == "LocalLength" ) {
630     hyp->myDim = 1;
631     hyp->myCreationMethod = "LocalLength";
632     hyp->myType = "Regular_1D";
633     hyp->myArgMethods.Append( "SetLength" );
634   }
635   else if ( hypType == "NumberOfSegments" ) {
636     hyp = new _pyNumberOfSegmentsHyp( theCreationCmd );
637     hyp->myDim = 1;
638     hyp->myCreationMethod = "NumberOfSegments";
639     hyp->myType = "Regular_1D";
640     hyp->myArgMethods.Append( "SetNumberOfSegments" );
641     hyp->myArgMethods.Append( "SetScaleFactor" );
642   }
643   else if ( hypType == "Arithmetic1D" ) {
644     hyp = new _pyComplexParamHypo( theCreationCmd );
645     hyp->myDim = 1;
646     hyp->myCreationMethod = "Arithmetic1D";
647     hyp->myType = "Regular_1D";
648   }
649   else if ( hypType == "StartEndLength" ) {
650     hyp = new _pyComplexParamHypo( theCreationCmd );
651     hyp->myDim = 1;
652     hyp->myCreationMethod = "StartEndLength";
653     hyp->myType = "Regular_1D";
654   }
655   else if ( hypType == "Deflection1D" ) {
656     hyp->myDim = 1;
657     hyp->myCreationMethod = "Deflection1D";
658     hyp->myArgMethods.Append( "SetDeflection" );
659     hyp->myType = "Regular_1D";
660   }
661   else if ( hypType == "Propagation" ) {
662     hyp->myDim = 1;
663     hyp->myCreationMethod = "Propagation";
664     hyp->myType = "Regular_1D";
665   }
666   else if ( hypType == "AutomaticLength" ) {
667     hyp->myDim = 1;
668     hyp->myCreationMethod = "AutomaticLength";
669     hyp->myType = "Regular_1D";
670     hyp->myArgMethods.Append( "SetFineness");
671   }
672   // 1D Python_1D ----------
673   else if ( hypType == "Python_1D" ) {
674     algo->myDim = 1;
675     algo->myCreationMethod = "Segment";
676     algo->myArgs.Append( "algo=smesh.PYTHON");
677   }
678   else if ( hypType == "PythonSplit1D" ) {
679     hyp->myDim = 1;
680     hyp->myCreationMethod = "PythonSplit1D";
681     hyp->myType = "Python_1D";
682     hyp->myArgMethods.Append( "SetNumberOfSegments");
683     hyp->myArgMethods.Append( "SetPythonLog10RatioFunction");
684   }
685   // 2D ----------
686   else if ( hypType == "MEFISTO_2D" ) {
687     algo->myDim = 2;
688     algo->myCreationMethod = "Triangle";
689   }
690   else if ( hypType == "MaxElementArea" ) {
691     hyp->myDim = 2;
692     hyp->myCreationMethod = "MaxElementArea";
693     hyp->myType = "MEFISTO_2D";
694     hyp->myArgMethods.Append( "SetMaxElementArea");
695   }
696   else if ( hypType == "LengthFromEdges" ) {
697     hyp->myDim = 2;
698     hyp->myCreationMethod = "LengthFromEdges";
699     hyp->myType = "MEFISTO_2D";
700   }
701   else if ( hypType == "Quadrangle_2D" ) {
702     algo->myDim = 2;
703     algo->myCreationMethod = "Quadrangle";
704   }
705   else if ( hypType == "QuadranglePreference" ) {
706     hyp->myDim = 2;
707     hyp->myCreationMethod = "QuadranglePreference";
708     hyp->myType = "Quadrangle_2D";
709   }
710   // 3D ----------
711   else if ( hypType == "NETGEN_3D") {
712     algo->myDim = 3;
713     algo->myCreationMethod = "Tetrahedron";
714     algo->myArgs.Append( "algo=smesh.NETGEN" );
715   }
716   else if ( hypType == "MaxElementVolume") {
717     hyp->myDim = 3;
718     hyp->myCreationMethod = "MaxElementVolume";
719     hyp->myType = "NETGEN_3D";
720     hyp->myArgMethods.Append( "SetMaxElementVolume" );
721   }
722   else if ( hypType == "GHS3D_3D" ) {
723     algo->myDim = 3;
724     algo->myCreationMethod = "Tetrahedron";
725     algo->myArgs.Append( "algo=smesh.GHS3D" );
726   }
727   else if ( hypType == "Hexa_3D" ) {
728     algo->myDim = 3;
729     algo->myCreationMethod = "Hexahedron";
730   }
731
732   if ( algo->GetDim() ) {
733     algo->myType = hypType;
734     return algo;
735   }
736   return hyp;
737 }
738
739 //================================================================================
740 /*!
741  * \brief Convert the command adding a hypothesis to mesh into a smesh command
742   * \param theCmd - The command like mesh.AddHypothesis( geom, hypo )
743   * \param theAlgo - The algo that can create this hypo
744   * \retval bool - false if the command cant be converted
745  */
746 //================================================================================
747
748 bool _pyHypothesis::Addition2Creation( const Handle(_pyCommand)& theCmd,
749                                        const _pyID&              theMesh)
750 {
751   ASSERT(( theCmd->GetMethod() == "AddHypothesis" ));
752
753   if ( !IsWrappable( theMesh ))
754     return false;
755
756   myIsWrapped = true;
757
758   if ( myIsWrapped )
759   {
760     // mesh.AddHypothesis(geom,hyp) --> hyp = theMesh.myCreationMethod(args)
761     theCmd->SetResultValue( GetID() );
762     theCmd->SetObject( theMesh );
763     theCmd->SetMethod( myCreationMethod );
764     // set args
765     theCmd->RemoveArgs();
766     for ( int i = 1; i <= myArgs.Length(); ++i ) {
767       if ( !myArgs( i ).IsEmpty() )
768         theCmd->SetArg( i, myArgs( i ));
769       else
770         theCmd->SetArg( i, "[]");
771     }
772
773     // clear commands setting arg values
774     list < Handle(_pyCommand) >::iterator argCmd = myArgCommands.begin();
775     for ( ; argCmd != myArgCommands.end(); ++argCmd )
776       (*argCmd)->Clear();
777   }
778   else
779   {
780 //     // set arg commands after hypo creation
781 //     list<Handle(_pyCommand)>::iterator argCmd = myArgCommands.begin();
782 //     for ( ; argCmd != myArgCommands.end(); ++argCmd )
783 //       if ( !(*argCmd)->IsEmpty() && GetCommandNb() > (*argCmd)->GetOrderNb() )
784 //         theGen->ExchangeCommands( GetCreationCmd(), *argCmd );
785   }
786
787   // set unknown arg commands after hypo creation
788   Handle(_pyCommand) afterCmd = myIsWrapped ? theCmd : GetCreationCmd();
789   list<Handle(_pyCommand)>::iterator cmd = myUnknownCommands.begin();
790   for ( ; cmd != myUnknownCommands.end(); ++cmd ) {
791     if ( !(*cmd)->IsEmpty() && afterCmd->GetOrderNb() > (*cmd)->GetOrderNb() ) {
792       theGen->SetCommandAfter( *cmd, afterCmd );
793       afterCmd = *cmd;
794     }
795   }
796   myArgCommands.clear();
797   myUnknownCommands.clear();
798
799   return myIsWrapped;
800 }
801
802 //================================================================================
803 /*!
804  * \brief Remember hypothesis parameter values
805  * \param theCommand - The called hypothesis method
806  */
807 //================================================================================
808
809 void _pyHypothesis::Process( const Handle(_pyCommand)& theCommand)
810 {
811   ASSERT( !myIsAlgo );
812   // set args
813   for ( int i = 1; i <= myArgMethods.Length(); ++i ) {
814     if ( myArgMethods( i ) == theCommand->GetMethod() ) {
815       while ( myArgs.Length() < i )
816         myArgs.Append( "[]" );
817       myArgs( i ) = theCommand->GetArg( 1 ); // arg value
818       myArgCommands.push_back( theCommand );
819       return;
820     }
821   }
822   myUnknownCommands.push_back( theCommand );
823 }
824
825 //================================================================================
826 /*!
827  * \brief Finish conversion
828  */
829 //================================================================================
830
831 void _pyHypothesis::Flush()
832 {
833   if ( IsWrapped() )
834     GetCreationCmd()->Clear();
835 }
836
837 //================================================================================
838 /*!
839  * \brief Remember hypothesis parameter values
840   * \param theCommand - The called hypothesis method
841  */
842 //================================================================================
843
844 void _pyComplexParamHypo::Process( const Handle(_pyCommand)& theCommand)
845 {
846   // ex: hyp.SetLength(start, 1)
847   //     hyp.SetLength(end,   0)
848   ASSERT(( theCommand->GetMethod() == "SetLength" ));
849   ASSERT(( theCommand->GetArg( 2 ).IsIntegerValue() ));
850   int i = 2 - theCommand->GetArg( 2 ).IntegerValue();
851   while ( myArgs.Length() < i )
852     myArgs.Append( "[]" );
853   myArgs( i ) = theCommand->GetArg( 1 ); // arg value
854   myArgCommands.push_back( theCommand );
855 }
856
857 //================================================================================
858 /*!
859  * \brief additionally to Addition2Creation, clears SetDistrType() command
860   * \param theCmd - AddHypothesis() command
861   * \param theMesh - mesh to which a hypothesis is added
862   * \retval bool - convertion result
863  */
864 //================================================================================
865
866 bool _pyNumberOfSegmentsHyp::Addition2Creation( const Handle(_pyCommand)& theCmd,
867                                                 const _pyID&              theMesh)
868 {
869   if ( IsWrappable( theMesh ) && myArgs.Length() > 1 ) {
870     list<Handle(_pyCommand)>::iterator cmd = myUnknownCommands.begin();
871     for ( ; cmd != myUnknownCommands.end(); ++cmd ) {
872       // clear SetDistrType()
873       if ( (*cmd)->GetString().Location( "SetDistrType", 1, (*cmd)->Length() ))
874         (*cmd)->Clear();
875     }
876   }
877   return _pyHypothesis::Addition2Creation( theCmd, theMesh );
878 }
879
880 //================================================================================
881 /*!
882  * \brief _pyAlgorithm constructor
883   * \param theCreationCmd - The command like "algo = smeshgen.CreateHypothesis(type,lib)"
884  */
885 //================================================================================
886
887 _pyAlgorithm::_pyAlgorithm(const Handle(_pyCommand)& theCreationCmd)
888   : _pyHypothesis( theCreationCmd )
889 {
890   myIsAlgo = true;
891 }
892
893 //================================================================================
894 /*!
895  * \brief Convert the command adding an algorithm to mesh
896   * \param theCmd - The command like mesh.AddHypothesis( geom, algo )
897   * \param theMesh - The mesh needing this algo 
898   * \retval bool - false if the command cant be converted
899  */
900 //================================================================================
901   
902 bool _pyAlgorithm::Addition2Creation( const Handle(_pyCommand)& theCmd,
903                                       const _pyID&              theMeshID)
904 {
905   if ( IsWrappable( theMeshID )) {
906
907     myGeom = theCmd->GetArg( 1 );
908
909     // mesh.AddHypothesis(geom,algo) --> theMeshID.myCreationMethod()
910     if ( _pyHypothesis::Addition2Creation( theCmd, theMeshID )) {
911       theGen->SetAccessorMethod( GetID(), "GetAlgorithm()" );
912       return true;
913     }
914   }
915   return false;
916 }
917
918 //================================================================================
919 /*!
920  * \brief Return starting position of a part of python command
921   * \param thePartIndex - The index of command part
922   * \retval int - Part position
923  */
924 //================================================================================
925
926 int _pyCommand::GetBegPos( int thePartIndex )
927 {
928   if ( IsEmpty() )
929     return EMPTY;
930   if ( myBegPos.Length() < thePartIndex )
931     return UNKNOWN;
932   return myBegPos( thePartIndex );
933 }
934
935 //================================================================================
936 /*!
937  * \brief Store starting position of a part of python command
938   * \param thePartIndex - The index of command part
939   * \param thePosition - Part position
940  */
941 //================================================================================
942
943 void _pyCommand::SetBegPos( int thePartIndex, int thePosition )
944 {
945   while ( myBegPos.Length() < thePartIndex )
946     myBegPos.Append( UNKNOWN );
947   myBegPos( thePartIndex ) = thePosition;
948 }
949
950 //================================================================================
951 /*!
952  * \brief Return substring of python command looking like ResultValue = Obj.Meth()
953   * \retval const TCollection_AsciiString & - ResultValue substring
954  */
955 //================================================================================
956
957 const TCollection_AsciiString & _pyCommand::GetResultValue()
958 {
959   if ( GetBegPos( RESULT_IND ) == UNKNOWN )
960   {
961     int begPos = myString.Location( "=", 1, Length() );
962     if ( begPos )
963       myRes = GetWord( myString, begPos, false );
964     else
965       begPos = EMPTY;
966     SetBegPos( RESULT_IND, begPos );
967   }
968   return myRes;
969 }
970
971 //================================================================================
972 /*!
973  * \brief Return substring of python command looking like ResVal = Object.Meth()
974   * \retval const TCollection_AsciiString & - Object substring
975  */
976 //================================================================================
977
978 const TCollection_AsciiString & _pyCommand::GetObject()
979 {
980   if ( GetBegPos( OBJECT_IND ) == UNKNOWN )
981   {
982     // beginning
983     int begPos = GetBegPos( RESULT_IND ) + myRes.Length();
984     if ( begPos < 1 )
985       begPos = myString.Location( "=", 1, Length() ) + 1;
986     // store
987     myObj = GetWord( myString, begPos, true );
988     SetBegPos( OBJECT_IND, begPos );
989   }
990   //SCRUTE(myObj);
991   return myObj;
992 }
993
994 //================================================================================
995 /*!
996  * \brief Return substring of python command looking like ResVal = Obj.Method()
997   * \retval const TCollection_AsciiString & - Method substring
998  */
999 //================================================================================
1000
1001 const TCollection_AsciiString & _pyCommand::GetMethod()
1002 {
1003   if ( GetBegPos( METHOD_IND ) == UNKNOWN )
1004   {
1005     // beginning
1006     int begPos = GetBegPos( OBJECT_IND ) + myObj.Length();
1007     bool forward = true;
1008     if ( begPos < 1 ) {
1009       begPos = myString.Location( "(", 1, Length() ) - 1;
1010       forward = false;
1011     }
1012     // store
1013     myMeth = GetWord( myString, begPos, forward );
1014     SetBegPos( METHOD_IND, begPos );
1015   }
1016   //SCRUTE(myMeth);
1017   return myMeth;
1018 }
1019
1020 //================================================================================
1021 /*!
1022  * \brief Return substring of python command looking like ResVal = Obj.Meth(Arg1,...)
1023   * \retval const TCollection_AsciiString & - Arg<index> substring
1024  */
1025 //================================================================================
1026
1027 const TCollection_AsciiString & _pyCommand::GetArg( int index )
1028 {
1029   if ( GetBegPos( ARG1_IND ) == UNKNOWN )
1030   {
1031     // find all args
1032     int begPos = GetBegPos( METHOD_IND ) + myMeth.Length();
1033     if ( begPos < 1 )
1034       begPos = myString.Location( "(", 1, Length() ) + 1;
1035
1036     int i = 0, prevLen = 0;
1037     while ( begPos != EMPTY ) {
1038       begPos += prevLen;
1039       // check if we are looking at the closing parenthesis
1040       while ( begPos <= Length() && isspace( myString.Value( begPos )))
1041         ++begPos;
1042       if ( begPos > Length() || myString.Value( begPos ) == ')' )
1043         break;
1044       myArgs.Append( GetWord( myString, begPos, true, true ));
1045       SetBegPos( ARG1_IND + i, begPos );
1046       prevLen = myArgs.Last().Length();
1047       if ( prevLen == 0 )
1048         myArgs.Remove( myArgs.Length() ); // no more args
1049       i++;
1050     }
1051   }
1052   if ( myArgs.Length() < index )
1053     return theEmptyString;
1054   return myArgs( index );
1055 }
1056
1057 //================================================================================
1058 /*!
1059  * \brief Check if char is a word part
1060   * \param c - The character to check
1061   * \retval bool - The check result
1062  */
1063 //================================================================================
1064
1065 static inline bool isWord(const char c, const bool dotIsWord)
1066 {
1067   return
1068     !isspace(c) && c != ',' && c != '=' && c != ')' && c != '(' && ( dotIsWord || c != '.');
1069 }
1070
1071 //================================================================================
1072 /*!
1073  * \brief Looks for a word in the string and returns word's beginning
1074   * \param theString - The input string
1075   * \param theStartPos - The position to start the search, returning word's beginning
1076   * \param theForward - The search direction
1077   * \retval TCollection_AsciiString - The found word
1078  */
1079 //================================================================================
1080
1081 TCollection_AsciiString _pyCommand::GetWord( const TCollection_AsciiString & theString,
1082                                             int &      theStartPos,
1083                                             const bool theForward,
1084                                             const bool dotIsWord )
1085 {
1086   int beg = theStartPos, end = theStartPos;
1087   theStartPos = EMPTY;
1088   if ( beg < 1 || beg > theString.Length() )
1089     return theEmptyString;
1090
1091   if ( theForward ) { // search forward
1092     // beg
1093     while ( beg <= theString.Length() && !isWord( theString.Value( beg ), dotIsWord))
1094       ++beg;
1095     if ( beg > theString.Length() )
1096       return theEmptyString; // no word found
1097     // end
1098     end = beg + 1;
1099     while ( end <= theString.Length() && isWord( theString.Value( end ), dotIsWord))
1100       ++end;
1101     --end;
1102   }
1103   else {  // search backward
1104     // end
1105     while ( end > 0 && !isWord( theString.Value( end ), dotIsWord))
1106       --end;
1107     if ( end == 0 )
1108       return theEmptyString; // no word found
1109     beg = end - 1;
1110     while ( beg > 0 && isWord( theString.Value( beg ), dotIsWord))
1111       --beg;
1112     ++beg;
1113   }
1114   theStartPos = beg;
1115   //cout << theString << " ---- " << beg << " - " << end << endl;
1116   return theString.SubString( beg, end );
1117 }
1118
1119 //================================================================================
1120 /*!
1121  * \brief Look for position where not space char is
1122   * \param theString - The string 
1123   * \param thePos - The position to search from and which returns result
1124   * \retval bool - false if there are only space after thePos in theString
1125  * 
1126  * 
1127  */
1128 //================================================================================
1129
1130 bool _pyCommand::SkipSpaces( const TCollection_AsciiString & theString, int & thePos )
1131 {
1132   if ( thePos < 1 || thePos > theString.Length() )
1133     return false;
1134
1135   while ( thePos <= theString.Length() && isspace( theString.Value( thePos )))
1136     ++thePos;
1137
1138   return thePos <= theString.Length();
1139 }
1140
1141 //================================================================================
1142 /*!
1143  * \brief Modify a part of the command
1144   * \param thePartIndex - The index of the part
1145   * \param thePart - The new part string
1146   * \param theOldPart - The old part
1147  */
1148 //================================================================================
1149
1150 void _pyCommand::SetPart(int thePartIndex, const TCollection_AsciiString& thePart,
1151                         TCollection_AsciiString& theOldPart)
1152 {
1153   int pos = GetBegPos( thePartIndex );
1154   if ( pos <= Length() && theOldPart != thePart)
1155   {
1156     TCollection_AsciiString seperator;
1157     if ( pos < 1 ) {
1158       pos = GetBegPos( thePartIndex + 1 );
1159       if ( pos < 1 ) return;
1160       switch ( thePartIndex ) {
1161       case RESULT_IND: seperator = " = "; break;
1162       case OBJECT_IND: seperator = "."; break;
1163       case METHOD_IND: seperator = "()"; break;
1164       default:;
1165       }
1166     }      
1167     myString.Remove( pos, theOldPart.Length() );
1168     if ( !seperator.IsEmpty() )
1169       myString.Insert( pos , seperator );
1170     myString.Insert( pos, thePart );
1171     // update starting positions of the following parts
1172     int posDelta = thePart.Length() + seperator.Length() - theOldPart.Length();
1173     for ( int i = thePartIndex + 1; i <= myBegPos.Length(); ++i ) {
1174       if ( myBegPos( i ) > 0 )
1175         myBegPos( i ) += posDelta;
1176     }
1177     theOldPart = thePart;
1178   }
1179 }
1180
1181 //================================================================================
1182 /*!
1183  * \brief Set agrument
1184   * \param index - The argument index, it counts from 1
1185   * \param theArg - The argument string
1186  */
1187 //================================================================================
1188
1189 void _pyCommand::SetArg( int index, const TCollection_AsciiString& theArg)
1190 {
1191   FindAllArgs();
1192   int argInd = ARG1_IND + index - 1;
1193   int pos = GetBegPos( argInd );
1194   if ( pos < 1 ) // no index-th arg exist, append inexistent args
1195   {
1196     // find a closing parenthesis
1197     pos = Length();
1198     while ( pos > 0 && myString.Value( pos ) != ')' )
1199       --pos;
1200     if ( pos == 0 ) { // no parentheses at all
1201       myString += "()";
1202       pos = Length();
1203     }
1204     while ( myArgs.Length() < index ) {
1205       if ( myArgs.Length() )
1206         myString.Insert( pos++, "," );
1207       myArgs.Append("None");
1208       myString.Insert( pos, myArgs.Last() );
1209       SetBegPos( ARG1_IND + myArgs.Length() - 1, pos );
1210       pos += myArgs.Last().Length();
1211     }
1212   }
1213   SetPart( argInd, theArg, myArgs( index ));
1214 }
1215
1216 //================================================================================
1217 /*!
1218  * \brief Empty arg list
1219  */
1220 //================================================================================
1221
1222 void _pyCommand::RemoveArgs()
1223 {
1224   if ( int pos = myString.Location( '(', 1, Length() ))
1225     myString.Trunc( pos );
1226   myString += ")";
1227   myArgs.Clear();
1228   if ( myBegPos.Length() >= ARG1_IND )
1229     myBegPos.Remove( ARG1_IND, myBegPos.Length() );
1230 }