Salome HOME
0022099: EDF 2307 SMESH: Apply a transformation to several meshes, sub-meshes or...
[modules/smesh.git] / src / SMESHGUI / SMESHGUI_TranslationDlg.cxx
index e3946ac21f542ff4ac7c299e1964e516926e9f18..ca047bcb8e591d56cd132939e8e216615fb8959c 100644 (file)
 #include <SVTK_ViewModel.h>
 #include <SVTK_ViewWindow.h>
 #include <SALOME_ListIO.hxx>
+#include <SALOME_ListIteratorOfListIO.hxx>
 
 // SALOME KERNEL includes
+#include <SALOMEDSClient.hxx>
 #include <SALOMEDSClient_SObject.hxx>
 
 // OCCT includes
@@ -110,10 +112,9 @@ private:
 // purpose  :
 //=================================================================================
 SMESHGUI_TranslationDlg::SMESHGUI_TranslationDlg( SMESHGUI* theModule ) : 
-  SMESHGUI_PreviewDlg( theModule ),
+  SMESHGUI_MultiPreviewDlg( theModule ),
   mySelectionMgr( SMESH::GetSelectionMgr( theModule ) ),
-  myFilterDlg(0),
-  mySelectedObject(SMESH::SMESH_IDSource::_nil())
+  myFilterDlg(0)
 {
   QPixmap image0 (SMESH::GetResourceMgr( mySMESHGUI )->loadPixmap("SMESH", tr("ICON_SMESH_TRANSLATION_POINTS")));
   QPixmap image1 (SMESH::GetResourceMgr( mySMESHGUI )->loadPixmap("SMESH", tr("ICON_SMESH_TRANSLATION_VECTOR")));
@@ -328,7 +329,6 @@ SMESHGUI_TranslationDlg::SMESHGUI_TranslationDlg( SMESHGUI* theModule ) :
   connect(SpinBox2_1,  SIGNAL(valueChanged(double)), this, SLOT(toDisplaySimulation()));
   connect(SpinBox2_2,  SIGNAL(valueChanged(double)), this, SLOT(toDisplaySimulation()));
   connect(SpinBox2_3,  SIGNAL(valueChanged(double)), this, SLOT(toDisplaySimulation()));
-  connect(SpinBox2_3,  SIGNAL(valueChanged(double)), this, SLOT(toDisplaySimulation()));
 
   
   //To Connect preview check box
@@ -359,6 +359,9 @@ SMESHGUI_TranslationDlg::~SMESHGUI_TranslationDlg()
 void SMESHGUI_TranslationDlg::Init (bool ResetControls)
 {
   myBusy = false;
+  myObjects.clear();
+  myObjectsNames.clear();
+  myMeshes.clear();
 
   myEditCurrentArgument = 0;
   LineEditElements->clear();
@@ -369,7 +372,6 @@ void SMESHGUI_TranslationDlg::Init (bool ResetControls)
   buttonApply->setEnabled(false);
 
   myActor = 0;
-  myMesh = SMESH::SMESH_Mesh::_nil();
 
   if (ResetControls) {
     SpinBox1_1->SetValue(0.0);
@@ -503,54 +505,96 @@ bool SMESHGUI_TranslationDlg::ClickOnApply()
     QStringList anEntryList;
     try {
       SUIT_OverrideCursor aWaitCursor;
-      SMESH::SMESH_MeshEditor_var aMeshEditor = myMesh->GetMeshEditor();
-
-      myMesh->SetParameters(aParameters.join(":").toLatin1().constData());
 
       switch ( actionButton ) {
       case MOVE_ELEMS_BUTTON:
         if(CheckBoxMesh->isChecked())
-          aMeshEditor->TranslateObject(mySelectedObject, aVector, false);
-        else
+         for ( int i = 0; i < myObjects.count(); i++ ) {
+           SMESH::SMESH_MeshEditor_var aMeshEditor = myMeshes[i]->GetMeshEditor();
+           myMeshes[i]->SetParameters(aParameters.join( ":" ).toLatin1().constData());
+           aMeshEditor->TranslateObject(myObjects[i], aVector, false);
+         }
+        else {
+         SMESH::SMESH_MeshEditor_var aMeshEditor = myMeshes[0]->GetMeshEditor();
+         myMeshes[0]->SetParameters(aParameters.join( ":" ).toLatin1().constData());
           aMeshEditor->Translate(anElementsId, aVector, false);
+       }
         break;
       case COPY_ELEMS_BUTTON:
         if ( makeGroups ) {
           SMESH::ListOfGroups_var groups;
-          if(CheckBoxMesh->isChecked())
-            groups = aMeshEditor->TranslateObjectMakeGroups(mySelectedObject,aVector);
-          else
+          if(CheckBoxMesh->isChecked()) {
+           for ( int i = 0; i < myObjects.count(); i++ ) {
+             SMESH::SMESH_MeshEditor_var aMeshEditor = myMeshes[i]->GetMeshEditor();
+             myMeshes[i]->SetParameters(aParameters.join( ":" ).toLatin1().constData());
+             groups = aMeshEditor->TranslateObjectMakeGroups(myObjects[i],aVector);
+           }
+         }
+          else {
+           SMESH::SMESH_MeshEditor_var aMeshEditor = myMeshes[0]->GetMeshEditor();
+           myMeshes[0]->SetParameters(aParameters.join( ":" ).toLatin1().constData());
             groups = aMeshEditor->TranslateMakeGroups(anElementsId, aVector);
+         }
         }
         else {
-          if(CheckBoxMesh->isChecked())
-            aMeshEditor->TranslateObject(mySelectedObject, aVector, true);
-          else
+          if(CheckBoxMesh->isChecked()) {
+           for ( int i = 0; i < myObjects.count(); i++ ) {
+             SMESH::SMESH_MeshEditor_var aMeshEditor = myMeshes[i]->GetMeshEditor();
+             myMeshes[i]->SetParameters(aParameters.join( ":" ).toLatin1().constData());
+             aMeshEditor->TranslateObject(myObjects[i], aVector, true);
+           }
+         }
+          else {
+           SMESH::SMESH_MeshEditor_var aMeshEditor = myMeshes[0]->GetMeshEditor();
+           myMeshes[0]->SetParameters(aParameters.join( ":" ).toLatin1().constData());
             aMeshEditor->Translate(anElementsId, aVector, true);
+         }
         }
         break;
       case MAKE_MESH_BUTTON: {
         SMESH::SMESH_Mesh_var mesh;
-        if (CheckBoxMesh->isChecked())
-          mesh = aMeshEditor->TranslateObjectMakeMesh(mySelectedObject, aVector, makeGroups,
-                                                      LineEditNewMesh->text().toLatin1().data());
-        else
-          mesh = aMeshEditor->TranslateMakeMesh(anElementsId, aVector, makeGroups,
+        if (CheckBoxMesh->isChecked()) {
+         for ( int i = 0; i < myObjects.count(); i++ ) {
+           QString aName = SMESH::UniqueMeshName( LineEditNewMesh->text().replace( "*", myObjectsNames[i] ) );
+           SMESH::SMESH_MeshEditor_var aMeshEditor = myMeshes[i]->GetMeshEditor();
+           myMeshes[i]->SetParameters(aParameters.join( ":" ).toLatin1().constData());
+           mesh = aMeshEditor->TranslateObjectMakeMesh(myObjects[i], aVector, makeGroups,
+                                                       aName.toLatin1().data());
+           if( _PTR(SObject) aSObject = SMESH::ObjectToSObject( mesh ) )
+             anEntryList.append( aSObject->GetID().c_str() );
+
+#ifdef WITHGENERICOBJ
+           // obj has been published in study. Its refcount has been incremented.
+           // It is safe to decrement its refcount
+           // so that it will be destroyed when the entry in study will be removed
+           mesh->UnRegister();
+#endif
+         }
+       }
+       else {
+         SMESH::SMESH_MeshEditor_var aMeshEditor = myMeshes[0]->GetMeshEditor();
+         myMeshes[0]->SetParameters(aParameters.join( ":" ).toLatin1().constData());
+         mesh = aMeshEditor->TranslateMakeMesh(anElementsId, aVector, makeGroups,
                                                 LineEditNewMesh->text().toLatin1().data());
           if( _PTR(SObject) aSObject = SMESH::ObjectToSObject( mesh ) )
             anEntryList.append( aSObject->GetID().c_str() );
+       
 #ifdef WITHGENERICOBJ
           // obj has been published in study. Its refcount has been incremented.
           // It is safe to decrement its refcount
           // so that it will be destroyed when the entry in study will be removed
           mesh->UnRegister();
 #endif
-        }
+       }
+       break;
+      }
       }
     } catch (...) {
     }
 
-    SMESH::UpdateView();
+    for ( int i = 0; i < myMeshes.count(); i++ )
+      SMESH::Update( (SMESH::FindActorByObject( myMeshes[i] ))->getIO(), true );
+
     if ( ( MakeGroupsCheck->isEnabled() && MakeGroupsCheck->isChecked() ) ||
          actionButton == MAKE_MESH_BUTTON ) {
       mySMESHGUI->updateObjBrowser(true); // new groups may appear
@@ -561,7 +605,6 @@ bool SMESHGUI_TranslationDlg::ClickOnApply()
       
     Init(false);
     ConstructorsClicked(GetConstructorId());
-    mySelectedObject = SMESH::SMESH_IDSource::_nil();
     SelectionIntoArgument();
 
     SMESHGUI::Modified();
@@ -686,6 +729,7 @@ void SMESHGUI_TranslationDlg::SelectionIntoArgument()
   // clear
   myActor = 0;
   QString aString = "";
+  onDisplaySimulation(false);
 
   if (myEditCurrentArgument == (QWidget*)LineEditElements) {
     LineEditElements->setText(aString);
@@ -702,32 +746,51 @@ void SMESHGUI_TranslationDlg::SelectionIntoArgument()
   mySelectionMgr->selectedObjects(aList,SVTK_Viewer::Type());
 
   int nbSel = aList.Extent();
-  if (nbSel != 1)
+  if (nbSel < 1)
     return;
 
-  Handle(SALOME_InteractiveObject) IO = aList.First();
-  SMESH::SMESH_Mesh_var aMesh = SMESH::GetMeshByIO(IO);
-  if (aMesh->_is_nil())
-    return;
-
-  SMESH_Actor* anActor = SMESH::FindActorByObject(aMesh);
-  if (!anActor)
-    anActor = SMESH::FindActorByEntry(IO->getEntry());
-
-  if (!anActor && !CheckBoxMesh->isChecked())
-      return;
-
   int aNbUnits = 0;
-
-  if (myEditCurrentArgument == (QWidget*)LineEditElements)
-  {
-    myMesh = aMesh;
-    myActor = anActor;
-
+  if (myEditCurrentArgument == (QWidget*)LineEditElements) {
     myElementsId = "";
+    myObjects.clear();
+    myObjectsNames.clear();
+    myMeshes.clear();
+
+    for ( SALOME_ListIteratorOfListIO it( aList ); it.More(); it.Next() ) {
+      Handle(SALOME_InteractiveObject) IO = it.Value();
+      SMESH::SMESH_Mesh_var aMesh = SMESH::GetMeshByIO( IO );
+      if ( aMesh->_is_nil() )
+       return;
+
+      SMESH_Actor* anActor = SMESH::FindActorByObject( aMesh );
+      if ( !anActor )
+       anActor = SMESH::FindActorByEntry( IO->getEntry() );
+
+      if ( !anActor && !CheckBoxMesh->isChecked() )
+       return;
+
+      if ( !SMESH::IObjectToInterface<SMESH::SMESH_IDSource>( IO )->_is_nil() ) {
+       if ( _PTR(Study) aStudy = SMESH::GetActiveStudyDocument() ) {
+         _PTR(SObject) obj = aStudy->FindObjectID( qPrintable( QString( IO->getEntry() ) ) );
+         _PTR(GenericAttribute) anAttr;
+         if ( obj && obj->FindAttribute( anAttr, "AttributeName" ) ) {
+           _PTR(AttributeName) aNameAttr( anAttr );
+           myObjects << SMESH::IObjectToInterface<SMESH::SMESH_IDSource>( IO );
+           myObjectsNames << aNameAttr->Value().c_str();
+           myMeshes << aMesh;
+         }
+       }
+      }
 
+      myActor = anActor;
+    }
+    
     // MakeGroups is available if there are groups and "Copy"
-    if ( myMesh->NbGroups() == 0 ) {
+    int aNbGroups = 0;
+    for ( int i = 0; i < myMeshes.count(); i++ )
+      aNbGroups += myMeshes[i]->NbGroups();
+
+    if ( aNbGroups == 0 ) {
       MakeGroupsCheck->setChecked(false);
       MakeGroupsCheck->setEnabled(false);
     }
@@ -737,14 +800,10 @@ void SMESHGUI_TranslationDlg::SelectionIntoArgument()
 
     if (CheckBoxMesh->isChecked()) {
       SMESH::GetNameOfSelectedIObjects( mySelectionMgr, aString );
-
-      if (!SMESH::IObjectToInterface<SMESH::SMESH_IDSource>(IO)->_is_nil()) { //MESH, SUBMESH, OR GROUP
-        mySelectedObject = SMESH::IObjectToInterface<SMESH::SMESH_IDSource>(IO);
-      }
-      else
-        return;
+      if (myMeshes.isEmpty())
+       return;
     } else {
-      aNbUnits = SMESH::GetNameOfSelectedElements(mySelector, IO, aString);
+      aNbUnits = SMESH::GetNameOfSelectedElements(mySelector, aList.First(), aString);
       myElementsId = aString;
       if (aNbUnits < 1)
         return;
@@ -752,6 +811,16 @@ void SMESHGUI_TranslationDlg::SelectionIntoArgument()
 
     myNbOkElements = true;
   } else {
+    Handle(SALOME_InteractiveObject) IO = aList.First();
+    if ((SMESH::GetMeshByIO(IO))->_is_nil())
+      return;
+
+    SMESH_Actor* anActor = SMESH::FindActorByObject(SMESH::GetMeshByIO(IO));
+    if (!anActor)
+      anActor = SMESH::FindActorByEntry(IO->getEntry());
+    if (!anActor && !CheckBoxMesh->isChecked())
+      return;
+
     aNbUnits = SMESH::GetNameOfSelectedNodes(mySelector, IO, aString);
     if (aNbUnits != 1)
       return;
@@ -928,6 +997,10 @@ void SMESHGUI_TranslationDlg::onSelectMesh (bool toSelectMesh)
 
 void SMESHGUI_TranslationDlg::onActionClicked(int button)
 {
+  int aNbGroups = 0;
+  for ( int i = 0; i < myMeshes.count(); i++ )
+    aNbGroups += myMeshes[i]->NbGroups();
+
   switch ( button ) {
   case MOVE_ELEMS_BUTTON:
     MakeGroupsCheck->setEnabled(false);
@@ -936,18 +1009,12 @@ void SMESHGUI_TranslationDlg::onActionClicked(int button)
   case COPY_ELEMS_BUTTON:
     LineEditNewMesh->setEnabled(false);
     MakeGroupsCheck->setText( tr("SMESH_MAKE_GROUPS"));
-    if ( myMesh->_is_nil() || myMesh->NbGroups() > 0)
-      MakeGroupsCheck->setEnabled(true);
-    else
-      MakeGroupsCheck->setEnabled(false);
+    MakeGroupsCheck->setEnabled( myMeshes.isEmpty() || aNbGroups > 0 );
     break;
   case MAKE_MESH_BUTTON:
     LineEditNewMesh->setEnabled(true);
     MakeGroupsCheck->setText( tr("SMESH_COPY_GROUPS"));
-    if ( myMesh->_is_nil() || myMesh->NbGroups() > 0)
-      MakeGroupsCheck->setEnabled(true);
-    else
-      MakeGroupsCheck->setEnabled(false);
+    MakeGroupsCheck->setEnabled( myMeshes.isEmpty() || aNbGroups > 0 );
     break;
   }
   setNewMeshName();
@@ -962,13 +1029,13 @@ void SMESHGUI_TranslationDlg::onActionClicked(int button)
 void SMESHGUI_TranslationDlg::setNewMeshName()
 {
   LineEditNewMesh->setText("");
-  if ( LineEditNewMesh->isEnabled() && !myMesh->_is_nil() ) {
+  if ( LineEditNewMesh->isEnabled() && !myMeshes.isEmpty() ) {
     QString name;
     if ( CheckBoxMesh->isChecked() ) {
-      name = LineEditElements->text();
+      name = myObjects.count() > 1 ? "*" : LineEditElements->text();
     }
     else {
-      _PTR(SObject) meshSO = SMESH::FindSObject( myMesh );
+      _PTR(SObject) meshSO = SMESH::FindSObject( myMeshes[0] );
       name = meshSO->GetName().c_str();
     }
     if ( !name.isEmpty() )
@@ -1007,7 +1074,7 @@ void SMESHGUI_TranslationDlg::keyPressEvent( QKeyEvent* e )
 //=================================================================================
 void SMESHGUI_TranslationDlg::setFilters()
 {
-  if(myMesh->_is_nil()) {
+  if(myMeshes.isEmpty()) {
     SUIT_MessageBox::critical(this,
                               tr("SMESH_ERROR"),
                               tr("NO_MESH_SELECTED"));
@@ -1017,7 +1084,7 @@ void SMESHGUI_TranslationDlg::setFilters()
     myFilterDlg = new SMESHGUI_FilterDlg( mySMESHGUI, SMESH::ALL );
 
   myFilterDlg->SetSelection();
-  myFilterDlg->SetMesh( myMesh );
+  myFilterDlg->SetMesh( myMeshes[0] );
   myFilterDlg->SetSourceWg( LineEditElements );
 
   myFilterDlg->show();
@@ -1082,14 +1149,19 @@ void SMESHGUI_TranslationDlg::onDisplaySimulation( bool toDisplayPreview ) {
         bool copy = ( ActionGroup->checkedId() == COPY_ELEMS_BUTTON ||
                       ActionGroup->checkedId() == MAKE_MESH_BUTTON );
         SUIT_OverrideCursor aWaitCursor;
-        SMESH::SMESH_MeshEditor_var aMeshEditor = myMesh->GetMeshEditPreviewer();
+       QList<SMESH::MeshPreviewStruct_var> aMeshPreviewStruct;
         if(CheckBoxMesh->isChecked())
-          aMeshEditor->TranslateObject(mySelectedObject, aVector, copy);
-        else
+         for ( int i = 0; i < myObjects.count(); i++ ) {
+           SMESH::SMESH_MeshEditor_var aMeshEditor = myMeshes[i]->GetMeshEditPreviewer();
+           aMeshEditor->TranslateObject(myObjects[i], aVector, copy);
+           aMeshPreviewStruct << aMeshEditor->GetPreviewData();
+         }
+        else {
+         SMESH::SMESH_MeshEditor_var aMeshEditor = myMeshes[0]->GetMeshEditPreviewer();
           aMeshEditor->Translate(anElementsId, aVector, copy);
-        
-        SMESH::MeshPreviewStruct_var aMeshPreviewStruct = aMeshEditor->GetPreviewData();
-        mySimulation->SetData(aMeshPreviewStruct._retn());      
+         aMeshPreviewStruct << aMeshEditor->GetPreviewData();
+       }
+       setSimulationPreview( aMeshPreviewStruct );
       } catch (...) {
         
       }