ForkMaster/geom
1
0
Fork 0
mirror of https://git.salome-platform.org/gitpub/modules/geom.git synced 2025-03-17 02:17:55 +05:00
Code Issues Packages Projects Releases Wiki Activity

"Dump Python". Add GetDumpName(), add global names to script

Browse Source
This commit is contained in:
eap 2005-03-29 08:42:50 +00:00
parent b4ab5222d6
commit f724bcc264
2 changed files with 54 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

67
src/GEOM/GEOM_Engine.cxx
View File

@ -368,6 +368,8 @@ TCollection_AsciiString GEOM_Engine::DumpPython(int theDocID,
aScript += "def RebuildData(theStudy):";
aScript += "\n\tgeompy.init_geom(theStudy)";
Standard_Integer posToInertGlobalVars = aScript.Length() + 1;
Handle(TDataStd_TreeNode) aNode, aRoot;
Handle(GEOM_Function) aFunction;
TColStd_MapOfTransient aMap;
@ -387,25 +389,22 @@ TCollection_AsciiString GEOM_Engine::DumpPython(int theDocID,
Resource_DataMapOfAsciiStringAsciiString aEntry2StEntry, aStEntry2Entry;
Resource_DataMapIteratorOfDataMapOfAsciiStringAsciiString anEntryToNameIt;
if ( isPublished )
// build maps entry <-> studyEntry
for (anEntryToNameIt.Initialize( theObjectNames );
anEntryToNameIt.More();
anEntryToNameIt.Next())
{
// build maps entry <-> studyEntry
for (anEntryToNameIt.Initialize( theObjectNames );
anEntryToNameIt.More();
anEntryToNameIt.Next())
{
const TCollection_AsciiString& aEntry = anEntryToNameIt.Key();
// look for an object by entry
TDF_Label L;
TDF_Tool::Label( aDoc->GetData(), aEntry, L );
if ( L.IsNull() ) continue;
Handle(GEOM_Object) obj = GEOM_Object::GetObject( L );
// fill maps
if ( !obj.IsNull() ) {
TCollection_AsciiString aStudyEntry (obj->GetAuxData());
aEntry2StEntry.Bind( aEntry, aStudyEntry);
aStEntry2Entry.Bind( aStudyEntry, aEntry );
}
const TCollection_AsciiString& aEntry = anEntryToNameIt.Key();
// look for an object by entry
TDF_Label L;
TDF_Tool::Label( aDoc->GetData(), aEntry, L );
if ( L.IsNull() ) continue;
Handle(GEOM_Object) obj = GEOM_Object::GetObject( L );
// fill maps
if ( !obj.IsNull() ) {
TCollection_AsciiString aStudyEntry (obj->GetAuxData());
aEntry2StEntry.Bind( aEntry, aStudyEntry);
aStEntry2Entry.Bind( aStudyEntry, aEntry );
}
}
@ -492,9 +491,41 @@ TCollection_AsciiString GEOM_Engine::DumpPython(int theDocID,
anUpdatedScript += "\n\tpass\n";
aValidScript = true;
// fill _studyEntry2NameMap and build globalVars
TCollection_AsciiString globalVars;
_studyEntry2NameMap.Clear();
Resource_DataMapIteratorOfDataMapOfAsciiStringAsciiString aStEntryToEntryIt;
for (aStEntryToEntryIt.Initialize( aStEntry2Entry );
aStEntryToEntryIt.More();
aStEntryToEntryIt.Next() )
{
const TCollection_AsciiString & name = theObjectNames( aStEntryToEntryIt.Value() );
_studyEntry2NameMap.Bind (aStEntryToEntryIt.Key(), name );
if ( !globalVars.IsEmpty() )
globalVars += ", ";
globalVars += name;
}
if ( !globalVars.IsEmpty() ) {
globalVars.Insert( 1, "\n\tglobal " );
anUpdatedScript.Insert( posToInertGlobalVars, globalVars );
}
return anUpdatedScript;
}
//=======================================================================
//function : GetDumpName
//purpose :
//=======================================================================
const char* GEOM_Engine::GetDumpName (const char* theStudyEntry) const
{
if ( _studyEntry2NameMap.IsBound( (char*)theStudyEntry ))
return _studyEntry2NameMap( (char*)theStudyEntry ).ToCString();
return NULL;
}
//===========================================================================
// Internal functions

4
src/GEOM/GEOM_Engine.hxx
View File

@ -69,6 +69,8 @@ class GEOM_Engine
bool isPublished,
bool& aValidScript);
const char* GetDumpName (const char* theStudyEntry) const;
protected:
static void SetEngine(GEOM_Engine* theEngine);
@ -78,6 +80,8 @@ class GEOM_Engine
Interface_DataMapOfIntegerTransient _mapIDDocument;
int _UndoLimit;
GEOM_DataMapOfAsciiStringTransient _objects;
Resource_DataMapOfAsciiStringAsciiString _studyEntry2NameMap;
};
#endif