Salome HOME
Parametrization of the NETGENPLUGIN.
[modules/smesh.git] / src / SMESH_I / SMESH_NoteBook.cxx
1 // Copyright (C) 2008  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
2 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
3 //
4 // This library is free software; you can redistribute it and/or
5 // modify it under the terms of the GNU Lesser General Public
6 // License as published by the Free Software Foundation; either
7 // version 2.1 of the License.
8 //
9 // This library is distributed in the hope that it will be useful
10 // but WITHOUT ANY WARRANTY; without even the implied warranty of
11 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12 // Lesser General Public License for more details.
13 //
14 // You should have received a copy of the GNU Lesser General Public
15 // License along with this library; if not, write to the Free Software
16 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
17 //
18 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
19 //
20 // File      : SMESH_NoteBook.cxx
21 // Author    : Roman NIKOLAEV
22
23 #include "SMESH_2smeshpy.hxx"
24 #include "SMESH_NoteBook.hxx"
25 #include "SMESH_Gen_i.hxx"
26 #include "SMESH_PythonDump.hxx"
27
28 #include <Resource_DataMapOfAsciiStringAsciiString.hxx>
29 #include <TColStd_SequenceOfAsciiString.hxx>
30 #include <TColStd_HSequenceOfInteger.hxx>
31
32 #include <vector>
33 #include <string>
34
35 #ifdef _DEBUG_
36 static int MYDEBUG = 1;
37 #else
38 static int MYDEBUG = 1;
39 #endif
40
41 using namespace std;
42
43
44
45 //================================================================================
46 /*!
47  * \brief Constructor
48  */
49 //================================================================================
50 ObjectStates::ObjectStates(TCollection_AsciiString theType)
51 {
52   _type = theType;
53   _dumpstate = 0;
54 }
55
56 //================================================================================
57 /*!
58  * \brief Destructor
59  */
60 //================================================================================
61 ObjectStates::~ObjectStates()
62 {
63 }
64
65 //================================================================================
66 /*!
67  * \brief Add new object state 
68  * \param theState - Object state (vector of notebook variable)
69  */
70 //================================================================================
71 void ObjectStates::AddState(const TState &theState)
72 {
73   _states.push_back(theState);
74 }
75
76 //================================================================================
77 /*!
78  * \brief Return current object state
79  * \\retval state - Object state (vector of notebook variable)
80  */
81 //================================================================================
82 TState ObjectStates::GetCurrectState() const
83 {
84   if(_states.size() > _dumpstate)
85     return _states[_dumpstate];
86   TState empty;
87   return empty;
88 }
89
90
91 //================================================================================
92 /*!
93  *
94  */
95 //================================================================================
96 TAllStates ObjectStates::GetAllStates() const
97 {
98   return _states;
99 }
100
101 //================================================================================
102 /*!
103  *
104  */
105 //================================================================================
106 void ObjectStates::IncrementState()
107 {
108   _dumpstate++;
109 }
110
111 //================================================================================
112 /*!
113  *
114  */
115 //================================================================================
116 TCollection_AsciiString ObjectStates::GetObjectType() const{
117   return _type;
118 }
119
120
121 //================================================================================
122 /*!
123  * \brief Constructor
124  */
125 //================================================================================
126 LayerDistributionStates::LayerDistributionStates():
127   ObjectStates("LayerDistribution")
128 {
129 }
130 //================================================================================
131 /*!
132  * \brief Destructor
133  */
134 //================================================================================
135 LayerDistributionStates::~LayerDistributionStates()
136 {
137 }
138
139
140 //================================================================================
141 /*!
142  * \brief AddDistribution
143  */
144 //================================================================================
145 void LayerDistributionStates::AddDistribution(const TCollection_AsciiString& theDistribution)
146 {
147   _distributions.insert(pair<TCollection_AsciiString,TCollection_AsciiString>(theDistribution,""));
148 }
149
150 //================================================================================
151 /*!
152  * \brief HasDistribution
153  */
154 //================================================================================
155 bool LayerDistributionStates::HasDistribution(const TCollection_AsciiString& theDistribution) const
156 {
157   return _distributions.find(theDistribution) != _distributions.end();
158 }
159
160 //================================================================================
161 /*!
162  * \brief SetDistributionType
163  */
164 //================================================================================
165 bool LayerDistributionStates::SetDistributionType(const TCollection_AsciiString& theDistribution,
166                                                   const TCollection_AsciiString& theType)
167 {
168   TDistributionMap::iterator it = _distributions.find(theDistribution);
169   if(it == _distributions.end())
170     return false;
171   (*it).second = theType;
172   return true;
173 }
174
175 //================================================================================
176 /*!
177  * \brief GetDistributionType
178  */
179 //================================================================================
180 TCollection_AsciiString LayerDistributionStates::
181 GetDistributionType(const TCollection_AsciiString& theDistribution) const
182 {
183   TDistributionMap::const_iterator it = _distributions.find(theDistribution);
184   return (it == _distributions.end()) ? TCollection_AsciiString() : (*it).second;
185 }
186
187 //================================================================================
188 /*!
189  * \brief Constructor
190  */
191 //================================================================================
192 SMESH_NoteBook::SMESH_NoteBook()
193 {
194   InitObjectMap();
195 }
196
197 //================================================================================
198 /*!
199  * \brief Destructor
200  */
201 //================================================================================
202 SMESH_NoteBook::~SMESH_NoteBook()
203 {
204   TVariablesMap::const_iterator it = _objectMap.begin();
205   for(;it!=_objectMap.end();it++) {
206     if((*it).second)
207       delete (*it).second;
208   }
209 }
210
211 //================================================================================
212 /*!
213  * \brief Replace parameters of the functions on the Salome NoteBook Variables
214  * \param theString - Input string
215  * \retval TCollection_AsciiString - Convertion result
216  */
217 //================================================================================
218 void SMESH_NoteBook::ReplaceVariables()
219 {
220   for(int i=0;i<_commands.size();i++) {
221     Handle(_pyCommand) aCmd = _commands[i];
222     TCollection_AsciiString aMethod = aCmd->GetMethod();
223     TCollection_AsciiString aObject = aCmd->GetObject();
224     TCollection_AsciiString aResultValue = aCmd->GetResultValue();
225     if(MYDEBUG) {
226       cout<<"Command before : "<< aCmd->GetString()<<endl;
227       cout<<"Method : "<< aMethod<<endl;
228       cout<<"Object : "<< aObject<<endl;
229       cout<<"Result : "<< aResultValue<<endl;
230     }
231     
232     // check if method modifies the object itself
233     TVariablesMap::const_iterator it = _objectMap.find(aObject);
234     if(it == _objectMap.end()) // check if method returns a new object
235       it = _objectMap.find(aResultValue);
236     
237     if(it == _objectMap.end()) { // check if method modifies a mesh using mesh editor
238       TMeshEditorMap::const_iterator meIt = myMeshEditors.find(aObject);
239       if(meIt != myMeshEditors.end()) {
240         TCollection_AsciiString aMesh = (*meIt).second;
241         it = _objectMap.find(aMesh);
242       }
243     }
244     
245     if(it != _objectMap.end()) {
246       ObjectStates *aStates = (*it).second;
247       // Case for LocalLength hypothesis
248       if(aStates->GetObjectType().IsEqual("LocalLength") && aStates->GetCurrectState().size() >= 2) {
249         if(aMethod.IsEqual("SetLength")) {
250           if(!aStates->GetCurrectState().at(0).IsEmpty() )
251             aCmd->SetArg(1,aStates->GetCurrectState().at(0));
252           aStates->IncrementState();
253         }
254         else if(aMethod.IsEqual("SetPrecision")) {
255           if(!aStates->GetCurrectState().at(1).IsEmpty() )
256             aCmd->SetArg(1,aStates->GetCurrectState().at(1));
257           aStates->IncrementState();
258         }
259       }
260       
261       // Case for SegmentLengthAroundVertex hypothesis
262       else if(aStates->GetObjectType().IsEqual("SegmentLengthAroundVertex")
263               && aStates->GetCurrectState().size() >= 1) {
264         if(aMethod == "SetLength") {
265           if(!aStates->GetCurrectState().at(0).IsEmpty() )
266             aCmd->SetArg(1,aStates->GetCurrectState().at(0));
267           aStates->IncrementState();
268         }
269       }
270
271       // Case for Arithmetic1D and StartEndLength hypothesis
272       else if(aStates->GetObjectType().IsEqual("Arithmetic1D") || 
273               aStates->GetObjectType().IsEqual("StartEndLength")) {
274         if(aMethod == "SetLength" &&
275            aStates->GetCurrectState().size() >= 2) {
276           if(aCmd->GetArg(2) == "1" && !aStates->GetCurrectState().at(0).IsEmpty())
277             aCmd->SetArg(1,aStates->GetCurrectState().at(0));
278           else if(!aStates->GetCurrectState().at(1).IsEmpty())
279             aCmd->SetArg(1,aStates->GetCurrectState().at(1));
280           aStates->IncrementState();
281         }
282       }
283       
284       //Case for Deflection1D hypothesis
285       else if(aStates->GetObjectType().IsEqual("Deflection1D")){
286         if(aMethod == "SetDeflection" && aStates->GetCurrectState().size() >= 1) {
287           if(!aStates->GetCurrectState().at(0).IsEmpty() )
288             aCmd->SetArg(1,aStates->GetCurrectState().at(0));
289           aStates->IncrementState();
290         }
291       }
292       
293       // Case for LayerDistribution hypothesis (not finished yet)
294       else if(aStates->GetObjectType() == "LayerDistribution") {
295         if(aMethod == "SetLayerDistribution"){
296           LayerDistributionStates* aLDStates = (LayerDistributionStates*)(aStates);
297           aLDStates->AddDistribution(aCmd->GetArg(1));
298         }
299       }
300       
301       // Case for MaxElementArea hypothesis
302       else if(aStates->GetObjectType().IsEqual("MaxElementArea")){
303         if(aMethod == "SetMaxElementArea" && aStates->GetCurrectState().size() >= 1) {
304           if(!aStates->GetCurrectState().at(0).IsEmpty() )
305             aCmd->SetArg(1,aStates->GetCurrectState().at(0));
306           aStates->IncrementState();
307         }
308       }
309
310       // Case for MaxElementVolume hypothesis
311       else if(aStates->GetObjectType().IsEqual("MaxElementVolume")){
312         if(aMethod == "SetMaxElementVolume" && aStates->GetCurrectState().size() >= 1) {
313           if(!aStates->GetCurrectState().at(0).IsEmpty() )
314             aCmd->SetArg(1,aStates->GetCurrectState().at(0));
315           aStates->IncrementState();
316         }
317       }
318       // Case for NETGEN_Parameters_2D hypothesis
319       else if(aStates->GetObjectType().IsEqual("NETGEN_Parameters_2D")){
320         if(aMethod == "SetMaxSize" && aStates->GetCurrectState().size() >= 1) {
321           if(!aStates->GetCurrectState().at(0).IsEmpty() )
322             aCmd->SetArg(1,aStates->GetCurrectState().at(0));
323           aStates->IncrementState();
324         }
325         else if(aMethod == "SetGrowthRate" && aStates->GetCurrectState().size() >= 2) {
326           if(!aStates->GetCurrectState().at(1).IsEmpty() )
327             aCmd->SetArg(1,aStates->GetCurrectState().at(1));
328           aStates->IncrementState();
329         }
330         else if(aMethod == "SetNbSegPerEdge" && aStates->GetCurrectState().size() >= 3) {
331           if(!aStates->GetCurrectState().at(2).IsEmpty() )
332             aCmd->SetArg(1,aStates->GetCurrectState().at(2));
333           aStates->IncrementState();
334         } 
335         else if(aMethod == "SetNbSegPerRadius" && aStates->GetCurrectState().size() >= 4) {
336           if(!aStates->GetCurrectState().at(3).IsEmpty() )
337             aCmd->SetArg(1,aStates->GetCurrectState().at(3));
338           aStates->IncrementState();
339         } 
340       }
341
342       // Case for NumberOfLayers hypothesis
343       else if(aStates->GetObjectType().IsEqual("NumberOfLayers")){
344         if(aMethod == "SetNumberOfLayers" && aStates->GetCurrectState().size() >= 1) {
345           if(!aStates->GetCurrectState().at(0).IsEmpty() )
346             aCmd->SetArg(1,aStates->GetCurrectState().at(0));
347           aStates->IncrementState();
348         }
349       }
350
351       // Case for NumberOfSegments hypothesis
352       else if(aStates->GetObjectType().IsEqual("NumberOfSegments")){
353         if(aMethod == "SetNumberOfSegments" && aStates->GetCurrectState().size() >= 1) {
354           if(!aStates->GetCurrectState().at(0).IsEmpty() )
355             aCmd->SetArg(1,aStates->GetCurrectState().at(0));
356           if(aStates->GetCurrectState().size()==1)
357             aStates->IncrementState();
358         }
359         else if (aMethod == "SetScaleFactor" && aStates->GetCurrectState().size() >= 2) {
360           if(!aStates->GetCurrectState().at(1).IsEmpty() )
361             aCmd->SetArg(1,aStates->GetCurrectState().at(1));
362           aStates->IncrementState();
363         }
364       }
365       
366       else if(aStates->GetObjectType().IsEqual("Mesh")) {
367         TState aCurrentState = aStates->GetCurrectState();
368         int aCurrentStateSize = aCurrentState.size();
369         if(aMethod.IsEqual("Translate") ||
370            aMethod.IsEqual("TranslateMakeGroups") ||
371            aMethod.IsEqual("TranslateMakeMesh")) {
372           bool isVariableFound = false;
373           int anArgIndex = 0;
374           for(int i = 1, n = aCmd->GetNbArgs(); i <= n; i++) {
375             if(aCmd->GetArg(i).IsEqual("SMESH.PointStruct")) {
376               anArgIndex = i+1;
377               break;
378             }
379           }
380           if(anArgIndex > 0) {
381             if(aCurrentStateSize == 3) { // translation by dx, dy, dz
382               for(int j = 0; j < aCurrentStateSize; j++) {
383                 if(!aCurrentState.at(j).IsEmpty()) {
384                   isVariableFound = true;
385                   aCmd->SetArg(anArgIndex+j, aCurrentState.at(j));
386                 }
387               }
388             }
389             else if(aCurrentStateSize == 6) { // translation by x1, x2, y1, y2, z1, z2
390               isVariableFound = true;
391               for(int j = 0; j < 3; j++) {
392                 TCollection_AsciiString anArg = aCmd->GetArg(anArgIndex+j);
393                 TCollection_AsciiString aValue1 = aCurrentState.at(2*j), aValue2 = aCurrentState.at(2*j+1);
394                 bool aV1 = !aValue1.IsEmpty();
395                 bool aV2 = !aValue2.IsEmpty();
396                 double aValue, aCurrentValue = anArg.IsRealValue() ? anArg.RealValue() : 0;
397                 if(aV1 && !aV2) {
398                   if(!GetReal(aValue1, aValue))
399                     aValue = 0;
400                   aValue2 = TCollection_AsciiString( aValue + aCurrentValue );
401                 }
402                 else if(!aV1 && aV2) {
403                   if(!GetReal(aValue2, aValue))
404                     aValue = 0;
405                   aValue1 = TCollection_AsciiString( aValue - aCurrentValue );
406                 }
407                 else if(!aV1 && !aV2) {
408                   aValue1 = TCollection_AsciiString( 0 );
409                   aValue2 = TCollection_AsciiString( aCurrentValue );
410                 }
411                 aCmd->SetArg(anArgIndex+j, aValue1 + ", " + aValue2 );
412               }
413             }
414           }
415           if(isVariableFound) {
416             TCollection_AsciiString aDim;
417             if(aCurrentStateSize == 6)
418               aDim = "6";
419             aCmd->SetArg(anArgIndex - 1, TCollection_AsciiString(SMESH_2smeshpy::SmeshpyName())+".PointStructStr"+aDim);
420             aCmd->SetArg(anArgIndex - 2, TCollection_AsciiString(SMESH_2smeshpy::SmeshpyName())+".DirStructStr");
421           }
422           aStates->IncrementState();
423         }
424         else if(aMethod.IsEqual("Rotate") ||
425                 aMethod.IsEqual("RotateMakeGroups") ||
426                 aMethod.IsEqual("RotateMakeMesh") ||
427                 aMethod.IsEqual("RotationSweep") ||
428                 aMethod.IsEqual("RotationSweepMakeGroups") ||
429                 aMethod.IsEqual("Mirror") ||
430                 aMethod.IsEqual("MirrorMakeGroups")) {
431           bool isSubstitute = false;
432           int anArgIndex = 0;
433           for(int i = 1, n = aCmd->GetNbArgs(); i <= n; i++) {
434             if(aCmd->GetArg(i).IsEqual("SMESH.AxisStruct")) {
435               anArgIndex = i+1;
436               break;
437             }
438           }
439           if(anArgIndex > 0) {
440             for(int j = 0; j < aCurrentStateSize; j++) {
441               if(!aCurrentState.at(j).IsEmpty()) {
442                 if(j < 6) // 0-5 - axis struct, 6 - angle (rotation & sweep), 7-8 - nbSteps and tolerance (sweep)
443                   isSubstitute = true;
444                 aCmd->SetArg(anArgIndex+j, aCurrentState.at(j));
445               }
446             }
447           }
448           if(isSubstitute)
449             aCmd->SetArg(anArgIndex - 1, TCollection_AsciiString(SMESH_2smeshpy::SmeshpyName())+".AxisStructStr");
450           aStates->IncrementState();
451         }
452         else if(aMethod.IsEqual("AddNode") ||
453                 aMethod.IsEqual("MoveClosestNodeToPoint")) {
454           for(int j = 0; j < aCurrentStateSize; j++) {
455             if(!aCurrentState.at(j).IsEmpty())
456               aCmd->SetArg(j+1, aCurrentState.at(j));
457           }
458           aStates->IncrementState();
459         }
460         else if(aMethod.IsEqual("MoveNode")) {
461           for(int j = 0; j < aCurrentStateSize; j++) {
462             if(!aCurrentState.at(j).IsEmpty())
463               aCmd->SetArg(j+2, aCurrentState.at(j));
464           }
465           aStates->IncrementState();
466         }
467         else if(aMethod.IsEqual("ExtrusionSweep") ||
468                 aMethod.IsEqual("ExtrusionSweepMakeGroups")) {
469           bool isSubstitute = false;
470           int anArgIndex = 0;
471           for(int i = 1, n = aCmd->GetNbArgs(); i <= n; i++) {
472             if(aCmd->GetArg(i).IsEqual("SMESH.PointStruct")) {
473               anArgIndex = i+1;
474               break;
475             }
476           }
477           if(anArgIndex > 0) {
478             for(int j = 0; j < aCurrentStateSize; j++) {
479               if(!aCurrentState.at(j).IsEmpty()) {
480                 if(j < 3) // 0-2 - dir struct, 3 - number of steps
481                   isSubstitute = true;
482                 aCmd->SetArg(anArgIndex+j, aCurrentState.at(j));
483               }
484             }
485           }
486           if(isSubstitute) {
487             aCmd->SetArg(anArgIndex - 1, TCollection_AsciiString(SMESH_2smeshpy::SmeshpyName())+".PointStructStr");
488             aCmd->SetArg(anArgIndex - 2, TCollection_AsciiString(SMESH_2smeshpy::SmeshpyName())+".DirStructStr");
489           }
490           aStates->IncrementState();
491         }
492         else if(aMethod.IsEqual("ExtrusionAlongPath") ||
493                 aMethod.IsEqual("ExtrusionAlongPathMakeGroups") ||
494                 /* workaround for a bug in the command parsing algorithm */
495                 aCmd->GetString().Search("ExtrusionAlongPathMakeGroups") != -1) {
496           int aNbAngles = aCurrentStateSize-3; // State looks like "Angle1:...:AngleN:X:Y:Z"
497           bool isSubstitute = false;
498           int anArgIndex = 0;
499           for(int i = 1, n = aCmd->GetNbArgs(); i <= n; i++) {
500             if(aCmd->GetArg(i).IsEqual("SMESH.PointStruct")) {
501               anArgIndex = i-1-aNbAngles;
502               break;
503             }
504           }
505           if(anArgIndex > 0) {
506             int j = 0;
507             for(; j < aNbAngles; j++) {
508               if(!aCurrentState.at(j).IsEmpty()) {
509                 aCmd->SetArg(anArgIndex+j-1, aCurrentState.at(j));
510               }
511             }
512             for(; j < aNbAngles+3; j++) {
513               if(!aCurrentState.at(j).IsEmpty()) {
514                 isSubstitute = true;
515                 aCmd->SetArg(anArgIndex+j+2, aCurrentState.at(j));
516               }
517             }
518           }
519           if(isSubstitute)
520             aCmd->SetArg(anArgIndex + aNbAngles + 1,
521                          TCollection_AsciiString(SMESH_2smeshpy::SmeshpyName())+".PointStructStr");
522           aStates->IncrementState();
523         }
524         else if(aMethod.IsEqual("TriToQuad") ||
525                 aMethod.IsEqual("Concatenate") ||
526                 aMethod.IsEqual("ConcatenateWithGroups")) {
527           if(aCurrentStateSize && !aCurrentState.at(0).IsEmpty())
528             aCmd->SetArg(aCmd->GetNbArgs(), aCurrentState.at(0));
529           aStates->IncrementState();
530         }
531         else if(aMethod.IsEqual("Smooth") ||
532                 aMethod.IsEqual("SmoothParametric")) {
533           int anArgIndex = aCmd->GetNbArgs() - 2;
534           for(int j = 0; j < aCurrentStateSize; j++) {
535             if(!aCurrentState.at(j).IsEmpty())
536               aCmd->SetArg(anArgIndex+j, aCurrentState.at(j));
537           }
538           aStates->IncrementState();
539         }
540       }
541     }
542     if(MYDEBUG) {
543       cout<<"Command after: "<< aCmd->GetString()<<endl;
544     }
545   }
546   //  ProcessLayerDistribution();
547 }
548 //================================================================================
549 /*!
550  * \brief Private method
551  */
552 //================================================================================
553 void SMESH_NoteBook::InitObjectMap()
554 {
555   SMESH_Gen_i *aGen = SMESH_Gen_i::GetSMESHGen();
556   if(!aGen)
557     return;
558   
559   SALOMEDS::Study_ptr aStudy = aGen->GetCurrentStudy();
560   if(aStudy->_is_nil())
561     return;
562   
563   SALOMEDS::SObject_var aSO = aStudy->FindComponent(aGen->ComponentDataType());
564   if(CORBA::is_nil(aSO))
565     return;
566   
567   SALOMEDS::ChildIterator_var Itr = aStudy->NewChildIterator(aSO);
568   char* aParameters;
569   for(Itr->InitEx(true); Itr->More(); Itr->Next()) {
570     SALOMEDS::SObject_var aSObject = Itr->Value();
571     SALOMEDS::GenericAttribute_var anAttr;
572     if ( aSObject->FindAttribute(anAttr, "AttributeString")) {
573       aParameters = SALOMEDS::AttributeString::_narrow(anAttr)->Value();
574       SALOMEDS::ListOfListOfStrings_var aSections = aStudy->ParseVariables(aParameters);
575       if(MYDEBUG) {
576         cout<<"Entry : "<< aSObject->GetID()<<endl;
577         cout<<"aParameters : "<<aParameters<<endl;
578       }      
579       TCollection_AsciiString anObjType;
580       CORBA::Object_var anObject = SMESH_Gen_i::SObjectToObject(aSObject);
581       SMESH::SMESH_Hypothesis_var aHyp = SMESH::SMESH_Hypothesis::_narrow(anObject);
582       if(!aHyp->_is_nil()) {
583         anObjType = TCollection_AsciiString(aHyp->GetName());
584       }
585       else if(SMESH::SMESH_Mesh_var aMesh = SMESH::SMESH_Mesh::_narrow(anObject)) {
586         anObjType = TCollection_AsciiString("Mesh");
587       }
588       if(MYDEBUG)
589         cout<<"The object Type : "<<anObjType<<endl;
590       ObjectStates *aState = NULL;
591       if(anObjType == "LayerDistribution")
592         aState = new LayerDistributionStates();
593       else
594         aState = new  ObjectStates(anObjType);
595       
596       for(int i = 0; i < aSections->length(); i++) {
597         TState aVars;
598         SALOMEDS::ListOfStrings aListOfVars = aSections[i];
599         for(int j = 0;j<aListOfVars.length();j++) {
600           TCollection_AsciiString aVar(aListOfVars[j].in());
601           if(!aVar.IsEmpty() && aStudy->IsVariable(aVar.ToCString())) {
602             aVar.InsertBefore(1,"\"");
603             aVar.InsertAfter(aVar.Length(),"\"");
604           }
605           aVars.push_back(aVar);
606           if(MYDEBUG) {
607             cout<<"Variable: '"<<aVar<<"'"<<endl;
608           }
609         }
610         aState->AddState(aVars);
611       }
612       _objectMap.insert(pair<TCollection_AsciiString,ObjectStates*>(TCollection_AsciiString(aSObject->GetID()),aState));
613     }
614   }
615 }
616
617 //================================================================================
618 /*!
619  * 
620  */
621 //================================================================================
622 void SMESH_NoteBook::AddCommand(const TCollection_AsciiString& theString)
623 {
624   if(MYDEBUG)
625     cout<<theString<<endl;
626   Handle(_pyCommand) aCommand = new _pyCommand( theString, -1);
627   _commands.push_back(aCommand);
628
629   if ( aCommand->GetMethod() == "GetMeshEditor" ) { // MeshEditor creation
630     myMeshEditors.insert( make_pair( aCommand->GetResultValue(),
631                                      aCommand->GetObject() ) );
632   }
633 }
634
635 //================================================================================
636 /*!
637  * 
638  */
639 //================================================================================
640 void SMESH_NoteBook::ProcessLayerDistribution()
641 {
642   // 1) Find all LayerDistribution states
643   vector<LayerDistributionStates*> aLDS;
644   TVariablesMap::const_iterator it = _objectMap.begin();
645   for(;it != _objectMap.end();it++)
646     if(LayerDistributionStates* aLDStates = (LayerDistributionStates*)((*it).second)) {
647       aLDS.push_back(aLDStates);
648     }
649   
650   // 2) Initialize all type of 1D Distribution hypothesis
651   for(int i=0;i<_commands.size();i++){
652     for(int j =0;j < aLDS.size();j++){
653       TCollection_AsciiString aResultValue = _commands[i]->GetResultValue();
654       if(_commands[i]->GetMethod() == "CreateHypothesis" &&
655          aLDS[j]->HasDistribution(aResultValue)){
656         TCollection_AsciiString aType = _commands[i]->GetArg(1);
657         aType.RemoveAll('\'');
658         aLDS[j]->SetDistributionType(aResultValue,aType);
659       }
660     }
661   }
662   // 3) ... and replase variables ...
663
664   for(int i=0;i<_commands.size();i++){
665     for(int j =0;j < aLDS.size();j++){
666       TCollection_AsciiString anObject = _commands[i]->GetObject();
667
668       if(aLDS[j]->HasDistribution(anObject)) {
669         TCollection_AsciiString aType = aLDS[j]->GetDistributionType(anObject);
670         TCollection_AsciiString aMethod = _commands[i]->GetMethod();
671         if(aType == "LocalLength") {
672           if(aMethod == "SetLength") {
673             if(!aLDS[j]->GetCurrectState().at(0).IsEmpty() )
674               _commands[i]->SetArg(1,aLDS[j]->GetCurrectState().at(0));
675             aLDS[j]->IncrementState();
676           }
677           else if(aMethod == "SetPrecision") {
678             if(!aLDS[j]->GetCurrectState().at(1).IsEmpty() )
679               _commands[i]->SetArg(1,aLDS[j]->GetCurrectState().at(1));
680             aLDS[j]->IncrementState();
681           }
682         }
683       }
684     }
685   }
686 }
687 //================================================================================
688 /*!
689  *  \brief Return result script
690  */
691 //================================================================================
692 TCollection_AsciiString SMESH_NoteBook::GetResultScript() const
693 {
694   TCollection_AsciiString aResult;
695   for(int i=0;i<_commands.size();i++)
696     aResult+=_commands[i]->GetString()+"\n";
697   return aResult;
698 }
699
700 //================================================================================
701 /*!
702  *  \brief Return value of the variable
703  */
704 //================================================================================
705 bool SMESH_NoteBook::GetReal(const TCollection_AsciiString& theVarName, double& theValue)
706 {
707   bool ok = false;
708
709   SMESH_Gen_i *aGen = SMESH_Gen_i::GetSMESHGen();
710   if(!aGen)
711     return ok;
712
713   SALOMEDS::Study_ptr aStudy = aGen->GetCurrentStudy();
714   if(aStudy->_is_nil())
715     return ok;
716
717   TCollection_AsciiString aVarName = theVarName;
718   aVarName.RemoveAll('\"');
719
720   if(aVarName.IsEmpty())
721     return ok;
722
723   const char* aName = aVarName.ToCString();
724   if(aStudy->IsVariable(aName) && (aStudy->IsReal(aName) || aStudy->IsInteger(aName))) {
725     theValue = aStudy->GetReal(aVarName.ToCString());
726     ok = true;
727   }
728
729   return ok;
730 }