-// Copyright (C) 2007-2013 CEA/DEN, EDF R&D, OPEN CASCADE
-//
-// Copyright (C) 2003-2007 OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
-// CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
-//
+// Copyright (C) 2014-2015 EDF-R&D
// This library is free software; you can redistribute it and/or
// modify it under the terms of the GNU Lesser General Public
// License as published by the Free Software Foundation; either
-// version 2.1 of the License.
+// version 2.1 of the License, or (at your option) any later version.
//
// This library is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
#include "HYDROGUI_ColorDlg.h"
#include "HYDROGUI_DataModel.h"
#include "HYDROGUI_Module.h"
-#include "HYDROGUI_Tool.h"
+#include "HYDROGUI_Tool2.h"
#include "HYDROGUI_UpdateFlags.h"
+#include <HYDROData_Object.h>
+#include <HYDROData_IPolyline.h>
+#include <HYDROData_LandCoverMap.h>
+
#include <LightApp_Application.h>
#include <LightApp_UpdateFlags.h>
{
}
+bool HYDROGUI_SetColorOp::CanObjectBeColored( const Handle(HYDROData_Entity)& theObject,
+ HYDROGUI_Module* theModule )
+{
+ if ( theObject.IsNull() )
+ return false;
+
+ bool isCanBeColored =
+ theObject->IsKind( STANDARD_TYPE(HYDROData_Object) ) ||
+ theObject->IsKind( STANDARD_TYPE(HYDROData_IPolyline) );
+
+ return isCanBeColored;
+}
+
void HYDROGUI_SetColorOp::startOperation()
{
HYDROGUI_Operation::startOperation();
// Get the selected object
- myEditedObject = Handle(HYDROData_Object)::DownCast( HYDROGUI_Tool::GetSelectedObject( module() ) );
-
- if( !myEditedObject.IsNull() ) {
- // Get colors from the object
- QColor aFillingColor = myEditedObject->GetFillingColor();
- QColor aBorderColor = myEditedObject->GetBorderColor();
+ //myEditedObject = Handle(HYDROData_Object)::DownCast( HYDROGUI_Tool::GetSelectedObject( module() ) );
+ myEditedObject = HYDROGUI_Tool::GetSelectedObject( module() );
+ if ( !CanObjectBeColored( myEditedObject, module() ) )
+ {
+ abort();
+ return;
+ }
- // Create color dialog
- myColorDlg = new HYDROGUI_ColorDlg( module()->getApp()->desktop() );
- myColorDlg->setModal( true );
- myColorDlg->setWindowTitle( getName() );
+ bool anIsOneColor = false;
+
+ // Get colors from the object
+ QColor aFirstColor, aSecondColor;
+ if ( myEditedObject->IsKind( STANDARD_TYPE(HYDROData_Object) ) )
+ {
+ Handle(HYDROData_Object) aGeomObject =
+ Handle(HYDROData_Object)::DownCast( myEditedObject );
+
+ if ( myEditedObject->GetKind() == KIND_POLYLINE ||
+ myEditedObject->GetKind() == KIND_PROFILE )
+ {
+ aFirstColor = aGeomObject->GetBorderColor();
+ anIsOneColor = true;
+ }
+ else
+ {
+ aFirstColor = aGeomObject->GetFillingColor();
+ aSecondColor = aGeomObject->GetBorderColor();
+ }
+ }
+ else if ( myEditedObject->IsKind( STANDARD_TYPE(HYDROData_IPolyline) ) )
+ {
+ Handle(HYDROData_IPolyline) aPolyObject =
+ Handle(HYDROData_IPolyline)::DownCast( myEditedObject );
- // Set colors from the object
- myColorDlg->setFillingColor( aFillingColor );
- myColorDlg->setBorderColor( aBorderColor );
+ //aFirstColor = aPolyObject->GetWireColor();
- // Connect the dialog to operation slots
- connect( myColorDlg, SIGNAL( accepted() ), this, SLOT( onApply() ) );
- connect( myColorDlg, SIGNAL( rejected() ), this, SLOT( onCancel() ) );
+ if (aPolyObject->NbSections() > 0)
+ aPolyObject->GetSectionColor(0, aFirstColor);
+ else
+ aFirstColor = aPolyObject->GetWireColor();
- // Show the dialog
- myColorDlg->exec();
+ anIsOneColor = true;
+ }
+ else if ( myEditedObject->IsKind( STANDARD_TYPE(HYDROData_LandCoverMap) ) )
+ {
+ Handle(HYDROData_LandCoverMap) aLandCoverMap =
+ Handle(HYDROData_LandCoverMap)::DownCast( myEditedObject );
+
+ /* TODO
+ aFirstColor = aLandCoverMap->GetFillingColor();
+ aSecondColor = aLandCoverMap->GetBorderColor();
+ */
}
-}
-bool HYDROGUI_SetColorOp::processApply( int& theUpdateFlags,
- QString& theErrorMsg )
-{
- bool anIsOk = false;
+ // Create color dialog
+ myColorDlg = new HYDROGUI_ColorDlg( module()->getApp()->desktop(), anIsOneColor );
+ myColorDlg->setModal( true );
+ myColorDlg->setWindowTitle( getName() );
- if ( myColorDlg && myEditedObject ) {
- QColor aFillingColor = myColorDlg->getFillingColor();
- QColor aBorderColor = myColorDlg->getBorderColor();
+ // Set colors from the object
+ myColorDlg->setFirstColor( aFirstColor );
+ myColorDlg->setSecondColor( aSecondColor );
- myEditedObject->SetFillingColor( aFillingColor );
- myEditedObject->SetBorderColor( aBorderColor );
+ // Connect the dialog to operation slots
+ connect( myColorDlg, SIGNAL( accepted() ), this, SLOT( onApply() ) );
+ connect( myColorDlg, SIGNAL( rejected() ), this, SLOT( onCancel() ) );
- theUpdateFlags = UF_Model | UF_OCCViewer | UF_OCC_Forced;
+ // Show the dialog
+ myColorDlg->exec();
+}
- anIsOk = true;
+bool HYDROGUI_SetColorOp::processApply( int& theUpdateFlags,
+ QString& theErrorMsg,
+ QStringList& theBrowseObjectsEntries )
+{
+ if ( !myColorDlg || myEditedObject.IsNull() )
+ return false;
+
+ QColor aFirstColor = myColorDlg->getFirstColor();
+ QColor aSecondColor = myColorDlg->getSecondColor();
+
+ if ( myEditedObject->IsKind( STANDARD_TYPE(HYDROData_Object) ) )
+ {
+ Handle(HYDROData_Object) aGeomObject =
+ Handle(HYDROData_Object)::DownCast( myEditedObject );
+
+ if ( myEditedObject->GetKind() == KIND_POLYLINE ||
+ myEditedObject->GetKind() == KIND_PROFILE )
+ {
+ aGeomObject->SetBorderColor( aFirstColor );
+ }
+ else
+ {
+ aGeomObject->SetFillingColor( aFirstColor );
+ aGeomObject->SetBorderColor( aSecondColor );
+ }
+ }
+ else if ( myEditedObject->IsKind( STANDARD_TYPE(HYDROData_IPolyline) ) )
+ {
+ Handle(HYDROData_IPolyline) aPolyObject =
+ Handle(HYDROData_IPolyline)::DownCast( myEditedObject );
+
+ //aPolyObject->SetWireColor( aFirstColor );
+ int nbSec = aPolyObject->NbSections();
+ if (nbSec > 0)
+ {
+ for (int i = 0; i < nbSec; i++)
+ aPolyObject->SetSectionColor(i, aFirstColor);
+ }
+ else
+ aPolyObject->SetWireColor( aFirstColor );
+ }
+ else if ( myEditedObject->IsKind( STANDARD_TYPE(HYDROData_LandCoverMap) ) )
+ {
+ Handle(HYDROData_LandCoverMap) aLandCoverMap =
+ Handle(HYDROData_LandCoverMap)::DownCast( myEditedObject );
+
+ /* TODO
+ aLandCoverMap->SetFillingColor( aFirstColor );
+ aLandCoverMap->SetBorderColor( aSecondColor );
+ */
}
+
+ module()->setIsToUpdate( myEditedObject );
+
+ theUpdateFlags = UF_Model | UF_OCCViewer | UF_OCC_Forced | UF_VTKViewer;
- return anIsOk;
+ return true;
}