src/HYDROGUI/HYDROGUI_Poly3DOp.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_Poly3DOp.h"
  24
  25 #include "HYDROGUI_Module.h"
  26 #include "HYDROGUI_Tool.h"
  27 #include "HYDROGUI_Poly3DDlg.h"
  28 #include "HYDROGUI_UpdateFlags.h"
  29
  30 #include <HYDROData_Document.h>
  31 #include <HYDROData_Polyline3D.h>
  32 #include <HYDROData_Profile.h>
  33 #include <HYDROData_PolylineXY.h>
  34 #include <HYDROData_Bathymetry.h>
  35
  36 #include <HYDROData_OperationsFactory.h>
  37
  38 HYDROGUI_Poly3DOp::HYDROGUI_Poly3DOp( HYDROGUI_Module* theModule,
  39                                             const bool theIsEdit )
  40 : HYDROGUI_Operation( theModule ),
  41   myIsEdit( theIsEdit ),
  42   myEditedObject( 0 )
  43 {
  44   setName( theIsEdit ? tr( "EDIT_POLYLINE_3D" ) : tr( "CREATE_POLYLINE_3D" ) );
  45 }
  46
  47 HYDROGUI_Poly3DOp::~HYDROGUI_Poly3DOp()
  48 {
  49 }
  50
  51 HYDROGUI_InputPanel* HYDROGUI_Poly3DOp::createInputPanel() const
  52 {
  53   return new HYDROGUI_Poly3DDlg( module(), getName() );
  54 }
  55
  56 void HYDROGUI_Poly3DOp::startOperation()
  57 {
  58   HYDROGUI_Operation::startOperation();
  59
  60   HYDROGUI_Poly3DDlg* aPanel = (HYDROGUI_Poly3DDlg*)inputPanel();
  61   aPanel->reset();
  62   aPanel->setMode( myIsEdit );
  63
  64   QString aPoly3DName;
  65   if( myIsEdit )
  66   {
  67     myEditedObject = Handle(HYDROData_Polyline3D)::DownCast( HYDROGUI_Tool::GetSelectedObject( module() ) );
  68     if( !myEditedObject.IsNull() )
  69       aPoly3DName = myEditedObject->GetName();
  70   }
  71   else
  72   {
  73     aPoly3DName = HYDROGUI_Tool::GenerateObjectName( module(), tr( "DEFAULT_POLYLINE_3D_NAME" ) );
  74   }
  75   aPanel->setResultName( aPoly3DName );
  76
  77   QString aPolyName, aProfileName, aBathName;
  78   if( myIsEdit && !myEditedObject.IsNull() )
  79   {
  80     Handle(HYDROData_Entity) aProfile = myEditedObject->GetProfileUZ()->GetFatherObject();
  81     if( !aProfile.IsNull() )
  82       aProfileName = aProfile->GetName();
  83
  84     Handle(HYDROData_Entity) aPoly = myEditedObject->GetPolylineXY();
  85     if( !aPoly.IsNull() )
  86       aPolyName = aPoly->GetName();
  87
  88     //TODO: use bathymetry from data model
  89
  90     aPanel->setSelectedObjects( aPolyName, aProfileName, aBathName );
  91   }
  92   else if( !myIsEdit )
  93   {
  94     Handle(HYDROData_Profile) aSelectedProfile =
  95       Handle(HYDROData_Profile)::DownCast( HYDROGUI_Tool::GetSelectedObject( module() ) );
  96     if( !aSelectedProfile.IsNull() )
  97     {
  98       QString aSelectedName = aSelectedProfile->GetName();
  99       aPanel->setPreselectedObject( aSelectedName );
 100     }
 101   }
 102 }
 103
 104 bool HYDROGUI_Poly3DOp::processApply( int& theUpdateFlags,
 105                                          QString& theErrorMsg )
 106 {
 107   HYDROGUI_Poly3DDlg* aPanel = dynamic_cast<HYDROGUI_Poly3DDlg*>( inputPanel() );
 108
 109   QString aResultName = aPanel->getResultName();
 110   if( aResultName.isEmpty() )
 111     return false;
 112
 113   QString aPolyName, aProfileName, aBathName;
 114   if( !aPanel->getSelectedObjects( aPolyName, aProfileName, aBathName ) )
 115     return false;
 116
 117   if( !myIsEdit || ( !myEditedObject.IsNull() && myEditedObject->GetName() != aResultName ) )
 118   {
 119     // check that there are no other objects with the same name in the document
 120     Handle(HYDROData_Entity) anObject = HYDROGUI_Tool::FindObjectByName( module(), aResultName );
 121     if( !anObject.IsNull() )
 122     {
 123       theErrorMsg = tr( "OBJECT_EXISTS_IN_DOCUMENT" ).arg( aResultName );
 124       return false;
 125     }
 126   }
 127
 128   Handle(HYDROData_Entity) aProfileEnt =
 129     HYDROGUI_Tool::FindObjectByName( module(), aProfileName, KIND_PROFILE ) ;
 130   Handle(HYDROData_Entity) aPolyEnt =
 131     HYDROGUI_Tool::FindObjectByName( module(), aPolyName, KIND_POLYLINEXY );
 132   Handle(HYDROData_Entity) aBathEnt =
 133     HYDROGUI_Tool::FindObjectByName( module(), aBathName, KIND_BATHYMETRY );
 134
 135   if( aPolyEnt.IsNull() || ( aPolyEnt.IsNull() && aBathEnt.IsNull() ) )
 136     return false;
 137
 138   Handle(HYDROData_Profile) aProfile = Handle(HYDROData_Profile)::DownCast( aProfileEnt );
 139   Handle(HYDROData_Bathymetry) aBath = Handle(HYDROData_Bathymetry)::DownCast( aBathEnt );
 140   Handle(HYDROData_PolylineXY) aPolyline = Handle(HYDROData_PolylineXY)::DownCast( aPolyEnt );
 141
 142   if( aProfile.IsNull() || ( aPolyline.IsNull() && aBath.IsNull() ) )
 143     return false;
 144
 145   Handle(HYDROData_ProfileUZ) aProfileUZ = aProfile->GetProfileUZ();
 146   if( aProfileUZ.IsNull() )
 147     return false;
 148
 149   Handle(HYDROData_Polyline3D) aResult;
 150   if( myIsEdit )
 151   {
 152     aResult = myEditedObject;
 153     aResult->RemoveProfileUZ();
 154     aResult->RemovePolylineXY();
 155   }
 156   else
 157   {
 158     aResult = Handle(HYDROData_Polyline3D)::DownCast( doc()->CreateObject( KIND_POLYLINE ) );
 159   }
 160
 161   if( aResult.IsNull() )
 162     return false;
 163
 164   aResult->SetName( aResultName );
 165   aResult->SetProfileUZ( aProfileUZ );
 166   aResult->SetPolylineXY( aPolyline );
 167   //TODO: set bathymetry
 168
 169   if( !myIsEdit )
 170   {
 171     aResult->SetBorderColor( HYDROData_Polyline3D::DefaultBorderColor() );
 172   }
 173
 174   aResult->Update();
 175
 176   if( !myIsEdit )
 177   {
 178     size_t aViewId = HYDROGUI_Tool::GetActiveOCCViewId( module() );
 179     module()->setObjectVisible( aViewId, aPolyline, false );
 180     module()->setObjectVisible( aViewId, aProfile, false );
 181     module()->setObjectVisible( aViewId, aResult, true );
 182   }
 183
 184   theUpdateFlags = UF_Model | UF_Viewer | UF_GV_Forced | UF_OCCViewer | UF_OCC_Forced;
 185   return true;
 186 }