mirror of
https://git.salome-platform.org/gitpub/modules/geom.git
synced 2025-01-28 10:30:34 +05:00
309 lines
11 KiB
C++
309 lines
11 KiB
C++
// Copyright (C) 2007-2015 CEA/DEN, EDF R&D, OPEN CASCADE
|
|
//
|
|
// Copyright (C) 2003-2007 OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
|
|
// CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
|
|
//
|
|
// This library is free software; you can redistribute it and/or
|
|
// modify it under the terms of the GNU Lesser General Public
|
|
// License as published by the Free Software Foundation; either
|
|
// version 2.1 of the License, or (at your option) any later version.
|
|
//
|
|
// This library is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
// Lesser General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU Lesser General Public
|
|
// License along with this library; if not, write to the Free Software
|
|
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
//
|
|
// See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
|
|
//
|
|
|
|
#ifdef WIN32
|
|
#pragma warning( disable:4786 )
|
|
#endif
|
|
|
|
#include <Standard_Stream.hxx>
|
|
|
|
#include "GEOM_Gen_i.hh"
|
|
|
|
#include <TCollection_AsciiString.hxx>
|
|
#include <TCollection_ExtendedString.hxx>
|
|
#include <TColStd_HSequenceOfAsciiString.hxx>
|
|
#include <Resource_DataMapOfAsciiStringAsciiString.hxx>
|
|
|
|
#include <vector>
|
|
#include <string>
|
|
|
|
//=======================================================================
|
|
//function : RemoveTabulation
|
|
//purpose :
|
|
//=======================================================================
|
|
void RemoveTabulation( TCollection_AsciiString& theScript )
|
|
{
|
|
std::string aString( theScript.ToCString() );
|
|
std::string::size_type aPos = 0;
|
|
while( aPos < aString.length() )
|
|
{
|
|
aPos = aString.find( "\n\t", aPos );
|
|
if( aPos == std::string::npos )
|
|
break;
|
|
aString.replace( aPos, 2, "\n" );
|
|
aPos++;
|
|
}
|
|
theScript = aString.c_str();
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : ConvertV6toV7
|
|
//purpose : dump elements are stored when study is saved,
|
|
// and kept when study is reloaded.
|
|
// When a study from an older version is loaded,
|
|
// the dump must be post processed in case of syntax modification.
|
|
// In V7.2, geompy.GEOM --> GEOM
|
|
//=======================================================================
|
|
void ConvertV6toV7( TCollection_AsciiString& theScript )
|
|
{
|
|
std::string aString( theScript.ToCString() );
|
|
std::string::size_type aPos = 0;
|
|
while( aPos < aString.length() )
|
|
{
|
|
aPos = aString.find( "geompy.GEOM", aPos );
|
|
if( aPos == std::string::npos )
|
|
break;
|
|
aString.replace( aPos, 11, "GEOM" );
|
|
aPos++;
|
|
}
|
|
theScript = aString.c_str();
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : DumpPython
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
Engines::TMPFile* GEOM_Gen_i::DumpPython(CORBA::Object_ptr theStudy,
|
|
CORBA::Boolean isPublished,
|
|
CORBA::Boolean isMultiFile,
|
|
CORBA::Boolean& isValidScript)
|
|
{
|
|
SALOMEDS::Study_var aStudy = SALOMEDS::Study::_narrow(theStudy);
|
|
if(CORBA::is_nil(aStudy))
|
|
return new Engines::TMPFile(0);
|
|
|
|
SALOMEDS::SObject_var aSO = aStudy->FindComponent(ComponentDataType());
|
|
if(CORBA::is_nil(aSO))
|
|
return new Engines::TMPFile(0);
|
|
|
|
TObjectData objData;
|
|
std::vector<TObjectData> objectDataVec;
|
|
|
|
TVariablesList aVariableMap;
|
|
|
|
SALOMEDS::ChildIterator_var Itr = aStudy->NewChildIterator(aSO);
|
|
for(Itr->InitEx(true); Itr->More(); Itr->Next()) {
|
|
SALOMEDS::SObject_var aValue = Itr->Value();
|
|
CORBA::String_var IOR = aValue->GetIOR();
|
|
if(strlen(IOR.in()) > 0) {
|
|
CORBA::Object_var obj = _orb->string_to_object(IOR);
|
|
GEOM::GEOM_BaseObject_var GO = GEOM::GEOM_BaseObject::_narrow(obj);
|
|
if(!CORBA::is_nil(GO)) {
|
|
CORBA::String_var aName = aValue->GetName();
|
|
CORBA::String_var anEntry = GO->GetEntry();
|
|
CORBA::String_var aStudyEntry = aValue->GetID();
|
|
objData._name = aName.in();
|
|
objData._entry = anEntry.in();
|
|
objData._studyEntry = aStudyEntry.in();
|
|
|
|
//Find Drawable Attribute
|
|
SALOMEDS::GenericAttribute_var aGenAttr;
|
|
if(aValue->FindAttribute(aGenAttr, "AttributeDrawable") ) {
|
|
SALOMEDS::AttributeDrawable_var aDrw = SALOMEDS::AttributeDrawable::_narrow(aGenAttr);
|
|
objData._unpublished = !aDrw->IsDrawable();
|
|
} else {
|
|
objData._unpublished = false;
|
|
}
|
|
|
|
objectDataVec.push_back( objData );
|
|
|
|
//Find attribute with list of used notebook variables
|
|
SALOMEDS::GenericAttribute_var anAttr;
|
|
SALOMEDS::AttributeString_var anAttrStr;
|
|
if(aValue->FindAttribute(anAttr,"AttributeString")){
|
|
anAttrStr = SALOMEDS::AttributeString::_narrow(anAttr);
|
|
SALOMEDS::ListOfListOfStrings_var aSections = aStudy->ParseVariables(anAttrStr->Value());
|
|
ObjectStates* aStates = new ObjectStates();
|
|
for(int i = 0; i < aSections->length(); i++) {
|
|
TState aState;
|
|
SALOMEDS::ListOfStrings aListOfVars = aSections[i];
|
|
for(int j = 0; j < aListOfVars.length(); j++) {
|
|
bool isVar = aStudy->IsVariable(aListOfVars[j].in());
|
|
TVariable aVar = TVariable( (char*)aListOfVars[j].in(), isVar );
|
|
aState.push_back(aVar);
|
|
}
|
|
aStates->AddState(aState);
|
|
}
|
|
aVariableMap.insert(std::make_pair(TCollection_AsciiString(anEntry),aStates));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
TCollection_AsciiString aScript;
|
|
aScript += _impl->DumpPython(aStudy->StudyId(), objectDataVec, aVariableMap, isPublished, isMultiFile, isValidScript);
|
|
|
|
if (isPublished)
|
|
{
|
|
|
|
SALOMEDS::AttributeParameter_var ap = aStudy->GetModuleParameters("Interface Applicative",
|
|
ComponentDataType(),
|
|
-1);
|
|
if(!CORBA::is_nil(ap)) {
|
|
//Add the id parameter of the object
|
|
std::vector<TObjectData>::iterator it = objectDataVec.begin();
|
|
for( ;it != objectDataVec.end(); it++ ) {
|
|
|
|
//1. Encode entry
|
|
if ( (*it)._studyEntry.Length() < 7 ) continue;
|
|
std::string tail( (*it)._studyEntry.ToCString(), 6, (*it)._studyEntry.Length()-1 );
|
|
std::string newEntry(ComponentDataType());
|
|
newEntry+=("_"+tail);
|
|
|
|
CORBA::String_var anEntry = CORBA::string_dup(newEntry.c_str());
|
|
|
|
if( ap->IsSet(anEntry, 6) ) { //6 Means string array, see SALOMEDS_Attributes.idl AttributeParameter interface
|
|
if ( GetDumpName((*it)._studyEntry.ToCString()) == NULL )
|
|
continue;
|
|
std::string idCommand = std::string("geompy.getObjectID(") + GetDumpName((*it)._studyEntry.ToCString()) + std::string(")");
|
|
SALOMEDS::StringSeq_var aSeq= ap->GetStrArray(anEntry);
|
|
int oldLenght = aSeq->length();
|
|
aSeq->length(oldLenght+2);
|
|
aSeq[oldLenght] = CORBA::string_dup("_PT_OBJECT_ID_");
|
|
aSeq[oldLenght + 1] = CORBA::string_dup(idCommand.c_str());
|
|
ap->SetStrArray( anEntry, aSeq );
|
|
}
|
|
}
|
|
}
|
|
|
|
TCollection_AsciiString aDirScript, aNodeName, aNodeEntry, aFatherName, aFatherEntry;
|
|
Resource_DataMapOfAsciiStringAsciiString aNameToEntry;
|
|
SALOMEDS::UseCaseBuilder_var useCaseBuilder = aStudy->GetUseCaseBuilder();
|
|
SALOMEDS::UseCaseIterator_var Itr = useCaseBuilder->GetUseCaseIterator(aSO);
|
|
SALOMEDS::SObject_var aNodeSO;
|
|
SALOMEDS::SObject_var aFatherSO;
|
|
SALOMEDS::GenericAttribute_var anIDAttr;
|
|
for(Itr->Init(true); Itr->More(); Itr->Next()) {
|
|
aFatherName.Clear();
|
|
aFatherEntry.Clear();
|
|
aNodeSO = Itr->Value();
|
|
// get father info
|
|
aFatherSO = useCaseBuilder->GetFather( aNodeSO );
|
|
if ( aFatherSO->FindAttribute(anIDAttr, "AttributeLocalID") ) {
|
|
SALOMEDS::AttributeLocalID_var aLocalID = SALOMEDS::AttributeLocalID::_narrow(anIDAttr);
|
|
if ( aLocalID->Value() == 999 ) {
|
|
aFatherName = aFatherSO->GetName();
|
|
aFatherEntry = aFatherSO->GetID();
|
|
_impl->healPyName( aFatherName, aFatherEntry, aNameToEntry);
|
|
}
|
|
aLocalID->UnRegister();
|
|
}
|
|
// get node info
|
|
if ( aNodeSO->FindAttribute(anIDAttr, "AttributeLocalID") ) {
|
|
SALOMEDS::AttributeLocalID_var aLocalID = SALOMEDS::AttributeLocalID::_narrow(anIDAttr);
|
|
if ( aLocalID->Value() == 999 ) {
|
|
// the node is folder
|
|
aNodeName = aNodeSO->GetName();
|
|
aNodeEntry = aNodeSO->GetID();
|
|
_impl->healPyName( aNodeName, aNodeEntry, aNameToEntry);
|
|
aDirScript += aNodeName;
|
|
aDirScript += " = geompy.NewFolder('";
|
|
aDirScript += aNodeSO->GetName();
|
|
aDirScript += "'";
|
|
if ( !aFatherName.IsEmpty() && !aFatherEntry.IsEmpty() ) {
|
|
// the folder takes place under another folder
|
|
aDirScript += ", ";
|
|
aDirScript += aFatherName;
|
|
}
|
|
aDirScript += ")\n";
|
|
aNameToEntry.Bind( aNodeName, aNodeEntry );
|
|
}
|
|
aLocalID->UnRegister();
|
|
} else if ( !aFatherName.IsEmpty() && !aFatherEntry.IsEmpty() ) {
|
|
// the node is Geom object under folder
|
|
aDirScript += "geompy.PutToFolder(";
|
|
if ( GetDumpName( aNodeSO->GetID() ) == NULL )
|
|
continue;
|
|
aDirScript += GetDumpName( aNodeSO->GetID() );
|
|
aDirScript += ", ";
|
|
aDirScript += aFatherName;
|
|
aDirScript += ")\n";
|
|
}
|
|
}
|
|
if ( !aDirScript.IsEmpty() )
|
|
{
|
|
aScript += "\n\t### Folders and it's content\n";
|
|
aScript += aDirScript;
|
|
}
|
|
//Output the script that sets up the visual parameters.
|
|
char* script = aStudy->GetDefaultScript(ComponentDataType(), "\t");
|
|
if (script && strlen(script) > 0) {
|
|
aScript += "\n\t### Store presentation parameters of displayed objects\n";
|
|
aScript += script;
|
|
CORBA::string_free(script);
|
|
}
|
|
}
|
|
|
|
if( isMultiFile )
|
|
aScript += "\n\tpass";
|
|
aScript += "\n";
|
|
|
|
if( !isMultiFile ) // remove unnecessary tabulation
|
|
RemoveTabulation( aScript );
|
|
|
|
ConvertV6toV7( aScript ); //convert scripts related to studies saved in SALOME V6 and older
|
|
|
|
int aLen = aScript.Length();
|
|
unsigned char* aBuffer = new unsigned char[aLen+1];
|
|
strcpy((char*)aBuffer, aScript.ToCString());
|
|
|
|
CORBA::Octet* anOctetBuf = (CORBA::Octet*)aBuffer;
|
|
Engines::TMPFile_var aStreamFile = new Engines::TMPFile(aLen+1, aLen+1, anOctetBuf, 1);
|
|
|
|
return aStreamFile._retn();
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : GetDumpName
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
char* GEOM_Gen_i::GetDumpName (const char* theStudyEntry)
|
|
{
|
|
const char* name = _impl->GetDumpName( theStudyEntry );
|
|
if ( name && strlen( name ) > 0 )
|
|
return strdup( name );
|
|
|
|
return NULL;
|
|
}
|
|
|
|
//=======================================================================
|
|
//function : GetAllDumpNames
|
|
//purpose :
|
|
//=======================================================================
|
|
|
|
GEOM::string_array* GEOM_Gen_i::GetAllDumpNames()
|
|
{
|
|
Handle(TColStd_HSequenceOfAsciiString) aHSeq = _impl->GetAllDumpNames();
|
|
int i = 0, aLen = aHSeq->Length();
|
|
|
|
GEOM::string_array_var seq = new GEOM::string_array();
|
|
seq->length(aLen);
|
|
|
|
for (; i < aLen; i++) {
|
|
seq[i] = CORBA::string_dup(aHSeq->Value(i + 1).ToCString());
|
|
}
|
|
|
|
return seq._retn();
|
|
}
|