#include "HYDROData_Zone.h"
+#include "HYDROData_ArtificialObject.h"
+#include "HYDROData_IAltitudeObject.h"
#include "HYDROData_Document.h"
-#include "HYDROData_Object.h"
+#include "HYDROData_NaturalObject.h"
#include <TNaming_Builder.hxx>
#include <TNaming_NamedShape.hxx>
#include <QStringList>
-#define PYTHON_ZONE_ID "KIND_ZONE"
+#include <TDataStd_Integer.hxx>
IMPLEMENT_STANDARD_HANDLE(HYDROData_Zone, HYDROData_Entity)
IMPLEMENT_STANDARD_RTTIEXT(HYDROData_Zone, HYDROData_Entity)
HYDROData_Zone::HYDROData_Zone()
: HYDROData_Entity()
{
+ myInterpolator = NULL;
}
HYDROData_Zone::~HYDROData_Zone()
{
}
-QStringList HYDROData_Zone::DumpToPython( MapOfTreatedObjects& theTreatedObjects ) const
+bool HYDROData_Zone::CanBeUpdated() const
{
- QStringList aResList;
+ return false;
+}
+
+bool HYDROData_Zone::CanRemove()
+{
+ return false;
+}
- Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( this );
- if ( aDocument.IsNull() )
- return aResList;
+HYDROData_SequenceOfObjects HYDROData_Zone::GetAllReferenceObjects() const
+{
+ HYDROData_SequenceOfObjects aResSeq = HYDROData_Entity::GetAllReferenceObjects();
- QString aDocName = aDocument->GetDocPyName();
- QString aZoneName = GetName();
+ HYDROData_SequenceOfObjects aSeqOfGeomObjects = GetGeometryObjects();
+ aResSeq.Append( aSeqOfGeomObjects );
- aResList << QString( "%1 = %2.CreateObject( %3 );" )
- .arg( aZoneName ).arg( aDocName ).arg( PYTHON_ZONE_ID );
- aResList << QString( "%1.SetName( \"%2\" );" )
- .arg( aZoneName ).arg( aZoneName );
- aResList << QString( "" );
+ return aResSeq;
+}
+
+void HYDROData_Zone::SetShape( const TopoDS_Shape& theShape )
+{
+ TNaming_Builder aBuilder( myLab.FindChild( DataTag_Shape ) );
+ aBuilder.Generated( theShape );
+}
+
+TopoDS_Shape HYDROData_Zone::GetShape() const
+{
+ TDF_Label aLabel = myLab.FindChild( DataTag_Shape, false );
+ if ( !aLabel.IsNull() )
+ {
+ Handle(TNaming_NamedShape) aNamedShape;
+ if( aLabel.FindAttribute( TNaming_NamedShape::GetID(), aNamedShape ) )
+ return aNamedShape->Get();
+ }
+
+ return TopoDS_Shape();
+}
+
+bool HYDROData_Zone::IsMergingNeed() const
+{
+ Handle(HYDROData_IAltitudeObject) aRefAltitude;
HYDROData_SequenceOfObjects aGeomObjects = GetGeometryObjects();
HYDROData_SequenceOfObjects::Iterator aGeomObjsIter( aGeomObjects );
{
Handle(HYDROData_Object) aRefGeomObj =
Handle(HYDROData_Object)::DownCast( aGeomObjsIter.Value() );
- if ( !aRefGeomObj.IsNull() )
- setPythonReferenceObject( theTreatedObjects, aResList, aRefGeomObj, "AddGeometryObject" );
+ if ( aRefGeomObj.IsNull() )
+ continue;
+
+ Handle(HYDROData_IAltitudeObject) anObjAltitude = aRefGeomObj->GetAltitudeObject();
+ if ( anObjAltitude.IsNull() )
+ continue;
+
+ if ( aRefAltitude.IsNull() )
+ {
+ aRefAltitude = anObjAltitude;
+ continue;
+ }
+
+ if ( !IsEqual( aRefAltitude, anObjAltitude ) )
+ return true;
}
- // How can we get the shape? Mb Update() method to intersect the shapes of reference objects?
- // TODO: TopoDS_Shape aRefShape = GetShape();
+ return false;
+}
- return aResList;
+void HYDROData_Zone::SetInterpolator( HYDROData_IInterpolator* theInter )
+{
+ myInterpolator = theInter;
}
-void HYDROData_Zone::SetShape( const TopoDS_Shape& theShape )
+HYDROData_IInterpolator* HYDROData_Zone::GetInterpolator() const
{
- TNaming_Builder aBuilder( myLab.FindChild( DataTag_Shape ) );
- aBuilder.Generated( theShape );
+ return myInterpolator;
}
-TopoDS_Shape HYDROData_Zone::GetShape() const
+void HYDROData_Zone::SetMergeType( const MergeAltitudesType& theType )
{
- Handle(TNaming_NamedShape) aNamedShape;
- if( myLab.FindChild( DataTag_Shape ).FindAttribute( TNaming_NamedShape::GetID(), aNamedShape ) )
- return aNamedShape->Get();
- return TopoDS_Shape();
+ TDataStd_Integer::Set( myLab.FindChild( DataTag_MergeType ), (int)theType );
}
-int HYDROData_Zone::NbGeometryObjects() const
+HYDROData_Zone::MergeAltitudesType HYDROData_Zone::GetMergeType() const
{
- return NbReferenceObjects( DataTag_GeometryObject );
+ MergeAltitudesType aMergeType = Merge_UNKNOWN;
+
+ TDF_Label aLabel = myLab.FindChild( DataTag_MergeType, false );
+ if ( !aLabel.IsNull() )
+ {
+ Handle(TDataStd_Integer) anInt;
+ if ( aLabel.FindAttribute( TDataStd_Integer::GetID(), anInt ) )
+ aMergeType = (MergeAltitudesType)anInt->Get();
+ }
+
+ return aMergeType;
}
-void HYDROData_Zone::AddGeometryObject( const Handle(HYDROData_Object)& theObject )
+void HYDROData_Zone::SetMergeAltitude( const Handle(HYDROData_IAltitudeObject)& theAltitude )
{
- AddReferenceObject( theObject, DataTag_GeometryObject );
+ SetReferenceObject( theAltitude, DataTag_MergeAltitude );
}
-void HYDROData_Zone::SetGeometryObject( const int theIndex,
- const Handle(HYDROData_Object)& theObject )
+Handle(HYDROData_IAltitudeObject) HYDROData_Zone::GetMergeAltitude() const
{
- SetReferenceObject( theObject, DataTag_GeometryObject, theIndex );
+ return Handle(HYDROData_IAltitudeObject)::DownCast(
+ GetReferenceObject( DataTag_MergeAltitude ) );
}
-void HYDROData_Zone::SetGeometryObjects( const HYDROData_SequenceOfObjects& theObjects )
+void HYDROData_Zone::RemoveMergeAltitude()
{
- SetReferenceObjects( theObjects, DataTag_GeometryObject );
+ ClearReferenceObjects( DataTag_MergeAltitude );
}
-Handle(HYDROData_Object) HYDROData_Zone::GetGeometryObject( const int theIndex ) const
+bool HYDROData_Zone::AddGeometryObject( const Handle(HYDROData_Object)& theObject )
{
- return Handle(HYDROData_Object)::DownCast(
- GetReferenceObject( DataTag_GeometryObject, theIndex ) );
+ if ( theObject.IsNull() )
+ return false;
+
+ if ( !theObject->IsKind( STANDARD_TYPE(HYDROData_ArtificialObject) ) &&
+ !theObject->IsKind( STANDARD_TYPE(HYDROData_NaturalObject) ) )
+ return false; // Wrong type of object
+
+ if ( HasReference( theObject, DataTag_GeometryObject ) )
+ return false; // Object is already in reference list
+
+ AddReferenceObject( theObject, DataTag_GeometryObject );
+
+ return true;
}
HYDROData_SequenceOfObjects HYDROData_Zone::GetGeometryObjects() const
-