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