1 // Copyright (C) 2007-2015 CEA/DEN, EDF R&D, OPEN CASCADE
3 // Copyright (C) 2003-2007 OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
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, or (at your option) any later version.
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.
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
20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
23 #include "HYDROGUI_SetColorOp.h"
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"
31 #include <HYDROData_Object.h>
32 #include <HYDROData_IPolyline.h>
34 #include <LightApp_Application.h>
35 #include <LightApp_UpdateFlags.h>
37 #include <SUIT_Desktop.h>
39 HYDROGUI_SetColorOp::HYDROGUI_SetColorOp( HYDROGUI_Module* theModule )
40 : HYDROGUI_Operation( theModule ),
43 setName( tr( "SET_COLOR" ) );
46 HYDROGUI_SetColorOp::~HYDROGUI_SetColorOp()
50 bool HYDROGUI_SetColorOp::CanObjectBeColored( const Handle(HYDROData_Entity)& theObject )
52 if ( theObject.IsNull() )
55 return theObject->IsKind( STANDARD_TYPE(HYDROData_Object) ) ||
56 theObject->IsKind( STANDARD_TYPE(HYDROData_IPolyline) );
59 void HYDROGUI_SetColorOp::startOperation()
61 HYDROGUI_Operation::startOperation();
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 ) )
72 bool anIsOneColor = false;
74 // Get colors from the object
75 QColor aFirstColor, aSecondColor;
76 if ( myEditedObject->IsKind( STANDARD_TYPE(HYDROData_Object) ) )
78 Handle(HYDROData_Object) aGeomObject =
79 Handle(HYDROData_Object)::DownCast( myEditedObject );
81 if ( myEditedObject->GetKind() == KIND_POLYLINE ||
82 myEditedObject->GetKind() == KIND_PROFILE )
84 aFirstColor = aGeomObject->GetBorderColor();
89 aFirstColor = aGeomObject->GetFillingColor();
90 aSecondColor = aGeomObject->GetBorderColor();
93 else if ( myEditedObject->IsKind( STANDARD_TYPE(HYDROData_IPolyline) ) )
95 Handle(HYDROData_IPolyline) aPolyObject =
96 Handle(HYDROData_IPolyline)::DownCast( myEditedObject );
98 aFirstColor = aPolyObject->GetWireColor();
103 // Create color dialog
104 myColorDlg = new HYDROGUI_ColorDlg( module()->getApp()->desktop(), anIsOneColor );
105 myColorDlg->setModal( true );
106 myColorDlg->setWindowTitle( getName() );
108 // Set colors from the object
109 myColorDlg->setFirstColor( aFirstColor );
110 myColorDlg->setSecondColor( aSecondColor );
112 // Connect the dialog to operation slots
113 connect( myColorDlg, SIGNAL( accepted() ), this, SLOT( onApply() ) );
114 connect( myColorDlg, SIGNAL( rejected() ), this, SLOT( onCancel() ) );
120 bool HYDROGUI_SetColorOp::processApply( int& theUpdateFlags,
121 QString& theErrorMsg,
122 QStringList& theBrowseObjectsEntries )
124 if ( !myColorDlg || myEditedObject.IsNull() )
127 QColor aFirstColor = myColorDlg->getFirstColor();
128 QColor aSecondColor = myColorDlg->getSecondColor();
130 if ( myEditedObject->IsKind( STANDARD_TYPE(HYDROData_Object) ) )
132 Handle(HYDROData_Object) aGeomObject =
133 Handle(HYDROData_Object)::DownCast( myEditedObject );
135 if ( myEditedObject->GetKind() == KIND_POLYLINE ||
136 myEditedObject->GetKind() == KIND_PROFILE )
138 aGeomObject->SetBorderColor( aFirstColor );
142 aGeomObject->SetFillingColor( aFirstColor );
143 aGeomObject->SetBorderColor( aSecondColor );
146 else if ( myEditedObject->IsKind( STANDARD_TYPE(HYDROData_IPolyline) ) )
148 Handle(HYDROData_IPolyline) aPolyObject =
149 Handle(HYDROData_IPolyline)::DownCast( myEditedObject );
151 aPolyObject->SetWireColor( aFirstColor );
154 module()->setIsToUpdate( myEditedObject );
156 theUpdateFlags = UF_Model | UF_OCCViewer | UF_OCC_Forced | UF_VTKViewer;