]> SALOME platform Git repositories - modules/geom.git/blob - src/GEOM_I/GEOM_DumpPython.cc
Salome HOME
Start potring to the VTK OpenGL2 backend.
[modules/geom.git] / src / GEOM_I / GEOM_DumpPython.cc
1 // Copyright (C) 2007-2016  CEA/DEN, EDF R&D, OPEN CASCADE
2 //
3 // Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
5 //
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, or (at your option) any later version.
10 //
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.
15 //
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
19 //
20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
21 //
22
23 #ifdef WIN32
24 #pragma warning( disable:4786 )
25 #endif
26
27 #include <Standard_Stream.hxx>
28
29 #include "GEOM_Gen_i.hh"
30
31 #include <TCollection_AsciiString.hxx>
32 #include <TCollection_ExtendedString.hxx>
33 #include <TColStd_HSequenceOfAsciiString.hxx>
34 #include <Resource_DataMapOfAsciiStringAsciiString.hxx>
35
36 #include <vector>
37 #include <string>
38
39 //=======================================================================
40 //function : RemoveTabulation
41 //purpose  : 
42 //=======================================================================
43 void RemoveTabulation( TCollection_AsciiString& theScript )
44 {
45   std::string aString( theScript.ToCString() );
46   std::string::size_type aPos = 0;
47   while( aPos < aString.length() )
48   {
49     aPos = aString.find( "\n\t", aPos );
50     if( aPos == std::string::npos )
51       break;
52     aString.replace( aPos, 2, "\n" );
53     aPos++;
54   }
55   theScript = aString.c_str();
56 }
57
58 //=======================================================================
59 //function : ConvertV6toV7
60 //purpose  : dump elements are stored when study is saved,
61 //           and kept when study is reloaded.
62 //           When a study from an older version is loaded,
63 //           the dump must be post processed in case of syntax modification.
64 //           In V7.2, geompy.GEOM --> GEOM
65 //=======================================================================
66 void ConvertV6toV7( TCollection_AsciiString& theScript )
67 {
68   std::string aString( theScript.ToCString() );
69   std::string::size_type aPos = 0;
70   while( aPos < aString.length() )
71   {
72     aPos = aString.find( "geompy.GEOM", aPos );
73     if( aPos == std::string::npos )
74       break;
75     aString.replace( aPos, 11, "GEOM" );
76     aPos++;
77   }
78   theScript = aString.c_str();
79 }
80
81 //=======================================================================
82 //function : DumpPython
83 //purpose  : 
84 //=======================================================================
85
86 Engines::TMPFile* GEOM_Gen_i::DumpPython(CORBA::Object_ptr theStudy, 
87                                          CORBA::Boolean isPublished, 
88                                          CORBA::Boolean isMultiFile,
89                                          CORBA::Boolean& isValidScript)
90 {
91   SALOMEDS::Study_var aStudy = SALOMEDS::Study::_narrow(theStudy);
92   if(CORBA::is_nil(aStudy))
93     return new Engines::TMPFile(0);   
94
95   SALOMEDS::SObject_var aSO = aStudy->FindComponent(ComponentDataType());
96   if(CORBA::is_nil(aSO))
97     return new Engines::TMPFile(0);  
98
99   TObjectData objData;
100   std::vector<TObjectData> objectDataVec;
101
102   TVariablesList aVariableMap;
103
104   SALOMEDS::ChildIterator_var Itr = aStudy->NewChildIterator(aSO);
105   for(Itr->InitEx(true); Itr->More(); Itr->Next()) {
106     SALOMEDS::SObject_var aValue = Itr->Value();
107     CORBA::String_var IOR = aValue->GetIOR();
108     if(strlen(IOR.in()) > 0) {
109       CORBA::Object_var obj = _orb->string_to_object(IOR);
110       GEOM::GEOM_BaseObject_var GO = GEOM::GEOM_BaseObject::_narrow(obj);
111       if(!CORBA::is_nil(GO)) {
112         CORBA::String_var aName       = aValue->GetName();
113         CORBA::String_var anEntry     = GO->GetEntry();
114         CORBA::String_var aStudyEntry = aValue->GetID();
115         objData._name       = aName.in();
116         objData._entry      = anEntry.in();
117         objData._studyEntry = aStudyEntry.in();
118
119         //Find Drawable Attribute
120         SALOMEDS::GenericAttribute_var aGenAttr;
121         if(aValue->FindAttribute(aGenAttr, "AttributeDrawable") ) {
122           SALOMEDS::AttributeDrawable_var aDrw = SALOMEDS::AttributeDrawable::_narrow(aGenAttr);
123           objData._unpublished = !aDrw->IsDrawable();
124         } else {
125           objData._unpublished = false;
126         }
127
128         objectDataVec.push_back( objData );
129
130         //Find attribute with list of used notebook variables
131         SALOMEDS::GenericAttribute_var anAttr;
132         SALOMEDS::AttributeString_var anAttrStr;
133         if(aValue->FindAttribute(anAttr,"AttributeString")){
134           anAttrStr = SALOMEDS::AttributeString::_narrow(anAttr);
135           SALOMEDS::ListOfListOfStrings_var aSections = aStudy->ParseVariables(anAttrStr->Value());
136           ObjectStates* aStates = new ObjectStates();
137           for(int i = 0; i < aSections->length(); i++) {
138             TState aState;
139             SALOMEDS::ListOfStrings aListOfVars = aSections[i];
140             for(int j = 0; j < aListOfVars.length(); j++) {
141               bool isVar = aStudy->IsVariable(aListOfVars[j].in());
142               TVariable aVar = TVariable( (char*)aListOfVars[j].in(), isVar );
143               aState.push_back(aVar);
144             }
145             aStates->AddState(aState);
146           }
147           aVariableMap.insert(std::make_pair(TCollection_AsciiString(anEntry),aStates));
148         }
149       }
150     }
151   }
152   
153   TCollection_AsciiString aScript;
154   aScript += _impl->DumpPython(aStudy->StudyId(), objectDataVec, aVariableMap, isPublished, isMultiFile, isValidScript);
155
156   if (isPublished)
157   {
158
159     SALOMEDS::AttributeParameter_var ap = aStudy->GetModuleParameters("Interface Applicative", 
160                                                                       ComponentDataType(),
161                                                                       -1);
162     if(!CORBA::is_nil(ap)) {
163       //Add the id parameter of the object
164       std::vector<TObjectData>::iterator it = objectDataVec.begin();
165       for( ;it != objectDataVec.end(); it++ ) {
166       
167         //1. Encode entry
168         if ( (*it)._studyEntry.Length() < 7 ) continue;
169         std::string tail( (*it)._studyEntry.ToCString(), 6, (*it)._studyEntry.Length()-1 );
170         std::string newEntry(ComponentDataType());
171         newEntry+=("_"+tail);
172         
173         CORBA::String_var anEntry = CORBA::string_dup(newEntry.c_str());
174         
175         if( ap->IsSet(anEntry, 6) ) { //6 Means string array, see SALOMEDS_Attributes.idl AttributeParameter interface
176           if ( GetDumpName((*it)._studyEntry.ToCString()) == NULL ) 
177             continue;
178           std::string idCommand = std::string("geompy.getObjectID(") + GetDumpName((*it)._studyEntry.ToCString()) + std::string(")");
179           SALOMEDS::StringSeq_var aSeq= ap->GetStrArray(anEntry);
180           int oldLenght = aSeq->length();       
181           aSeq->length(oldLenght+2);
182           aSeq[oldLenght] = CORBA::string_dup("_PT_OBJECT_ID_");
183           aSeq[oldLenght + 1] = CORBA::string_dup(idCommand.c_str());
184           ap->SetStrArray( anEntry, aSeq );
185         }        
186       }
187     }
188
189     TCollection_AsciiString aDirScript, aNodeName, aNodeEntry, aFatherName, aFatherEntry;
190     Resource_DataMapOfAsciiStringAsciiString aNameToEntry;
191     SALOMEDS::UseCaseBuilder_var useCaseBuilder = aStudy->GetUseCaseBuilder();
192     SALOMEDS::UseCaseIterator_var Itr = useCaseBuilder->GetUseCaseIterator(aSO);
193     SALOMEDS::SObject_var aNodeSO;
194     SALOMEDS::SObject_var aFatherSO;
195     SALOMEDS::GenericAttribute_var anIDAttr;
196     for(Itr->Init(true); Itr->More(); Itr->Next()) {
197       aFatherName.Clear();
198       aFatherEntry.Clear();
199       aNodeSO = Itr->Value();
200       // get father info
201       aFatherSO = useCaseBuilder->GetFather( aNodeSO );
202       if ( aFatherSO->FindAttribute(anIDAttr, "AttributeLocalID") ) {
203         SALOMEDS::AttributeLocalID_var aLocalID = SALOMEDS::AttributeLocalID::_narrow(anIDAttr);
204         if ( aLocalID->Value() == 999 ) {
205           aFatherName = aFatherSO->GetName();
206           aFatherEntry = aFatherSO->GetID();
207           _impl->healPyName( aFatherName, aFatherEntry, aNameToEntry);
208         }
209         aLocalID->UnRegister();
210       }
211       // get node info
212       if ( aNodeSO->FindAttribute(anIDAttr, "AttributeLocalID") ) {
213         SALOMEDS::AttributeLocalID_var aLocalID = SALOMEDS::AttributeLocalID::_narrow(anIDAttr);
214         if ( aLocalID->Value() == 999 ) {
215           // the node is folder
216           aNodeName = aNodeSO->GetName();
217           aNodeEntry = aNodeSO->GetID();
218           _impl->healPyName( aNodeName, aNodeEntry, aNameToEntry);
219           aDirScript += aNodeName;
220           aDirScript += " = geompy.NewFolder('";
221           aDirScript += aNodeSO->GetName();
222           aDirScript += "'";
223           if ( !aFatherName.IsEmpty() && !aFatherEntry.IsEmpty() ) {
224             // the folder takes place under another folder
225             aDirScript += ", ";
226             aDirScript += aFatherName;
227           }
228           aDirScript += ")\n";
229           aNameToEntry.Bind( aNodeName, aNodeEntry );
230         }
231         aLocalID->UnRegister();
232       } else if ( !aFatherName.IsEmpty() && !aFatherEntry.IsEmpty() ) {
233         // the node is Geom object under folder
234         aDirScript += "geompy.PutToFolder(";
235         if ( GetDumpName( aNodeSO->GetID() ) == NULL )
236           continue;
237         aDirScript += GetDumpName( aNodeSO->GetID() );
238         aDirScript += ", ";
239         aDirScript += aFatherName;
240         aDirScript += ")\n";
241       }
242     }
243     if ( !aDirScript.IsEmpty() )
244     {
245       aScript += "\n\t### Folders and it's content\n";
246       aScript += aDirScript;
247     }
248     //Output the script that sets up the visual parameters.
249     char* script = aStudy->GetDefaultScript(ComponentDataType(), "\t");
250     if (script && strlen(script) > 0) {
251       aScript += "\n\t### Store presentation parameters of displayed objects\n";
252       aScript += script;
253       CORBA::string_free(script);
254     }
255   }
256
257   if( isMultiFile )
258     aScript += "\n\tpass";
259   aScript += "\n";
260
261   if( !isMultiFile ) // remove unnecessary tabulation
262     RemoveTabulation( aScript );
263
264   ConvertV6toV7( aScript ); //convert scripts related to studies saved in SALOME V6 and older
265   
266   int aLen = aScript.Length(); 
267   unsigned char* aBuffer = new unsigned char[aLen+1];
268   strcpy((char*)aBuffer, aScript.ToCString());
269
270   CORBA::Octet* anOctetBuf =  (CORBA::Octet*)aBuffer;
271   Engines::TMPFile_var aStreamFile = new Engines::TMPFile(aLen+1, aLen+1, anOctetBuf, 1); 
272
273   return aStreamFile._retn(); 
274 }
275
276 //=======================================================================
277 //function : GetDumpName
278 //purpose  : 
279 //=======================================================================
280
281 char* GEOM_Gen_i::GetDumpName (const char* theStudyEntry)
282 {
283   const char* name = _impl->GetDumpName( theStudyEntry );
284   if ( name && strlen( name ) > 0 )
285     return strdup( name );
286
287   return NULL;
288 }
289
290 //=======================================================================
291 //function : GetAllDumpNames
292 //purpose  : 
293 //=======================================================================
294
295 GEOM::string_array* GEOM_Gen_i::GetAllDumpNames()
296 {
297   Handle(TColStd_HSequenceOfAsciiString) aHSeq = _impl->GetAllDumpNames();
298   int i = 0, aLen = aHSeq->Length();
299
300   GEOM::string_array_var seq = new GEOM::string_array();
301   seq->length(aLen);
302
303   for (; i < aLen; i++) {
304     seq[i] = CORBA::string_dup(aHSeq->Value(i + 1).ToCString());
305   }
306
307   return seq._retn();
308 }