1 // Copyright (C) 2008 OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
2 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
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.
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.
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
18 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
20 // File : SMESH_NoteBook.cxx
21 // Author : Roman NIKOLAEV
23 #include "SMESH_2smeshpy.hxx"
24 #include "SMESH_NoteBook.hxx"
25 #include "SMESH_Gen_i.hxx"
26 #include "SMESH_PythonDump.hxx"
28 #include <Resource_DataMapOfAsciiStringAsciiString.hxx>
29 #include <TColStd_SequenceOfAsciiString.hxx>
30 #include <TColStd_HSequenceOfInteger.hxx>
36 static int MYDEBUG = 0;
38 static int MYDEBUG = 0;
45 //================================================================================
49 //================================================================================
50 ObjectStates::ObjectStates(TCollection_AsciiString theType)
56 //================================================================================
60 //================================================================================
61 ObjectStates::~ObjectStates()
65 //================================================================================
67 * \brief Add new object state
68 * \param theState - Object state (vector of notebook variable)
70 //================================================================================
71 void ObjectStates::AddState(const TState &theState)
73 _states.push_back(theState);
76 //================================================================================
78 * \brief Return current object state
79 * \\retval state - Object state (vector of notebook variable)
81 //================================================================================
82 TState ObjectStates::GetCurrectState() const
84 if(_states.size() > _dumpstate)
85 return _states[_dumpstate];
91 //================================================================================
95 //================================================================================
96 TAllStates ObjectStates::GetAllStates() const
101 //================================================================================
105 //================================================================================
106 void ObjectStates::IncrementState()
111 //================================================================================
115 //================================================================================
116 TCollection_AsciiString ObjectStates::GetObjectType() const{
121 //================================================================================
125 //================================================================================
126 LayerDistributionStates::LayerDistributionStates():
127 ObjectStates("LayerDistribution")
130 //================================================================================
134 //================================================================================
135 LayerDistributionStates::~LayerDistributionStates()
140 //================================================================================
142 * \brief AddDistribution
144 //================================================================================
145 void LayerDistributionStates::AddDistribution(const TCollection_AsciiString& theDistribution)
147 _distributions.insert(pair<TCollection_AsciiString,TCollection_AsciiString>(theDistribution,""));
150 //================================================================================
152 * \brief HasDistribution
154 //================================================================================
155 bool LayerDistributionStates::HasDistribution(const TCollection_AsciiString& theDistribution) const
157 return _distributions.find(theDistribution) != _distributions.end();
160 //================================================================================
162 * \brief SetDistributionType
164 //================================================================================
165 bool LayerDistributionStates::SetDistributionType(const TCollection_AsciiString& theDistribution,
166 const TCollection_AsciiString& theType)
168 TDistributionMap::iterator it = _distributions.find(theDistribution);
169 if(it == _distributions.end())
171 (*it).second = theType;
175 //================================================================================
177 * \brief GetDistributionType
179 //================================================================================
180 TCollection_AsciiString LayerDistributionStates::
181 GetDistributionType(const TCollection_AsciiString& theDistribution) const
183 TDistributionMap::const_iterator it = _distributions.find(theDistribution);
184 return (it == _distributions.end()) ? TCollection_AsciiString() : (*it).second;
187 //================================================================================
191 //================================================================================
192 SMESH_NoteBook::SMESH_NoteBook()
197 //================================================================================
201 //================================================================================
202 SMESH_NoteBook::~SMESH_NoteBook()
204 TVariablesMap::const_iterator it = _objectMap.begin();
205 for(;it!=_objectMap.end();it++) {
211 //================================================================================
213 * \brief Replace parameters of the functions on the Salome NoteBook Variables
214 * \param theString - Input string
215 * \retval TCollection_AsciiString - Convertion result
217 //================================================================================
218 void SMESH_NoteBook::ReplaceVariables()
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();
226 cout<<"Command before : "<< aCmd->GetString()<<endl;
227 cout<<"Method : "<< aMethod<<endl;
228 cout<<"Object : "<< aObject<<endl;
229 cout<<"Result : "<< aResultValue<<endl;
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);
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);
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();
254 else if(aMethod.IsEqual("SetPrecision")) {
255 if(!aStates->GetCurrectState().at(1).IsEmpty() )
256 aCmd->SetArg(1,aStates->GetCurrectState().at(1));
257 aStates->IncrementState();
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();
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();
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();
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));
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();
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();
319 // Case for NumberOfLayers hypothesis
320 else if(aStates->GetObjectType().IsEqual("NumberOfLayers")){
321 if(aMethod == "SetNumberOfLayers" && aStates->GetCurrectState().size() >= 1) {
322 if(!aStates->GetCurrectState().at(0).IsEmpty() )
323 aCmd->SetArg(1,aStates->GetCurrectState().at(0));
324 aStates->IncrementState();
328 // Case for NumberOfSegments hypothesis
329 else if(aStates->GetObjectType().IsEqual("NumberOfSegments")){
330 if(aMethod == "SetNumberOfSegments" && aStates->GetCurrectState().size() >= 1) {
331 if(!aStates->GetCurrectState().at(0).IsEmpty() )
332 aCmd->SetArg(1,aStates->GetCurrectState().at(0));
333 if(aStates->GetCurrectState().size()==1)
334 aStates->IncrementState();
336 else if (aMethod == "SetScaleFactor" && aStates->GetCurrectState().size() >= 2) {
337 if(!aStates->GetCurrectState().at(1).IsEmpty() )
338 aCmd->SetArg(1,aStates->GetCurrectState().at(1));
339 aStates->IncrementState();
343 else if(aStates->GetObjectType().IsEqual("Mesh")) {
344 TState aCurrentState = aStates->GetCurrectState();
345 int aCurrentStateSize = aCurrentState.size();
346 if(aMethod.IsEqual("Translate") ||
347 aMethod.IsEqual("TranslateMakeGroups") ||
348 aMethod.IsEqual("TranslateMakeMesh")) {
349 bool isVariableFound = false;
351 for(int i = 1, n = aCmd->GetNbArgs(); i <= n; i++) {
352 if(aCmd->GetArg(i).IsEqual("SMESH.PointStruct")) {
358 if(aCurrentStateSize == 3) { // translation by dx, dy, dz
359 for(int j = 0; j < aCurrentStateSize; j++) {
360 if(!aCurrentState.at(j).IsEmpty()) {
361 isVariableFound = true;
362 aCmd->SetArg(anArgIndex+j, aCurrentState.at(j));
366 else if(aCurrentStateSize == 6) { // translation by x1, x2, y1, y2, z1, z2
367 isVariableFound = true;
368 for(int j = 0; j < 3; j++) {
369 TCollection_AsciiString anArg = aCmd->GetArg(anArgIndex+j);
370 TCollection_AsciiString aValue1 = aCurrentState.at(2*j), aValue2 = aCurrentState.at(2*j+1);
371 bool aV1 = !aValue1.IsEmpty();
372 bool aV2 = !aValue2.IsEmpty();
373 double aValue, aCurrentValue = anArg.IsRealValue() ? anArg.RealValue() : 0;
375 if(!GetReal(aValue1, aValue))
377 aValue2 = TCollection_AsciiString( aValue + aCurrentValue );
379 else if(!aV1 && aV2) {
380 if(!GetReal(aValue2, aValue))
382 aValue1 = TCollection_AsciiString( aValue - aCurrentValue );
384 else if(!aV1 && !aV2) {
385 aValue1 = TCollection_AsciiString( 0 );
386 aValue2 = TCollection_AsciiString( aCurrentValue );
388 aCmd->SetArg(anArgIndex+j, aValue1 + ", " + aValue2 );
392 if(isVariableFound) {
393 TCollection_AsciiString aDim;
394 if(aCurrentStateSize == 6)
396 aCmd->SetArg(anArgIndex - 1, TCollection_AsciiString(SMESH_2smeshpy::SmeshpyName())+".PointStructStr"+aDim);
397 aCmd->SetArg(anArgIndex - 2, TCollection_AsciiString(SMESH_2smeshpy::SmeshpyName())+".DirStructStr");
399 aStates->IncrementState();
401 else if(aMethod.IsEqual("Rotate") ||
402 aMethod.IsEqual("RotateMakeGroups") ||
403 aMethod.IsEqual("RotateMakeMesh") ||
404 aMethod.IsEqual("RotationSweep") ||
405 aMethod.IsEqual("RotationSweepMakeGroups") ||
406 aMethod.IsEqual("Mirror") ||
407 aMethod.IsEqual("MirrorMakeGroups")) {
408 bool isSubstitute = false;
410 for(int i = 1, n = aCmd->GetNbArgs(); i <= n; i++) {
411 if(aCmd->GetArg(i).IsEqual("SMESH.AxisStruct")) {
417 for(int j = 0; j < aCurrentStateSize; j++) {
418 if(!aCurrentState.at(j).IsEmpty()) {
419 if(j < 6) // 0-5 - axis struct, 6 - angle (rotation & sweep), 7-8 - nbSteps and tolerance (sweep)
421 aCmd->SetArg(anArgIndex+j, aCurrentState.at(j));
426 aCmd->SetArg(anArgIndex - 1, TCollection_AsciiString(SMESH_2smeshpy::SmeshpyName())+".AxisStructStr");
427 aStates->IncrementState();
429 else if(aMethod.IsEqual("AddNode") ||
430 aMethod.IsEqual("MoveClosestNodeToPoint")) {
431 for(int j = 0; j < aCurrentStateSize; j++) {
432 if(!aCurrentState.at(j).IsEmpty())
433 aCmd->SetArg(j+1, aCurrentState.at(j));
435 aStates->IncrementState();
437 else if(aMethod.IsEqual("MoveNode")) {
438 for(int j = 0; j < aCurrentStateSize; j++) {
439 if(!aCurrentState.at(j).IsEmpty())
440 aCmd->SetArg(j+2, aCurrentState.at(j));
442 aStates->IncrementState();
444 else if(aMethod.IsEqual("ExtrusionSweep") ||
445 aMethod.IsEqual("ExtrusionSweepMakeGroups")) {
446 bool isSubstitute = false;
448 for(int i = 1, n = aCmd->GetNbArgs(); i <= n; i++) {
449 if(aCmd->GetArg(i).IsEqual("SMESH.PointStruct")) {
455 for(int j = 0; j < aCurrentStateSize; j++) {
456 if(!aCurrentState.at(j).IsEmpty()) {
457 if(j < 3) // 0-2 - dir struct, 3 - number of steps
459 aCmd->SetArg(anArgIndex+j, aCurrentState.at(j));
464 aCmd->SetArg(anArgIndex - 1, TCollection_AsciiString(SMESH_2smeshpy::SmeshpyName())+".PointStructStr");
465 aCmd->SetArg(anArgIndex - 2, TCollection_AsciiString(SMESH_2smeshpy::SmeshpyName())+".DirStructStr");
467 aStates->IncrementState();
469 else if(aMethod.IsEqual("ExtrusionAlongPath") ||
470 aMethod.IsEqual("ExtrusionAlongPathMakeGroups") ||
471 /* workaround for a bug in the command parsing algorithm */
472 aCmd->GetString().Search("ExtrusionAlongPathMakeGroups") != -1) {
473 int aNbAngles = aCurrentStateSize-3; // State looks like "Angle1:...:AngleN:X:Y:Z"
474 bool isSubstitute = false;
476 for(int i = 1, n = aCmd->GetNbArgs(); i <= n; i++) {
477 if(aCmd->GetArg(i).IsEqual("SMESH.PointStruct")) {
478 anArgIndex = i-1-aNbAngles;
484 for(; j < aNbAngles; j++) {
485 if(!aCurrentState.at(j).IsEmpty()) {
486 aCmd->SetArg(anArgIndex+j-1, aCurrentState.at(j));
489 for(; j < aNbAngles+3; j++) {
490 if(!aCurrentState.at(j).IsEmpty()) {
492 aCmd->SetArg(anArgIndex+j+2, aCurrentState.at(j));
497 aCmd->SetArg(anArgIndex + aNbAngles + 1,
498 TCollection_AsciiString(SMESH_2smeshpy::SmeshpyName())+".PointStructStr");
499 aStates->IncrementState();
501 else if(aMethod.IsEqual("TriToQuad") ||
502 aMethod.IsEqual("Concatenate") ||
503 aMethod.IsEqual("ConcatenateWithGroups")) {
504 if(aCurrentStateSize && !aCurrentState.at(0).IsEmpty())
505 aCmd->SetArg(aCmd->GetNbArgs(), aCurrentState.at(0));
506 aStates->IncrementState();
508 else if(aMethod.IsEqual("Smooth") ||
509 aMethod.IsEqual("SmoothParametric")) {
510 int anArgIndex = aCmd->GetNbArgs() - 2;
511 for(int j = 0; j < aCurrentStateSize; j++) {
512 if(!aCurrentState.at(j).IsEmpty())
513 aCmd->SetArg(anArgIndex+j, aCurrentState.at(j));
515 aStates->IncrementState();
520 cout<<"Command after: "<< aCmd->GetString()<<endl;
523 // ProcessLayerDistribution();
525 //================================================================================
527 * \brief Private method
529 //================================================================================
530 void SMESH_NoteBook::InitObjectMap()
532 SMESH_Gen_i *aGen = SMESH_Gen_i::GetSMESHGen();
536 SALOMEDS::Study_ptr aStudy = aGen->GetCurrentStudy();
537 if(aStudy->_is_nil())
540 SALOMEDS::SObject_var aSO = aStudy->FindComponent(aGen->ComponentDataType());
541 if(CORBA::is_nil(aSO))
544 SALOMEDS::ChildIterator_var Itr = aStudy->NewChildIterator(aSO);
546 for(Itr->InitEx(true); Itr->More(); Itr->Next()) {
547 SALOMEDS::SObject_var aSObject = Itr->Value();
548 SALOMEDS::GenericAttribute_var anAttr;
549 if ( aSObject->FindAttribute(anAttr, "AttributeString")) {
550 aParameters = SALOMEDS::AttributeString::_narrow(anAttr)->Value();
551 SALOMEDS::ListOfListOfStrings_var aSections = aStudy->ParseVariables(aParameters);
553 cout<<"Entry : "<< aSObject->GetID()<<endl;
554 cout<<"aParameters : "<<aParameters<<endl;
556 TCollection_AsciiString anObjType;
557 CORBA::Object_var anObject = SMESH_Gen_i::SObjectToObject(aSObject);
558 SMESH::SMESH_Hypothesis_var aHyp = SMESH::SMESH_Hypothesis::_narrow(anObject);
559 if(!aHyp->_is_nil()) {
560 anObjType = TCollection_AsciiString(aHyp->GetName());
562 else if(SMESH::SMESH_Mesh_var aMesh = SMESH::SMESH_Mesh::_narrow(anObject)) {
563 anObjType = TCollection_AsciiString("Mesh");
566 cout<<"The object Type : "<<anObjType<<endl;
567 ObjectStates *aState = NULL;
568 if(anObjType == "LayerDistribution")
569 aState = new LayerDistributionStates();
571 aState = new ObjectStates(anObjType);
573 for(int i = 0; i < aSections->length(); i++) {
575 SALOMEDS::ListOfStrings aListOfVars = aSections[i];
576 for(int j = 0;j<aListOfVars.length();j++) {
577 TCollection_AsciiString aVar(aListOfVars[j].in());
578 if(!aVar.IsEmpty() && aStudy->IsVariable(aVar.ToCString())) {
579 aVar.InsertBefore(1,"\"");
580 aVar.InsertAfter(aVar.Length(),"\"");
582 aVars.push_back(aVar);
584 cout<<"Variable: '"<<aVar<<"'"<<endl;
587 aState->AddState(aVars);
589 _objectMap.insert(pair<TCollection_AsciiString,ObjectStates*>(TCollection_AsciiString(aSObject->GetID()),aState));
594 //================================================================================
598 //================================================================================
599 void SMESH_NoteBook::AddCommand(const TCollection_AsciiString& theString)
602 cout<<theString<<endl;
603 Handle(_pyCommand) aCommand = new _pyCommand( theString, -1);
604 _commands.push_back(aCommand);
606 if ( aCommand->GetMethod() == "GetMeshEditor" ) { // MeshEditor creation
607 myMeshEditors.insert( make_pair( aCommand->GetResultValue(),
608 aCommand->GetObject() ) );
612 //================================================================================
616 //================================================================================
617 void SMESH_NoteBook::ProcessLayerDistribution()
619 // 1) Find all LayerDistribution states
620 vector<LayerDistributionStates*> aLDS;
621 TVariablesMap::const_iterator it = _objectMap.begin();
622 for(;it != _objectMap.end();it++)
623 if(LayerDistributionStates* aLDStates = (LayerDistributionStates*)((*it).second)) {
624 aLDS.push_back(aLDStates);
627 // 2) Initialize all type of 1D Distribution hypothesis
628 for(int i=0;i<_commands.size();i++){
629 for(int j =0;j < aLDS.size();j++){
630 TCollection_AsciiString aResultValue = _commands[i]->GetResultValue();
631 if(_commands[i]->GetMethod() == "CreateHypothesis" &&
632 aLDS[j]->HasDistribution(aResultValue)){
633 TCollection_AsciiString aType = _commands[i]->GetArg(1);
634 aType.RemoveAll('\'');
635 aLDS[j]->SetDistributionType(aResultValue,aType);
639 // 3) ... and replase variables ...
641 for(int i=0;i<_commands.size();i++){
642 for(int j =0;j < aLDS.size();j++){
643 TCollection_AsciiString anObject = _commands[i]->GetObject();
645 if(aLDS[j]->HasDistribution(anObject)) {
646 TCollection_AsciiString aType = aLDS[j]->GetDistributionType(anObject);
647 TCollection_AsciiString aMethod = _commands[i]->GetMethod();
648 if(aType == "LocalLength") {
649 if(aMethod == "SetLength") {
650 if(!aLDS[j]->GetCurrectState().at(0).IsEmpty() )
651 _commands[i]->SetArg(1,aLDS[j]->GetCurrectState().at(0));
652 aLDS[j]->IncrementState();
654 else if(aMethod == "SetPrecision") {
655 if(!aLDS[j]->GetCurrectState().at(1).IsEmpty() )
656 _commands[i]->SetArg(1,aLDS[j]->GetCurrectState().at(1));
657 aLDS[j]->IncrementState();
664 //================================================================================
666 * \brief Return result script
668 //================================================================================
669 TCollection_AsciiString SMESH_NoteBook::GetResultScript() const
671 TCollection_AsciiString aResult;
672 for(int i=0;i<_commands.size();i++)
673 aResult+=_commands[i]->GetString()+"\n";
677 //================================================================================
679 * \brief Return value of the variable
681 //================================================================================
682 bool SMESH_NoteBook::GetReal(const TCollection_AsciiString& theVarName, double& theValue)
686 SMESH_Gen_i *aGen = SMESH_Gen_i::GetSMESHGen();
690 SALOMEDS::Study_ptr aStudy = aGen->GetCurrentStudy();
691 if(aStudy->_is_nil())
694 TCollection_AsciiString aVarName = theVarName;
695 aVarName.RemoveAll('\"');
697 if(aVarName.IsEmpty())
700 const char* aName = aVarName.ToCString();
701 if(aStudy->IsVariable(aName) && (aStudy->IsReal(aName) || aStudy->IsInteger(aName))) {
702 theValue = aStudy->GetReal(aVarName.ToCString());