2 #include "HYDROData_Region.h"
4 #include "HYDROData_Document.h"
5 #include "HYDROData_Iterator.h"
6 #include "HYDROData_Zone.h"
10 #define PYTHON_REGION_ID "KIND_REGION"
12 IMPLEMENT_STANDARD_HANDLE(HYDROData_Region, HYDROData_Entity)
13 IMPLEMENT_STANDARD_RTTIEXT(HYDROData_Region, HYDROData_Entity)
16 HYDROData_Region::HYDROData_Region()
21 HYDROData_Region::~HYDROData_Region()
25 QStringList HYDROData_Region::DumpToPython( MapOfTreatedObjects& theTreatedObjects ) const
29 Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( myLab );
30 if ( aDocument.IsNull() )
33 QString aDocName = aDocument->GetDocPyName();
34 QString aRegionName = GetName();
36 aResList << QString( "%1 = %2.CreateObject( %3 );" )
37 .arg( aRegionName ).arg( aDocName ).arg( PYTHON_REGION_ID );
38 aResList << QString( "%1.SetName( \"%2\" );" )
39 .arg( aRegionName ).arg( aRegionName );
40 aResList << QString( "" );
42 HYDROData_SequenceOfObjects aZones = GetZones();
43 HYDROData_SequenceOfObjects::Iterator anIter( aZones );
44 for ( ; anIter.More(); anIter.Next() )
46 Handle(HYDROData_Zone) aRefZone =
47 Handle(HYDROData_Zone)::DownCast( anIter.Value() );
48 if ( !aRefZone.IsNull() )
49 setPythonReferenceObject( theTreatedObjects, aResList, aRefZone, "AddZone" );
51 aResList << QString( "" );
56 bool HYDROData_Region::AddZone( const Handle(HYDROData_Zone)& theZone )
58 if ( theZone.IsNull() )
61 if ( HasReference( theZone, DataTag_Zone ) )
62 return false; // Object is already in reference list
64 // Move the zone from other region
65 Handle(HYDROData_Region) aFatherRegion =
66 Handle(HYDROData_Region)::DownCast( theZone->GetFatherObject() );
67 if ( !aFatherRegion.IsNull() && aFatherRegion->Label() != myLab )
69 Handle(HYDROData_Zone) aNewZone = addNewZone();
70 theZone->CopyTo( aNewZone );
72 aFatherRegion->RemoveZone( theZone );
74 theZone->SetLabel( aNewZone->Label() );
78 AddReferenceObject( theZone, DataTag_Zone );
84 HYDROData_SequenceOfObjects HYDROData_Region::GetZones() const
86 return GetReferenceObjects( DataTag_Zone );
89 void HYDROData_Region::RemoveZone( const Handle(HYDROData_Zone)& theZone )
91 if ( theZone.IsNull() )
94 RemoveReferenceObject( theZone->Label(), DataTag_Zone );
96 // Remove zone from data model
97 Handle(HYDROData_Region) aFatherRegion =
98 Handle(HYDROData_Region)::DownCast( theZone->GetFatherObject() );
99 if ( !aFatherRegion.IsNull() && aFatherRegion->Label() == myLab )
102 // If the last zone has been removed from region we remove this region
103 HYDROData_SequenceOfObjects aRefZones = GetZones();
104 if ( aRefZones.IsEmpty() )
108 void HYDROData_Region::RemoveZones()
110 ClearReferenceObjects( DataTag_Zone );
111 myLab.FindChild( DataTag_ChildZone ).ForgetAllAttributes( true );
114 Handle(HYDROData_Zone) HYDROData_Region::addNewZone()
116 TDF_Label aNewLab = myLab.FindChild( DataTag_ChildZone ).NewChild();
118 Handle(HYDROData_Zone) aNewZone =
119 Handle(HYDROData_Zone)::DownCast( HYDROData_Iterator::CreateObject( aNewLab, KIND_ZONE ) );