src/HYDROGUI/HYDROGUI_TranslateObstacleOp.cxx

   1 // Copyright (C) 2007-2013  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.
  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 #include "HYDROGUI_TranslateObstacleOp.h"
  24
  25 #include "HYDROGUI_TranslateObstacleDlg.h"
  26 #include "HYDROGUI_Module.h"
  27 #include "HYDROGUI_Shape.h"
  28 #include "HYDROGUI_Tool.h"
  29 #include "HYDROGUI_UpdateFlags.h"
  30
  31 #include <HYDROData_ShapesTool.h>
  32
  33 #include <LightApp_Application.h>
  34 #include <LightApp_SelectionMgr.h>
  35 #include <LightApp_UpdateFlags.h>
  36
  37 #include <SUIT_MessageBox.h>
  38 #include <SUIT_Desktop.h>
  39
  40 #include <OCCViewer_ViewManager.h>
  41 #include <OCCViewer_ViewModel.h>
  42 #include <OCCViewer_ViewWindow.h>
  43
  44
  45 HYDROGUI_TranslateObstacleOp::HYDROGUI_TranslateObstacleOp( HYDROGUI_Module* theModule )
  46 : HYDROGUI_Operation( theModule ),
  47   myPreviewPrs( NULL )
  48 {
  49   setName( tr( "TRANSLATE_OBSTACLE" ) );
  50 }
  51
  52 HYDROGUI_TranslateObstacleOp::~HYDROGUI_TranslateObstacleOp()
  53 {
  54   erasePreview();
  55 }
  56
  57 void HYDROGUI_TranslateObstacleOp::startOperation()
  58 {
  59   HYDROGUI_Operation::startOperation();
  60
  61   // Get panel and reset its state
  62   HYDROGUI_TranslateObstacleDlg* aPanel = (HYDROGUI_TranslateObstacleDlg*)inputPanel();
  63   aPanel->reset();
  64
  65   // Get the edited object
  66   myEditedObject = Handle(HYDROData_Obstacle)::DownCast( HYDROGUI_Tool::GetSelectedObject( module() ) );
  67   if ( myEditedObject.IsNull() ) {
  68     abort();
  69     return;
  70   }
  71   else if ( myEditedObject->IsMustBeUpdated() ) {
  72     myEditedObject->Update();
  73   }
  74
  75   // Set the edited object name to the panel
  76   aPanel->setName( myEditedObject->GetName() );
  77
  78   // Create preview
  79   createPreview();
  80 }
  81
  82 void HYDROGUI_TranslateObstacleOp::abortOperation()
  83 {
  84   erasePreview();
  85   abortDocOperation();
  86
  87   HYDROGUI_Operation::abortOperation();
  88 }
  89
  90 void HYDROGUI_TranslateObstacleOp::commitOperation()
  91 {
  92   erasePreview();
  93
  94   HYDROGUI_Operation::commitOperation();
  95 }
  96
  97 HYDROGUI_InputPanel* HYDROGUI_TranslateObstacleOp::createInputPanel() const
  98 {
  99   HYDROGUI_TranslateObstacleDlg* aPanel = new HYDROGUI_TranslateObstacleDlg( module(), getName() );
 100
 101   connect( aPanel, SIGNAL( argumentsChanged() ), this, SLOT( onArgumentsChanged() ) );
 102
 103   return aPanel;
 104 }
 105
 106 bool HYDROGUI_TranslateObstacleOp::processApply( int& theUpdateFlags,
 107                                                  QString& theErrorMsg )
 108 {
 109   HYDROGUI_TranslateObstacleDlg* aPanel = ::qobject_cast<HYDROGUI_TranslateObstacleDlg*>( inputPanel() );
 110   if ( !aPanel || myEditedObject.IsNull() ) {
 111     return false;
 112   }
 113
 114   // Erase preview
 115   erasePreview();
 116
 117   // Get the translated shape
 118   double aDx = aPanel->getDx();
 119   double aDy = aPanel->getDy();
 120   double aDz = aPanel->getDz();
 121
 122   // Translate the obstacle
 123   myEditedObject->Translate( aDx, aDy, aDz );
 124   myEditedObject->Update();
 125
 126   module()->setIsToUpdate( myEditedObject );
 127
 128   // Set update flags
 129   theUpdateFlags = UF_Model | UF_OCCViewer | UF_OCC_Forced | UF_VTKViewer;
 130
 131   return true;
 132 }
 133
 134 void HYDROGUI_TranslateObstacleOp::createPreview()
 135 {
 136   HYDROGUI_TranslateObstacleDlg* aPanel = (HYDROGUI_TranslateObstacleDlg*)inputPanel();
 137   if ( myEditedObject.IsNull() || !aPanel )
 138     return;
 139
 140   // Create preview presentation if necessary
 141   if ( !myPreviewPrs ) {
 142     LightApp_Application* anApp = module()->getApp();
 143     OCCViewer_ViewManager* aViewManager = ::qobject_cast<OCCViewer_ViewManager*>(
 144       anApp->getViewManager( OCCViewer_Viewer::Type(), true ) );
 145
 146     if ( aViewManager ) {
 147       if ( OCCViewer_Viewer* aViewer = aViewManager->getOCCViewer() ) {
 148         Handle(AIS_InteractiveContext) aCtx = aViewer->getAISContext();
 149         if ( !aCtx.IsNull() ) {
 150           myPreviewPrs = new HYDROGUI_Shape( aCtx, NULL );
 151           myPreviewPrs->setFillingColor( myEditedObject->GetFillingColor(), false, false );
 152           myPreviewPrs->setBorderColor( myEditedObject->GetBorderColor(), false, false );
 153         }
 154       }
 155     }
 156   }
 157
 158   // Set the translated shape to the preview presentation
 159   if ( myPreviewPrs )
 160   {
 161     double aDx = aPanel->getDx();
 162     double aDy = aPanel->getDy();
 163     double aDz = aPanel->getDz();
 164     TopoDS_Shape anOriShape = myEditedObject->GetShape3D();
 165
 166     TopoDS_Shape aTranslatedShape = HYDROData_ShapesTool::Translated( anOriShape, aDx, aDy, aDz );
 167     myPreviewPrs->setShape( aTranslatedShape );
 168   }
 169 }
 170
 171 void HYDROGUI_TranslateObstacleOp::erasePreview()
 172 {
 173   if( myPreviewPrs ) {
 174     delete myPreviewPrs;
 175     myPreviewPrs = 0;
 176   }
 177 }
 178
 179 void HYDROGUI_TranslateObstacleOp::onArgumentsChanged()
 180 {
 181   // Update preview
 182   createPreview();
 183 }
 184
 185 TopoDS_Shape HYDROGUI_TranslateObstacleOp::getTranslatedShape() const
 186 {
 187   TopoDS_Shape aTranslatedShape;
 188
 189   HYDROGUI_TranslateObstacleDlg* aPanel = (HYDROGUI_TranslateObstacleDlg*)inputPanel();
 190   if ( aPanel && !myEditedObject.IsNull() ) {
 191     double aDx = aPanel->getDx();
 192     double aDy = aPanel->getDy();
 193     double aDz = aPanel->getDz();
 194
 195     TopoDS_Shape aShape = myEditedObject->GetShape3D();
 196     gp_Trsf aTrsf;
 197     gp_Vec aVec( aDx, aDy, aDz );
 198     aTrsf.SetTranslation(aVec);
 199     TopLoc_Location aLocOrig = aShape.Location();
 200     gp_Trsf aTrsfOrig = aLocOrig.Transformation();
 201     TopLoc_Location aLocRes( aTrsf * aTrsfOrig );
 202     aTranslatedShape = aShape.Located( aLocRes );
 203   }
 204
 205   return aTranslatedShape;
 206 }