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 #define PYTHON_ZONE_ID "KIND_ZONE"
20 IMPLEMENT_STANDARD_HANDLE(HYDROData_Zone, HYDROData_Entity)
21 IMPLEMENT_STANDARD_RTTIEXT(HYDROData_Zone, HYDROData_Entity)
24 HYDROData_Zone::HYDROData_Zone()
29 HYDROData_Zone::~HYDROData_Zone()
33 QStringList HYDROData_Zone::DumpToPython( MapOfTreatedObjects& theTreatedObjects ) const
37 Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( myLab );
38 if ( aDocument.IsNull() )
41 QString aDocName = aDocument->GetDocPyName();
42 QString aZoneName = GetName();
44 aResList << QString( "%1 = %2.CreateObject( %3 );" )
45 .arg( aZoneName ).arg( aDocName ).arg( PYTHON_ZONE_ID );
46 aResList << QString( "%1.SetName( \"%2\" );" )
47 .arg( aZoneName ).arg( aZoneName );
48 aResList << QString( "" );
50 HYDROData_SequenceOfObjects aGeomObjects = GetGeometryObjects();
51 HYDROData_SequenceOfObjects::Iterator aGeomObjsIter( aGeomObjects );
52 for ( ; aGeomObjsIter.More(); aGeomObjsIter.Next() )
54 Handle(HYDROData_Object) aRefGeomObj =
55 Handle(HYDROData_Object)::DownCast( aGeomObjsIter.Value() );
56 if ( !aRefGeomObj.IsNull() )
57 setPythonReferenceObject( theTreatedObjects, aResList, aRefGeomObj, "AddGeometryObject" );
60 // How can we get the shape? Mb Update() method to intersect the shapes of reference objects?
61 // TODO: TopoDS_Shape aRefShape = GetShape();
66 bool HYDROData_Zone::CanBeUpdated() const
71 bool HYDROData_Zone::CanRemove()
76 HYDROData_SequenceOfObjects HYDROData_Zone::GetAllReferenceObjects() const
78 HYDROData_SequenceOfObjects aResSeq = HYDROData_Entity::GetAllReferenceObjects();
80 HYDROData_SequenceOfObjects aSeqOfGeomObjects = GetGeometryObjects();
81 aResSeq.Append( aSeqOfGeomObjects );
86 void HYDROData_Zone::SetShape( const TopoDS_Shape& theShape )
88 TNaming_Builder aBuilder( myLab.FindChild( DataTag_Shape ) );
89 aBuilder.Generated( theShape );
92 TopoDS_Shape HYDROData_Zone::GetShape() const
94 TDF_Label aLabel = myLab.FindChild( DataTag_Shape, false );
95 if ( !aLabel.IsNull() )
97 Handle(TNaming_NamedShape) aNamedShape;
98 if( aLabel.FindAttribute( TNaming_NamedShape::GetID(), aNamedShape ) )
99 return aNamedShape->Get();
102 return TopoDS_Shape();
105 bool HYDROData_Zone::IsMergingNeed() const
107 Handle(HYDROData_IAltitudeObject) aRefAltitude;
109 HYDROData_SequenceOfObjects aGeomObjects = GetGeometryObjects();
110 HYDROData_SequenceOfObjects::Iterator aGeomObjsIter( aGeomObjects );
111 for ( ; aGeomObjsIter.More(); aGeomObjsIter.Next() )
113 Handle(HYDROData_Object) aRefGeomObj =
114 Handle(HYDROData_Object)::DownCast( aGeomObjsIter.Value() );
115 if ( aRefGeomObj.IsNull() )
118 Handle(HYDROData_IAltitudeObject) anObjAltitude = aRefGeomObj->GetAltitudeObject();
119 if ( anObjAltitude.IsNull() )
122 if ( aRefAltitude.IsNull() )
124 aRefAltitude = anObjAltitude;
128 if ( !IsEqual( aRefAltitude, anObjAltitude ) )
135 void HYDROData_Zone::SetMergeType( const MergeAltitudesType& theType )
137 Handle(TDataStd_Integer) anInt;
138 if ( myLab.FindChild( DataTag_MergeType ).FindAttribute( TDataStd_Integer::GetID(), anInt ) )
140 anInt->Set( (int)theType );
144 anInt = TDataStd_Integer::Set( myLab.FindChild( DataTag_MergeType ), (int)theType );
148 HYDROData_Zone::MergeAltitudesType HYDROData_Zone::GetMergeType() const
150 MergeAltitudesType aMergeType = Merge_UNKNOWN;
152 TDF_Label aLabel = myLab.FindChild( DataTag_MergeType, false );
153 if ( !aLabel.IsNull() )
155 Handle(TDataStd_Integer) anInt;
156 if ( aLabel.FindAttribute( TDataStd_Integer::GetID(), anInt ) )
157 aMergeType = (MergeAltitudesType)anInt->Get();
163 void HYDROData_Zone::SetMergeAltitude( const Handle(HYDROData_IAltitudeObject)& theAltitude )
165 SetReferenceObject( theAltitude, DataTag_MergeAltitude );
168 Handle(HYDROData_IAltitudeObject) HYDROData_Zone::GetMergeAltitude() const
170 return Handle(HYDROData_IAltitudeObject)::DownCast(
171 GetReferenceObject( DataTag_MergeAltitude ) );
174 void HYDROData_Zone::RemoveMergeAltitude()
176 ClearReferenceObjects( DataTag_MergeAltitude );
179 bool HYDROData_Zone::AddGeometryObject( const Handle(HYDROData_Object)& theObject )
181 if ( theObject.IsNull() )
184 if ( !theObject->IsKind( STANDARD_TYPE(HYDROData_ArtificialObject) ) &&
185 !theObject->IsKind( STANDARD_TYPE(HYDROData_NaturalObject) ) )
186 return false; // Wrong type of object
188 if ( HasReference( theObject, DataTag_GeometryObject ) )
189 return false; // Object is already in reference list
191 AddReferenceObject( theObject, DataTag_GeometryObject );
196 HYDROData_SequenceOfObjects HYDROData_Zone::GetGeometryObjects() const
198 return GetReferenceObjects( DataTag_GeometryObject );
201 void HYDROData_Zone::RemoveGeometryObjects()
203 ClearReferenceObjects( DataTag_GeometryObject );