src/HYDROGUI/HYDROGUI_SetColorOp.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_SetColorOp.h"
  24
  25 #include "HYDROGUI_ColorDlg.h"
  26 #include "HYDROGUI_DataModel.h"
  27 #include "HYDROGUI_Module.h"
  28 #include "HYDROGUI_Tool.h"
  29 #include "HYDROGUI_UpdateFlags.h"
  30
  31 #include <HYDROData_Object.h>
  32 #include <HYDROData_IPolyline.h>
  33
  34 #include <LightApp_Application.h>
  35 #include <LightApp_UpdateFlags.h>
  36
  37 #include <SUIT_Desktop.h>
  38
  39 HYDROGUI_SetColorOp::HYDROGUI_SetColorOp( HYDROGUI_Module* theModule )
  40 : HYDROGUI_Operation( theModule ),
  41   myColorDlg( 0 )
  42 {
  43   setName( tr( "SET_COLOR" ) );
  44 }
  45
  46 HYDROGUI_SetColorOp::~HYDROGUI_SetColorOp()
  47 {
  48 }
  49
  50 bool HYDROGUI_SetColorOp::CanObjectBeColored( const Handle(HYDROData_Entity)& theObject )
  51 {
  52   if ( theObject.IsNull() )
  53     return false;
  54
  55   return theObject->IsKind( STANDARD_TYPE(HYDROData_Object) ) ||
  56          theObject->IsKind( STANDARD_TYPE(HYDROData_IPolyline) );
  57 }
  58
  59 void HYDROGUI_SetColorOp::startOperation()
  60 {
  61   HYDROGUI_Operation::startOperation();
  62
  63   // Get the selected object
  64   //myEditedObject = Handle(HYDROData_Object)::DownCast( HYDROGUI_Tool::GetSelectedObject( module() ) );
  65   myEditedObject = HYDROGUI_Tool::GetSelectedObject( module() );
  66   if ( !CanObjectBeColored( myEditedObject ) )
  67   {
  68     abort();
  69     return;
  70   }
  71
  72   bool anIsOneColor = false;
  73
  74   // Get colors from the object
  75   QColor aFirstColor, aSecondColor;
  76   if ( myEditedObject->IsKind( STANDARD_TYPE(HYDROData_Object) ) )
  77   {
  78     Handle(HYDROData_Object) aGeomObject =
  79       Handle(HYDROData_Object)::DownCast( myEditedObject );
  80
  81     if ( myEditedObject->GetKind() == KIND_POLYLINE ||
  82          myEditedObject->GetKind() == KIND_PROFILE )
  83     {
  84       aFirstColor = aGeomObject->GetBorderColor();
  85       anIsOneColor = true;
  86     }
  87     else
  88     {
  89       aFirstColor = aGeomObject->GetFillingColor();
  90       aSecondColor = aGeomObject->GetBorderColor();
  91     }
  92   }
  93   else if ( myEditedObject->IsKind( STANDARD_TYPE(HYDROData_IPolyline) ) )
  94   {
  95     Handle(HYDROData_IPolyline) aPolyObject =
  96       Handle(HYDROData_IPolyline)::DownCast( myEditedObject );
  97
  98     aFirstColor = aPolyObject->GetWireColor();
  99
 100     anIsOneColor = true;
 101   }
 102
 103   // Create color dialog
 104   myColorDlg = new HYDROGUI_ColorDlg( module()->getApp()->desktop(), anIsOneColor );
 105   myColorDlg->setModal( true );
 106   myColorDlg->setWindowTitle( getName() );
 107
 108   // Set colors from the object
 109   myColorDlg->setFirstColor( aFirstColor );
 110   myColorDlg->setSecondColor( aSecondColor );
 111
 112   // Connect the dialog to operation slots
 113   connect( myColorDlg, SIGNAL( accepted() ), this, SLOT( onApply()  ) );
 114   connect( myColorDlg, SIGNAL( rejected() ), this, SLOT( onCancel() ) );
 115
 116   // Show the dialog
 117   myColorDlg->exec();
 118 }
 119
 120 bool HYDROGUI_SetColorOp::processApply( int& theUpdateFlags,
 121                                         QString& theErrorMsg,
 122                                         QStringList& theBrowseObjectsEntries )
 123 {
 124   if ( !myColorDlg || myEditedObject.IsNull() )
 125     return false;
 126
 127   QColor aFirstColor = myColorDlg->getFirstColor();
 128   QColor aSecondColor = myColorDlg->getSecondColor();
 129
 130   if ( myEditedObject->IsKind( STANDARD_TYPE(HYDROData_Object) ) )
 131   {
 132     Handle(HYDROData_Object) aGeomObject =
 133       Handle(HYDROData_Object)::DownCast( myEditedObject );
 134
 135     if ( myEditedObject->GetKind() == KIND_POLYLINE ||
 136          myEditedObject->GetKind() == KIND_PROFILE  )
 137     {
 138       aGeomObject->SetBorderColor( aFirstColor );
 139     }
 140     else
 141     {
 142       aGeomObject->SetFillingColor( aFirstColor );
 143       aGeomObject->SetBorderColor( aSecondColor );
 144     }
 145   }
 146   else if ( myEditedObject->IsKind( STANDARD_TYPE(HYDROData_IPolyline) ) )
 147   {
 148     Handle(HYDROData_IPolyline) aPolyObject =
 149       Handle(HYDROData_IPolyline)::DownCast( myEditedObject );
 150
 151     aPolyObject->SetWireColor( aFirstColor );
 152   }
 153
 154   module()->setIsToUpdate( myEditedObject );
 155
 156   theUpdateFlags = UF_Model | UF_OCCViewer | UF_OCC_Forced | UF_VTKViewer;
 157
 158   return true;
 159 }