// File : SMESH_2smeshpy.cxx // Created : Fri Nov 18 13:20:10 2005 // Author : Edward AGAPOV (eap) #include "SMESH_2smeshpy.hxx" #include "SMESH_Gen_i.hxx" #include "utilities.h" #include "SMESH_PythonDump.hxx" #include "Resource_DataMapOfAsciiStringAsciiString.hxx" IMPLEMENT_STANDARD_HANDLE (_pyObject ,Standard_Transient); IMPLEMENT_STANDARD_HANDLE (_pyCommand ,Standard_Transient); IMPLEMENT_STANDARD_HANDLE (_pyGen ,_pyObject); IMPLEMENT_STANDARD_HANDLE (_pyMesh ,_pyObject); IMPLEMENT_STANDARD_HANDLE (_pyHypothesis ,_pyObject); IMPLEMENT_STANDARD_HANDLE (_pyAlgorithm ,_pyHypothesis); IMPLEMENT_STANDARD_HANDLE (_pyComplexParamHypo,_pyHypothesis); IMPLEMENT_STANDARD_HANDLE (_pyNumberOfSegmentsHyp,_pyHypothesis); IMPLEMENT_STANDARD_RTTIEXT(_pyObject ,Standard_Transient); IMPLEMENT_STANDARD_RTTIEXT(_pyCommand ,Standard_Transient); IMPLEMENT_STANDARD_RTTIEXT(_pyGen ,_pyObject); IMPLEMENT_STANDARD_RTTIEXT(_pyMesh ,_pyObject); IMPLEMENT_STANDARD_RTTIEXT(_pyHypothesis ,_pyObject); IMPLEMENT_STANDARD_RTTIEXT(_pyAlgorithm ,_pyHypothesis); IMPLEMENT_STANDARD_RTTIEXT(_pyComplexParamHypo,_pyHypothesis); IMPLEMENT_STANDARD_RTTIEXT(_pyNumberOfSegmentsHyp,_pyHypothesis); using namespace std; using SMESH::TPythonDump; /*! * \brief Container of commands into which the initial script is split. * It also contains data coresponding to SMESH_Gen contents */ static Handle(_pyGen) theGen; static TCollection_AsciiString theEmptyString; //#define DUMP_CONVERSION #if !defined(_DEBUG_) && defined(DUMP_CONVERSION) #undef DUMP_CONVERSION #endif //================================================================================ /*! * \brief Convert python script using commands of smesh.py * \param theScript - Input script * \retval TCollection_AsciiString - Convertion result */ //================================================================================ TCollection_AsciiString SMESH_2smeshpy::ConvertScript(const TCollection_AsciiString& theScript, Resource_DataMapOfAsciiStringAsciiString& theEntry2AccessorMethod) { theGen = new _pyGen( theEntry2AccessorMethod ); // split theScript into separate commands int from = 1, end = theScript.Length(), to; while ( from < end && ( to = theScript.Location( "\n", from, end ))) { if ( to != from ) // cut out and store a command theGen->AddCommand( theScript.SubString( from, to - 1 )); from = to + 1; } // finish conversion theGen->Flush(); #ifdef DUMP_CONVERSION cout << endl << " ######## RESULT ######## " << endl<< endl; #endif // concat commands back into a script TCollection_AsciiString aScript; list< Handle(_pyCommand) >::iterator cmd = theGen->GetCommands().begin(); for ( ; cmd != theGen->GetCommands().end(); ++cmd ) { #ifdef DUMP_CONVERSION cout << "## COM " << (*cmd)->GetOrderNb() << ": "<< (*cmd)->GetString() << endl; #endif if ( !(*cmd)->IsEmpty() ) { aScript += "\n"; aScript += (*cmd)->GetString(); } } aScript += "\n"; theGen.Nullify(); return aScript; } //================================================================================ /*! * \brief _pyGen constructor */ //================================================================================ _pyGen::_pyGen(Resource_DataMapOfAsciiStringAsciiString& theEntry2AccessorMethod) : _pyObject( new _pyCommand( TPythonDump::SMESHGenName(), 0 )), myID2AccessorMethod( theEntry2AccessorMethod ) { myNbCommands = 0; // make that GetID() to return TPythonDump::SMESHGenName() GetCreationCmd()->GetString() += "="; } //================================================================================ /*! * \brief Convert a command using a specific converter * \param theCommand - the command to convert * \retval bool - convertion result */ //================================================================================ void _pyGen::AddCommand( const TCollection_AsciiString& theCommand) { // store theCommand in the sequence myCommands.push_back( new _pyCommand( theCommand, ++myNbCommands )); Handle(_pyCommand) aCommand = myCommands.back(); #ifdef DUMP_CONVERSION cout << "## COM " << myNbCommands << ": "<< aCommand->GetString() << endl; #endif _pyID objID = aCommand->GetObject(); if ( objID.IsEmpty() ) return; // SMESH_Gen method? if ( objID == this->GetID() ) { this->Process( aCommand ); return; } // SMESH_Mesh method? map< _pyID, Handle(_pyMesh) >::iterator id_mesh = myMeshes.find( objID ); if ( id_mesh != myMeshes.end() ) { id_mesh->second->Process( aCommand ); return; } // SMESH_Hypothesis method list< Handle(_pyHypothesis) >::iterator hyp = myHypos.begin(); for ( ; hyp != myHypos.end(); ++hyp ) if ( !(*hyp)->IsAlgo() && objID == (*hyp)->GetID() ) (*hyp)->Process( aCommand ); // Mesh provides SMESH_IDSource interface used in SMESH_MeshEditor. // Add access to wrapped mesh if ( objID == TPythonDump::MeshEditorName() ) { // in all SMESH_MeshEditor's commands, a SMESH_IDSource is the first arg id_mesh = myMeshes.find( aCommand->GetArg( 1 )); if ( id_mesh != myMeshes.end() ) aCommand->SetArg( 1 , aCommand->GetArg( 1 ) + ".GetMesh()" ); } } //================================================================================ /*! * \brief Convert the command or remember it for later conversion * \param theCommand - The python command calling a method of SMESH_Gen */ //================================================================================ void _pyGen::Process( const Handle(_pyCommand)& theCommand ) { // there are methods to convert: // CreateMesh( shape ) // CreateHypothesis( theHypType, theLibName ) // Compute( mesh, geom ) if ( theCommand->GetMethod() == "CreateMesh" ) { Handle(_pyMesh) mesh = new _pyMesh( theCommand ); myMeshes.insert( make_pair( mesh->GetID(), mesh )); return; } // CreateHypothesis() if ( theCommand->GetMethod() == "CreateHypothesis" ) { myHypos.push_back( _pyHypothesis::NewHypothesis( theCommand )); return; } // smeshgen.Compute( mesh, geom ) --> mesh.Compute() if ( theCommand->GetMethod() == "Compute" ) { const _pyID& meshID = theCommand->GetArg( 1 ); map< _pyID, Handle(_pyMesh) >::iterator id_mesh = myMeshes.find( meshID ); if ( id_mesh != myMeshes.end() ) { theCommand->SetObject( meshID ); theCommand->RemoveArgs(); id_mesh->second->Flush(); return; } } // smeshgen.Method() --> smesh.smesh.Method() theCommand->SetObject( SMESH_2smeshpy::GenName() ); } //================================================================================ /*! * \brief Convert the remembered commands */ //================================================================================ void _pyGen::Flush() { map< _pyID, Handle(_pyMesh) >::iterator id_mesh = myMeshes.begin(); for ( ; id_mesh != myMeshes.end(); ++id_mesh ) if ( ! id_mesh->second.IsNull() ) id_mesh->second->Flush(); list< Handle(_pyHypothesis) >::iterator hyp = myHypos.begin(); for ( ; hyp != myHypos.end(); ++hyp ) if ( !hyp->IsNull() ) { (*hyp)->Flush(); // smeshgen.CreateHypothesis() --> smesh.smesh.CreateHypothesis() if ( !(*hyp)->GetCreationCmd()->IsEmpty() ) (*hyp)->GetCreationCmd()->SetObject( SMESH_2smeshpy::GenName() ); } } //================================================================================ /*! * \brief Find hypothesis by ID (entry) * \param theHypID - The hypothesis ID * \retval Handle(_pyHypothesis) - The found hypothesis */ //================================================================================ Handle(_pyHypothesis) _pyGen::FindHyp( const _pyID& theHypID ) { list< Handle(_pyHypothesis) >::iterator hyp = myHypos.begin(); for ( ; hyp != myHypos.end(); ++hyp ) if ( !hyp->IsNull() && theHypID == (*hyp)->GetID() ) return *hyp; return Handle(_pyHypothesis)(); } //================================================================================ /*! * \brief Find algorithm the created algorithm * \param theGeom - The shape ID the algorithm was created on * \param theMesh - The mesh ID that created the algorithm * \param dim - The algo dimension * \retval Handle(_pyHypothesis) - The found algo */ //================================================================================ Handle(_pyHypothesis) _pyGen::FindAlgo( const _pyID& theGeom, const _pyID& theMesh, const TCollection_AsciiString& theAlgoType ) { list< Handle(_pyHypothesis) >::iterator hyp = myHypos.begin(); for ( ; hyp != myHypos.end(); ++hyp ) if ( !hyp->IsNull() && (*hyp)->IsAlgo() && (*hyp)->GetType() == theAlgoType && (*hyp)->GetGeom() == theGeom && (*hyp)->GetMesh() == theMesh ) return *hyp; return 0; } //================================================================================ /*! * \brief Change order of commands in the script * \param theCmd1 - One command * \param theCmd2 - Another command */ //================================================================================ void _pyGen::ExchangeCommands( Handle(_pyCommand) theCmd1, Handle(_pyCommand) theCmd2 ) { list< Handle(_pyCommand) >::iterator pos1, pos2; pos1 = find( myCommands.begin(), myCommands.end(), theCmd1 ); pos2 = find( myCommands.begin(), myCommands.end(), theCmd2 ); myCommands.insert( pos1, theCmd2 ); myCommands.insert( pos2, theCmd1 ); myCommands.erase( pos1 ); myCommands.erase( pos2 ); int nb1 = theCmd1->GetOrderNb(); theCmd1->SetOrderNb( theCmd2->GetOrderNb() ); theCmd2->SetOrderNb( nb1 ); } //================================================================================ /*! * \brief Set one command after the other * \param theCmd - Command to move * \param theAfterCmd - Command ater which to insert the first one */ //================================================================================ void _pyGen::SetCommandAfter( Handle(_pyCommand) theCmd, Handle(_pyCommand) theAfterCmd ) { list< Handle(_pyCommand) >::iterator pos; pos = find( myCommands.begin(), myCommands.end(), theCmd ); myCommands.erase( pos ); pos = find( myCommands.begin(), myCommands.end(), theAfterCmd ); myCommands.insert( ++pos, theCmd ); int i = 1; for ( pos = myCommands.begin(); pos != myCommands.end(); ++pos) (*pos)->SetOrderNb( i++ ); } //================================================================================ /*! * \brief Set method to access to object wrapped with python class * \param theID - The wrapped object entry * \param theMethod - The accessor method */ //================================================================================ void _pyGen::SetAccessorMethod(const _pyID& theID, const char* theMethod ) { myID2AccessorMethod.Bind( theID, (char*) theMethod ); } //================================================================================ /*! * \brief Find out type of geom group * \param grpID - The geom group entry * \retval int - The type */ //================================================================================ static bool sameGroupType( const _pyID& grpID, const TCollection_AsciiString& theType) { // define group type as smesh.Mesh.Group() does int type = -1; SALOMEDS::Study_var study = SMESH_Gen_i::GetSMESHGen()->GetCurrentStudy(); SALOMEDS::SObject_var aSObj = study->FindObjectID( grpID.ToCString() ); if ( !aSObj->_is_nil() ) { GEOM::GEOM_Object_var aGeomObj = GEOM::GEOM_Object::_narrow( aSObj->GetObject() ); if ( !aGeomObj->_is_nil() ) { switch ( aGeomObj->GetShapeType() ) { case GEOM::VERTEX: type = SMESH::NODE; break; case GEOM::EDGE: type = SMESH::EDGE; break; case GEOM::FACE: type = SMESH::FACE; break; case GEOM::SOLID: case GEOM::SHELL: type = SMESH::VOLUME; break; case GEOM::COMPOUND: { GEOM::GEOM_Gen_var aGeomGen = SMESH_Gen_i::GetSMESHGen()->GetGeomEngine(); if ( !aGeomGen->_is_nil() ) { GEOM::GEOM_IGroupOperations_var aGrpOp = aGeomGen->GetIGroupOperations( study->StudyId() ); if ( !aGrpOp->_is_nil() ) { switch ( aGrpOp->GetType( aGeomObj )) { case TopAbs_VERTEX: type = SMESH::NODE; break; case TopAbs_EDGE: type = SMESH::EDGE; break; case TopAbs_FACE: type = SMESH::FACE; break; case TopAbs_SOLID: type = SMESH::VOLUME; break; default:; } } } } default:; } } } if ( type < 0 ) { MESSAGE("Type of the group " << grpID << " not found"); return false; } if ( theType.IsIntegerValue() ) return type == theType.IntegerValue(); switch ( type ) { case SMESH::NODE: return theType.Location( "NODE", 1, theType.Length() ); case SMESH::EDGE: return theType.Location( "EDGE", 1, theType.Length() ); case SMESH::FACE: return theType.Location( "FACE", 1, theType.Length() ); case SMESH::VOLUME: return theType.Location( "VOLUME", 1, theType.Length() ); default:; } return false; } //================================================================================ /*! * \brief * \param theCreationCmd - */ //================================================================================ _pyMesh::_pyMesh(const Handle(_pyCommand) theCreationCmd): _pyObject(theCreationCmd) { // convert my creation command Handle(_pyCommand) creationCmd = GetCreationCmd(); creationCmd->SetObject( SMESH_2smeshpy::SmeshpyName() ); creationCmd->SetMethod( "Mesh" ); theGen->SetAccessorMethod( GetID(), "GetMesh()" ); } //================================================================================ /*! case brief: default: * \param theCommand - Engine method called for this mesh */ //================================================================================ void _pyMesh::Process( const Handle(_pyCommand)& theCommand ) { // smesh.py wraps the following methods: // // 1. GetSubMesh(geom, name) + AddHypothesis(geom, algo) // --> in Mesh_Algorithm.Create(mesh, geom, hypo, so) // 2. AddHypothesis(geom, hyp) // --> in Mesh_Algorithm.Hypothesis(hyp, args, so) // 3. CreateGroupFromGEOM(type, name, grp) // --> in Mesh.Group(grp, name="") // 4. ExportToMED(f, opt, version) // --> in Mesh.ExportToMED( f, version, opt=0 ) // 5. ExportMED(f, opt) // --> in Mesh.ExportMED( f,opt=0 ) // 6. ExportDAT(f) // --> in Mesh.ExportDAT( f ) // 7. ExportUNV(f) // --> in Mesh.ExportUNV(f) // 8. ExportSTL(f, ascii) // --> in Mesh.ExportSTL(f, ascii=1) const TCollection_AsciiString method = theCommand->GetMethod(); if ( method == "GetSubMesh" ) { mySubmeshes.push_back( theCommand ); } else if ( method == "AddHypothesis" ) { // mesh.AddHypothesis(geom, HYPO ) myAddHypCmds.push_back( theCommand ); // set mesh to hypo const _pyID& hypID = theCommand->GetArg( 2 ); Handle(_pyHypothesis) hyp = theGen->FindHyp( hypID ); if ( !hyp.IsNull() && hyp->GetMesh().IsEmpty() ) hyp->SetMesh( this->GetID() ); } else if ( method == "CreateGroupFromGEOM" ) {// (type, name, grp) _pyID grp = theCommand->GetArg( 3 ); if ( sameGroupType( grp, theCommand->GetArg( 1 )) ) { // --> Group(grp) theCommand->SetMethod( "Group" ); theCommand->RemoveArgs(); theCommand->SetArg( 1, grp ); } else { AddMeshAccess( theCommand ); } } else if ( method == "ExportToMED" ) {//(f, opt, version) // --> (f, version, opt) _pyID opt = theCommand->GetArg( 2 ); _pyID ver = theCommand->GetArg( 3 ); theCommand->SetArg( 2, ver ); theCommand->SetArg( 3, opt ); } else if ( method == "RemoveHypothesis" ) // (geom, hyp) { const _pyID & hypID = theCommand->GetArg( 2 ); // check if this mesh still has corresponding addition command bool hasAddCmd = false; list< Handle(_pyCommand) >::iterator cmd = myAddHypCmds.begin(); while ( cmd != myAddHypCmds.end() ) { // AddHypothesis(geom, hyp) if ( hypID == (*cmd)->GetArg( 2 )) { // erase both commands theCommand->Clear(); (*cmd)->Clear(); cmd = myAddHypCmds.erase( cmd ); hasAddCmd = true; } else { ++cmd; } } if ( ! hasAddCmd ) { // access to wrapped mesh AddMeshAccess( theCommand ); // access to wrapped algo Handle(_pyHypothesis) hyp = theGen->FindHyp( hypID ); if ( !hyp.IsNull() && hyp->IsAlgo() && hyp->IsWrapped() ) theCommand->SetArg( 2, theCommand->GetArg( 2 ) + ".GetAlgorithm()" ); } } else { // apply theCommand to the mesh wrapped by smeshpy mesh AddMeshAccess( theCommand ); } } //================================================================================ /*! * \brief Convert creation and addition of all algos and hypos */ //================================================================================ void _pyMesh::Flush() { list < Handle(_pyCommand) >::iterator cmd, cmd2; map< _pyID, Handle(_pyCommand) > algo2additionCmd; // try to convert algo addition like this: // mesh.AddHypothesis(geom, ALGO ) --> ALGO = mesh.Algo() for ( cmd = myAddHypCmds.begin(); cmd != myAddHypCmds.end(); ++cmd ) { Handle(_pyCommand) addCmd = *cmd; const _pyID& algoID = addCmd->GetArg( 2 ); Handle(_pyHypothesis) algo = theGen->FindHyp( algoID ); if ( algo.IsNull() || !algo->IsAlgo() ) continue; // try to convert _pyID geom = addCmd->GetArg( 1 ); if ( algo->Addition2Creation( addCmd, this->GetID() )) // OK { algo2additionCmd[ algo->GetID() ] = addCmd; if ( geom != GetGeom() ) // local algo { // mesh.AddHypothesis(geom, ALGO ) --> mesh.AlgoMethod(geom) addCmd->SetArg( addCmd->GetNbArgs() + 1, TCollection_AsciiString( "geom=" ) + geom ); // sm = mesh.GetSubMesh(geom, name) --> sm = ALGO.GetSubMesh() for ( cmd2 = mySubmeshes.begin(); cmd2 != mySubmeshes.end(); ++cmd2 ) { Handle(_pyCommand) subCmd = *cmd2; if ( geom == subCmd->GetArg( 1 )) { subCmd->SetObject( algo->GetID() ); subCmd->RemoveArgs(); if ( addCmd->GetOrderNb() > subCmd->GetOrderNb() ) theGen->SetCommandAfter( subCmd, addCmd ); } } } } else // ALGO was already created { // mesh.AddHypothesis(geom, ALGO ) --> mesh.GetMesh().AddHypothesis(geom, ALGO ) AddMeshAccess( addCmd ); // mesh.GetMesh().AddHypothesis(geom, ALGO ) -> // mesh.GetMesh().AddHypothesis(geom, ALGO.GetAlgorithm() ) addCmd->SetArg( 2, addCmd->GetArg( 2 ) + ".GetAlgorithm()" ); } } // try to convert hypo addition like this: // mesh.AddHypothesis(geom, HYPO ) --> HYPO = algo.Hypo() for ( cmd = myAddHypCmds.begin(); cmd != myAddHypCmds.end(); ++cmd ) { Handle(_pyCommand) addCmd = *cmd; const _pyID& hypID = addCmd->GetArg( 2 ); Handle(_pyHypothesis) hyp = theGen->FindHyp( hypID ); if ( hyp.IsNull() || hyp->IsAlgo() ) continue; const _pyID& geom = addCmd->GetArg( 1 ); // find algo created on for this mesh Handle(_pyHypothesis) algo = theGen->FindAlgo( geom, this->GetID(), hyp->GetType() ); //_pyID algoID = algo.IsNull() ? "" : algo->GetID(); if ( !algo.IsNull() && hyp->Addition2Creation( addCmd, this->GetID() )) // OK { addCmd->SetObject( algo->GetID() ); Handle(_pyCommand) algoCmd = algo2additionCmd[ algo->GetID() ]; if ( !algoCmd.IsNull() && algoCmd->GetOrderNb() > addCmd->GetOrderNb() ) // algo was created later than hyp theGen->ExchangeCommands( algoCmd, addCmd ); } else { AddMeshAccess( addCmd ); } } // sm = mesh.GetSubMesh(geom, name) --> sm = mesh.GetMesh().GetSubMesh(geom, name) for ( cmd = mySubmeshes.begin(); cmd != mySubmeshes.end(); ++cmd ) { Handle(_pyCommand) subCmd = *cmd; if ( subCmd->GetNbArgs() > 0 ) AddMeshAccess( subCmd ); } myAddHypCmds.clear(); mySubmeshes.clear(); } //================================================================================ /*! * \brief _pyHypothesis constructor * \param theCreationCmd - */ //================================================================================ _pyHypothesis::_pyHypothesis(const Handle(_pyCommand)& theCreationCmd): _pyObject( theCreationCmd ) { myDim = myIsAlgo = /*myIsLocal = */myIsWrapped = myIsConverted = false; } //================================================================================ /*! * \brief Creates algorithm or hypothesis * \param theCreationCmd - The engine command creating a hypothesis * \retval Handle(_pyHypothesis) - Result _pyHypothesis */ //================================================================================ Handle(_pyHypothesis) _pyHypothesis::NewHypothesis( const Handle(_pyCommand)& theCreationCmd) { // theCreationCmd: CreateHypothesis( "theHypType", "theLibName" ) ASSERT (( theCreationCmd->GetMethod() == "CreateHypothesis")); Handle(_pyHypothesis) hyp, algo; // "theHypType" const TCollection_AsciiString & hypTypeWithQuotes = theCreationCmd->GetArg( 1 ); if ( hypTypeWithQuotes.IsEmpty() ) return hyp; // theHypType TCollection_AsciiString hypType = hypTypeWithQuotes.SubString( 2, hypTypeWithQuotes.Length() - 1 ); algo = new _pyAlgorithm( theCreationCmd ); hyp = new _pyHypothesis( theCreationCmd ); // 1D Regular_1D ---------- if ( hypType == "Regular_1D" ) { algo->myDim = 1; algo->myCreationMethod = "Segment"; } else if ( hypType == "LocalLength" ) { hyp->myDim = 1; hyp->myCreationMethod = "LocalLength"; hyp->myType = "Regular_1D"; hyp->myArgMethods.Append( "SetLength" ); } else if ( hypType == "NumberOfSegments" ) { hyp = new _pyNumberOfSegmentsHyp( theCreationCmd ); hyp->myDim = 1; hyp->myCreationMethod = "NumberOfSegments"; hyp->myType = "Regular_1D"; hyp->myArgMethods.Append( "SetNumberOfSegments" ); hyp->myArgMethods.Append( "SetScaleFactor" ); } else if ( hypType == "Arithmetic1D" ) { hyp = new _pyComplexParamHypo( theCreationCmd ); hyp->myDim = 1; hyp->myCreationMethod = "Arithmetic1D"; hyp->myType = "Regular_1D"; } else if ( hypType == "StartEndLength" ) { hyp = new _pyComplexParamHypo( theCreationCmd ); hyp->myDim = 1; hyp->myCreationMethod = "StartEndLength"; hyp->myType = "Regular_1D"; } else if ( hypType == "Deflection1D" ) { hyp->myDim = 1; hyp->myCreationMethod = "Deflection1D"; hyp->myArgMethods.Append( "SetDeflection" ); hyp->myType = "Regular_1D"; } else if ( hypType == "Propagation" ) { hyp->myDim = 1; hyp->myCreationMethod = "Propagation"; hyp->myType = "Regular_1D"; } else if ( hypType == "AutomaticLength" ) { hyp->myDim = 1; hyp->myCreationMethod = "AutomaticLength"; hyp->myType = "Regular_1D"; } // 1D Python_1D ---------- else if ( hypType == "Python_1D" ) { algo->myDim = 1; algo->myCreationMethod = "Segment"; algo->myArgs.Append( "algo=smesh.PYTHON"); } else if ( hypType == "PythonSplit1D" ) { hyp->myDim = 1; hyp->myCreationMethod = "PythonSplit1D"; hyp->myType = "Python_1D"; hyp->myArgMethods.Append( "SetNumberOfSegments"); hyp->myArgMethods.Append( "SetPythonLog10RatioFunction"); } // 2D ---------- else if ( hypType == "MEFISTO_2D" ) { algo->myDim = 2; algo->myCreationMethod = "Triangle"; } else if ( hypType == "MaxElementArea" ) { hyp->myDim = 2; hyp->myCreationMethod = "MaxElementArea"; hyp->myType = "MEFISTO_2D"; hyp->myArgMethods.Append( "SetMaxElementArea"); } else if ( hypType == "LengthFromEdges" ) { hyp->myDim = 2; hyp->myCreationMethod = "LengthFromEdges"; hyp->myType = "MEFISTO_2D"; } else if ( hypType == "Quadrangle_2D" ) { algo->myDim = 2; algo->myCreationMethod = "Quadrangle"; } else if ( hypType == "QuadranglePreference" ) { hyp->myDim = 2; hyp->myCreationMethod = "QuadranglePreference"; hyp->myType = "MEFISTO_2D"; } // 3D ---------- else if ( hypType == "NETGEN_3D") { algo->myDim = 3; algo->myCreationMethod = "Tetrahedron"; algo->myArgs.Append( "algo=smesh.NETGEN" ); } else if ( hypType == "MaxElementVolume") { hyp->myDim = 3; hyp->myCreationMethod = "MaxElementVolume"; hyp->myType = "NETGEN_3D"; hyp->myArgMethods.Append( "SetMaxElementVolume" ); } else if ( hypType == "GHS3D_3D" ) { algo->myDim = 3; algo->myCreationMethod = "Tetrahedron"; algo->myArgs.Append( "algo=smesh.GHS3D" ); } else if ( hypType == "Hexa_3D" ) { algo->myDim = 3; algo->myCreationMethod = "Hexahedron"; } if ( algo->GetDim() ) { algo->myType = hypType; return algo; } return hyp; } //================================================================================ /*! * \brief Convert the command adding a hypothesis to mesh into a smesh command * \param theCmd - The command like mesh.AddHypothesis( geom, hypo ) * \param theAlgo - The algo that can create this hypo * \retval bool - false if the command cant be converted */ //================================================================================ bool _pyHypothesis::Addition2Creation( const Handle(_pyCommand)& theCmd, const _pyID& theMesh) { ASSERT(( theCmd->GetMethod() == "AddHypothesis" )); if ( !IsWrappable( theMesh )) return false; myIsWrapped = true; if ( myIsWrapped ) { // mesh.AddHypothesis(geom,hyp) --> hyp = theMesh.myCreationMethod(args) theCmd->SetResultValue( GetID() ); theCmd->SetObject( theMesh ); theCmd->SetMethod( myCreationMethod ); // set args theCmd->RemoveArgs(); for ( int i = 1; i <= myArgs.Length(); ++i ) { if ( !myArgs( i ).IsEmpty() ) theCmd->SetArg( i, myArgs( i )); else theCmd->SetArg( i, "[]"); } // clear commands setting arg values list < Handle(_pyCommand) >::iterator argCmd = myArgCommands.begin(); for ( ; argCmd != myArgCommands.end(); ++argCmd ) (*argCmd)->Clear(); } else { // // set arg commands after hypo creation // list::iterator argCmd = myArgCommands.begin(); // for ( ; argCmd != myArgCommands.end(); ++argCmd ) // if ( !(*argCmd)->IsEmpty() && GetCommandNb() > (*argCmd)->GetOrderNb() ) // theGen->ExchangeCommands( GetCreationCmd(), *argCmd ); } // set unknown arg commands after hypo creation Handle(_pyCommand) afterCmd = myIsWrapped ? theCmd : GetCreationCmd(); list::iterator cmd = myUnknownCommands.begin(); for ( ; cmd != myUnknownCommands.end(); ++cmd ) { if ( !(*cmd)->IsEmpty() && afterCmd->GetOrderNb() > (*cmd)->GetOrderNb() ) { theGen->SetCommandAfter( *cmd, afterCmd ); afterCmd = *cmd; } } myArgCommands.clear(); myUnknownCommands.clear(); return myIsWrapped; } //================================================================================ /*! * \brief Remember hypothesis parameter values * \param theCommand - The called hypothesis method */ //================================================================================ void _pyHypothesis::Process( const Handle(_pyCommand)& theCommand) { ASSERT( !myIsAlgo ); // set args for ( int i = 1; i <= myArgMethods.Length(); ++i ) { if ( myArgMethods( i ) == theCommand->GetMethod() ) { while ( myArgs.Length() < i ) myArgs.Append( "[]" ); myArgs( i ) = theCommand->GetArg( 1 ); // arg value myArgCommands.push_back( theCommand ); return; } } myUnknownCommands.push_back( theCommand ); } //================================================================================ /*! * \brief Finish conversion */ //================================================================================ void _pyHypothesis::Flush() { if ( IsWrapped() ) GetCreationCmd()->Clear(); } //================================================================================ /*! * \brief Remember hypothesis parameter values * \param theCommand - The called hypothesis method */ //================================================================================ void _pyComplexParamHypo::Process( const Handle(_pyCommand)& theCommand) { // ex: hyp.SetLength(start, 1) // hyp.SetLength(end, 0) ASSERT(( theCommand->GetMethod() == "SetLength" )); ASSERT(( theCommand->GetArg( 2 ).IsIntegerValue() )); int i = 2 - theCommand->GetArg( 2 ).IntegerValue(); while ( myArgs.Length() < i ) myArgs.Append( "[]" ); myArgs( i ) = theCommand->GetArg( 1 ); // arg value myArgCommands.push_back( theCommand ); } //================================================================================ /*! * \brief additionally to Addition2Creation, clears SetDistrType() command * \param theCmd - AddHypothesis() command * \param theMesh - mesh to which a hypothesis is added * \retval bool - convertion result */ //================================================================================ bool _pyNumberOfSegmentsHyp::Addition2Creation( const Handle(_pyCommand)& theCmd, const _pyID& theMesh) { if ( IsWrappable( theMesh ) && myArgs.Length() > 1 ) { list::iterator cmd = myUnknownCommands.begin(); for ( ; cmd != myUnknownCommands.end(); ++cmd ) { // clear SetDistrType() if ( (*cmd)->GetString().Location( "SetDistrType", 1, (*cmd)->Length() )) (*cmd)->Clear(); } } return _pyHypothesis::Addition2Creation( theCmd, theMesh ); } //================================================================================ /*! * \brief _pyAlgorithm constructor * \param theCreationCmd - The command like "algo = smeshgen.CreateHypothesis(type,lib)" */ //================================================================================ _pyAlgorithm::_pyAlgorithm(const Handle(_pyCommand)& theCreationCmd) : _pyHypothesis( theCreationCmd ) { myIsAlgo = true; } //================================================================================ /*! * \brief Convert the command adding an algorithm to mesh * \param theCmd - The command like mesh.AddHypothesis( geom, algo ) * \param theMesh - The mesh needing this algo * \retval bool - false if the command cant be converted */ //================================================================================ bool _pyAlgorithm::Addition2Creation( const Handle(_pyCommand)& theCmd, const _pyID& theMeshID) { if ( IsWrappable( theMeshID )) { myGeom = theCmd->GetArg( 1 ); // mesh.AddHypothesis(geom,algo) --> theMeshID.myCreationMethod() if ( _pyHypothesis::Addition2Creation( theCmd, theMeshID )) { theGen->SetAccessorMethod( GetID(), "GetAlgorithm()" ); return true; } } return false; } //================================================================================ /*! * \brief Return starting position of a part of python command * \param thePartIndex - The index of command part * \retval int - Part position */ //================================================================================ int _pyCommand::GetBegPos( int thePartIndex ) { if ( IsEmpty() ) return EMPTY; if ( myBegPos.Length() < thePartIndex ) return UNKNOWN; return myBegPos( thePartIndex ); } //================================================================================ /*! * \brief Store starting position of a part of python command * \param thePartIndex - The index of command part * \param thePosition - Part position */ //================================================================================ void _pyCommand::SetBegPos( int thePartIndex, int thePosition ) { while ( myBegPos.Length() < thePartIndex ) myBegPos.Append( UNKNOWN ); myBegPos( thePartIndex ) = thePosition; } //================================================================================ /*! * \brief Return substring of python command looking like ResultValue = Obj.Meth() * \retval const TCollection_AsciiString & - ResultValue substring */ //================================================================================ const TCollection_AsciiString & _pyCommand::GetResultValue() { if ( GetBegPos( RESULT_IND ) == UNKNOWN ) { int begPos = myString.Location( "=", 1, Length() ); if ( begPos ) myRes = GetWord( myString, begPos, false ); else begPos = EMPTY; SetBegPos( RESULT_IND, begPos ); } return myRes; } //================================================================================ /*! * \brief Return substring of python command looking like ResVal = Object.Meth() * \retval const TCollection_AsciiString & - Object substring */ //================================================================================ const TCollection_AsciiString & _pyCommand::GetObject() { if ( GetBegPos( OBJECT_IND ) == UNKNOWN ) { // beginning int begPos = GetBegPos( RESULT_IND ) + myRes.Length(); if ( begPos < 1 ) begPos = myString.Location( "=", 1, Length() ) + 1; // store myObj = GetWord( myString, begPos, true ); SetBegPos( OBJECT_IND, begPos ); } //SCRUTE(myObj); return myObj; } //================================================================================ /*! * \brief Return substring of python command looking like ResVal = Obj.Method() * \retval const TCollection_AsciiString & - Method substring */ //================================================================================ const TCollection_AsciiString & _pyCommand::GetMethod() { if ( GetBegPos( METHOD_IND ) == UNKNOWN ) { // beginning int begPos = GetBegPos( OBJECT_IND ) + myObj.Length(); bool forward = true; if ( begPos < 1 ) { begPos = myString.Location( "(", 1, Length() ) - 1; forward = false; } // store myMeth = GetWord( myString, begPos, forward ); SetBegPos( METHOD_IND, begPos ); } //SCRUTE(myMeth); return myMeth; } //================================================================================ /*! * \brief Return substring of python command looking like ResVal = Obj.Meth(Arg1,...) * \retval const TCollection_AsciiString & - Arg substring */ //================================================================================ const TCollection_AsciiString & _pyCommand::GetArg( int index ) { if ( GetBegPos( ARG1_IND ) == UNKNOWN ) { // find all args int begPos = GetBegPos( METHOD_IND ) + myMeth.Length(); if ( begPos < 1 ) begPos = myString.Location( "(", 1, Length() ) + 1; int i = 0, prevLen = 0; while ( begPos != EMPTY ) { begPos += prevLen; // check if we are looking at the closing parenthesis while ( begPos <= Length() && isspace( myString.Value( begPos ))) ++begPos; if ( begPos > Length() || myString.Value( begPos ) == ')' ) break; myArgs.Append( GetWord( myString, begPos, true, true )); SetBegPos( ARG1_IND + i, begPos ); prevLen = myArgs.Last().Length(); if ( prevLen == 0 ) myArgs.Remove( myArgs.Length() ); // no more args i++; } } if ( myArgs.Length() < index ) return theEmptyString; return myArgs( index ); } //================================================================================ /*! * \brief Check if char is a word part * \param c - The character to check * \retval bool - The check result */ //================================================================================ static inline bool isWord(const char c, const bool dotIsWord) { return !isspace(c) && c != ',' && c != '=' && c != ')' && c != '(' && ( dotIsWord || c != '.'); } //================================================================================ /*! * \brief Looks for a word in the string and returns word's beginning * \param theString - The input string * \param theStartPos - The position to start the search, returning word's beginning * \param theForward - The search direction * \retval TCollection_AsciiString - The found word */ //================================================================================ TCollection_AsciiString _pyCommand::GetWord( const TCollection_AsciiString & theString, int & theStartPos, const bool theForward, const bool dotIsWord ) { int beg = theStartPos, end = theStartPos; theStartPos = EMPTY; if ( beg < 1 || beg > theString.Length() ) return theEmptyString; if ( theForward ) { // search forward // beg while ( beg <= theString.Length() && !isWord( theString.Value( beg ), dotIsWord)) ++beg; if ( beg > theString.Length() ) return theEmptyString; // no word found // end end = beg + 1; while ( end <= theString.Length() && isWord( theString.Value( end ), dotIsWord)) ++end; --end; } else { // search backward // end while ( end > 0 && !isWord( theString.Value( end ), dotIsWord)) --end; if ( end == 0 ) return theEmptyString; // no word found beg = end - 1; while ( beg > 0 && isWord( theString.Value( beg ), dotIsWord)) --beg; ++beg; } theStartPos = beg; //cout << theString << " ---- " << beg << " - " << end << endl; return theString.SubString( beg, end ); } //================================================================================ /*! * \brief Look for position where not space char is * \param theString - The string * \param thePos - The position to search from and which returns result * \retval bool - false if there are only space after thePos in theString * * */ //================================================================================ bool _pyCommand::SkipSpaces( const TCollection_AsciiString & theString, int & thePos ) { if ( thePos < 1 || thePos > theString.Length() ) return false; while ( thePos <= theString.Length() && isspace( theString.Value( thePos ))) ++thePos; return thePos <= theString.Length(); } //================================================================================ /*! * \brief Modify a part of the command * \param thePartIndex - The index of the part * \param thePart - The new part string * \param theOldPart - The old part */ //================================================================================ void _pyCommand::SetPart(int thePartIndex, const TCollection_AsciiString& thePart, TCollection_AsciiString& theOldPart) { int pos = GetBegPos( thePartIndex ); if ( pos <= Length() && theOldPart != thePart) { TCollection_AsciiString seperator; if ( pos < 1 ) { pos = GetBegPos( thePartIndex + 1 ); if ( pos < 1 ) return; switch ( thePartIndex ) { case RESULT_IND: seperator = " = "; break; case OBJECT_IND: seperator = "."; break; case METHOD_IND: seperator = "()"; break; default:; } } myString.Remove( pos, theOldPart.Length() ); if ( !seperator.IsEmpty() ) myString.Insert( pos , seperator ); myString.Insert( pos, thePart ); // update starting positions of the following parts int posDelta = thePart.Length() + seperator.Length() - theOldPart.Length(); for ( int i = thePartIndex + 1; i <= myBegPos.Length(); ++i ) { if ( myBegPos( i ) > 0 ) myBegPos( i ) += posDelta; } theOldPart = thePart; } } //================================================================================ /*! * \brief Set agrument * \param index - The argument index, it counts from 1 * \param theArg - The argument string */ //================================================================================ void _pyCommand::SetArg( int index, const TCollection_AsciiString& theArg) { FindAllArgs(); int argInd = ARG1_IND + index - 1; int pos = GetBegPos( argInd ); if ( pos < 1 ) // no index-th arg exist, append inexistent args { // find a closing parenthesis pos = Length(); while ( pos > 0 && myString.Value( pos ) != ')' ) --pos; if ( pos == 0 ) { // no parentheses at all myString += "()"; pos = Length(); } while ( myArgs.Length() < index ) { if ( myArgs.Length() ) myString.Insert( pos++, "," ); myArgs.Append("None"); myString.Insert( pos, myArgs.Last() ); SetBegPos( ARG1_IND + myArgs.Length() - 1, pos ); pos += myArgs.Last().Length(); } } SetPart( argInd, theArg, myArgs( index )); } //================================================================================ /*! * \brief Empty arg list */ //================================================================================ void _pyCommand::RemoveArgs() { if ( int pos = myString.Location( '(', 1, Length() )) myString.Trunc( pos ); myString += ")"; myArgs.Clear(); if ( myBegPos.Length() >= ARG1_IND ) myBegPos.Remove( ARG1_IND, myBegPos.Length() ); }