2 #include "HYDROData_Zone.h"
4 #include "HYDROData_ArtificialObject.h"
5 #include "HYDROData_IAltitudeObject.h"
6 #include "HYDROData_Document.h"
7 #include "HYDROData_NaturalObject.h"
9 #include <TNaming_Builder.hxx>
10 #include <TNaming_NamedShape.hxx>
12 #include <TopoDS_Shape.hxx>
14 #include <QStringList>
16 #include <TDataStd_Integer.hxx>
18 IMPLEMENT_STANDARD_HANDLE(HYDROData_Zone, HYDROData_Entity)
19 IMPLEMENT_STANDARD_RTTIEXT(HYDROData_Zone, HYDROData_Entity)
22 HYDROData_Zone::HYDROData_Zone()
27 HYDROData_Zone::~HYDROData_Zone()
31 QStringList HYDROData_Zone::DumpToPython( MapOfTreatedObjects& theTreatedObjects ) const
33 QStringList aResList = HYDROData_Entity::DumpToPython( theTreatedObjects );
35 HYDROData_SequenceOfObjects aGeomObjects = GetGeometryObjects();
36 HYDROData_SequenceOfObjects::Iterator aGeomObjsIter( aGeomObjects );
37 for ( ; aGeomObjsIter.More(); aGeomObjsIter.Next() )
39 Handle(HYDROData_Object) aRefGeomObj =
40 Handle(HYDROData_Object)::DownCast( aGeomObjsIter.Value() );
41 if ( !aRefGeomObj.IsNull() )
42 setPythonReferenceObject( theTreatedObjects, aResList, aRefGeomObj, "AddGeometryObject" );
45 // How can we get the shape? Mb Update() method to intersect the shapes of reference objects?
46 // TODO: TopoDS_Shape aRefShape = GetShape();
51 bool HYDROData_Zone::CanBeUpdated() const
56 bool HYDROData_Zone::CanRemove()
61 HYDROData_SequenceOfObjects HYDROData_Zone::GetAllReferenceObjects() const
63 HYDROData_SequenceOfObjects aResSeq = HYDROData_Entity::GetAllReferenceObjects();
65 HYDROData_SequenceOfObjects aSeqOfGeomObjects = GetGeometryObjects();
66 aResSeq.Append( aSeqOfGeomObjects );
71 void HYDROData_Zone::SetShape( const TopoDS_Shape& theShape )
73 TNaming_Builder aBuilder( myLab.FindChild( DataTag_Shape ) );
74 aBuilder.Generated( theShape );
77 TopoDS_Shape HYDROData_Zone::GetShape() const
79 TDF_Label aLabel = myLab.FindChild( DataTag_Shape, false );
80 if ( !aLabel.IsNull() )
82 Handle(TNaming_NamedShape) aNamedShape;
83 if( aLabel.FindAttribute( TNaming_NamedShape::GetID(), aNamedShape ) )
84 return aNamedShape->Get();
87 return TopoDS_Shape();
90 bool HYDROData_Zone::IsMergingNeed() const
92 Handle(HYDROData_IAltitudeObject) aRefAltitude;
94 HYDROData_SequenceOfObjects aGeomObjects = GetGeometryObjects();
95 HYDROData_SequenceOfObjects::Iterator aGeomObjsIter( aGeomObjects );
96 for ( ; aGeomObjsIter.More(); aGeomObjsIter.Next() )
98 Handle(HYDROData_Object) aRefGeomObj =
99 Handle(HYDROData_Object)::DownCast( aGeomObjsIter.Value() );
100 if ( aRefGeomObj.IsNull() )
103 Handle(HYDROData_IAltitudeObject) anObjAltitude = aRefGeomObj->GetAltitudeObject();
104 if ( anObjAltitude.IsNull() )
107 if ( aRefAltitude.IsNull() )
109 aRefAltitude = anObjAltitude;
113 if ( !IsEqual( aRefAltitude, anObjAltitude ) )
120 void HYDROData_Zone::SetMergeType( const MergeAltitudesType& theType )
122 Handle(TDataStd_Integer) anInt;
123 if ( myLab.FindChild( DataTag_MergeType ).FindAttribute( TDataStd_Integer::GetID(), anInt ) )
125 anInt->Set( (int)theType );
129 anInt = TDataStd_Integer::Set( myLab.FindChild( DataTag_MergeType ), (int)theType );
133 HYDROData_Zone::MergeAltitudesType HYDROData_Zone::GetMergeType() const
135 MergeAltitudesType aMergeType = Merge_UNKNOWN;
137 TDF_Label aLabel = myLab.FindChild( DataTag_MergeType, false );
138 if ( !aLabel.IsNull() )
140 Handle(TDataStd_Integer) anInt;
141 if ( aLabel.FindAttribute( TDataStd_Integer::GetID(), anInt ) )
142 aMergeType = (MergeAltitudesType)anInt->Get();
148 void HYDROData_Zone::SetMergeAltitude( const Handle(HYDROData_IAltitudeObject)& theAltitude )
150 SetReferenceObject( theAltitude, DataTag_MergeAltitude );
153 Handle(HYDROData_IAltitudeObject) HYDROData_Zone::GetMergeAltitude() const
155 return Handle(HYDROData_IAltitudeObject)::DownCast(
156 GetReferenceObject( DataTag_MergeAltitude ) );
159 void HYDROData_Zone::RemoveMergeAltitude()
161 ClearReferenceObjects( DataTag_MergeAltitude );
164 bool HYDROData_Zone::AddGeometryObject( const Handle(HYDROData_Object)& theObject )
166 if ( theObject.IsNull() )
169 if ( !theObject->IsKind( STANDARD_TYPE(HYDROData_ArtificialObject) ) &&
170 !theObject->IsKind( STANDARD_TYPE(HYDROData_NaturalObject) ) )
171 return false; // Wrong type of object
173 if ( HasReference( theObject, DataTag_GeometryObject ) )
174 return false; // Object is already in reference list
176 AddReferenceObject( theObject, DataTag_GeometryObject );
181 HYDROData_SequenceOfObjects HYDROData_Zone::GetGeometryObjects() const
183 return GetReferenceObjects( DataTag_GeometryObject );
186 void HYDROData_Zone::RemoveGeometryObjects()
188 ClearReferenceObjects( DataTag_GeometryObject );