src/HYDROGUI/HYDROGUI_SetColorOp.cxx

   1 // Copyright (C) 2014-2015  EDF-R&D
   2 // This library is free software; you can redistribute it and/or
   3 // modify it under the terms of the GNU Lesser General Public
   4 // License as published by the Free Software Foundation; either
   5 // version 2.1 of the License, or (at your option) any later version.
   6 //
   7 // This library is distributed in the hope that it will be useful,
   8 // but WITHOUT ANY WARRANTY; without even the implied warranty of
   9 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  10 // Lesser General Public License for more details.
  11 //
  12 // You should have received a copy of the GNU Lesser General Public
  13 // License along with this library; if not, write to the Free Software
  14 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  15 //
  16 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  17 //
  18
  19 #include "HYDROGUI_SetColorOp.h"
  20
  21 #include "HYDROGUI_ColorDlg.h"
  22 #include "HYDROGUI_DataModel.h"
  23 #include "HYDROGUI_Module.h"
  24 #include "HYDROGUI_Tool2.h"
  25 #include "HYDROGUI_UpdateFlags.h"
  26
  27 #include <HYDROData_Object.h>
  28 #include <HYDROData_IPolyline.h>
  29 #include <HYDROData_LandCoverMap.h>
  30
  31 #include <LightApp_Application.h>
  32 #include <LightApp_UpdateFlags.h>
  33
  34 #include <SUIT_Desktop.h>
  35
  36 HYDROGUI_SetColorOp::HYDROGUI_SetColorOp( HYDROGUI_Module* theModule )
  37 : HYDROGUI_Operation( theModule ),
  38   myColorDlg( 0 )
  39 {
  40   setName( tr( "SET_COLOR" ) );
  41 }
  42
  43 HYDROGUI_SetColorOp::~HYDROGUI_SetColorOp()
  44 {
  45 }
  46
  47 bool HYDROGUI_SetColorOp::CanObjectBeColored( const Handle(HYDROData_Entity)& theObject,
  48                                               HYDROGUI_Module* theModule )
  49 {
  50   if ( theObject.IsNull() )
  51     return false;
  52
  53   bool isCanBeColored =
  54     theObject->IsKind( STANDARD_TYPE(HYDROData_Object) ) ||
  55     theObject->IsKind( STANDARD_TYPE(HYDROData_IPolyline) );
  56
  57   return isCanBeColored;
  58 }
  59
  60 void HYDROGUI_SetColorOp::startOperation()
  61 {
  62   HYDROGUI_Operation::startOperation();
  63
  64   // Get the selected object
  65   //myEditedObject = Handle(HYDROData_Object)::DownCast( HYDROGUI_Tool::GetSelectedObject( module() ) );
  66   myEditedObject = HYDROGUI_Tool::GetSelectedObject( module() );
  67   if ( !CanObjectBeColored( myEditedObject, module() ) )
  68   {
  69     abort();
  70     return;
  71   }
  72
  73   bool anIsOneColor = false;
  74
  75   // Get colors from the object
  76   QColor aFirstColor, aSecondColor;
  77   if ( myEditedObject->IsKind( STANDARD_TYPE(HYDROData_Object) ) )
  78   {
  79     Handle(HYDROData_Object) aGeomObject =
  80       Handle(HYDROData_Object)::DownCast( myEditedObject );
  81
  82     if ( myEditedObject->GetKind() == KIND_POLYLINE ||
  83          myEditedObject->GetKind() == KIND_PROFILE )
  84     {
  85       aFirstColor = aGeomObject->GetBorderColor();
  86       anIsOneColor = true;
  87     }
  88     else
  89     {
  90       aFirstColor = aGeomObject->GetFillingColor();
  91       aSecondColor = aGeomObject->GetBorderColor();
  92     }
  93   }
  94   else if ( myEditedObject->IsKind( STANDARD_TYPE(HYDROData_IPolyline) ) )
  95   {
  96     Handle(HYDROData_IPolyline) aPolyObject =
  97       Handle(HYDROData_IPolyline)::DownCast( myEditedObject );
  98
  99     //aFirstColor = aPolyObject->GetWireColor();
 100
 101     if (aPolyObject->NbSections() > 0)
 102       aPolyObject->GetSectionColor(0, aFirstColor);
 103     else
 104       aFirstColor = aPolyObject->GetWireColor();
 105
 106     anIsOneColor = true;
 107   }
 108   else if ( myEditedObject->IsKind( STANDARD_TYPE(HYDROData_LandCoverMap) ) )
 109   {
 110     Handle(HYDROData_LandCoverMap) aLandCoverMap =
 111       Handle(HYDROData_LandCoverMap)::DownCast( myEditedObject );
 112
 113     /* TODO
 114     aFirstColor = aLandCoverMap->GetFillingColor();
 115     aSecondColor = aLandCoverMap->GetBorderColor();
 116     */
 117   }
 118
 119   // Create color dialog
 120   myColorDlg = new HYDROGUI_ColorDlg( module()->getApp()->desktop(), anIsOneColor );
 121   myColorDlg->setModal( true );
 122   myColorDlg->setWindowTitle( getName() );
 123
 124   // Set colors from the object
 125   myColorDlg->setFirstColor( aFirstColor );
 126   myColorDlg->setSecondColor( aSecondColor );
 127
 128   // Connect the dialog to operation slots
 129   connect( myColorDlg, SIGNAL( accepted() ), this, SLOT( onApply()  ) );
 130   connect( myColorDlg, SIGNAL( rejected() ), this, SLOT( onCancel() ) );
 131
 132   // Show the dialog
 133   myColorDlg->exec();
 134 }
 135
 136 bool HYDROGUI_SetColorOp::processApply( int& theUpdateFlags,
 137                                         QString& theErrorMsg,
 138                                         QStringList& theBrowseObjectsEntries )
 139 {
 140   if ( !myColorDlg || myEditedObject.IsNull() )
 141     return false;
 142
 143   QColor aFirstColor = myColorDlg->getFirstColor();
 144   QColor aSecondColor = myColorDlg->getSecondColor();
 145
 146   if ( myEditedObject->IsKind( STANDARD_TYPE(HYDROData_Object) ) )
 147   {
 148     Handle(HYDROData_Object) aGeomObject =
 149       Handle(HYDROData_Object)::DownCast( myEditedObject );
 150
 151     if ( myEditedObject->GetKind() == KIND_POLYLINE ||
 152          myEditedObject->GetKind() == KIND_PROFILE  )
 153     {
 154       aGeomObject->SetBorderColor( aFirstColor );
 155     }
 156     else
 157     {
 158       aGeomObject->SetFillingColor( aFirstColor );
 159       aGeomObject->SetBorderColor( aSecondColor );
 160     }
 161   }
 162   else if ( myEditedObject->IsKind( STANDARD_TYPE(HYDROData_IPolyline) ) )
 163   {
 164     Handle(HYDROData_IPolyline) aPolyObject =
 165       Handle(HYDROData_IPolyline)::DownCast( myEditedObject );
 166
 167     //aPolyObject->SetWireColor( aFirstColor );
 168     int nbSec = aPolyObject->NbSections();
 169     if (nbSec > 0)
 170     {
 171       for (int i = 0; i < nbSec; i++)
 172         aPolyObject->SetSectionColor(i, aFirstColor);
 173     }
 174     else
 175       aPolyObject->SetWireColor( aFirstColor );
 176   }
 177   else if ( myEditedObject->IsKind( STANDARD_TYPE(HYDROData_LandCoverMap) ) )
 178   {
 179     Handle(HYDROData_LandCoverMap) aLandCoverMap =
 180       Handle(HYDROData_LandCoverMap)::DownCast( myEditedObject );
 181
 182     /* TODO
 183     aLandCoverMap->SetFillingColor( aFirstColor );
 184     aLandCoverMap->SetBorderColor( aSecondColor );
 185     */
 186   }
 187
 188   module()->setIsToUpdate( myEditedObject );
 189
 190   theUpdateFlags = UF_Model | UF_OCCViewer | UF_OCC_Forced | UF_VTKViewer;
 191
 192   return true;
 193 }