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.
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.
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
16 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
19 #include "HYDROGUI_SetColorOp.h"
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"
27 #include <HYDROData_Object.h>
28 #include <HYDROData_IPolyline.h>
29 #include <HYDROData_LandCoverMap.h>
31 #include <LightApp_Application.h>
32 #include <LightApp_UpdateFlags.h>
34 #include <SUIT_Desktop.h>
36 HYDROGUI_SetColorOp::HYDROGUI_SetColorOp( HYDROGUI_Module* theModule )
37 : HYDROGUI_Operation( theModule ),
40 setName( tr( "SET_COLOR" ) );
43 HYDROGUI_SetColorOp::~HYDROGUI_SetColorOp()
47 bool HYDROGUI_SetColorOp::CanObjectBeColored( const Handle(HYDROData_Entity)& theObject,
48 HYDROGUI_Module* theModule )
50 if ( theObject.IsNull() )
54 theObject->IsKind( STANDARD_TYPE(HYDROData_Object) ) ||
55 theObject->IsKind( STANDARD_TYPE(HYDROData_IPolyline) );
57 return isCanBeColored;
60 void HYDROGUI_SetColorOp::startOperation()
62 HYDROGUI_Operation::startOperation();
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() ) )
73 bool anIsOneColor = false;
75 // Get colors from the object
76 QColor aFirstColor, aSecondColor;
77 if ( myEditedObject->IsKind( STANDARD_TYPE(HYDROData_Object) ) )
79 Handle(HYDROData_Object) aGeomObject =
80 Handle(HYDROData_Object)::DownCast( myEditedObject );
82 if ( myEditedObject->GetKind() == KIND_POLYLINE ||
83 myEditedObject->GetKind() == KIND_PROFILE )
85 aFirstColor = aGeomObject->GetBorderColor();
90 aFirstColor = aGeomObject->GetFillingColor();
91 aSecondColor = aGeomObject->GetBorderColor();
94 else if ( myEditedObject->IsKind( STANDARD_TYPE(HYDROData_IPolyline) ) )
96 Handle(HYDROData_IPolyline) aPolyObject =
97 Handle(HYDROData_IPolyline)::DownCast( myEditedObject );
99 //aFirstColor = aPolyObject->GetWireColor();
101 if (aPolyObject->NbSections() > 0)
102 aPolyObject->GetSectionColor(0, aFirstColor);
104 aFirstColor = aPolyObject->GetWireColor();
108 else if ( myEditedObject->IsKind( STANDARD_TYPE(HYDROData_LandCoverMap) ) )
110 Handle(HYDROData_LandCoverMap) aLandCoverMap =
111 Handle(HYDROData_LandCoverMap)::DownCast( myEditedObject );
114 aFirstColor = aLandCoverMap->GetFillingColor();
115 aSecondColor = aLandCoverMap->GetBorderColor();
119 // Create color dialog
120 myColorDlg = new HYDROGUI_ColorDlg( module()->getApp()->desktop(), anIsOneColor );
121 myColorDlg->setModal( true );
122 myColorDlg->setWindowTitle( getName() );
124 // Set colors from the object
125 myColorDlg->setFirstColor( aFirstColor );
126 myColorDlg->setSecondColor( aSecondColor );
128 // Connect the dialog to operation slots
129 connect( myColorDlg, SIGNAL( accepted() ), this, SLOT( onApply() ) );
130 connect( myColorDlg, SIGNAL( rejected() ), this, SLOT( onCancel() ) );
136 bool HYDROGUI_SetColorOp::processApply( int& theUpdateFlags,
137 QString& theErrorMsg,
138 QStringList& theBrowseObjectsEntries )
140 if ( !myColorDlg || myEditedObject.IsNull() )
143 QColor aFirstColor = myColorDlg->getFirstColor();
144 QColor aSecondColor = myColorDlg->getSecondColor();
146 if ( myEditedObject->IsKind( STANDARD_TYPE(HYDROData_Object) ) )
148 Handle(HYDROData_Object) aGeomObject =
149 Handle(HYDROData_Object)::DownCast( myEditedObject );
151 if ( myEditedObject->GetKind() == KIND_POLYLINE ||
152 myEditedObject->GetKind() == KIND_PROFILE )
154 aGeomObject->SetBorderColor( aFirstColor );
158 aGeomObject->SetFillingColor( aFirstColor );
159 aGeomObject->SetBorderColor( aSecondColor );
162 else if ( myEditedObject->IsKind( STANDARD_TYPE(HYDROData_IPolyline) ) )
164 Handle(HYDROData_IPolyline) aPolyObject =
165 Handle(HYDROData_IPolyline)::DownCast( myEditedObject );
167 //aPolyObject->SetWireColor( aFirstColor );
168 int nbSec = aPolyObject->NbSections();
171 for (int i = 0; i < nbSec; i++)
172 aPolyObject->SetSectionColor(i, aFirstColor);
175 aPolyObject->SetWireColor( aFirstColor );
177 else if ( myEditedObject->IsKind( STANDARD_TYPE(HYDROData_LandCoverMap) ) )
179 Handle(HYDROData_LandCoverMap) aLandCoverMap =
180 Handle(HYDROData_LandCoverMap)::DownCast( myEditedObject );
183 aLandCoverMap->SetFillingColor( aFirstColor );
184 aLandCoverMap->SetBorderColor( aSecondColor );
188 module()->setIsToUpdate( myEditedObject );
190 theUpdateFlags = UF_Model | UF_OCCViewer | UF_OCC_Forced | UF_VTKViewer;