src/HYDROData/HYDROData_Zone.cxx

   1
   2 #include "HYDROData_Zone.h"
   3
   4 #include "HYDROData_ArtificialObject.h"
   5 #include "HYDROData_IAltitudeObject.h"
   6 #include "HYDROData_Document.h"
   7 #include "HYDROData_NaturalObject.h"
   8
   9 #include <TNaming_Builder.hxx>
  10 #include <TNaming_NamedShape.hxx>
  11
  12 #include <TopoDS_Shape.hxx>
  13
  14 #include <QStringList>
  15
  16 #include <TDataStd_Integer.hxx>
  17
  18 IMPLEMENT_STANDARD_HANDLE(HYDROData_Zone, HYDROData_Entity)
  19 IMPLEMENT_STANDARD_RTTIEXT(HYDROData_Zone, HYDROData_Entity)
  20
  21
  22 HYDROData_Zone::HYDROData_Zone()
  23 : HYDROData_Entity()
  24 {
  25 }
  26
  27 HYDROData_Zone::~HYDROData_Zone()
  28 {
  29 }
  30
  31 QStringList HYDROData_Zone::DumpToPython( MapOfTreatedObjects& theTreatedObjects ) const
  32 {
  33   QStringList aResList = dumpObjectCreation( theTreatedObjects );
  34
  35   HYDROData_SequenceOfObjects aGeomObjects = GetGeometryObjects();
  36   HYDROData_SequenceOfObjects::Iterator aGeomObjsIter( aGeomObjects );
  37   for ( ; aGeomObjsIter.More(); aGeomObjsIter.Next() )
  38   {
  39     Handle(HYDROData_Object) aRefGeomObj =
  40       Handle(HYDROData_Object)::DownCast( aGeomObjsIter.Value() );
  41     if ( !aRefGeomObj.IsNull() )
  42       setPythonReferenceObject( theTreatedObjects, aResList, aRefGeomObj, "AddGeometryObject" );
  43   }
  44
  45   // How can we get the shape? Mb Update() method to intersect the shapes of reference objects?
  46   // TODO:  TopoDS_Shape aRefShape = GetShape();
  47   QString aName = GetName();
  48   aResList << QString( "" );
  49   aResList << QString( "%1.Update();" ).arg( aName );
  50   aResList << QString( "" );
  51
  52   return aResList;
  53 }
  54
  55 bool HYDROData_Zone::CanBeUpdated() const
  56 {
  57   return false;
  58 }
  59
  60 bool HYDROData_Zone::CanRemove()
  61 {
  62   return false;
  63 }
  64
  65 HYDROData_SequenceOfObjects HYDROData_Zone::GetAllReferenceObjects() const
  66 {
  67   HYDROData_SequenceOfObjects aResSeq = HYDROData_Entity::GetAllReferenceObjects();
  68
  69   HYDROData_SequenceOfObjects aSeqOfGeomObjects = GetGeometryObjects();
  70   aResSeq.Append( aSeqOfGeomObjects );
  71
  72   return aResSeq;
  73 }
  74
  75 void HYDROData_Zone::SetShape( const TopoDS_Shape& theShape )
  76 {
  77   TNaming_Builder aBuilder( myLab.FindChild( DataTag_Shape ) );
  78   aBuilder.Generated( theShape );
  79 }
  80
  81 TopoDS_Shape HYDROData_Zone::GetShape() const
  82 {
  83   TDF_Label aLabel = myLab.FindChild( DataTag_Shape, false );
  84   if ( !aLabel.IsNull() )
  85   {
  86     Handle(TNaming_NamedShape) aNamedShape;
  87     if( aLabel.FindAttribute( TNaming_NamedShape::GetID(), aNamedShape ) )
  88       return aNamedShape->Get();
  89   }
  90
  91   return TopoDS_Shape();
  92 }
  93
  94 bool HYDROData_Zone::IsMergingNeed() const
  95 {
  96   Handle(HYDROData_IAltitudeObject) aRefAltitude;
  97
  98   HYDROData_SequenceOfObjects aGeomObjects = GetGeometryObjects();
  99   HYDROData_SequenceOfObjects::Iterator aGeomObjsIter( aGeomObjects );
 100   for ( ; aGeomObjsIter.More(); aGeomObjsIter.Next() )
 101   {
 102     Handle(HYDROData_Object) aRefGeomObj =
 103       Handle(HYDROData_Object)::DownCast( aGeomObjsIter.Value() );
 104     if ( aRefGeomObj.IsNull() )
 105       continue;
 106
 107     Handle(HYDROData_IAltitudeObject) anObjAltitude = aRefGeomObj->GetAltitudeObject();
 108     if ( anObjAltitude.IsNull() )
 109       continue;
 110
 111     if ( aRefAltitude.IsNull() )
 112     {
 113       aRefAltitude = anObjAltitude;
 114       continue;
 115     }
 116
 117     if ( !IsEqual( aRefAltitude, anObjAltitude ) )
 118       return true;
 119   }
 120
 121   return false;
 122 }
 123
 124 void HYDROData_Zone::SetMergeType( const MergeAltitudesType& theType )
 125 {
 126   Handle(TDataStd_Integer) anInt;
 127   if ( myLab.FindChild( DataTag_MergeType ).FindAttribute( TDataStd_Integer::GetID(), anInt ) )
 128   {
 129     anInt->Set( (int)theType );
 130   }
 131   else
 132   {
 133     anInt = TDataStd_Integer::Set( myLab.FindChild( DataTag_MergeType ), (int)theType );
 134   }
 135 }
 136
 137 HYDROData_Zone::MergeAltitudesType HYDROData_Zone::GetMergeType() const
 138 {
 139   MergeAltitudesType aMergeType = Merge_UNKNOWN;
 140
 141   TDF_Label aLabel = myLab.FindChild( DataTag_MergeType, false );
 142   if ( !aLabel.IsNull() )
 143   {
 144     Handle(TDataStd_Integer) anInt;
 145     if ( aLabel.FindAttribute( TDataStd_Integer::GetID(), anInt ) )
 146       aMergeType = (MergeAltitudesType)anInt->Get();
 147   }
 148
 149   return aMergeType;
 150 }
 151
 152 void HYDROData_Zone::SetMergeAltitude( const Handle(HYDROData_IAltitudeObject)& theAltitude )
 153 {
 154   SetReferenceObject( theAltitude, DataTag_MergeAltitude );
 155 }
 156
 157 Handle(HYDROData_IAltitudeObject) HYDROData_Zone::GetMergeAltitude() const
 158 {
 159   return Handle(HYDROData_IAltitudeObject)::DownCast(
 160            GetReferenceObject( DataTag_MergeAltitude ) );
 161 }
 162
 163 void HYDROData_Zone::RemoveMergeAltitude()
 164 {
 165   ClearReferenceObjects( DataTag_MergeAltitude );
 166 }
 167
 168 bool HYDROData_Zone::AddGeometryObject( const Handle(HYDROData_Object)& theObject )
 169 {
 170   if ( theObject.IsNull() )
 171     return false;
 172
 173   if ( !theObject->IsKind( STANDARD_TYPE(HYDROData_ArtificialObject) ) &&
 174        !theObject->IsKind( STANDARD_TYPE(HYDROData_NaturalObject) ) )
 175     return false; // Wrong type of object
 176
 177   if ( HasReference( theObject, DataTag_GeometryObject ) )
 178     return false; // Object is already in reference list
 179
 180   AddReferenceObject( theObject, DataTag_GeometryObject );
 181
 182   return true;
 183 }
 184
 185 HYDROData_SequenceOfObjects HYDROData_Zone::GetGeometryObjects() const
 186 {
 187   return GetReferenceObjects( DataTag_GeometryObject );
 188 }
 189
 190 void HYDROData_Zone::RemoveGeometryObjects()
 191 {
 192   ClearReferenceObjects( DataTag_GeometryObject );
 193 }
 194
 195
 196