17.12.2013. Added Partition algorithm (draft version).

[modules/hydro.git] / src / HYDROData / HYDROData_Object.cxx

diff --git a/src/HYDROData/HYDROData_Object.cxx b/src/HYDROData/HYDROData_Object.cxx
index 87ffb2dfbdd881d30748c7f100de859e02104ad9..7dfbe8134ae4531f34a0a390c5ec1bc35c2fa972 100644 (file)
--- a/src/HYDROData/HYDROData_Object.cxx
+++ b/src/HYDROData/HYDROData_Object.cxx
@@ -2,12 +2,17 @@
 #include "HYDROData_Object.h"
 
 #include "HYDROData_Bathymetry.h"
+#include "HYDROData_DummyObject3D.h"
+#include "HYDROData_ShapesGroup.h"
+#include "HYDROData_Iterator.h"
 
 #include <TNaming_Builder.hxx>
 #include <TNaming_NamedShape.hxx>
 
 #include <TopoDS_Shape.hxx>
 
+#include <QColor>
+
 IMPLEMENT_STANDARD_HANDLE(HYDROData_Object,HYDROData_Entity)
 IMPLEMENT_STANDARD_RTTIEXT(HYDROData_Object,HYDROData_Entity)
 
@@ -20,6 +25,34 @@ HYDROData_Object::~HYDROData_Object()
 {
 }
 
+void HYDROData_Object::Update()
+{
+  HYDROData_Entity::Update();
+  removeTopShape();
+  removeShape3D();
+  removeGroupObjects();
+}
+
+HYDROData_SequenceOfObjects HYDROData_Object::GetAllReferenceObjects() const
+{
+  HYDROData_SequenceOfObjects aResSeq = HYDROData_Entity::GetAllReferenceObjects();
+
+  Handle(HYDROData_Bathymetry) aRefBathymetry = GetBathymetry();
+  if ( !aRefBathymetry.IsNull() )
+    aResSeq.Append( aRefBathymetry );
+
+  return aResSeq;
+}
+
+void HYDROData_Object::SetToUpdate( bool theFlag )
+{
+  HYDROData_Entity::SetToUpdate( theFlag );
+
+  Handle(HYDROData_DummyObject3D) anObject3D = GetObject3D();
+  if ( !anObject3D.IsNull() )
+    anObject3D->SetToUpdate( theFlag );
+}
+
 void HYDROData_Object::SetTopShape( const TopoDS_Shape& theShape )
 {
   TNaming_Builder aBuilder( myLab.FindChild( DataTag_TopShape ) );
@@ -30,13 +63,62 @@ void HYDROData_Object::SetShape3D( const TopoDS_Shape& theShape )
 {
   TNaming_Builder aBuilder( myLab.FindChild( DataTag_Shape3D ) );
   aBuilder.Generated( theShape );
+  
+  // Check the object 3D existance
+  checkAndSetObject3D();
 }
 
-void HYDROData_Object::SetBathymetry( const Handle(HYDROData_Bathymetry)& theBathymetry )
+Handle(HYDROData_DummyObject3D) HYDROData_Object::GetObject3D() const
 {
+  Handle(HYDROData_DummyObject3D) anObject;
+  
+  TDF_Label aLabel = myLab.FindChild( DataTag_Object3D, false );
+  if ( !aLabel.IsNull() )
+  {
+    TDF_Label aChildLabel = aLabel.FindChild( 0, false );
+    if ( !aChildLabel.IsNull() )
+    {
+      anObject = Handle(HYDROData_DummyObject3D)::DownCast(
+        HYDROData_Iterator::Object( aChildLabel ) );
+    }
+  }
+
+  return anObject;
+}
+
+void HYDROData_Object::checkAndSetObject3D()
+{
+  TDF_Label aLabel = myLab.FindChild( DataTag_Object3D, false );
+  if ( !aLabel.IsNull() )
+    return;
+
+  TDF_Label aChildLabel = myLab.FindChild( DataTag_Object3D ).FindChild( 0 );
+  HYDROData_Iterator::CreateObject( aChildLabel, KIND_DUMMY_3D );
+}
+
+HYDROData_SequenceOfObjects HYDROData_Object::GetGroups() const
+{
+  return GetReferenceObjects( DataTag_EdgesGroup );
+}
+
+bool HYDROData_Object::SetBathymetry( const Handle(HYDROData_Bathymetry)& theBathymetry )
+{
+  if ( theBathymetry.IsNull() )
+    return false;
+  
+  Handle(HYDROData_Bathymetry) aPrevBathymetry = GetBathymetry();
+  if ( IsEqual( aPrevBathymetry, theBathymetry ) )
+    return true;
+
   SetReferenceObject( theBathymetry, DataTag_Bathymetry );
+
+  // Indicate model of the need to update object
+  SetToUpdate( true );
+
+  return true;
 }
 
+
 Handle(HYDROData_Bathymetry) HYDROData_Object::GetBathymetry() const
 {
   return Handle(HYDROData_Bathymetry)::DownCast( 
@@ -45,22 +127,115 @@ Handle(HYDROData_Bathymetry) HYDROData_Object::GetBathymetry() const
 
 void HYDROData_Object::RemoveBathymetry()
 {
+  Handle(HYDROData_Bathymetry) aPrevBathymetry = GetBathymetry();
+  if ( aPrevBathymetry.IsNull() )
+    return;
+
   ClearReferenceObjects( DataTag_Bathymetry );
+
+  // 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::DefaultFillingColor()
+{
+  return QColor( Qt::yellow );
+}
+
+QColor HYDROData_Object::DefaultBorderColor()
+{
+  return QColor( Qt::transparent );
+}
+
+QColor HYDROData_Object::getDefaultFillingColor() const
+{
+  return DefaultFillingColor();
+}
+
+QColor HYDROData_Object::getDefaultBorderColor() const
+{
+  return DefaultBorderColor();
+}
+
+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::createGroupObjects()
+{
+}
+
+void HYDROData_Object::removeGroupObjects()
+{
+  TDF_Label aLabel = myLab.FindChild( DataTag_EdgesGroup, false );
+  if ( !aLabel.IsNull() )
+    aLabel.ForgetAllAttributes();
 }
 
 TopoDS_Shape HYDROData_Object::getTopShape() const
 {
-  Handle(TNaming_NamedShape) aNamedShape;
-  if( myLab.FindChild( DataTag_TopShape ).FindAttribute( TNaming_NamedShape::GetID(), aNamedShape ) )
-    return aNamedShape->Get();
+  TDF_Label aLabel = myLab.FindChild( DataTag_TopShape, false );
+  if ( !aLabel.IsNull() )
+  {
+    Handle(TNaming_NamedShape) aNamedShape;
+    if( aLabel.FindAttribute( TNaming_NamedShape::GetID(), aNamedShape ) )
+      return aNamedShape->Get();
+  }
+
   return TopoDS_Shape();
 }
 
+void HYDROData_Object::removeTopShape()
+{
+  TDF_Label aLabel = myLab.FindChild( DataTag_TopShape, false );
+  if ( !aLabel.IsNull() )
+    aLabel.ForgetAllAttributes();
+}
+
 TopoDS_Shape HYDROData_Object::getShape3D() const
 {
-  Handle(TNaming_NamedShape) aNamedShape;
-  if( myLab.FindChild( DataTag_Shape3D ).FindAttribute( TNaming_NamedShape::GetID(), aNamedShape ) )
-    return aNamedShape->Get();
+  TDF_Label aLabel = myLab.FindChild( DataTag_Shape3D, false );
+  if ( !aLabel.IsNull() )
+  {
+    Handle(TNaming_NamedShape) aNamedShape;
+    if( aLabel.FindAttribute( TNaming_NamedShape::GetID(), aNamedShape ) )
+      return aNamedShape->Get();
+  }
+
   return TopoDS_Shape();
 }
 
+void HYDROData_Object::removeShape3D()
+{
+  TDF_Label aLabel = myLab.FindChild( DataTag_Shape3D, false );
+  if ( !aLabel.IsNull() )
+    aLabel.ForgetAllAttributes();
+}
+