X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=src%2FHYDROData%2FHYDROData_Object.cxx;h=25b9e2625187447f1a6a2ad224eb99c530b2da13;hb=23a6f7b3e29e74735d71d7e1448465a9920f91dd;hp=5349d354169574155e9a35e97f47d6984f779571;hpb=89c9794998495fcd82e49e22a75502367de46bd5;p=modules%2Fhydro.git diff --git a/src/HYDROData/HYDROData_Object.cxx b/src/HYDROData/HYDROData_Object.cxx index 5349d354..25b9e262 100644 --- a/src/HYDROData/HYDROData_Object.cxx +++ b/src/HYDROData/HYDROData_Object.cxx @@ -1,14 +1,34 @@ +// 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, 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 +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +// Lesser General Public License for more details. +// +// You should have received a copy of the GNU Lesser General Public +// License along with this library; if not, write to the Free Software +// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA +// +// See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com +// #include "HYDROData_Object.h" -#include "HYDROData_Bathymetry.h" #include "HYDROData_DummyObject3D.h" +#include "HYDROData_ShapesGroup.h" +#include "HYDROData_Tool.h" #include "HYDROData_Iterator.h" +#include "HYDROData_IAltitudeObject.h" #include #include #include +#include #include @@ -24,13 +44,51 @@ HYDROData_Object::~HYDROData_Object() { } +void HYDROData_Object::SetName( const QString& theName ) +{ + QString anOldObjName = GetName(); + if ( anOldObjName != theName ) + { + HYDROData_SequenceOfObjects aGroups = GetGroups(); + HYDROData_SequenceOfObjects::Iterator anIter( aGroups ); + for ( ; anIter.More(); anIter.Next() ) + { + Handle(HYDROData_ShapesGroup) aGroup = + Handle(HYDROData_ShapesGroup)::DownCast( anIter.Value() ); + if ( aGroup.IsNull() ) + continue; + + HYDROData_Tool::UpdateChildObjectName( anOldObjName, theName, aGroup ); + } + + Handle(HYDROData_DummyObject3D) anObject3D = GetObject3D(); + if ( !anObject3D.IsNull() ) + HYDROData_Tool::UpdateChildObjectName( anOldObjName, theName, anObject3D ); + + Handle(HYDROData_IAltitudeObject) anAltitudeObj = getChildAltitudeObject(); + if ( !anAltitudeObj.IsNull() ) + HYDROData_Tool::UpdateChildObjectName( anOldObjName, theName, anAltitudeObj ); + } + + HYDROData_Entity::SetName( theName ); +} + +void HYDROData_Object::Update() +{ + removeTopShape(); + removeShape3D(); + removeGroupObjects(); + checkAndSetAltitudeObject(); + HYDROData_Entity::Update(); +} + HYDROData_SequenceOfObjects HYDROData_Object::GetAllReferenceObjects() const { HYDROData_SequenceOfObjects aResSeq = HYDROData_Entity::GetAllReferenceObjects(); - Handle(HYDROData_Bathymetry) aRefBathymetry = GetBathymetry(); - if ( !aRefBathymetry.IsNull() ) - aResSeq.Append( aRefBathymetry ); + Handle(HYDROData_IAltitudeObject) aRefAltitude = GetAltitudeObject(); + if ( !aRefAltitude.IsNull() ) + aResSeq.Append( aRefAltitude ); return aResSeq; } @@ -87,31 +145,198 @@ void HYDROData_Object::checkAndSetObject3D() HYDROData_Iterator::CreateObject( aChildLabel, KIND_DUMMY_3D ); } -void HYDROData_Object::Update() +HYDROData_SequenceOfObjects HYDROData_Object::GetGroups() const { - HYDROData_Entity::Update(); - removeTopShape(); - removeShape3D(); + return GetReferenceObjects( DataTag_EdgesGroup ); } -void HYDROData_Object::SetBathymetry( const Handle(HYDROData_Bathymetry)& theBathymetry ) +Handle(HYDROData_ShapesGroup) HYDROData_Object::GetGroup( const int theGroupId ) const { - SetReferenceObject( theBathymetry, DataTag_Bathymetry ); + Handle(HYDROData_ShapesGroup) aResGroup; + + HYDROData_SequenceOfObjects aGroups = GetGroups(); + if ( theGroupId < 0 || theGroupId >= aGroups.Length() ) + return aResGroup; + + aResGroup = Handle(HYDROData_ShapesGroup)::DownCast( aGroups.Value( theGroupId + 1 ) ); + + return aResGroup; +} + +int HYDROData_Object::GetGroupId( const Handle(HYDROData_ShapesGroup)& theGroup ) const +{ + int aRes = -1; + + HYDROData_SequenceOfObjects aGroups = GetGroups(); + for ( int i = 1, n = aGroups.Length(); i <= n; ++i ) + { + Handle(HYDROData_ShapesGroup) aGroup = + Handle(HYDROData_ShapesGroup)::DownCast( aGroups.Value( i ) ); + if ( IsEqual( theGroup, aGroup ) ) + { + aRes = i - 1; + break; + } + } + + return aRes; +} + +bool HYDROData_Object::SetAltitudeObject( + const Handle(HYDROData_IAltitudeObject)& theAltitude ) +{ + if ( theAltitude.IsNull() ) + return false; + + Handle(HYDROData_IAltitudeObject) aPrevAltitude = GetAltitudeObject(); + if ( IsEqual( aPrevAltitude, theAltitude ) ) + return true; + + SetReferenceObject( theAltitude, DataTag_AltitudeObject ); + + // Indicate model of the need to update object SetToUpdate( true ); + + return true; } -Handle(HYDROData_Bathymetry) HYDROData_Object::GetBathymetry() const +Handle(HYDROData_IAltitudeObject) HYDROData_Object::GetAltitudeObject() const { - return Handle(HYDROData_Bathymetry)::DownCast( - GetReferenceObject( DataTag_Bathymetry ) ); + return Handle(HYDROData_IAltitudeObject)::DownCast( + GetReferenceObject( DataTag_AltitudeObject ) ); } -void HYDROData_Object::RemoveBathymetry() +void HYDROData_Object::RemoveAltitudeObject() { - ClearReferenceObjects( DataTag_Bathymetry ); + Handle(HYDROData_IAltitudeObject) aPrevAltitude = GetAltitudeObject(); + if ( aPrevAltitude.IsNull() ) + return; + + ClearReferenceObjects( DataTag_AltitudeObject ); + + // Indicate model of the need to update object SetToUpdate( true ); } +void HYDROData_Object::SetFillingColor( const QColor& theColor ) +{ + SetColor( theColor, DataTag_FillingColor ); +} + +QColor HYDROData_Object::GetFillingColor() const +{ + return GetColor( getDefaultFillingColor(), DataTag_FillingColor ); +} + +void HYDROData_Object::SetBorderColor( const QColor& theColor ) +{ + SetColor( theColor, DataTag_BorderColor ); +} + +QColor HYDROData_Object::GetBorderColor() const +{ + return GetColor( getDefaultBorderColor(), DataTag_BorderColor ); +} + +QColor HYDROData_Object::getDefaultFillingColor() const +{ + return QColor( Qt::yellow ); +} + +QColor HYDROData_Object::getDefaultBorderColor() const +{ + return QColor( Qt::transparent ); +} + +QStringList HYDROData_Object::dumpObjectCreation( MapOfTreatedObjects& theTreatedObjects ) const +{ + QStringList aResList = HYDROData_Entity::dumpObjectCreation( theTreatedObjects ); + if ( aResList.isEmpty() ) + return aResList; //Object was not created + + QStringList aColorsDef; + + QColor aFillingColor = GetFillingColor(); + setPythonObjectColor( aColorsDef, aFillingColor, getDefaultFillingColor(), "SetFillingColor" ); + + QColor aBorderColor = GetBorderColor(); + setPythonObjectColor( aColorsDef, aBorderColor, getDefaultBorderColor(), "SetBorderColor" ); + + if ( !aColorsDef.isEmpty() ) + { + aResList << aColorsDef; + aResList << QString( "" ); + } + + return aResList; +} + +ObjectKind HYDROData_Object::getAltitudeObjectType() const +{ + return KIND_UNKNOWN; +} + +Handle(HYDROData_IAltitudeObject) HYDROData_Object::getChildAltitudeObject() const +{ + Handle(HYDROData_IAltitudeObject) anObject; + + TDF_Label aLabel = myLab.FindChild( DataTag_ChildAltitudeObject, false ); + if ( !aLabel.IsNull() ) + { + TDF_Label aChildLabel = aLabel.FindChild( 0, false ); + if ( !aChildLabel.IsNull() ) + { + anObject = Handle(HYDROData_IAltitudeObject)::DownCast( + HYDROData_Iterator::Object( aChildLabel ) ); + } + } + + return anObject; +} + +void HYDROData_Object::checkAndSetAltitudeObject() +{ + Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( myLab ); + + ObjectKind anAltitudeObjectType = getAltitudeObjectType(); + if ( anAltitudeObjectType == KIND_UNKNOWN ) + return; // No need to create altitude object + + TDF_Label aLabel = myLab.FindChild( DataTag_ChildAltitudeObject, false ); + if ( !aLabel.IsNull() ) + return; + + TDF_Label aChildLabel = myLab.FindChild( DataTag_ChildAltitudeObject ).FindChild( 0 ); + + Handle(HYDROData_IAltitudeObject) anAltitudeObject = + Handle(HYDROData_IAltitudeObject)::DownCast( + HYDROData_Iterator::CreateObject( aChildLabel, anAltitudeObjectType ) ); + + QString anAltitudePref = GetName() + "_Altitude"; + QString anAltitudeName = HYDROData_Tool::GenerateObjectName( aDocument, anAltitudePref ); + anAltitudeObject->SetName( anAltitudeName ); + + SetAltitudeObject( anAltitudeObject ); +} + +Handle(HYDROData_ShapesGroup) HYDROData_Object::createGroupObject() +{ + TDF_Label aNewLab = myLab.FindChild( DataTag_EdgesGroup ).NewChild(); + + Handle(HYDROData_ShapesGroup) aNewGroup = + Handle(HYDROData_ShapesGroup)::DownCast( HYDROData_Iterator::CreateObject( aNewLab, KIND_SHAPES_GROUP ) ); + AddReferenceObject( aNewGroup, DataTag_EdgesGroup ); + + return aNewGroup; +} + +void HYDROData_Object::removeGroupObjects() +{ + TDF_Label aLabel = myLab.FindChild( DataTag_EdgesGroup, false ); + if ( !aLabel.IsNull() ) + aLabel.ForgetAllAttributes(); +} + TopoDS_Shape HYDROData_Object::getTopShape() const { TDF_Label aLabel = myLab.FindChild( DataTag_TopShape, false ); @@ -152,32 +377,20 @@ void HYDROData_Object::removeShape3D() aLabel.ForgetAllAttributes(); } -void HYDROData_Object::SetFillingColor( const QColor& theColor ) -{ - SetColor( theColor, DataTag_FillingColor ); -} - -QColor HYDROData_Object::GetFillingColor() const +bool HYDROData_Object::IsSubmersible() const { - return GetColor( DefaultFillingColor(), DataTag_FillingColor ); -} - -void HYDROData_Object::SetBorderColor( const QColor& theColor ) -{ - SetColor( theColor, DataTag_BorderColor ); -} + Handle(TDataStd_Integer) aSubMersibleAttr; -QColor HYDROData_Object::GetBorderColor() const -{ - return GetColor( DefaultBorderColor(), DataTag_BorderColor ); + bool isSubmersible = true; //default + if( myLab.FindAttribute(TDataStd_Integer::GetID(), aSubMersibleAttr ) ) + { + int aValue = aSubMersibleAttr->Get(); + isSubmersible = ( aValue != 0 ); + } + return isSubmersible; } -QColor HYDROData_Object::DefaultFillingColor() +void HYDROData_Object::SetIsSubmersible( bool isSubmersible ) const { - return QColor( Qt::yellow ); + TDataStd_Integer::Set( myLab, isSubmersible ? 1 : 0 ); } - -QColor HYDROData_Object::DefaultBorderColor() -{ - return QColor( Qt::transparent ); -} \ No newline at end of file