1 // Copyright (C) 2007-2010 CEA/DEN, EDF R&D, OPEN CASCADE
3 // Copyright (C) 2003-2007 OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
6 // This library is free software; you can redistribute it and/or
7 // modify it under the terms of the GNU Lesser General Public
8 // License as published by the Free Software Foundation; either
9 // version 2.1 of the License.
11 // This library is distributed in the hope that it will be useful,
12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 // Lesser General Public License for more details.
16 // You should have received a copy of the GNU Lesser General Public
17 // License along with this library; if not, write to the Free Software
18 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
23 // SMESH SMESH_I : idl implementation based on 'SMESH' unit's calsses
24 // File : SMESH_2smeshpy.cxx
25 // Created : Fri Nov 18 13:20:10 2005
26 // Author : Edward AGAPOV (eap)
28 #include "SMESH_2smeshpy.hxx"
30 #include "utilities.h"
31 #include "SMESH_PythonDump.hxx"
32 #include "SMESH_NoteBook.hxx"
33 #include "Resource_DataMapOfAsciiStringAsciiString.hxx"
35 #include "SMESH_Gen_i.hxx"
36 /* SALOME headers that include CORBA headers that include windows.h
37 * that defines GetObject symbol as GetObjectA should stand before SALOME headers
38 * that declare methods named GetObject - to apply the same rules of GetObject renaming
39 * and thus to avoid mess with GetObject symbol on Windows */
41 IMPLEMENT_STANDARD_HANDLE (_pyObject ,Standard_Transient);
42 IMPLEMENT_STANDARD_HANDLE (_pyCommand ,Standard_Transient);
43 IMPLEMENT_STANDARD_HANDLE (_pyGen ,_pyObject);
44 IMPLEMENT_STANDARD_HANDLE (_pyMesh ,_pyObject);
45 IMPLEMENT_STANDARD_HANDLE (_pySubMesh ,_pyObject);
46 IMPLEMENT_STANDARD_HANDLE (_pyMeshEditor ,_pyObject);
47 IMPLEMENT_STANDARD_HANDLE (_pyHypothesis ,_pyObject);
48 IMPLEMENT_STANDARD_HANDLE (_pySelfEraser ,_pyObject);
49 IMPLEMENT_STANDARD_HANDLE (_pyAlgorithm ,_pyHypothesis);
50 IMPLEMENT_STANDARD_HANDLE (_pyComplexParamHypo,_pyHypothesis);
51 IMPLEMENT_STANDARD_HANDLE (_pyNumberOfSegmentsHyp,_pyHypothesis);
53 IMPLEMENT_STANDARD_RTTIEXT(_pyObject ,Standard_Transient);
54 IMPLEMENT_STANDARD_RTTIEXT(_pyCommand ,Standard_Transient);
55 IMPLEMENT_STANDARD_RTTIEXT(_pyGen ,_pyObject);
56 IMPLEMENT_STANDARD_RTTIEXT(_pyMesh ,_pyObject);
57 IMPLEMENT_STANDARD_RTTIEXT(_pySubMesh ,_pyObject);
58 IMPLEMENT_STANDARD_RTTIEXT(_pyMeshEditor ,_pyObject);
59 IMPLEMENT_STANDARD_RTTIEXT(_pyHypothesis ,_pyObject);
60 IMPLEMENT_STANDARD_RTTIEXT(_pySelfEraser ,_pyObject);
61 IMPLEMENT_STANDARD_RTTIEXT(_pyAlgorithm ,_pyHypothesis);
62 IMPLEMENT_STANDARD_RTTIEXT(_pyComplexParamHypo,_pyHypothesis);
63 IMPLEMENT_STANDARD_RTTIEXT(_pyNumberOfSegmentsHyp,_pyHypothesis);
64 IMPLEMENT_STANDARD_RTTIEXT(_pyLayerDistributionHypo,_pyHypothesis);
65 IMPLEMENT_STANDARD_RTTIEXT(_pySegmentLengthAroundVertexHyp,_pyHypothesis);
68 using SMESH::TPythonDump;
71 * \brief Container of commands into which the initial script is split.
72 * It also contains data coresponding to SMESH_Gen contents
74 static Handle(_pyGen) theGen;
76 static TCollection_AsciiString theEmptyString;
78 //#define DUMP_CONVERSION
80 #if !defined(_DEBUG_) && defined(DUMP_CONVERSION)
81 #undef DUMP_CONVERSION
87 //================================================================================
89 * \brief Set of TCollection_AsciiString initialized by C array of C strings
91 //================================================================================
93 struct TStringSet: public set<TCollection_AsciiString>
96 * \brief Filling. The last string must be ""
98 void Insert(const char* names[]) {
99 for ( int i = 0; names[i][0] ; ++i )
100 insert( (char*) names[i] );
103 * \brief Check if a string is in
105 bool Contains(const TCollection_AsciiString& name ) {
106 return find( name ) != end();
111 //================================================================================
113 * \brief Convert python script using commands of smesh.py
114 * \param theScript - Input script
115 * \retval TCollection_AsciiString - Convertion result
117 * Class SMESH_2smeshpy declared in SMESH_PythonDump.hxx
119 //================================================================================
121 TCollection_AsciiString
122 SMESH_2smeshpy::ConvertScript(const TCollection_AsciiString& theScript,
123 Resource_DataMapOfAsciiStringAsciiString& theEntry2AccessorMethod,
124 Resource_DataMapOfAsciiStringAsciiString& theObjectNames)
126 theGen = new _pyGen( theEntry2AccessorMethod, theObjectNames );
128 // split theScript into separate commands
130 SMESH_NoteBook * aNoteBook = new SMESH_NoteBook();
132 int from = 1, end = theScript.Length(), to;
133 while ( from < end && ( to = theScript.Location( "\n", from, end )))
136 // cut out and store a command
137 aNoteBook->AddCommand( theScript.SubString( from, to - 1 ));
141 aNoteBook->ReplaceVariables();
143 TCollection_AsciiString aNoteScript = aNoteBook->GetResultScript();
147 // split theScript into separate commands
148 from = 1, end = aNoteScript.Length();
149 while ( from < end && ( to = aNoteScript.Location( "\n", from, end )))
152 // cut out and store a command
153 theGen->AddCommand( aNoteScript.SubString( from, to - 1 ));
159 #ifdef DUMP_CONVERSION
160 MESSAGE_BEGIN ( std::endl << " ######## RESULT ######## " << std::endl<< std::endl );
163 // reorder commands after conversion
164 list< Handle(_pyCommand) >::iterator cmd;
167 orderChanges = false;
168 for ( cmd = theGen->GetCommands().begin(); cmd != theGen->GetCommands().end(); ++cmd )
169 if ( (*cmd)->SetDependentCmdsAfter() )
171 } while ( orderChanges );
173 // concat commands back into a script
174 TCollection_AsciiString aScript;
175 for ( cmd = theGen->GetCommands().begin(); cmd != theGen->GetCommands().end(); ++cmd )
177 #ifdef DUMP_CONVERSION
178 MESSAGE_ADD ( "## COM " << (*cmd)->GetOrderNb() << ": "<< (*cmd)->GetString() << std::endl );
180 if ( !(*cmd)->IsEmpty() ) {
182 aScript += (*cmd)->GetString();
192 //================================================================================
194 * \brief _pyGen constructor
196 //================================================================================
198 _pyGen::_pyGen(Resource_DataMapOfAsciiStringAsciiString& theEntry2AccessorMethod,
199 Resource_DataMapOfAsciiStringAsciiString& theObjectNames)
200 : _pyObject( new _pyCommand( TPythonDump::SMESHGenName(), 0 )),
201 myID2AccessorMethod( theEntry2AccessorMethod ),
202 myObjectNames( theObjectNames )
205 // make that GetID() to return TPythonDump::SMESHGenName()
206 GetCreationCmd()->GetString() += "=";
209 //================================================================================
211 * \brief name of SMESH_Gen in smesh.py
213 //================================================================================
215 const char* _pyGen::AccessorMethod() const
217 return SMESH_2smeshpy::GenName();
220 //================================================================================
222 * \brief Convert a command using a specific converter
223 * \param theCommand - the command to convert
225 //================================================================================
227 Handle(_pyCommand) _pyGen::AddCommand( const TCollection_AsciiString& theCommand)
229 // store theCommand in the sequence
230 myCommands.push_back( new _pyCommand( theCommand, ++myNbCommands ));
232 Handle(_pyCommand) aCommand = myCommands.back();
233 #ifdef DUMP_CONVERSION
234 MESSAGE ( "## COM " << myNbCommands << ": "<< aCommand->GetString() );
237 _pyID objID = aCommand->GetObject();
239 if ( objID.IsEmpty() )
243 if ( objID == this->GetID() || objID == SMESH_2smeshpy::GenName()) {
244 this->Process( aCommand );
248 // SMESH_Mesh method?
249 map< _pyID, Handle(_pyMesh) >::iterator id_mesh = myMeshes.find( objID );
250 if ( id_mesh != myMeshes.end() ) {
251 // check for mesh editor object
252 if ( aCommand->GetMethod() == "GetMeshEditor" ) { // MeshEditor creation
253 _pyID editorID = aCommand->GetResultValue();
254 Handle(_pyMeshEditor) editor = new _pyMeshEditor( aCommand );
255 myMeshEditors.insert( make_pair( editorID, editor ));
258 // check for SubMesh objects
259 else if ( aCommand->GetMethod() == "GetSubMesh" ) { // SubMesh creation
260 _pyID subMeshID = aCommand->GetResultValue();
261 Handle(_pySubMesh) subMesh = new _pySubMesh( aCommand );
262 myObjects.insert( make_pair( subMeshID, subMesh ));
265 id_mesh->second->Process( aCommand );
269 // SMESH_MeshEditor method?
270 map< _pyID, Handle(_pyMeshEditor) >::iterator id_editor = myMeshEditors.find( objID );
271 if ( id_editor != myMeshEditors.end() ) {
272 id_editor->second->Process( aCommand );
273 TCollection_AsciiString processedCommand = aCommand->GetString();
274 // some commands of SMESH_MeshEditor create meshes
275 if ( aCommand->GetMethod().Search("MakeMesh") != -1 ) {
276 Handle(_pyMesh) mesh = new _pyMesh( aCommand, aCommand->GetResultValue() );
277 aCommand->GetString() = processedCommand; // discard changes made by _pyMesh
278 myMeshes.insert( make_pair( mesh->GetID(), mesh ));
282 // SMESH_Hypothesis method?
283 list< Handle(_pyHypothesis) >::iterator hyp = myHypos.begin();
284 for ( ; hyp != myHypos.end(); ++hyp )
285 if ( !(*hyp)->IsAlgo() && objID == (*hyp)->GetID() ) {
286 (*hyp)->Process( aCommand );
290 // other object method?
291 map< _pyID, Handle(_pyObject) >::iterator id_obj = myObjects.find( objID );
292 if ( id_obj != myObjects.end() ) {
293 id_obj->second->Process( aCommand );
296 // if ( theCommand.Search( "aFilterManager" ) != -1 ) {
297 // if ( theCommand.Search( "CreateFilterManager" ) != -1 )
298 // myFilterManager = new _pySelfEraser( aCommand );
299 // else if ( !myFilterManager.IsNull() )
300 // myFilterManager->Process( aCommand );
304 // Add access to a wrapped mesh
305 AddMeshAccessorMethod( aCommand );
307 // Add access to a wrapped algorithm
308 // AddAlgoAccessorMethod( aCommand ); // ??? what if algo won't be wrapped at all ???
310 // PAL12227. PythonDump was not updated at proper time; result is
311 // aCriteria.append(SMESH.Filter.Criterion(17,26,0,'L1',26,25,1e-07,SMESH.EDGE,-1))
312 // TypeError: __init__() takes exactly 11 arguments (10 given)
313 char wrongCommand[] = "SMESH.Filter.Criterion(";
314 if ( int beg = theCommand.Location( wrongCommand, 1, theCommand.Length() ))
316 _pyCommand tmpCmd( theCommand.SubString( beg, theCommand.Length() ), -1);
317 // there must be 10 arguments, 5-th arg ThresholdID is missing,
318 const int wrongNbArgs = 9, missingArg = 5;
319 if ( tmpCmd.GetNbArgs() == wrongNbArgs )
321 for ( int i = wrongNbArgs; i > missingArg; --i )
322 tmpCmd.SetArg( i + 1, tmpCmd.GetArg( i ));
323 tmpCmd.SetArg( missingArg, "''");
324 aCommand->GetString().Trunc( beg - 1 );
325 aCommand->GetString() += tmpCmd.GetString();
331 //================================================================================
333 * \brief Convert the command or remember it for later conversion
334 * \param theCommand - The python command calling a method of SMESH_Gen
336 //================================================================================
338 void _pyGen::Process( const Handle(_pyCommand)& theCommand )
340 // there are methods to convert:
341 // CreateMesh( shape )
342 // Concatenate( [mesh1, ...], ... )
343 // CreateHypothesis( theHypType, theLibName )
344 // Compute( mesh, geom )
346 TCollection_AsciiString method = theCommand->GetMethod();
348 if ( method == "CreateMesh" || method == "CreateEmptyMesh")
350 Handle(_pyMesh) mesh = new _pyMesh( theCommand );
351 myMeshes.insert( make_pair( mesh->GetID(), mesh ));
354 if ( method == "CreateMeshesFromUNV" || method == "CreateMeshesFromSTL")
356 Handle(_pyMesh) mesh = new _pyMesh( theCommand, theCommand->GetResultValue() );
357 myMeshes.insert( make_pair( mesh->GetID(), mesh ));
360 if( method == "CreateMeshesFromMED")
362 for(int ind = 0;ind<theCommand->GetNbResultValues();ind++)
364 Handle(_pyMesh) mesh = new _pyMesh( theCommand, theCommand->GetResultValue(ind));
365 myMeshes.insert( make_pair( theCommand->GetResultValue(ind), mesh ));
369 // CreateHypothesis()
370 if ( method == "CreateHypothesis" )
372 // issue 199929, remove standard library name (default parameter)
373 const TCollection_AsciiString & aLibName = theCommand->GetArg( 2 );
374 if ( aLibName.Search( "StdMeshersEngine" ) != -1 ) {
375 // keep first argument
376 TCollection_AsciiString arg = theCommand->GetArg( 1 );
377 theCommand->RemoveArgs();
378 theCommand->SetArg( 1, arg );
381 myHypos.push_back( _pyHypothesis::NewHypothesis( theCommand ));
385 // smeshgen.Compute( mesh, geom ) --> mesh.Compute()
386 if ( method == "Compute" )
388 const _pyID& meshID = theCommand->GetArg( 1 );
389 map< _pyID, Handle(_pyMesh) >::iterator id_mesh = myMeshes.find( meshID );
390 if ( id_mesh != myMeshes.end() ) {
391 theCommand->SetObject( meshID );
392 theCommand->RemoveArgs();
393 id_mesh->second->Flush();
398 // objects erasing creation command if no more it's commands invoked:
399 // SMESH_Pattern, FilterManager
400 if ( method == "GetPattern" || method == "CreateFilterManager" ) {
401 Handle(_pyObject) obj = new _pySelfEraser( theCommand );
402 if ( !myObjects.insert( make_pair( obj->GetID(), obj )).second )
403 theCommand->Clear(); // already created
406 // Concatenate( [mesh1, ...], ... )
407 if ( method == "Concatenate" || method == "ConcatenateWithGroups")
409 if ( method == "ConcatenateWithGroups" ) {
410 theCommand->SetMethod( "Concatenate" );
411 theCommand->SetArg( theCommand->GetNbArgs() + 1, "True" );
413 Handle(_pyMesh) mesh = new _pyMesh( theCommand, theCommand->GetResultValue() );
414 myMeshes.insert( make_pair( mesh->GetID(), mesh ));
415 AddMeshAccessorMethod( theCommand );
418 // Replace name of SMESH_Gen
420 // names of SMESH_Gen methods fully equal to methods defined in smesh.py
421 static TStringSet smeshpyMethods;
422 if ( smeshpyMethods.empty() ) {
423 const char * names[] =
424 { "SetEmbeddedMode","IsEmbeddedMode","SetCurrentStudy","GetCurrentStudy",
425 "GetPattern","GetSubShapesId",
426 "" }; // <- mark of array end
427 smeshpyMethods.Insert( names );
429 if ( smeshpyMethods.Contains( theCommand->GetMethod() ))
430 // smeshgen.Method() --> smesh.Method()
431 theCommand->SetObject( SMESH_2smeshpy::SmeshpyName() );
433 // smeshgen.Method() --> smesh.smesh.Method()
434 theCommand->SetObject( SMESH_2smeshpy::GenName() );
437 //================================================================================
439 * \brief Convert the remembered commands
441 //================================================================================
445 // create empty command
446 myLastCommand = new _pyCommand();
448 map< _pyID, Handle(_pyMesh) >::iterator id_mesh = myMeshes.begin();
449 for ( ; id_mesh != myMeshes.end(); ++id_mesh )
450 if ( ! id_mesh->second.IsNull() )
451 id_mesh->second->Flush();
453 list< Handle(_pyHypothesis) >::iterator hyp = myHypos.begin();
454 for ( ; hyp != myHypos.end(); ++hyp )
455 if ( !hyp->IsNull() ) {
457 // smeshgen.CreateHypothesis() --> smesh.smesh.CreateHypothesis()
458 if ( !(*hyp)->IsWrapped() )
459 (*hyp)->GetCreationCmd()->SetObject( SMESH_2smeshpy::GenName() );
462 map< _pyID, Handle(_pyObject) >::iterator id_obj = myObjects.begin();
463 for ( ; id_obj != myObjects.end(); ++id_obj )
464 if ( ! id_obj->second.IsNull() )
465 id_obj->second->Flush();
467 myLastCommand->SetOrderNb( ++myNbCommands );
468 myCommands.push_back( myLastCommand );
471 //================================================================================
473 * \brief Add access method to mesh that is an argument
474 * \param theCmd - command to add access method
475 * \retval bool - true if added
477 //================================================================================
479 bool _pyGen::AddMeshAccessorMethod( Handle(_pyCommand) theCmd ) const
482 map< _pyID, Handle(_pyMesh) >::const_iterator id_mesh = myMeshes.begin();
483 for ( ; id_mesh != myMeshes.end(); ++id_mesh ) {
484 if ( theCmd->AddAccessorMethod( id_mesh->first, id_mesh->second->AccessorMethod() ))
490 //================================================================================
492 * \brief Add access method to algo that is an object or an argument
493 * \param theCmd - command to add access method
494 * \retval bool - true if added
496 //================================================================================
498 bool _pyGen::AddAlgoAccessorMethod( Handle(_pyCommand) theCmd ) const
501 list< Handle(_pyHypothesis) >::const_iterator hyp = myHypos.begin();
502 for ( ; hyp != myHypos.end(); ++hyp ) {
503 if ( (*hyp)->IsAlgo() && /*(*hyp)->IsWrapped() &&*/
504 theCmd->AddAccessorMethod( (*hyp)->GetID(), (*hyp)->AccessorMethod() ))
510 //================================================================================
512 * \brief Find hypothesis by ID (entry)
513 * \param theHypID - The hypothesis ID
514 * \retval Handle(_pyHypothesis) - The found hypothesis
516 //================================================================================
518 Handle(_pyHypothesis) _pyGen::FindHyp( const _pyID& theHypID )
520 list< Handle(_pyHypothesis) >::iterator hyp = myHypos.begin();
521 for ( ; hyp != myHypos.end(); ++hyp )
522 if ( !hyp->IsNull() && theHypID == (*hyp)->GetID() )
524 return Handle(_pyHypothesis)();
527 //================================================================================
529 * \brief Find algorithm the created algorithm
530 * \param theGeom - The shape ID the algorithm was created on
531 * \param theMesh - The mesh ID that created the algorithm
532 * \param dim - The algo dimension
533 * \retval Handle(_pyHypothesis) - The found algo
535 //================================================================================
537 Handle(_pyHypothesis) _pyGen::FindAlgo( const _pyID& theGeom, const _pyID& theMesh,
538 const Handle(_pyHypothesis)& theHypothesis )
540 list< Handle(_pyHypothesis) >::iterator hyp = myHypos.begin();
541 for ( ; hyp != myHypos.end(); ++hyp )
542 if ( !hyp->IsNull() &&
544 theHypothesis->CanBeCreatedBy( (*hyp)->GetAlgoType() ) &&
545 (*hyp)->GetGeom() == theGeom &&
546 (*hyp)->GetMesh() == theMesh )
551 //================================================================================
553 * \brief Find subMesh by ID (entry)
554 * \param theSubMeshID - The subMesh ID
555 * \retval Handle(_pySubMesh) - The found subMesh
557 //================================================================================
559 Handle(_pySubMesh) _pyGen::FindSubMesh( const _pyID& theSubMeshID )
561 map< _pyID, Handle(_pyObject) >::iterator id_subMesh = myObjects.find(theSubMeshID);
562 if ( id_subMesh != myObjects.end() )
563 return Handle(_pySubMesh)::DownCast( id_subMesh->second );
564 return Handle(_pySubMesh)();
568 //================================================================================
570 * \brief Change order of commands in the script
571 * \param theCmd1 - One command
572 * \param theCmd2 - Another command
574 //================================================================================
576 void _pyGen::ExchangeCommands( Handle(_pyCommand) theCmd1, Handle(_pyCommand) theCmd2 )
578 list< Handle(_pyCommand) >::iterator pos1, pos2;
579 pos1 = find( myCommands.begin(), myCommands.end(), theCmd1 );
580 pos2 = find( myCommands.begin(), myCommands.end(), theCmd2 );
581 myCommands.insert( pos1, theCmd2 );
582 myCommands.insert( pos2, theCmd1 );
583 myCommands.erase( pos1 );
584 myCommands.erase( pos2 );
586 int nb1 = theCmd1->GetOrderNb();
587 theCmd1->SetOrderNb( theCmd2->GetOrderNb() );
588 theCmd2->SetOrderNb( nb1 );
589 // cout << "BECOME " << theCmd1->GetOrderNb() << "\t" << theCmd1->GetString() << endl
590 // << "BECOME " << theCmd2->GetOrderNb() << "\t" << theCmd2->GetString() << endl << endl;
593 //================================================================================
595 * \brief Set one command after the other
596 * \param theCmd - Command to move
597 * \param theAfterCmd - Command ater which to insert the first one
599 //================================================================================
601 void _pyGen::SetCommandAfter( Handle(_pyCommand) theCmd, Handle(_pyCommand) theAfterCmd )
603 setNeighbourCommand( theCmd, theAfterCmd, true );
606 //================================================================================
608 * \brief Set one command before the other
609 * \param theCmd - Command to move
610 * \param theBeforeCmd - Command before which to insert the first one
612 //================================================================================
614 void _pyGen::SetCommandBefore( Handle(_pyCommand) theCmd, Handle(_pyCommand) theBeforeCmd )
616 setNeighbourCommand( theCmd, theBeforeCmd, false );
619 //================================================================================
621 * \brief Set one command before or after the other
622 * \param theCmd - Command to move
623 * \param theOtherCmd - Command ater or before which to insert the first one
625 //================================================================================
627 void _pyGen::setNeighbourCommand( Handle(_pyCommand)& theCmd,
628 Handle(_pyCommand)& theOtherCmd,
629 const bool theIsAfter )
631 list< Handle(_pyCommand) >::iterator pos;
632 pos = find( myCommands.begin(), myCommands.end(), theCmd );
633 myCommands.erase( pos );
634 pos = find( myCommands.begin(), myCommands.end(), theOtherCmd );
635 myCommands.insert( (theIsAfter ? ++pos : pos), theCmd );
638 for ( pos = myCommands.begin(); pos != myCommands.end(); ++pos)
639 (*pos)->SetOrderNb( i++ );
642 //================================================================================
644 * \brief Set command be last in list of commands
645 * \param theCmd - Command to be last
647 //================================================================================
649 Handle(_pyCommand)& _pyGen::GetLastCommand()
651 return myLastCommand;
654 //================================================================================
656 * \brief Set method to access to object wrapped with python class
657 * \param theID - The wrapped object entry
658 * \param theMethod - The accessor method
660 //================================================================================
662 void _pyGen::SetAccessorMethod(const _pyID& theID, const char* theMethod )
664 myID2AccessorMethod.Bind( theID, (char*) theMethod );
667 //================================================================================
669 * \brief Generated new ID for object and assign with existing name
670 * \param theID - ID of existing object
672 //================================================================================
674 _pyID _pyGen::GenerateNewID( const _pyID& theID )
679 aNewID = theID + _pyID( ":" ) + _pyID( index++ );
681 while ( myObjectNames.IsBound( aNewID ) );
683 myObjectNames.Bind( aNewID, myObjectNames.IsBound( theID )
684 ? (myObjectNames.Find( theID ) + _pyID( "_" ) + _pyID( index-1 ))
685 : _pyID( "A" ) + aNewID );
689 //================================================================================
691 * \brief Find out type of geom group
692 * \param grpID - The geom group entry
693 * \retval int - The type
695 //================================================================================
697 static bool sameGroupType( const _pyID& grpID,
698 const TCollection_AsciiString& theType)
700 // define group type as smesh.Mesh.Group() does
702 SALOMEDS::Study_var study = SMESH_Gen_i::GetSMESHGen()->GetCurrentStudy();
703 SALOMEDS::SObject_var aSObj = study->FindObjectID( grpID.ToCString() );
704 if ( !aSObj->_is_nil() ) {
705 GEOM::GEOM_Object_var aGeomObj = GEOM::GEOM_Object::_narrow( aSObj->GetObject() );
706 if ( !aGeomObj->_is_nil() ) {
707 switch ( aGeomObj->GetShapeType() ) {
708 case GEOM::VERTEX: type = SMESH::NODE; break;
709 case GEOM::EDGE: type = SMESH::EDGE; break;
710 case GEOM::FACE: type = SMESH::FACE; break;
712 case GEOM::SHELL: type = SMESH::VOLUME; break;
713 case GEOM::COMPOUND: {
714 GEOM::GEOM_Gen_ptr aGeomGen = SMESH_Gen_i::GetSMESHGen()->GetGeomEngine();
715 if ( !aGeomGen->_is_nil() ) {
716 GEOM::GEOM_IGroupOperations_var aGrpOp =
717 aGeomGen->GetIGroupOperations( study->StudyId() );
718 if ( !aGrpOp->_is_nil() ) {
719 switch ( aGrpOp->GetType( aGeomObj )) {
720 case TopAbs_VERTEX: type = SMESH::NODE; break;
721 case TopAbs_EDGE: type = SMESH::EDGE; break;
722 case TopAbs_FACE: type = SMESH::FACE; break;
723 case TopAbs_SOLID: type = SMESH::VOLUME; break;
734 MESSAGE("Type of the group " << grpID << " not found");
737 if ( theType.IsIntegerValue() )
738 return type == theType.IntegerValue();
741 case SMESH::NODE: return theType.Location( "NODE", 1, theType.Length() );
742 case SMESH::EDGE: return theType.Location( "EDGE", 1, theType.Length() );
743 case SMESH::FACE: return theType.Location( "FACE", 1, theType.Length() );
744 case SMESH::VOLUME: return theType.Location( "VOLUME", 1, theType.Length() );
750 //================================================================================
753 * \param theCreationCmd -
755 //================================================================================
757 _pyMesh::_pyMesh(const Handle(_pyCommand) theCreationCmd)
758 : _pyObject(theCreationCmd), myHasEditor(false)
760 // convert my creation command
761 Handle(_pyCommand) creationCmd = GetCreationCmd();
762 //TCollection_AsciiString str = creationCmd->GetMethod();
763 // if(str != "CreateMeshesFromUNV" &&
764 // str != "CreateMeshesFromMED" &&
765 // str != "CreateMeshesFromSTL")
766 creationCmd->SetObject( SMESH_2smeshpy::SmeshpyName() );
767 creationCmd->SetMethod( "Mesh" );
769 theGen->SetAccessorMethod( GetID(), "GetMesh()" );
772 //================================================================================
775 * \param theCreationCmd -
777 //================================================================================
778 _pyMesh::_pyMesh(const Handle(_pyCommand) theCreationCmd, const TCollection_AsciiString& id):
779 _pyObject(theCreationCmd), myHasEditor(false)
781 // convert my creation command
782 Handle(_pyCommand) creationCmd = GetCreationCmd();
783 creationCmd->SetObject( SMESH_2smeshpy::SmeshpyName() );
784 theGen->SetAccessorMethod( id, "GetMesh()" );
787 //================================================================================
789 * \brief Convert a IDL API command of SMESH::Mesh to a method call of python Mesh
790 * \param theCommand - Engine method called for this mesh
792 //================================================================================
794 void _pyMesh::Process( const Handle(_pyCommand)& theCommand )
796 // some methods of SMESH_Mesh interface needs special conversion
797 // to methods of Mesh python class
799 // 1. GetSubMesh(geom, name) + AddHypothesis(geom, algo)
800 // --> in Mesh_Algorithm.Create(mesh, geom, hypo, so)
801 // 2. AddHypothesis(geom, hyp)
802 // --> in Mesh_Algorithm.Hypothesis(hyp, args, so)
803 // 3. CreateGroupFromGEOM(type, name, grp)
804 // --> in Mesh.Group(grp, name="")
805 // 4. ExportToMED(f, auto_groups, version)
806 // --> in Mesh.ExportMED( f, auto_groups, version )
809 const TCollection_AsciiString method = theCommand->GetMethod();
810 // ----------------------------------------------------------------------
811 if ( method == "GetSubMesh" ) {
812 Handle(_pySubMesh) subMesh = theGen->FindSubMesh( theCommand->GetResultValue() );
813 if ( !subMesh.IsNull() ) {
814 subMesh->SetCreator( this );
815 mySubmeshes.push_back( subMesh );
818 // ----------------------------------------------------------------------
819 else if ( method == "AddHypothesis" ) { // mesh.AddHypothesis(geom, HYPO )
820 myAddHypCmds.push_back( theCommand );
822 const _pyID& hypID = theCommand->GetArg( 2 );
823 Handle(_pyHypothesis) hyp = theGen->FindHyp( hypID );
824 if ( !hyp.IsNull() ) {
825 myHypos.push_back( hyp );
826 if ( hyp->GetMesh().IsEmpty() )
827 hyp->SetMesh( this->GetID() );
830 // ----------------------------------------------------------------------
831 else if ( method == "CreateGroupFromGEOM" ) {// (type, name, grp)
832 _pyID grp = theCommand->GetArg( 3 );
833 if ( sameGroupType( grp, theCommand->GetArg( 1 )) ) { // --> Group(grp)
834 theCommand->SetMethod( "Group" );
835 theCommand->RemoveArgs();
836 theCommand->SetArg( 1, grp );
839 _pyID type = theCommand->GetArg( 1 );
840 _pyID name = theCommand->GetArg( 2 );
841 theCommand->SetMethod( "GroupOnGeom" );
842 theCommand->RemoveArgs();
843 theCommand->SetArg( 1, grp );
844 theCommand->SetArg( 2, name );
845 theCommand->SetArg( 3, type );
848 // ----------------------------------------------------------------------
849 else if ( method == "ExportToMED" || // ExportToMED() --> ExportMED()
850 method == "ExportToMEDX" ) { // ExportToMEDX() --> ExportMED()
851 theCommand->SetMethod( "ExportMED" );
853 // ----------------------------------------------------------------------
854 else if ( method == "CreateGroup" ) { // CreateGroup() --> CreateEmptyGroup()
855 theCommand->SetMethod( "CreateEmptyGroup" );
857 // ----------------------------------------------------------------------
858 else if ( method == "RemoveHypothesis" ) // (geom, hyp)
860 _pyID hypID = theCommand->GetArg( 2 );
862 // check if this mesh still has corresponding addition command
863 bool hasAddCmd = false;
864 list< Handle(_pyCommand) >::iterator cmd = myAddHypCmds.begin();
865 while ( cmd != myAddHypCmds.end() )
867 // AddHypothesis(geom, hyp)
868 if ( hypID == (*cmd)->GetArg( 2 )) { // erase both (add and remove) commands
871 cmd = myAddHypCmds.erase( cmd );
878 Handle(_pyHypothesis) hyp = theGen->FindHyp( hypID );
879 if ( ! hasAddCmd && hypID.Length() != 0 ) { // hypo addition already wrapped
880 // RemoveHypothesis(geom, hyp) --> RemoveHypothesis( hyp, geom=0 )
881 _pyID geom = theCommand->GetArg( 1 );
882 theCommand->RemoveArgs();
883 theCommand->SetArg( 1, hypID );
884 if ( geom != GetGeom() )
885 theCommand->SetArg( 2, geom );
887 // remove hyp from myHypos
888 myHypos.remove( hyp );
890 // check for SubMesh order commands
891 else if ( theCommand->GetMethod() == "GetMeshOrder" ||
892 theCommand->GetMethod() == "SetMeshOrder" ) {
893 // In fact arguments and result values does not support complex containers
894 // such as list of list
895 // So, here we parse it manually
897 //for(int ind = 0, n = theCommand->GetNbResultValues();ind<n;ind++) {
898 // Handle(_pySubMesh) subMesh = theGen->FindSubMesh( theCommand->GetResultValue(ind) );
900 //for(int ind = 0, n = theCommand->GetNbArgs();ind<n;ind++) {
901 // Handle(_pySubMesh) subMesh = theGen->FindSubMesh( theCommand->GetArg(ind) );
902 const bool isArg = theCommand->GetMethod() == "SetMeshOrder";
903 const TCollection_AsciiString& cmdStr = theCommand->GetString();
904 int begPos = (/*isArg ? cmdStr.Search( "(" ) :*/ cmdStr.Search( "[" )) + 1;
905 int endPos = (isArg ? cmdStr.Search( ")" ) : cmdStr.Search( "=" )) - 1;
906 if ( begPos != -1 && begPos < endPos && endPos <= cmdStr.Length() ) {
907 TCollection_AsciiString aSubStr = cmdStr.SubString( begPos, endPos );
908 Standard_Integer index = 1;
909 TCollection_AsciiString anIDStr = aSubStr.Token("\t ,[]", index++);
910 while ( !anIDStr.IsEmpty() ) {
911 Handle(_pySubMesh) subMesh = theGen->FindSubMesh( anIDStr );
912 if ( !subMesh.IsNull() )
913 subMesh->Process( theCommand );
914 anIDStr = aSubStr.Token("\t ,[]", index++);
918 // add accessor method if necessary
921 if ( NeedMeshAccess( theCommand ))
922 // apply theCommand to the mesh wrapped by smeshpy mesh
923 AddMeshAccess( theCommand );
927 //================================================================================
929 * \brief Return True if addition of accesor method is needed
931 //================================================================================
933 bool _pyMesh::NeedMeshAccess( const Handle(_pyCommand)& theCommand )
935 // names of SMESH_Mesh methods fully equal to methods of class Mesh, so
936 // no conversion is needed for them at all:
937 static TStringSet sameMethods;
938 if ( sameMethods.empty() ) {
939 const char * names[] =
940 { "ExportDAT","ExportUNV","ExportSTL", "RemoveGroup","RemoveGroupWithContents",
941 "GetGroups","UnionGroups","IntersectGroups","CutGroups","GetLog","GetId","ClearLog",
942 "GetStudyId","HasDuplicatedGroupNamesMED","GetMEDMesh","NbNodes","NbElements",
943 "NbEdges","NbEdgesOfOrder","NbFaces","NbFacesOfOrder","NbTriangles",
944 "NbTrianglesOfOrder","NbQuadrangles","NbQuadranglesOfOrder","NbPolygons","NbVolumes",
945 "NbVolumesOfOrder","NbTetras","NbTetrasOfOrder","NbHexas","NbHexasOfOrder",
946 "NbPyramids","NbPyramidsOfOrder","NbPrisms","NbPrismsOfOrder","NbPolyhedrons",
947 "NbSubMesh","GetElementsId","GetElementsByType","GetNodesId","GetElementType",
948 "GetSubMeshElementsId","GetSubMeshNodesId","GetSubMeshElementType","Dump","GetNodeXYZ",
949 "GetNodeInverseElements","GetShapeID","GetShapeIDForElem","GetElemNbNodes",
950 "GetElemNode","IsMediumNode","IsMediumNodeOfAnyElem","ElemNbEdges","ElemNbFaces",
951 "IsPoly","IsQuadratic","BaryCenter","GetHypothesisList", "SetAutoColor", "GetAutoColor",
952 "Clear", "ConvertToStandalone", "GetMeshOrder", "SetMeshOrder"
953 ,"" }; // <- mark of end
954 sameMethods.Insert( names );
957 return !sameMethods.Contains( theCommand->GetMethod() );
960 //================================================================================
962 * \brief Convert creation and addition of all algos and hypos
964 //================================================================================
966 void _pyMesh::Flush()
968 list < Handle(_pyCommand) >::iterator cmd;
970 // try to convert algo addition like this:
971 // mesh.AddHypothesis(geom, ALGO ) --> ALGO = mesh.Algo()
972 for ( cmd = myAddHypCmds.begin(); cmd != myAddHypCmds.end(); ++cmd )
974 Handle(_pyCommand) addCmd = *cmd;
976 _pyID algoID = addCmd->GetArg( 2 );
977 Handle(_pyHypothesis) algo = theGen->FindHyp( algoID );
978 if ( algo.IsNull() || !algo->IsAlgo() )
981 // check and create new algorithm instance if it is already wrapped
982 if ( algo->IsWrapped() ) {
983 _pyID localAlgoID = theGen->GenerateNewID( algoID );
984 TCollection_AsciiString aNewCmdStr = localAlgoID +
985 TCollection_AsciiString( " = " ) + theGen->GetID() +
986 TCollection_AsciiString( ".CreateHypothesis( \"" ) + algo->GetAlgoType() +
987 TCollection_AsciiString( "\" )" );
989 Handle(_pyCommand) newCmd = theGen->AddCommand( aNewCmdStr );
990 Handle(_pyAlgorithm) newAlgo = Handle(_pyAlgorithm)::DownCast(theGen->FindHyp( localAlgoID ));
991 if ( !newAlgo.IsNull() ) {
992 newAlgo->Assign( algo, this->GetID() );
993 newAlgo->SetCreationCmd( newCmd );
995 // set algorithm creation
996 theGen->SetCommandBefore( newCmd, addCmd );
1001 _pyID geom = addCmd->GetArg( 1 );
1002 bool isLocalAlgo = ( geom != GetGeom() );
1005 if ( algo->Addition2Creation( addCmd, this->GetID() )) // OK
1007 // wrapped algo is created atfer mesh creation
1008 GetCreationCmd()->AddDependantCmd( addCmd );
1010 if ( isLocalAlgo ) {
1011 // mesh.AddHypothesis(geom, ALGO ) --> mesh.AlgoMethod(geom)
1012 addCmd->SetArg( addCmd->GetNbArgs() + 1,
1013 TCollection_AsciiString( "geom=" ) + geom );
1014 // sm = mesh.GetSubMesh(geom, name) --> sm = ALGO.GetSubMesh()
1015 list < Handle(_pySubMesh) >::iterator smIt;
1016 for ( smIt = mySubmeshes.begin(); smIt != mySubmeshes.end(); ++smIt ) {
1017 Handle(_pySubMesh) subMesh = *smIt;
1018 Handle(_pyCommand) subCmd = subMesh->GetCreationCmd();
1019 if ( geom == subCmd->GetArg( 1 )) {
1020 subCmd->SetObject( algo->GetID() );
1021 subCmd->RemoveArgs();
1022 subMesh->SetCreator( algo );
1027 else // KO - ALGO was already created
1029 // mesh.AddHypothesis(geom, ALGO) --> mesh.AddHypothesis(ALGO, geom=0)
1030 addCmd->RemoveArgs();
1031 addCmd->SetArg( 1, algoID );
1033 addCmd->SetArg( 2, geom );
1037 // try to convert hypo addition like this:
1038 // mesh.AddHypothesis(geom, HYPO ) --> HYPO = algo.Hypo()
1039 for ( cmd = myAddHypCmds.begin(); cmd != myAddHypCmds.end(); ++cmd )
1041 Handle(_pyCommand) addCmd = *cmd;
1042 _pyID hypID = addCmd->GetArg( 2 );
1043 Handle(_pyHypothesis) hyp = theGen->FindHyp( hypID );
1044 if ( hyp.IsNull() || hyp->IsAlgo() )
1046 bool converted = hyp->Addition2Creation( addCmd, this->GetID() );
1048 // mesh.AddHypothesis(geom, HYP) --> mesh.AddHypothesis(HYP, geom=0)
1049 _pyID geom = addCmd->GetArg( 1 );
1050 addCmd->RemoveArgs();
1051 addCmd->SetArg( 1, hypID );
1052 if ( geom != GetGeom() )
1053 addCmd->SetArg( 2, geom );
1057 // sm = mesh.GetSubMesh(geom, name) --> sm = mesh.GetMesh().GetSubMesh(geom, name)
1058 // for ( cmd = mySubmeshes.begin(); cmd != mySubmeshes.end(); ++cmd ) {
1059 // Handle(_pyCommand) subCmd = *cmd;
1060 // if ( subCmd->GetNbArgs() > 0 )
1061 // AddMeshAccess( subCmd );
1063 myAddHypCmds.clear();
1064 mySubmeshes.clear();
1067 list< Handle(_pyHypothesis) >::iterator hyp = myHypos.begin();
1068 for ( ; hyp != myHypos.end(); ++hyp )
1072 //================================================================================
1074 * \brief MeshEditor convert its commands to ones of mesh
1076 //================================================================================
1078 _pyMeshEditor::_pyMeshEditor(const Handle(_pyCommand)& theCreationCmd):
1079 _pyObject( theCreationCmd )
1081 myMesh = theCreationCmd->GetObject();
1082 myCreationCmdStr = theCreationCmd->GetString();
1083 theCreationCmd->Clear();
1086 //================================================================================
1088 * \brief convert its commands to ones of mesh
1090 //================================================================================
1092 void _pyMeshEditor::Process( const Handle(_pyCommand)& theCommand)
1094 // names of SMESH_MeshEditor methods fully equal to methods of python class Mesh, so
1095 // commands calling this methods are converted to calls of methods of Mesh
1096 static TStringSet sameMethods;
1097 if ( sameMethods.empty() ) {
1098 const char * names[] = {
1099 "RemoveElements","RemoveNodes","RemoveOrphanNodes","AddNode","Add0DElement","AddEdge","AddFace","AddPolygonalFace",
1100 "AddVolume","AddPolyhedralVolume","AddPolyhedralVolumeByFaces","MoveNode", "MoveClosestNodeToPoint",
1101 "InverseDiag","DeleteDiag","Reorient","ReorientObject","TriToQuad","SplitQuad","SplitQuadObject",
1102 "BestSplit","Smooth","SmoothObject","SmoothParametric","SmoothParametricObject",
1103 "ConvertToQuadratic","ConvertFromQuadratic","RenumberNodes","RenumberElements",
1104 "RotationSweep","RotationSweepObject","RotationSweepObject1D","RotationSweepObject2D",
1105 "ExtrusionSweep","AdvancedExtrusion","ExtrusionSweepObject","ExtrusionSweepObject1D","ExtrusionSweepObject2D",
1106 "ExtrusionAlongPath","ExtrusionAlongPathObject","ExtrusionAlongPathObject1D","ExtrusionAlongPathObject2D",
1107 "Mirror","MirrorObject","Translate","TranslateObject","Rotate","RotateObject",
1108 "FindCoincidentNodes",/*"FindCoincidentNodesOnPart",*/"MergeNodes","FindEqualElements",
1109 "MergeElements","MergeEqualElements","SewFreeBorders","SewConformFreeBorders",
1110 "SewBorderToSide","SewSideElements","ChangeElemNodes","GetLastCreatedNodes",
1111 "GetLastCreatedElems",
1112 "MirrorMakeMesh","MirrorObjectMakeMesh","TranslateMakeMesh",
1113 "TranslateObjectMakeMesh","RotateMakeMesh","RotateObjectMakeMesh"
1114 ,"" }; // <- mark of the end
1115 sameMethods.Insert( names );
1118 // names of SMESH_MeshEditor methods which differ from methods of class Mesh
1119 // only by last two arguments
1120 static TStringSet diffLastTwoArgsMethods;
1121 if (diffLastTwoArgsMethods.empty() ) {
1122 const char * names[] = {
1123 "MirrorMakeGroups","MirrorObjectMakeGroups",
1124 "TranslateMakeGroups","TranslateObjectMakeGroups",
1125 "RotateMakeGroups","RotateObjectMakeGroups",
1126 ""};// <- mark of the end
1127 diffLastTwoArgsMethods.Insert( names );
1130 const TCollection_AsciiString & method = theCommand->GetMethod();
1131 bool isPyMeshMethod = sameMethods.Contains( method );
1132 if ( !isPyMeshMethod )
1134 //Replace SMESH_MeshEditor "MakeGroups" functions on the Mesh
1135 //functions with the flag "theMakeGroups = True" like:
1136 //SMESH_MeshEditor.CmdMakeGroups => Mesh.Cmd(...,True)
1137 int pos = method.Search("MakeGroups");
1140 isPyMeshMethod = true;
1142 // 1. Remove "MakeGroups" from the Command
1143 TCollection_AsciiString aMethod = theCommand->GetMethod();
1144 int nbArgsToAdd = diffLastTwoArgsMethods.Contains(aMethod) ? 2 : 1;
1145 aMethod.Trunc(pos-1);
1146 theCommand->SetMethod(aMethod);
1148 // 2. And add last "True" argument(s)
1149 while(nbArgsToAdd--)
1150 theCommand->SetArg(theCommand->GetNbArgs()+1,"True");
1154 // set "FindCoincidentNodesOnPart()" instead of "FindCoincidentNodesOnPartBut()"
1155 if ( !isPyMeshMethod && method == "FindCoincidentNodesOnPartBut")
1157 isPyMeshMethod=true;
1158 theCommand->SetMethod("FindCoincidentNodesOnPart");
1161 // meshes made by *MakeMesh() methods are not wrapped by _pyMesh,
1162 // so let _pyMesh care of it (TMP?)
1163 // if ( theCommand->GetMethod().Search("MakeMesh") != -1 )
1164 // _pyMesh( new _pyCommand( theCommand->GetString(), 0 )); // for theGen->SetAccessorMethod()
1165 if ( isPyMeshMethod )
1167 theCommand->SetObject( myMesh );
1171 // editor creation command is needed only if any editor function is called
1172 theGen->AddMeshAccessorMethod( theCommand ); // for *Object()
1173 if ( !myCreationCmdStr.IsEmpty() ) {
1174 GetCreationCmd()->GetString() = myCreationCmdStr;
1175 myCreationCmdStr.Clear();
1180 //================================================================================
1182 * \brief _pyHypothesis constructor
1183 * \param theCreationCmd -
1185 //================================================================================
1187 _pyHypothesis::_pyHypothesis(const Handle(_pyCommand)& theCreationCmd):
1188 _pyObject( theCreationCmd )
1190 myIsAlgo = myIsWrapped = /*myIsConverted = myIsLocal = myDim = */false;
1193 //================================================================================
1195 * \brief Creates algorithm or hypothesis
1196 * \param theCreationCmd - The engine command creating a hypothesis
1197 * \retval Handle(_pyHypothesis) - Result _pyHypothesis
1199 //================================================================================
1201 Handle(_pyHypothesis) _pyHypothesis::NewHypothesis( const Handle(_pyCommand)& theCreationCmd)
1203 // theCreationCmd: CreateHypothesis( "theHypType", "theLibName" )
1204 ASSERT (( theCreationCmd->GetMethod() == "CreateHypothesis"));
1206 Handle(_pyHypothesis) hyp, algo;
1209 const TCollection_AsciiString & hypTypeQuoted = theCreationCmd->GetArg( 1 );
1210 if ( hypTypeQuoted.IsEmpty() )
1213 TCollection_AsciiString hypType =
1214 hypTypeQuoted.SubString( 2, hypTypeQuoted.Length() - 1 );
1216 algo = new _pyAlgorithm( theCreationCmd );
1217 hyp = new _pyHypothesis( theCreationCmd );
1219 // 1D Regular_1D ----------
1220 if ( hypType == "Regular_1D" ) {
1221 // set mesh's method creating algo,
1222 // i.e. convertion result will be "regular1d = Mesh.Segment()",
1223 // and set hypType by which algo creating a hypothesis is searched for
1224 algo->SetConvMethodAndType("Segment", hypType.ToCString());
1226 else if ( hypType == "CompositeSegment_1D" ) {
1227 algo->SetConvMethodAndType("Segment", "Regular_1D");
1228 algo->myArgs.Append( "algo=smesh.COMPOSITE");
1230 else if ( hypType == "LocalLength" ) {
1231 // set algo's method creating hyp, and algo type
1232 hyp->SetConvMethodAndType( "LocalLength", "Regular_1D");
1233 // set method whose 1 arg will become the 1-st arg of hyp creation command
1234 // i.e. convertion result will be "locallength = regular1d.LocalLength(<arg of SetLength()>)"
1235 hyp->AddArgMethod( "SetLength" );
1237 else if ( hypType == "MaxLength" ) {
1238 // set algo's method creating hyp, and algo type
1239 hyp->SetConvMethodAndType( "MaxSize", "Regular_1D");
1240 // set method whose 1 arg will become the 1-st arg of hyp creation command
1241 // i.e. convertion result will be "maxsize = regular1d.MaxSize(<arg of SetLength()>)"
1242 hyp->AddArgMethod( "SetLength" );
1244 else if ( hypType == "NumberOfSegments" ) {
1245 hyp = new _pyNumberOfSegmentsHyp( theCreationCmd );
1246 hyp->SetConvMethodAndType( "NumberOfSegments", "Regular_1D");
1247 // arg of SetNumberOfSegments() will become the 1-st arg of hyp creation command
1248 hyp->AddArgMethod( "SetNumberOfSegments" );
1249 // arg of SetScaleFactor() will become the 2-nd arg of hyp creation command
1250 hyp->AddArgMethod( "SetScaleFactor" );
1251 hyp->AddArgMethod( "SetReversedEdges" );
1253 else if ( hypType == "Arithmetic1D" ) {
1254 hyp = new _pyComplexParamHypo( theCreationCmd );
1255 hyp->SetConvMethodAndType( "Arithmetic1D", "Regular_1D");
1256 hyp->AddArgMethod( "SetStartLength" );
1257 hyp->AddArgMethod( "SetEndLength" );
1258 hyp->AddArgMethod( "SetReversedEdges" );
1260 else if ( hypType == "StartEndLength" ) {
1261 hyp = new _pyComplexParamHypo( theCreationCmd );
1262 hyp->SetConvMethodAndType( "StartEndLength", "Regular_1D");
1263 hyp->AddArgMethod( "SetStartLength" );
1264 hyp->AddArgMethod( "SetEndLength" );
1265 hyp->AddArgMethod( "SetReversedEdges" );
1267 else if ( hypType == "Deflection1D" ) {
1268 hyp->SetConvMethodAndType( "Deflection1D", "Regular_1D");
1269 hyp->AddArgMethod( "SetDeflection" );
1271 else if ( hypType == "Propagation" ) {
1272 hyp->SetConvMethodAndType( "Propagation", "Regular_1D");
1274 else if ( hypType == "QuadraticMesh" ) {
1275 hyp->SetConvMethodAndType( "QuadraticMesh", "Regular_1D");
1277 else if ( hypType == "AutomaticLength" ) {
1278 hyp->SetConvMethodAndType( "AutomaticLength", "Regular_1D");
1279 hyp->AddArgMethod( "SetFineness");
1281 else if ( hypType == "SegmentLengthAroundVertex" ) {
1282 hyp = new _pySegmentLengthAroundVertexHyp( theCreationCmd );
1283 hyp->SetConvMethodAndType( "LengthNearVertex", "Regular_1D" );
1284 hyp->AddArgMethod( "SetLength" );
1286 // 1D Python_1D ----------
1287 else if ( hypType == "Python_1D" ) {
1288 algo->SetConvMethodAndType( "Segment", hypType.ToCString());
1289 algo->myArgs.Append( "algo=smesh.PYTHON");
1291 else if ( hypType == "PythonSplit1D" ) {
1292 hyp->SetConvMethodAndType( "PythonSplit1D", "Python_1D");
1293 hyp->AddArgMethod( "SetNumberOfSegments");
1294 hyp->AddArgMethod( "SetPythonLog10RatioFunction");
1296 // MEFISTO_2D ----------
1297 else if ( hypType == "MEFISTO_2D" ) { // MEFISTO_2D
1298 algo->SetConvMethodAndType( "Triangle", hypType.ToCString());
1300 else if ( hypType == "MaxElementArea" ) {
1301 hyp->SetConvMethodAndType( "MaxElementArea", "MEFISTO_2D");
1302 hyp->SetConvMethodAndType( "MaxElementArea", "NETGEN_2D_ONLY");
1303 hyp->AddArgMethod( "SetMaxElementArea");
1305 else if ( hypType == "LengthFromEdges" ) {
1306 hyp->SetConvMethodAndType( "LengthFromEdges", "MEFISTO_2D");
1307 hyp->SetConvMethodAndType( "LengthFromEdges", "NETGEN_2D_ONLY");
1309 // Quadrangle_2D ----------
1310 else if ( hypType == "Quadrangle_2D" ) {
1311 algo->SetConvMethodAndType( "Quadrangle" , hypType.ToCString());
1313 else if ( hypType == "QuadranglePreference" ) {
1314 hyp->SetConvMethodAndType( "QuadranglePreference", "Quadrangle_2D");
1315 hyp->SetConvMethodAndType( "SetQuadAllowed", "NETGEN_2D_ONLY");
1317 else if ( hypType == "TrianglePreference" ) {
1318 hyp->SetConvMethodAndType( "TrianglePreference", "Quadrangle_2D");
1320 // RadialQuadrangle_1D2D ----------
1321 else if ( hypType == "RadialQuadrangle_1D2D" ) {
1322 algo->SetConvMethodAndType( "Quadrangle" , hypType.ToCString());
1323 algo->myArgs.Append( "algo=smesh.RADIAL_QUAD" );
1325 else if ( hypType == "NumberOfLayers2D" ) {
1326 hyp->SetConvMethodAndType( "NumberOfLayers", "RadialQuadrangle_1D2D");
1327 hyp->AddArgMethod( "SetNumberOfLayers" );
1329 else if ( hypType == "LayerDistribution2D" ) {
1330 hyp = new _pyLayerDistributionHypo( theCreationCmd, "Get2DHypothesis" );
1331 hyp->SetConvMethodAndType( "LayerDistribution", "RadialQuadrangle_1D2D");
1333 // BLSURF ----------
1334 else if ( hypType == "BLSURF" ) {
1335 algo->SetConvMethodAndType( "Triangle", hypType.ToCString());
1336 algo->myArgs.Append( "algo=smesh.BLSURF" );
1338 else if ( hypType == "BLSURF_Parameters") {
1339 hyp->SetConvMethodAndType( "Parameters", "BLSURF");
1341 // NETGEN ----------
1342 else if ( hypType == "NETGEN_2D") { // 1D-2D
1343 algo->SetConvMethodAndType( "Triangle" , hypType.ToCString());
1344 algo->myArgs.Append( "algo=smesh.NETGEN" );
1346 else if ( hypType == "NETGEN_Parameters_2D") {
1347 hyp->SetConvMethodAndType( "Parameters", "NETGEN_2D");
1349 else if ( hypType == "NETGEN_SimpleParameters_2D") {
1350 hyp->SetConvMethodAndType( "Parameters", "NETGEN_2D");
1351 hyp->myArgs.Append( "which=smesh.SIMPLE" );
1353 else if ( hypType == "NETGEN_2D3D") { // 1D-2D-3D
1354 algo->SetConvMethodAndType( "Tetrahedron" , hypType.ToCString());
1355 algo->myArgs.Append( "algo=smesh.FULL_NETGEN" );
1357 else if ( hypType == "NETGEN_Parameters") {
1358 hyp->SetConvMethodAndType( "Parameters", "NETGEN_2D3D");
1360 else if ( hypType == "NETGEN_SimpleParameters_3D") {
1361 hyp->SetConvMethodAndType( "Parameters", "NETGEN_2D3D");
1362 hyp->myArgs.Append( "which=smesh.SIMPLE" );
1364 else if ( hypType == "NETGEN_2D_ONLY") { // 2D
1365 algo->SetConvMethodAndType( "Triangle" , hypType.ToCString());
1366 algo->myArgs.Append( "algo=smesh.NETGEN_2D" );
1368 else if ( hypType == "NETGEN_3D") { // 3D
1369 algo->SetConvMethodAndType( "Tetrahedron" , hypType.ToCString());
1370 algo->myArgs.Append( "algo=smesh.NETGEN" );
1372 else if ( hypType == "MaxElementVolume") {
1373 hyp->SetConvMethodAndType( "MaxElementVolume", "NETGEN_3D");
1374 hyp->AddArgMethod( "SetMaxElementVolume" );
1376 // GHS3D_3D ----------
1377 else if ( hypType == "GHS3D_3D" ) {
1378 algo->SetConvMethodAndType( "Tetrahedron", hypType.ToCString());
1379 algo->myArgs.Append( "algo=smesh.GHS3D" );
1381 else if ( hypType == "GHS3D_Parameters") {
1382 hyp->SetConvMethodAndType( "Parameters", "GHS3D_3D");
1384 // Hexa_3D ---------
1385 else if ( hypType == "BLSURF" ) {
1386 algo->SetConvMethodAndType( "Hexahedron", hypType.ToCString());
1388 // Repetitive Projection_1D ---------
1389 else if ( hypType == "Projection_1D" ) {
1390 algo->SetConvMethodAndType( "Projection1D", hypType.ToCString());
1392 else if ( hypType == "ProjectionSource1D" ) {
1393 hyp->SetConvMethodAndType( "SourceEdge", "Projection_1D");
1394 hyp->AddArgMethod( "SetSourceEdge");
1395 hyp->AddArgMethod( "SetSourceMesh");
1396 // 2 args of SetVertexAssociation() will become the 3-th and 4-th args of hyp creation command
1397 hyp->AddArgMethod( "SetVertexAssociation", 2 );
1399 // Projection_2D ---------
1400 else if ( hypType == "Projection_2D" ) {
1401 algo->SetConvMethodAndType( "Projection2D", hypType.ToCString());
1403 else if ( hypType == "ProjectionSource2D" ) {
1404 hyp->SetConvMethodAndType( "SourceFace", "Projection_2D");
1405 hyp->AddArgMethod( "SetSourceFace");
1406 hyp->AddArgMethod( "SetSourceMesh");
1407 hyp->AddArgMethod( "SetVertexAssociation", 4 );
1409 // Projection_3D ---------
1410 else if ( hypType == "Projection_3D" ) {
1411 algo->SetConvMethodAndType( "Projection3D", hypType.ToCString());
1413 else if ( hypType == "ProjectionSource3D" ) {
1414 hyp->SetConvMethodAndType( "SourceShape3D", "Projection_3D");
1415 hyp->AddArgMethod( "SetSource3DShape");
1416 hyp->AddArgMethod( "SetSourceMesh");
1417 hyp->AddArgMethod( "SetVertexAssociation", 4 );
1419 // Prism_3D ---------
1420 else if ( hypType == "Prism_3D" ) {
1421 algo->SetConvMethodAndType( "Prism", hypType.ToCString());
1423 // RadialPrism_3D ---------
1424 else if ( hypType == "RadialPrism_3D" ) {
1425 algo->SetConvMethodAndType( "Prism", hypType.ToCString());
1427 else if ( hypType == "NumberOfLayers" ) {
1428 hyp->SetConvMethodAndType( "NumberOfLayers", "RadialPrism_3D");
1429 hyp->AddArgMethod( "SetNumberOfLayers" );
1431 else if ( hypType == "LayerDistribution" ) {
1432 hyp = new _pyLayerDistributionHypo( theCreationCmd, "Get3DHypothesis" );
1433 hyp->SetConvMethodAndType( "LayerDistribution", "RadialPrism_3D");
1436 return algo->IsValid() ? algo : hyp;
1439 //================================================================================
1441 * \brief Convert the command adding a hypothesis to mesh into a smesh command
1442 * \param theCmd - The command like mesh.AddHypothesis( geom, hypo )
1443 * \param theAlgo - The algo that can create this hypo
1444 * \retval bool - false if the command cant be converted
1446 //================================================================================
1448 bool _pyHypothesis::Addition2Creation( const Handle(_pyCommand)& theCmd,
1449 const _pyID& theMesh)
1451 ASSERT(( theCmd->GetMethod() == "AddHypothesis" ));
1453 if ( !IsWrappable( theMesh ))
1456 myGeom = theCmd->GetArg( 1 );
1458 Handle(_pyHypothesis) algo;
1460 // find algo created on myGeom in theMesh
1461 algo = theGen->FindAlgo( myGeom, theMesh, this );
1462 if ( algo.IsNull() )
1464 // attach hypothesis creation command to be after algo creation command
1465 // because it can be new created instance of algorithm
1466 algo->GetCreationCmd()->AddDependantCmd( theCmd );
1470 // mesh.AddHypothesis(geom,hyp) --> hyp = <theMesh or algo>.myCreationMethod(args)
1471 theCmd->SetResultValue( GetID() );
1472 theCmd->SetObject( IsAlgo() ? theMesh : algo->GetID());
1473 theCmd->SetMethod( IsAlgo() ? GetAlgoCreationMethod() : GetCreationMethod( algo->GetAlgoType() ));
1475 theCmd->RemoveArgs();
1476 for ( int i = 1; i <= myArgs.Length(); ++i ) {
1477 if ( !myArgs( i ).IsEmpty() )
1478 theCmd->SetArg( i, myArgs( i ));
1480 theCmd->SetArg( i, "[]");
1482 // set a new creation command
1483 GetCreationCmd()->Clear();
1484 // replace creation command by wrapped instance
1485 // please note, that hypothesis attaches to algo creation command (see upper)
1486 SetCreationCmd( theCmd );
1489 // clear commands setting arg values
1490 list < Handle(_pyCommand) >::iterator argCmd = myArgCommands.begin();
1491 for ( ; argCmd != myArgCommands.end(); ++argCmd )
1494 // set unknown arg commands after hypo creation
1495 Handle(_pyCommand) afterCmd = myIsWrapped ? theCmd : GetCreationCmd();
1496 list<Handle(_pyCommand)>::iterator cmd = myUnknownCommands.begin();
1497 for ( ; cmd != myUnknownCommands.end(); ++cmd ) {
1498 afterCmd->AddDependantCmd( *cmd );
1504 //================================================================================
1506 * \brief Remember hypothesis parameter values
1507 * \param theCommand - The called hypothesis method
1509 //================================================================================
1511 void _pyHypothesis::Process( const Handle(_pyCommand)& theCommand)
1513 ASSERT( !myIsAlgo );
1516 for ( int i = 1; i <= myArgMethods.Length(); ++i ) {
1517 if ( myArgMethods( i ) == theCommand->GetMethod() ) {
1518 while ( myArgs.Length() < nbArgs + myNbArgsByMethod( i ))
1519 myArgs.Append( "[]" );
1520 for ( int iArg = 1; iArg <= myNbArgsByMethod( i ); ++iArg )
1521 myArgs( nbArgs + iArg ) = theCommand->GetArg( iArg ); // arg value
1522 myArgCommands.push_back( theCommand );
1525 nbArgs += myNbArgsByMethod( i );
1527 myUnknownCommands.push_back( theCommand );
1530 //================================================================================
1532 * \brief Finish conversion
1534 //================================================================================
1536 void _pyHypothesis::Flush()
1538 if ( IsWrapped() ) {
1541 list < Handle(_pyCommand) >::iterator cmd = myArgCommands.begin();
1542 for ( ; cmd != myArgCommands.end(); ++cmd ) {
1543 // Add access to a wrapped mesh
1544 theGen->AddMeshAccessorMethod( *cmd );
1545 // Add access to a wrapped algorithm
1546 theGen->AddAlgoAccessorMethod( *cmd );
1548 cmd = myUnknownCommands.begin();
1549 for ( ; cmd != myUnknownCommands.end(); ++cmd ) {
1550 // Add access to a wrapped mesh
1551 theGen->AddMeshAccessorMethod( *cmd );
1552 // Add access to a wrapped algorithm
1553 theGen->AddAlgoAccessorMethod( *cmd );
1556 // forget previous hypothesis modifications
1557 myArgCommands.clear();
1558 myUnknownCommands.clear();
1561 //================================================================================
1563 * \brief clear creation, arg and unkown commands
1565 //================================================================================
1567 void _pyHypothesis::ClearAllCommands()
1569 GetCreationCmd()->Clear();
1570 list<Handle(_pyCommand)>::iterator cmd = myArgCommands.begin();
1571 for ( ; cmd != myArgCommands.end(); ++cmd )
1573 cmd = myUnknownCommands.begin();
1574 for ( ; cmd != myUnknownCommands.end(); ++cmd )
1579 //================================================================================
1581 * \brief Assign fields of theOther to me except myIsWrapped
1583 //================================================================================
1585 void _pyHypothesis::Assign( const Handle(_pyHypothesis)& theOther,
1586 const _pyID& theMesh )
1588 myIsWrapped = false;
1591 // myCreationCmd = theOther->myCreationCmd;
1592 myIsAlgo = theOther->myIsAlgo;
1593 myGeom = theOther->myGeom;
1594 myType2CreationMethod = theOther->myType2CreationMethod;
1595 myArgs = theOther->myArgs;
1596 myArgMethods = theOther->myArgMethods;
1597 myNbArgsByMethod = theOther->myNbArgsByMethod;
1598 myArgCommands = theOther->myArgCommands;
1599 myUnknownCommands = theOther->myUnknownCommands;
1602 //================================================================================
1604 * \brief Remember hypothesis parameter values
1605 * \param theCommand - The called hypothesis method
1607 //================================================================================
1609 void _pyComplexParamHypo::Process( const Handle(_pyCommand)& theCommand)
1611 if( theCommand->GetMethod() == "SetLength" )
1613 // NOW it becomes OBSOLETE
1614 // ex: hyp.SetLength(start, 1)
1615 // hyp.SetLength(end, 0)
1616 ASSERT(( theCommand->GetArg( 2 ).IsIntegerValue() ));
1617 int i = 2 - theCommand->GetArg( 2 ).IntegerValue();
1618 while ( myArgs.Length() < i )
1619 myArgs.Append( "[]" );
1620 myArgs( i ) = theCommand->GetArg( 1 ); // arg value
1621 myArgCommands.push_back( theCommand );
1625 _pyHypothesis::Process( theCommand );
1628 //================================================================================
1630 * \brief Clear SetObjectEntry() as it is called by methods of Mesh_Segment
1632 //================================================================================
1634 void _pyComplexParamHypo::Flush()
1638 list < Handle(_pyCommand) >::iterator cmd = myUnknownCommands.begin();
1639 for ( ; cmd != myUnknownCommands.end(); ++cmd )
1640 if ((*cmd)->GetMethod() == "SetObjectEntry" )
1645 //================================================================================
1647 * \brief Convert methods of 1D hypotheses to my own methods
1648 * \param theCommand - The called hypothesis method
1650 //================================================================================
1652 void _pyLayerDistributionHypo::Process( const Handle(_pyCommand)& theCommand)
1654 if ( theCommand->GetMethod() != "SetLayerDistribution" )
1657 _pyID newName; // name for 1D hyp = "HypType" + "_Distribution"
1659 const _pyID& hyp1dID = theCommand->GetArg( 1 );
1660 Handle(_pyHypothesis) hyp1d = theGen->FindHyp( hyp1dID );
1661 if ( hyp1d.IsNull() ) // apparently hypId changed at study restoration
1663 else if ( !my1dHyp.IsNull() && hyp1dID != my1dHyp->GetID() ) {
1664 // 1D hypo is already set, so distribution changes and the old
1665 // 1D hypo is thrown away
1666 my1dHyp->ClearAllCommands();
1670 if ( !myArgCommands.empty() )
1671 myArgCommands.front()->Clear();
1672 myArgCommands.push_back( theCommand );
1675 //================================================================================
1678 * \param theAdditionCmd - command to be converted
1679 * \param theMesh - mesh instance
1680 * \retval bool - status
1682 //================================================================================
1684 bool _pyLayerDistributionHypo::Addition2Creation( const Handle(_pyCommand)& theAdditionCmd,
1685 const _pyID& theMesh)
1687 myIsWrapped = false;
1689 if ( my1dHyp.IsNull() )
1692 // set "SetLayerDistribution()" after addition cmd
1693 theAdditionCmd->AddDependantCmd( myArgCommands.front() );
1695 _pyID geom = theAdditionCmd->GetArg( 1 );
1697 Handle(_pyHypothesis) algo = theGen->FindAlgo( geom, theMesh, this );
1698 if ( !algo.IsNull() )
1700 my1dHyp->SetMesh( theMesh );
1701 my1dHyp->SetConvMethodAndType(my1dHyp->GetAlgoCreationMethod().ToCString(),
1702 algo->GetAlgoType().ToCString());
1703 if ( !my1dHyp->Addition2Creation( theAdditionCmd, theMesh ))
1706 // clear "SetLayerDistribution()" cmd
1707 myArgCommands.back()->Clear();
1709 // Convert my creation => me = RadialPrismAlgo.Get3DHypothesis()
1711 // find RadialPrism algo created on <geom> for theMesh
1712 GetCreationCmd()->SetObject( algo->GetID() );
1713 GetCreationCmd()->SetMethod( myAlgoMethod );
1714 GetCreationCmd()->RemoveArgs();
1715 theAdditionCmd->AddDependantCmd( GetCreationCmd() );
1721 //================================================================================
1725 //================================================================================
1727 void _pyLayerDistributionHypo::Flush()
1729 // as creation of 1D hyp was written later then it's edition,
1730 // we need to find all it's edition calls and process them
1731 if ( !my1dHyp.IsNull() )
1733 _pyID hyp1dID = my1dHyp->GetCreationCmd()->GetResultValue();
1735 // make a new name for 1D hyp = "HypType" + "_Distribution"
1737 if ( my1dHyp->IsWrapped() ) {
1738 newName = my1dHyp->GetCreationCmd()->GetMethod();
1741 TCollection_AsciiString hypTypeQuoted = my1dHyp->GetCreationCmd()->GetArg(1);
1742 newName = hypTypeQuoted.SubString( 2, hypTypeQuoted.Length() - 1 );
1744 newName += "_Distribution";
1745 my1dHyp->GetCreationCmd()->SetResultValue( newName );
1747 list< Handle(_pyCommand) >& cmds = theGen->GetCommands();
1748 list< Handle(_pyCommand) >::iterator cmdIt = cmds.begin();
1749 for ( ; cmdIt != cmds.end(); ++cmdIt ) {
1750 const _pyID& objID = (*cmdIt)->GetObject();
1751 if ( objID == hyp1dID ) {
1752 my1dHyp->Process( *cmdIt );
1753 my1dHyp->GetCreationCmd()->AddDependantCmd( *cmdIt );
1754 ( *cmdIt )->SetObject( newName );
1757 // Set new hyp name to SetLayerDistribution() cmd
1758 if ( !myArgCommands.empty() && !myArgCommands.back()->IsEmpty() )
1759 myArgCommands.back()->SetArg( 1, newName );
1763 //================================================================================
1765 * \brief additionally to Addition2Creation, clears SetDistrType() command
1766 * \param theCmd - AddHypothesis() command
1767 * \param theMesh - mesh to which a hypothesis is added
1768 * \retval bool - convertion result
1770 //================================================================================
1772 bool _pyNumberOfSegmentsHyp::Addition2Creation( const Handle(_pyCommand)& theCmd,
1773 const _pyID& theMesh)
1775 if ( IsWrappable( theMesh ) && myArgs.Length() > 1 ) {
1776 // scale factor (2-nd arg) is provided: clear SetDistrType(1) command
1777 bool scaleDistrType = false;
1778 list<Handle(_pyCommand)>::reverse_iterator cmd = myUnknownCommands.rbegin();
1779 for ( ; cmd != myUnknownCommands.rend(); ++cmd ) {
1780 if ( (*cmd)->GetMethod() == "SetDistrType" ) {
1781 if ( (*cmd)->GetArg( 1 ) == "1" ) {
1782 scaleDistrType = true;
1785 else if ( !scaleDistrType ) {
1786 // distribution type changed: remove scale factor from args
1787 myArgs.Remove( 2, myArgs.Length() );
1793 return _pyHypothesis::Addition2Creation( theCmd, theMesh );
1796 //================================================================================
1798 * \brief remove repeated commands defining distribution
1800 //================================================================================
1802 void _pyNumberOfSegmentsHyp::Flush()
1804 // find number of the last SetDistrType() command
1805 list<Handle(_pyCommand)>::reverse_iterator cmd = myUnknownCommands.rbegin();
1806 int distrTypeNb = 0;
1807 for ( ; !distrTypeNb && cmd != myUnknownCommands.rend(); ++cmd )
1808 if ( (*cmd)->GetMethod() == "SetDistrType" )
1809 distrTypeNb = (*cmd)->GetOrderNb();
1810 else if (IsWrapped() && (*cmd)->GetMethod() == "SetObjectEntry" )
1813 // clear commands before the last SetDistrType()
1814 list<Handle(_pyCommand)> * cmds[2] = { &myArgCommands, &myUnknownCommands };
1815 for ( int i = 0; i < 2; ++i ) {
1816 set<TCollection_AsciiString> uniqueMethods;
1817 list<Handle(_pyCommand)> & cmdList = *cmds[i];
1818 for ( cmd = cmdList.rbegin(); cmd != cmdList.rend(); ++cmd )
1820 bool clear = ( (*cmd)->GetOrderNb() < distrTypeNb );
1821 const TCollection_AsciiString& method = (*cmd)->GetMethod();
1822 if ( !clear || method == "SetNumberOfSegments" ) {
1823 bool isNewInSet = uniqueMethods.insert( method ).second;
1824 clear = !isNewInSet;
1833 //================================================================================
1835 * \brief Convert the command adding "SegmentLengthAroundVertex" to mesh
1836 * into regular1D.LengthNearVertex( length, vertex )
1837 * \param theCmd - The command like mesh.AddHypothesis( vertex, SegmentLengthAroundVertex )
1838 * \param theMesh - The mesh needing this hypo
1839 * \retval bool - false if the command cant be converted
1841 //================================================================================
1843 bool _pySegmentLengthAroundVertexHyp::Addition2Creation( const Handle(_pyCommand)& theCmd,
1844 const _pyID& theMeshID)
1846 if ( IsWrappable( theMeshID )) {
1848 _pyID vertex = theCmd->GetArg( 1 );
1850 // the problem here is that segment algo will not be found
1851 // by pyHypothesis::Addition2Creation() for <vertex>, so we try to find
1852 // geometry where segment algorithm is assigned
1853 Handle(_pyHypothesis) algo;
1854 _pyID geom = vertex;
1855 while ( algo.IsNull() && !geom.IsEmpty()) {
1856 // try to find geom as a father of <vertex>
1857 geom = FatherID( geom );
1858 algo = theGen->FindAlgo( geom, theMeshID, this );
1860 if ( algo.IsNull() )
1861 return false; // also possible to find geom as brother of veretex...
1862 // set geom instead of vertex
1863 theCmd->SetArg( 1, geom );
1865 // set vertex as a second arg
1866 if ( myArgs.Length() < 1) myArgs.Append( "1" ); // :(
1867 myArgs.Append( vertex );
1869 // mesh.AddHypothesis(vertex, SegmentLengthAroundVertex) -->
1870 // theMeshID.LengthNearVertex( length, vertex )
1871 return _pyHypothesis::Addition2Creation( theCmd, theMeshID );
1876 //================================================================================
1878 * \brief _pyAlgorithm constructor
1879 * \param theCreationCmd - The command like "algo = smeshgen.CreateHypothesis(type,lib)"
1881 //================================================================================
1883 _pyAlgorithm::_pyAlgorithm(const Handle(_pyCommand)& theCreationCmd)
1884 : _pyHypothesis( theCreationCmd )
1889 //================================================================================
1891 * \brief Convert the command adding an algorithm to mesh
1892 * \param theCmd - The command like mesh.AddHypothesis( geom, algo )
1893 * \param theMesh - The mesh needing this algo
1894 * \retval bool - false if the command cant be converted
1896 //================================================================================
1898 bool _pyAlgorithm::Addition2Creation( const Handle(_pyCommand)& theCmd,
1899 const _pyID& theMeshID)
1901 // mesh.AddHypothesis(geom,algo) --> theMeshID.myCreationMethod()
1902 if ( _pyHypothesis::Addition2Creation( theCmd, theMeshID )) {
1903 theGen->SetAccessorMethod( GetID(), "GetAlgorithm()" );
1909 //================================================================================
1911 * \brief Return starting position of a part of python command
1912 * \param thePartIndex - The index of command part
1913 * \retval int - Part position
1915 //================================================================================
1917 int _pyCommand::GetBegPos( int thePartIndex )
1921 if ( myBegPos.Length() < thePartIndex )
1923 return myBegPos( thePartIndex );
1926 //================================================================================
1928 * \brief Store starting position of a part of python command
1929 * \param thePartIndex - The index of command part
1930 * \param thePosition - Part position
1932 //================================================================================
1934 void _pyCommand::SetBegPos( int thePartIndex, int thePosition )
1936 while ( myBegPos.Length() < thePartIndex )
1937 myBegPos.Append( UNKNOWN );
1938 myBegPos( thePartIndex ) = thePosition;
1941 //================================================================================
1943 * \brief Returns whitespace symbols at the line beginning
1944 * \retval TCollection_AsciiString - result
1946 //================================================================================
1948 TCollection_AsciiString _pyCommand::GetIndentation()
1951 if ( GetBegPos( RESULT_IND ) == UNKNOWN )
1952 GetWord( myString, end, true );
1954 end = GetBegPos( RESULT_IND );
1955 return myString.SubString( 1, end - 1 );
1958 //================================================================================
1960 * \brief Return substring of python command looking like ResultValue = Obj.Meth()
1961 * \retval const TCollection_AsciiString & - ResultValue substring
1963 //================================================================================
1965 const TCollection_AsciiString & _pyCommand::GetResultValue()
1967 if ( GetBegPos( RESULT_IND ) == UNKNOWN )
1969 int begPos = myString.Location( "=", 1, Length() );
1971 myRes = GetWord( myString, begPos, false );
1974 SetBegPos( RESULT_IND, begPos );
1979 //================================================================================
1981 * \brief Return number of python command result value ResultValue = Obj.Meth()
1984 //================================================================================
1986 const int _pyCommand::GetNbResultValues()
1990 int endPos = myString.Location( "=", 1, Length() );
1991 TCollection_AsciiString str = "";
1992 while ( begPos < endPos) {
1993 str = GetWord( myString, begPos, true );
1994 begPos = begPos+ str.Length();
2001 //================================================================================
2003 * \brief Return substring of python command looking like
2004 * ResultValue1 , ResultValue1,... = Obj.Meth() with res index
2005 * \retval const TCollection_AsciiString & - ResultValue with res index substring
2007 //================================================================================
2008 const TCollection_AsciiString & _pyCommand::GetResultValue(int res)
2012 int endPos = myString.Location( "=", 1, Length() );
2013 while ( begPos < endPos) {
2014 myRes = GetWord( myString, begPos, true );
2015 begPos = begPos + myRes.Length();
2018 myRes.RemoveAll('[');myRes.RemoveAll(']');
2024 return theEmptyString;
2027 //================================================================================
2029 * \brief Return substring of python command looking like ResVal = Object.Meth()
2030 * \retval const TCollection_AsciiString & - Object substring
2032 //================================================================================
2034 const TCollection_AsciiString & _pyCommand::GetObject()
2036 if ( GetBegPos( OBJECT_IND ) == UNKNOWN )
2039 int begPos = GetBegPos( RESULT_IND ) + myRes.Length();
2041 begPos = myString.Location( "=", 1, Length() ) + 1;
2042 // is '=' in the string argument (for example, name) or not
2043 int nb1 = 0; // number of ' character at the left of =
2044 int nb2 = 0; // number of " character at the left of =
2045 for ( int i = 1; i < begPos-1; i++ ) {
2046 if ( myString.Value( i )=='\'' )
2048 else if ( myString.Value( i )=='"' )
2051 // if number of ' or " is not divisible by 2,
2052 // then get an object at the start of the command
2053 if ( nb1 % 2 != 0 || nb2 % 2 != 0 )
2056 myObj = GetWord( myString, begPos, true );
2057 // check if object is complex,
2058 // so far consider case like "smesh.smesh.Method()"
2059 if ( int bracketPos = myString.Location( "(", begPos, Length() )) {
2060 //if ( bracketPos==0 ) bracketPos = Length();
2061 int dotPos = begPos+myObj.Length();
2062 while ( dotPos+1 < bracketPos ) {
2063 if ( int pos = myString.Location( ".", dotPos+1, bracketPos ))
2068 if ( dotPos > begPos+myObj.Length() )
2069 myObj = myString.SubString( begPos, dotPos-1 );
2072 SetBegPos( OBJECT_IND, begPos );
2078 //================================================================================
2080 * \brief Return substring of python command looking like ResVal = Obj.Method()
2081 * \retval const TCollection_AsciiString & - Method substring
2083 //================================================================================
2085 const TCollection_AsciiString & _pyCommand::GetMethod()
2087 if ( GetBegPos( METHOD_IND ) == UNKNOWN )
2090 int begPos = GetBegPos( OBJECT_IND ) + myObj.Length();
2091 bool forward = true;
2093 begPos = myString.Location( "(", 1, Length() ) - 1;
2097 myMeth = GetWord( myString, begPos, forward );
2098 SetBegPos( METHOD_IND, begPos );
2104 //================================================================================
2106 * \brief Return substring of python command looking like ResVal = Obj.Meth(Arg1,...)
2107 * \retval const TCollection_AsciiString & - Arg<index> substring
2109 //================================================================================
2111 const TCollection_AsciiString & _pyCommand::GetArg( int index )
2113 if ( GetBegPos( ARG1_IND ) == UNKNOWN )
2116 int begPos = GetBegPos( METHOD_IND ) + myMeth.Length();
2118 begPos = myString.Location( "(", 1, Length() ) + 1;
2120 int i = 0, prevLen = 0, nbNestings = 0;
2121 while ( begPos != EMPTY ) {
2123 if( myString.Value( begPos ) == '(' )
2125 // check if we are looking at the closing parenthesis
2126 while ( begPos <= Length() && isspace( myString.Value( begPos )))
2128 if ( begPos > Length() )
2130 if ( myString.Value( begPos ) == ')' ) {
2132 if( nbNestings == 0 )
2135 myArgs.Append( GetWord( myString, begPos, true, true ));
2136 SetBegPos( ARG1_IND + i, begPos );
2137 prevLen = myArgs.Last().Length();
2139 myArgs.Remove( myArgs.Length() ); // no more args
2143 if ( myArgs.Length() < index )
2144 return theEmptyString;
2145 return myArgs( index );
2148 //================================================================================
2150 * \brief Check if char is a word part
2151 * \param c - The character to check
2152 * \retval bool - The check result
2154 //================================================================================
2156 static inline bool isWord(const char c, const bool dotIsWord)
2159 !isspace(c) && c != ',' && c != '=' && c != ')' && c != '(' && ( dotIsWord || c != '.');
2162 //================================================================================
2164 * \brief Looks for a word in the string and returns word's beginning
2165 * \param theString - The input string
2166 * \param theStartPos - The position to start the search, returning word's beginning
2167 * \param theForward - The search direction
2168 * \retval TCollection_AsciiString - The found word
2170 //================================================================================
2172 TCollection_AsciiString _pyCommand::GetWord( const TCollection_AsciiString & theString,
2174 const bool theForward,
2175 const bool dotIsWord )
2177 int beg = theStartPos, end = theStartPos;
2178 theStartPos = EMPTY;
2179 if ( beg < 1 || beg > theString.Length() )
2180 return theEmptyString;
2182 if ( theForward ) { // search forward
2184 while ( beg <= theString.Length() && !isWord( theString.Value( beg ), dotIsWord))
2186 if ( beg > theString.Length() )
2187 return theEmptyString; // no word found
2190 char begChar = theString.Value( beg );
2191 if ( begChar == '"' || begChar == '\'' || begChar == '[') {
2192 char endChar = ( begChar == '[' ) ? ']' : begChar;
2193 // end is at the corresponding quoting mark or bracket
2194 while ( end < theString.Length() &&
2195 ( theString.Value( end ) != endChar || theString.Value( end-1 ) == '\\'))
2199 while ( end <= theString.Length() && isWord( theString.Value( end ), dotIsWord))
2204 else { // search backward
2206 while ( end > 0 && !isWord( theString.Value( end ), dotIsWord))
2209 return theEmptyString; // no word found
2211 char endChar = theString.Value( end );
2212 if ( endChar == '"' || endChar == '\'' ) {
2213 // beg is at the corresponding quoting mark
2215 ( theString.Value( beg ) != endChar || theString.Value( beg-1 ) == '\\'))
2219 while ( beg > 0 && isWord( theString.Value( beg ), dotIsWord))
2225 //cout << theString << " ---- " << beg << " - " << end << endl;
2226 return theString.SubString( beg, end );
2229 //================================================================================
2231 * \brief Look for position where not space char is
2232 * \param theString - The string
2233 * \param thePos - The position to search from and which returns result
2234 * \retval bool - false if there are only space after thePos in theString
2238 //================================================================================
2240 bool _pyCommand::SkipSpaces( const TCollection_AsciiString & theString, int & thePos )
2242 if ( thePos < 1 || thePos > theString.Length() )
2245 while ( thePos <= theString.Length() && isspace( theString.Value( thePos )))
2248 return thePos <= theString.Length();
2251 //================================================================================
2253 * \brief Modify a part of the command
2254 * \param thePartIndex - The index of the part
2255 * \param thePart - The new part string
2256 * \param theOldPart - The old part
2258 //================================================================================
2260 void _pyCommand::SetPart(int thePartIndex, const TCollection_AsciiString& thePart,
2261 TCollection_AsciiString& theOldPart)
2263 int pos = GetBegPos( thePartIndex );
2264 if ( pos <= Length() && theOldPart != thePart)
2266 TCollection_AsciiString seperator;
2268 pos = GetBegPos( thePartIndex + 1 );
2269 if ( pos < 1 ) return;
2270 switch ( thePartIndex ) {
2271 case RESULT_IND: seperator = " = "; break;
2272 case OBJECT_IND: seperator = "."; break;
2273 case METHOD_IND: seperator = "()"; break;
2277 myString.Remove( pos, theOldPart.Length() );
2278 if ( !seperator.IsEmpty() )
2279 myString.Insert( pos , seperator );
2280 myString.Insert( pos, thePart );
2281 // update starting positions of the following parts
2282 int posDelta = thePart.Length() + seperator.Length() - theOldPart.Length();
2283 for ( int i = thePartIndex + 1; i <= myBegPos.Length(); ++i ) {
2284 if ( myBegPos( i ) > 0 )
2285 myBegPos( i ) += posDelta;
2287 theOldPart = thePart;
2291 //================================================================================
2293 * \brief Set agrument
2294 * \param index - The argument index, it counts from 1
2295 * \param theArg - The argument string
2297 //================================================================================
2299 void _pyCommand::SetArg( int index, const TCollection_AsciiString& theArg)
2302 int argInd = ARG1_IND + index - 1;
2303 int pos = GetBegPos( argInd );
2304 if ( pos < 1 ) // no index-th arg exist, append inexistent args
2306 // find a closing parenthesis
2307 if ( GetNbArgs() != 0 && index <= GetNbArgs() ) {
2308 int lastArgInd = GetNbArgs();
2309 pos = GetBegPos( ARG1_IND + lastArgInd - 1 ) + GetArg( lastArgInd ).Length();
2310 while ( pos > 0 && pos <= Length() && myString.Value( pos ) != ')' )
2315 while ( pos > 0 && myString.Value( pos ) != ')' )
2318 if ( pos < 1 || myString.Value( pos ) != ')' ) { // no parentheses at all
2322 while ( myArgs.Length() < index ) {
2323 if ( myArgs.Length() )
2324 myString.Insert( pos++, "," );
2325 myArgs.Append("None");
2326 myString.Insert( pos, myArgs.Last() );
2327 SetBegPos( ARG1_IND + myArgs.Length() - 1, pos );
2328 pos += myArgs.Last().Length();
2331 SetPart( argInd, theArg, myArgs( index ));
2334 //================================================================================
2336 * \brief Empty arg list
2338 //================================================================================
2340 void _pyCommand::RemoveArgs()
2342 if ( int pos = myString.Location( '(', 1, Length() ))
2343 myString.Trunc( pos );
2346 if ( myBegPos.Length() >= ARG1_IND )
2347 myBegPos.Remove( ARG1_IND, myBegPos.Length() );
2350 //================================================================================
2352 * \brief Set dependent commands after this one
2354 //================================================================================
2356 bool _pyCommand::SetDependentCmdsAfter() const
2358 bool orderChanged = false;
2359 list< Handle(_pyCommand)>::const_reverse_iterator cmd = myDependentCmds.rbegin();
2360 for ( ; cmd != myDependentCmds.rend(); ++cmd ) {
2361 if ( (*cmd)->GetOrderNb() < GetOrderNb() ) {
2362 orderChanged = true;
2363 theGen->SetCommandAfter( *cmd, this );
2364 (*cmd)->SetDependentCmdsAfter();
2367 return orderChanged;
2369 //================================================================================
2371 * \brief Insert accessor method after theObjectID
2372 * \param theObjectID - id of the accessed object
2373 * \param theAcsMethod - name of the method giving access to the object
2374 * \retval bool - false if theObjectID is not found in the command string
2376 //================================================================================
2378 bool _pyCommand::AddAccessorMethod( _pyID theObjectID, const char* theAcsMethod )
2380 if ( !theAcsMethod )
2382 // start object search from the object, i.e. ignore result
2384 int beg = GetBegPos( OBJECT_IND );
2385 if ( beg < 1 || beg > Length() )
2388 while (( beg = myString.Location( theObjectID, beg, Length() )))
2390 // check that theObjectID is not just a part of a longer ID
2391 int afterEnd = beg + theObjectID.Length();
2392 Standard_Character c = myString.Value( afterEnd );
2393 if ( !isalnum( c ) && c != ':' ) {
2394 // check if accessor method already present
2396 myString.Location( (char*) theAcsMethod, afterEnd, Length() ) != afterEnd+1) {
2398 int oldLen = Length();
2399 myString.Insert( afterEnd, (char*) theAcsMethod );
2400 myString.Insert( afterEnd, "." );
2401 // update starting positions of the parts following the modified one
2402 int posDelta = Length() - oldLen;
2403 for ( int i = 1; i <= myBegPos.Length(); ++i ) {
2404 if ( myBegPos( i ) > afterEnd )
2405 myBegPos( i ) += posDelta;
2410 beg = afterEnd; // is a part - next search
2415 //================================================================================
2417 * \brief Return method name giving access to an interaface object wrapped by python class
2418 * \retval const char* - method name
2420 //================================================================================
2422 const char* _pyObject::AccessorMethod() const
2426 //================================================================================
2428 * \brief Return ID of a father
2430 //================================================================================
2432 _pyID _pyObject::FatherID(const _pyID & childID)
2434 int colPos = childID.SearchFromEnd(':');
2436 return childID.SubString( 1, colPos-1 );
2440 //================================================================================
2442 * \brief SelfEraser erases creation command if no more it's commands invoked
2444 //================================================================================
2446 void _pySelfEraser::Flush()
2448 if ( GetNbCalls() == 0 )
2449 GetCreationCmd()->Clear();
2452 //================================================================================
2454 * \brief count invoked commands
2456 //================================================================================
2458 void _pySubMesh::Process( const Handle(_pyCommand)& theCommand )
2460 _pyObject::Process(theCommand); // count calls of Process()
2461 GetCreationCmd()->AddDependantCmd( theCommand );
2464 //================================================================================
2466 * \brief Clear creatin command if no commands invoked
2468 //================================================================================
2470 void _pySubMesh::Flush()
2472 if ( GetNbCalls() == 0 ) // move to the end of all commands
2473 theGen->GetLastCommand()->AddDependantCmd( GetCreationCmd() );
2474 else if ( !myCreator.IsNull() )
2475 // move to be just after creator
2476 myCreator->GetCreationCmd()->AddDependantCmd( GetCreationCmd() );