ForkMaster/smesh
1
0
Fork 0
mirror of https://git.salome-platform.org/gitpub/modules/smesh.git synced 2024-12-28 02:10:35 +05:00
Code Issues Packages Projects Releases Wiki Activity

fix dumping colors:

Browse Source 
use valid python names instead of names in Study
This commit is contained in:
eap 2009-04-21 10:01:50 +00:00
parent 216883b953
commit 99c8d3b659
2 changed files with 28 additions and 44 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

70
src/SMESH_I/SMESH_DumpPython.cxx
View File

@ -461,7 +461,7 @@ Engines::TMPFile* SMESH_Gen_i::DumpPython (CORBA::Object_ptr theStudy,
// Add trace of API methods calls and replace study entries by names // Add trace of API methods calls and replace study entries by names
TCollection_AsciiString aScript = TCollection_AsciiString aScript =
"### This file is generated by SALOME automatically by dump python functionality of SMESH component\n\n"; "### This file is generated by SALOME automatically by dump python functionality of SMESH component\n\n";
aScript += DumpPython_impl(aStudy->StudyId(), aMap, aMapNames, aScript += DumpPython_impl(aStudy, aMap, aMapNames,
isPublished, isValidScript, aSavedTrace); isPublished, isValidScript, aSavedTrace);
//inserting export for Object Names //inserting export for Object Names
@ -650,7 +650,7 @@ namespace {
*/ */
//============================================================================= //=============================================================================
TCollection_AsciiString SMESH_Gen_i::DumpPython_impl TCollection_AsciiString SMESH_Gen_i::DumpPython_impl
(int theStudyID, (SALOMEDS::Study_ptr theStudy,
Resource_DataMapOfAsciiStringAsciiString& theObjectNames, Resource_DataMapOfAsciiStringAsciiString& theObjectNames,
Resource_DataMapOfAsciiStringAsciiString& theNames, Resource_DataMapOfAsciiStringAsciiString& theNames,
bool isPublished, bool isPublished,
@ -707,7 +707,7 @@ TCollection_AsciiString SMESH_Gen_i::DumpPython_impl
} }
// Dump trace of API methods calls // Dump trace of API methods calls
TCollection_AsciiString aNewLines = GetNewPythonLines(theStudyID); TCollection_AsciiString aNewLines = GetNewPythonLines(theStudy->StudyId());
if (aNewLines.Length() > 0) { if (aNewLines.Length() > 0) {
aScript += helper + "\n" + aNewLines; aScript += helper + "\n" + aNewLines;
} }
@ -801,49 +801,33 @@ TCollection_AsciiString SMESH_Gen_i::DumpPython_impl
if (aSeq->Value(aLen) < aScriptLength) if (aSeq->Value(aLen) < aScriptLength)
anUpdatedScript += aScript.SubString(aSeq->Value(aLen) + 1, aScriptLength); anUpdatedScript += aScript.SubString(aSeq->Value(aLen) + 1, aScriptLength);
// Set colors
SMESH_Gen_i* aSMESHGenI = SMESH_Gen_i::GetSMESHGen(); SALOMEDS::SObject_var aComp = theStudy->FindComponent(ComponentDataType());
SALOMEDS::Study_ptr aStudy = aSMESHGenI->GetCurrentStudy(); if( !CORBA::is_nil(aComp) )
if( !CORBA::is_nil(aStudy) )
{ {
SALOMEDS::SObject_var aComp = aStudy->FindComponent(ComponentDataType()); SALOMEDS::ChildIterator_var Itr = theStudy->NewChildIterator(aComp);
if( !CORBA::is_nil(aComp) ) for( Itr->InitEx(true); Itr->More(); Itr->Next() )
{ {
SALOMEDS::ChildIterator_var Itr = aStudy->NewChildIterator(aComp); SALOMEDS::SObject_var aSObj = Itr->Value();
for( Itr->InitEx(true); Itr->More(); Itr->Next() ) SMESH::SMESH_Mesh_var aMesh = SMESH::SMESH_Mesh::_narrow( SObjectToObject( aSObj ) );
// mesh auto color
if( !CORBA::is_nil(aMesh) && aMesh->GetAutoColor() )
{ {
SALOMEDS::SObject_var aSObj = Itr->Value(); CORBA::String_var anEntry = aSObj->GetID();
CORBA::String_var aName = aSObj->GetName(); anUpdatedScript +=
SMESH_Comment("\n\t") << theObjectNames(anEntry.inout()) << ".SetAutoColor(1)";
SMESH::SMESH_Mesh_var aMesh = SMESH::SMESH_Mesh::_narrow( SMESH_Gen_i::SObjectToObject( aSObj ) ); }
if( !CORBA::is_nil(aMesh) ) SMESH::SMESH_GroupBase_var aGroup = SMESH::SMESH_GroupBase::_narrow( SObjectToObject(aSObj));
{ if( !CORBA::is_nil(aGroup) )
bool isAutoColor = aMesh->GetAutoColor(); {
if( isAutoColor ) SALOMEDS::Color aColor = aGroup->GetColor();
{ if ( aColor.R > 0 || aColor.G > 0 || aColor.B > 0 )
anUpdatedScript += "\n\t"; {
anUpdatedScript += (char*)aName.in(); CORBA::String_var anEntry = aSObj->GetID();
anUpdatedScript += ".SetAutoColor(1)"; anUpdatedScript += SMESH_Comment("\n\t")
} << theObjectNames(anEntry.inout()) << ".SetColor(SALOMEDS.Color("
} << aColor.R <<", "<< aColor.G <<", "<< aColor.B <<" ))";
}
SMESH::SMESH_GroupBase_var aGroup = SMESH::SMESH_GroupBase::_narrow( SMESH_Gen_i::SObjectToObject( aSObj ) );
if( !CORBA::is_nil(aGroup) )
{
SALOMEDS::Color aColor = aGroup->GetColor();
if ( aColor.R > 0 || aColor.G > 0 || aColor.B > 0 )
{
anUpdatedScript += "\n\t";
anUpdatedScript += (char*)aName.in();
anUpdatedScript += ".SetColor(SALOMEDS.Color(";
anUpdatedScript += aColor.R;
anUpdatedScript += ",";
anUpdatedScript += aColor.G;
anUpdatedScript += ",";
anUpdatedScript += aColor.B;
anUpdatedScript += "))";
}
}
} }
} }
} }

2
src/SMESH_I/SMESH_Gen_i.hxx
View File

@ -369,7 +369,7 @@ public:
void SavePython (SALOMEDS::Study_ptr theStudy); void SavePython (SALOMEDS::Study_ptr theStudy);
TCollection_AsciiString DumpPython_impl (int theStudyID, TCollection_AsciiString DumpPython_impl (SALOMEDS::Study_ptr theStudy,
Resource_DataMapOfAsciiStringAsciiString& theObjectNames, Resource_DataMapOfAsciiStringAsciiString& theObjectNames,
Resource_DataMapOfAsciiStringAsciiString& theNames, Resource_DataMapOfAsciiStringAsciiString& theNames,
bool isPublished, bool isPublished,