Salome HOME
0022099: EDF 2307 SMESH: Apply a transformation to several meshes, sub-meshes or...
[modules/smesh.git] / src / SMESHGUI / SMESHGUI_RotationDlg.cxx
index 4128a876324763ce28ad355fe13e9f551e707fca..4c495d65e24c83a390e483210df7bf0e32fc6ba7 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
@@ -92,10 +94,9 @@ enum { MOVE_ELEMS_BUTTON = 0, COPY_ELEMS_BUTTON, MAKE_MESH_BUTTON }; //!< action
 // purpose  :
 //=================================================================================
 SMESHGUI_RotationDlg::SMESHGUI_RotationDlg( 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_DLG_MESH_ROTATION")));
   QPixmap image1 (SMESH::GetResourceMgr( mySMESHGUI )->loadPixmap("SMESH", tr("ICON_SELECT")));
@@ -351,6 +352,9 @@ SMESHGUI_RotationDlg::~SMESHGUI_RotationDlg()
 void SMESHGUI_RotationDlg::Init (bool ResetControls)
 {
   myBusy = false;
+  myMeshes.clear();
+  myObjects.clear();
+  myObjectsNames.clear();
 
   myEditCurrentArgument = (QWidget*)LineEditElements;
   LineEditElements->setFocus();
@@ -361,7 +365,6 @@ void SMESHGUI_RotationDlg::Init (bool ResetControls)
   buttonApply->setEnabled(false);
 
   myActor = 0;
-  myMesh = SMESH::SMESH_Mesh::_nil();
 
   if (ResetControls) {
     SpinBox_X->SetValue(0.0);
@@ -432,57 +435,99 @@ bool SMESHGUI_RotationDlg::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->RotateObject(mySelectedObject, anAxis, anAngle, false);
-        else
-            aMeshEditor->Rotate(anElementsId, anAxis, anAngle, false);
+        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->RotateObject(myObjects[i], anAxis, anAngle, false);
+         }
+       }
+       else {
+         SMESH::SMESH_MeshEditor_var aMeshEditor = myMeshes[0]->GetMeshEditor();
+         myMeshes[0]->SetParameters(aParameters.join( ":" ).toLatin1().constData());
+         aMeshEditor->Rotate(anElementsId, anAxis, anAngle, false);
+       }
         break;
       case COPY_ELEMS_BUTTON:
         if ( makeGroups ) {
           SMESH::ListOfGroups_var groups;
-          if(CheckBoxMesh->isChecked())
-            groups = aMeshEditor->RotateObjectMakeGroups(mySelectedObject, anAxis, anAngle);
-          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->RotateObjectMakeGroups(myObjects[i], anAxis, anAngle);
+           }
+         }
+          else {
+           SMESH::SMESH_MeshEditor_var aMeshEditor = myMeshes[0]->GetMeshEditor();
+           myMeshes[0]->SetParameters(aParameters.join( ":" ).toLatin1().constData());
             groups = aMeshEditor->RotateMakeGroups(anElementsId, anAxis, anAngle);
+         }
         }
         else {
-          if(CheckBoxMesh->isChecked())
-            aMeshEditor->RotateObject(mySelectedObject, anAxis, anAngle, true);
-          else
-            aMeshEditor->Rotate(anElementsId, anAxis, anAngle, true);
+          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->RotateObject(myObjects[i], anAxis, anAngle, true);
+           }
+         }
+          else {
+           SMESH::SMESH_MeshEditor_var aMeshEditor = myMeshes[0]->GetMeshEditor();
+           myMeshes[0]->SetParameters(aParameters.join( ":" ).toLatin1().constData());
+           aMeshEditor->Rotate(anElementsId, anAxis, anAngle, true);
+         }
         }
         break;
       case MAKE_MESH_BUTTON: {
         SMESH::SMESH_Mesh_var mesh;
-        if (CheckBoxMesh->isChecked())
-          mesh = aMeshEditor->RotateObjectMakeMesh(mySelectedObject, anAxis, anAngle, makeGroups,
-                                                   LineEditNewMesh->text().toLatin1().data());
-        else
+        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->RotateObjectMakeMesh(myObjects[i], anAxis, anAngle, makeGroups,
+                                                    aName.toLatin1().data());
+           if (!mesh->_is_nil()) {
+             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->RotateMakeMesh(anElementsId, anAxis, anAngle, makeGroups,
                                              LineEditNewMesh->text().toLatin1().data());
-        if (!mesh->_is_nil()) {
-          if( _PTR(SObject) aSObject = SMESH::ObjectToSObject( mesh ) )
-            anEntryList.append( aSObject->GetID().c_str() );
+         if (!mesh->_is_nil()) {
+           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();
+           // 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
@@ -491,7 +536,6 @@ bool SMESHGUI_RotationDlg::ClickOnApply()
         anApp->browseObjects( anEntryList, isApplyAndClose() );
     }
     Init(false);
-    mySelectedObject = SMESH::SMESH_IDSource::_nil();
     SelectionIntoArgument();
 
     SMESHGUI::Modified();
@@ -617,6 +661,7 @@ void SMESHGUI_RotationDlg::SelectionIntoArgument()
   // clear
   myActor = 0;
   QString aString = "";
+  onDisplaySimulation(false);
 
   myBusy = true;
   if (myEditCurrentArgument == (QWidget*)LineEditElements) {
@@ -635,27 +680,48 @@ void SMESHGUI_RotationDlg::SelectionIntoArgument()
   mySelectionMgr->selectedObjects(aList,SVTK_Viewer::Type());
 
   int nbSel = aList.Extent();
-  if (nbSel != 1)
-    return;
-
-  Handle(SALOME_InteractiveObject) IO = aList.First();
-  myMesh = SMESH::GetMeshByIO(IO);
-  if (myMesh->_is_nil())
-    return;
-
-  myActor = SMESH::FindActorByObject(myMesh);
-  if (!myActor)
-    myActor = SMESH::FindActorByEntry(IO->getEntry());
-  if (!myActor && !CheckBoxMesh->isChecked())
+  if (nbSel < 1)
     return;
 
   int aNbUnits = 0;
 
   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;
+      
+      myActor = SMESH::FindActorByObject( aMesh );
+      if ( !myActor )
+       myActor = SMESH::FindActorByEntry( IO->getEntry() );
+      if ( !myActor && !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;
+         }
+       }
+      }
+    }
 
     // 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);
     }
@@ -665,12 +731,9 @@ void SMESHGUI_RotationDlg::SelectionIntoArgument()
     if (CheckBoxMesh->isChecked()) {
       SMESH::GetNameOfSelectedIObjects(mySelectionMgr, aString);
 
-      if (!SMESH::IObjectToInterface<SMESH::SMESH_IDSource>(IO)->_is_nil()) { //MESH
-        mySelectedObject = SMESH::IObjectToInterface<SMESH::SMESH_IDSource>(IO);
-      }
-      else
-        return;
-        // get IDs from mesh
+      if ( myMeshes.isEmpty() )
+       return;
+      // get IDs from mesh
         /*
           SMDS_Mesh* aSMDSMesh = myActor->GetObject()->GetMesh();
           if (!aSMDSMesh)
@@ -711,7 +774,7 @@ void SMESHGUI_RotationDlg::SelectionIntoArgument()
         }
         */
       } else {
-      aNbUnits = SMESH::GetNameOfSelectedElements(mySelector, IO, aString);
+      aNbUnits = SMESH::GetNameOfSelectedElements(mySelector, aList.First(), aString);
       myElementsId = aString;
       if (aNbUnits < 1)
         return;
@@ -720,11 +783,21 @@ void SMESHGUI_RotationDlg::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;
 
-    SMDS_Mesh* aMesh =  myActor->GetObject()->GetMesh();
+    SMDS_Mesh* aMesh =  anActor->GetObject()->GetMesh();
     if (!aMesh)
       return;
 
@@ -929,6 +1002,10 @@ void SMESHGUI_RotationDlg::onVectorChanged()
 
 void SMESHGUI_RotationDlg::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);
@@ -937,18 +1014,12 @@ void SMESHGUI_RotationDlg::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();
@@ -963,13 +1034,13 @@ void SMESHGUI_RotationDlg::onActionClicked(int button)
 void SMESHGUI_RotationDlg::setNewMeshName()
 {
   LineEditNewMesh->setText("");
-  if ( LineEditNewMesh->isEnabled() && !myMesh->_is_nil() ) {
+  if ( LineEditNewMesh->isEnabled() && !myMeshes.isEmpty() ) {
     QString name;
     if ( CheckBoxMesh->isChecked() ) {
-      name = LineEditElements->text();
+      name = myMeshes.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() )
@@ -999,7 +1070,7 @@ void SMESHGUI_RotationDlg::keyPressEvent( QKeyEvent* e )
 //=================================================================================
 void SMESHGUI_RotationDlg::setFilters()
 {
-  if(myMesh->_is_nil()) {
+  if ( myMeshes.isEmpty() ) {
     SUIT_MessageBox::critical(this,
                               tr("SMESH_ERROR"),
                               tr("NO_MESH_SELECTED"));
@@ -1009,7 +1080,7 @@ void SMESHGUI_RotationDlg::setFilters()
     myFilterDlg = new SMESHGUI_FilterDlg( mySMESHGUI, SMESH::ALL );
 
   myFilterDlg->SetSelection();
-  myFilterDlg->SetMesh( myMesh );
+  myFilterDlg->SetMesh( myMeshes[0] );
   myFilterDlg->SetSourceWg( LineEditElements );
 
   myFilterDlg->show();
@@ -1071,14 +1142,19 @@ void SMESHGUI_RotationDlg::onDisplaySimulation( bool toDisplayPreview ) {
         SUIT_OverrideCursor aWaitCursor;
         bool copy = ( ActionGroup->checkedId() == COPY_ELEMS_BUTTON  ||
                       ActionGroup->checkedId() == MAKE_MESH_BUTTON );
-        SMESH::SMESH_MeshEditor_var aMeshEditor = myMesh->GetMeshEditPreviewer();
+       QList<SMESH::MeshPreviewStruct_var> aMeshPreviewStruct;
         if(CheckBoxMesh->isChecked())
-          aMeshEditor->RotateObject(mySelectedObject, anAxis, anAngle, copy);
-        else
+         for ( int i = 0; i < myObjects.count(); i++ ) {
+           SMESH::SMESH_MeshEditor_var aMeshEditor = myMeshes[i]->GetMeshEditPreviewer();
+           aMeshEditor->RotateObject(myObjects[i], anAxis, anAngle, copy);
+           aMeshPreviewStruct << aMeshEditor->GetPreviewData();
+         }
+       else {
+         SMESH::SMESH_MeshEditor_var aMeshEditor = myMeshes[0]->GetMeshEditPreviewer();
           aMeshEditor->Rotate(anElementsId, anAxis, anAngle, copy);
-
-        SMESH::MeshPreviewStruct_var aMeshPreviewStruct = aMeshEditor->GetPreviewData();
-        mySimulation->SetData(aMeshPreviewStruct._retn());      
+         aMeshPreviewStruct << aMeshEditor->GetPreviewData();
+       }
+       setSimulationPreview( aMeshPreviewStruct );
       } catch (...) {
         hidePreview();
       }