Salome HOME
refs #651: wrapping land cover map by Python
[modules/hydro.git] / src / HYDROData / HYDROData_Zone.cxx
index 91765f5f79dbe4ad8f98cb05426a17dda04609cc..661e23c9b6a757a4ba1df0d1047fad084edf98c1 100644 (file)
+// Copyright (C) 2014-2015  EDF-R&D
+// This library is free software; you can redistribute it and/or
+// modify it under the terms of the GNU Lesser General Public
+// License as published by the Free Software Foundation; either
+// version 2.1 of the License, or (at your option) any later version.
+//
+// This library is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+// Lesser General Public License for more details.
+//
+// You should have received a copy of the GNU Lesser General Public
+// License along with this library; if not, write to the Free Software
+// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
+//
+// See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
+//
 
 #include "HYDROData_Zone.h"
 
-#include "HYDROData_Bathymetry.h"
+#include "HYDROData_ArtificialObject.h"
+#include "HYDROData_IAltitudeObject.h"
 #include "HYDROData_Document.h"
-#include "HYDROData_Polyline.h"
-#include "HYDROData_Iterator.h"
-
-#include <TDataStd_IntegerArray.hxx>
-
-#include <QColor>
+#include "HYDROData_NaturalObject.h"
 #include <QStringList>
+#include <TDataStd_Integer.hxx>
+#include <TopoDS_Shape.hxx>
 
-#define PYTHON_ZONE_ID "KIND_ZONE"
+IMPLEMENT_STANDARD_HANDLE(HYDROData_Zone, HYDROData_Entity)
+IMPLEMENT_STANDARD_RTTIEXT(HYDROData_Zone, HYDROData_Entity)
 
-IMPLEMENT_STANDARD_HANDLE(HYDROData_Zone, HYDROData_Object)
-IMPLEMENT_STANDARD_RTTIEXT(HYDROData_Zone, HYDROData_Object)
 
 HYDROData_Zone::HYDROData_Zone()
+: HYDROData_Entity( Geom_2d )
 {
+  myInterpolator = NULL;
 }
 
 HYDROData_Zone::~HYDROData_Zone()
 {
 }
 
-QStringList HYDROData_Zone::DumpToPython( MapOfTreatedObjects& theTreatedObjects ) const
+bool HYDROData_Zone::CanBeUpdated() const
 {
-  QStringList aResList;
-
-  Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( this );
-  if ( aDocument.IsNull() )
-    return aResList;
-
-  QString aDocName = aDocument->GetDocPyName();
-  QString aZoneName = GetName();
-
-  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( "" );
-
-  QColor aFillingColor = GetFillingColor();
-  aResList << QString( "filling_color = QColor( %1, %2, %3, %4 );" )
-              .arg( aFillingColor.red()  ).arg( aFillingColor.green() )
-              .arg( aFillingColor.blue() ).arg( aFillingColor.alpha() );
-  aResList << QString( "%1.SetFillingColor( filling_color );" ).arg( aZoneName );
-  aResList << QString( "" );
-
-  QColor aBorderColor = GetBorderColor();
-  aResList << QString( "border_color = QColor( %1, %2, %3, %4 );" )
-              .arg( aBorderColor.red()  ).arg( aBorderColor.green() )
-              .arg( aBorderColor.blue() ).arg( aBorderColor.alpha() );
-  aResList << QString( "%1.SetBorderColor( border_color );" ).arg( aZoneName );
-  aResList << QString( "" );
-
-  Handle(HYDROData_Polyline) aRefPolyline = GetPolyline();
-  setPythonReferenceObject( theTreatedObjects, aResList, aRefPolyline, "SetPolyline" );
-  aResList << QString( "" );
-
-  HYDROData_SequenceOfObjects aZoneBaths = GetBathymetries();
-  HYDROData_SequenceOfObjects::Iterator aBathsIter( aZoneBaths );
-  for ( ; aBathsIter.More(); aBathsIter.Next() )
-  {
-    Handle(HYDROData_Bathymetry) aRefBath =
-      Handle(HYDROData_Bathymetry)::DownCast( aBathsIter.Value() );
-    if ( !aRefBath.IsNull() )
-      setPythonReferenceObject( theTreatedObjects, aResList, aRefBath, "AddBathymetry" );
-  }
-
-  return aResList;
+  return false;
 }
 
-QColor HYDROData_Zone::DefaultFillingColor()
+bool HYDROData_Zone::IsHas2dPrs() const
 {
-  return QColor( Qt::green );
+  return true;
 }
 
-void HYDROData_Zone::SetFillingColor( const QColor& theColor )
+bool HYDROData_Zone::CanRemove()
 {
-  return SetColor( theColor, DataTag_FillingColor );
+  return false;
 }
 
-QColor HYDROData_Zone::GetFillingColor() const
+HYDROData_SequenceOfObjects HYDROData_Zone::GetAllReferenceObjects() const
 {
-  return GetColor( DefaultFillingColor(), DataTag_FillingColor );
+  HYDROData_SequenceOfObjects aResSeq = HYDROData_Entity::GetAllReferenceObjects();
+
+  HYDROData_SequenceOfObjects aSeqOfObjects = GetObjects();
+  aResSeq.Append( aSeqOfObjects );
+
+  return aResSeq;
 }
 
-QColor HYDROData_Zone::DefaultBorderColor()
+void HYDROData_Zone::SetShape( const TopoDS_Shape& theShape )
 {
-  return QColor( Qt::transparent );
+  HYDROData_Entity::SetShape( DataTag_Shape, theShape );
 }
 
-void HYDROData_Zone::SetBorderColor( const QColor& theColor )
+TopoDS_Shape HYDROData_Zone::GetShape() const
 {
-  return SetColor( theColor, DataTag_BorderColor );
+  return HYDROData_Entity::GetShape( DataTag_Shape );
 }
 
-QColor HYDROData_Zone::GetBorderColor() const
+bool HYDROData_Zone::IsMergingNeed() const
 {
-  return GetColor( DefaultBorderColor(), DataTag_BorderColor );
+  // Zones based on geometry objects (compare altitudes)
+  Handle(HYDROData_IAltitudeObject) aRefAltitude;
+
+  HYDROData_SequenceOfObjects anObjects = GetObjects();
+  HYDROData_SequenceOfObjects::Iterator anObjsIter( anObjects );
+  for ( ; anObjsIter.More(); anObjsIter.Next() )
+  {
+    Handle(HYDROData_Object) aRefGeomObj =
+      Handle(HYDROData_Object)::DownCast( anObjsIter.Value() );
+    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;
+  }
+
+  return false;
 }
 
-void HYDROData_Zone::SetPolyline( const Handle(HYDROData_Polyline)& thePolyline )
+void HYDROData_Zone::SetInterpolator( HYDROData_IInterpolator* theInter )
 {
-  SetReferenceObject( thePolyline, DataTag_Polyline );
+  myInterpolator = theInter;
 }
 
-Handle(HYDROData_Polyline) HYDROData_Zone::GetPolyline() const
+HYDROData_IInterpolator* HYDROData_Zone::GetInterpolator() const
 {
-  return Handle(HYDROData_Polyline)::DownCast( 
-           GetReferenceObject( DataTag_Polyline ) );
+  return myInterpolator;
 }
 
-void HYDROData_Zone::RemovePolyline()
+void HYDROData_Zone::SetMergeType( const MergeType& theType )
 {
-  ClearReferenceObjects( DataTag_Polyline );
+  TDataStd_Integer::Set( myLab.FindChild( DataTag_MergeType ), (int)theType );
 }
 
-int HYDROData_Zone::NbBathymetries() const
+HYDROData_Zone::MergeType HYDROData_Zone::GetMergeType() const
 {
-  return NbReferenceObjects( DataTag_Bathymetry );
+  MergeType 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 = (MergeType)anInt->Get();
+  }
+
+  return aMergeType;
 }
 
-void HYDROData_Zone::AddBathymetry( const Handle(HYDROData_Bathymetry)& theBathymetry )
+void HYDROData_Zone::SetMergeObject( const Handle(HYDROData_Entity)& theObject )
 {
-  AddReferenceObject( theBathymetry, DataTag_Bathymetry );
+  SetReferenceObject( theObject, DataTag_MergeObject );
 }
 
-void HYDROData_Zone::SetBathymetry( const int                           theIndex,
-                                    const Handle(HYDROData_Bathymetry)& theBathymetry )
+Handle(HYDROData_Entity) HYDROData_Zone::GetMergeObject() const
 {
-  SetReferenceObject( theBathymetry, DataTag_Bathymetry, theIndex );
+  return Handle(HYDROData_Entity)::DownCast( 
+           GetReferenceObject( DataTag_MergeObject ) );
 }
 
-Handle(HYDROData_Bathymetry) HYDROData_Zone::GetBathymetry( const int theIndex ) const
+void HYDROData_Zone::RemoveMergeObject()
 {
-  return Handle(HYDROData_Bathymetry)::DownCast( 
-           GetReferenceObject( DataTag_Bathymetry, theIndex ) );
+  ClearReferenceObjects( DataTag_MergeObject );
 }
 
-HYDROData_SequenceOfObjects HYDROData_Zone::GetBathymetries() const
+bool HYDROData_Zone::AddObject( const Handle(HYDROData_Entity)& theObject )
 {
-  return GetReferenceObjects( DataTag_Bathymetry );
+  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_Object ) )
+    return false; // Object is already in reference list
+
+  AddReferenceObject( theObject, DataTag_Object );
+
+  return true;
 }
 
-void HYDROData_Zone::RemoveBathymetries()
+HYDROData_SequenceOfObjects HYDROData_Zone::GetObjects() const
 {
-  ClearReferenceObjects( DataTag_Bathymetry );
+  return GetReferenceObjects( DataTag_Object );
 }
 
-QPainterPath HYDROData_Zone::GetPainterPath() const
+void HYDROData_Zone::RemoveObjects()
 {
-  QPainterPath aPath;
+  ClearReferenceObjects( DataTag_Object );
+}
 
-  Handle(HYDROData_Polyline) aPolyline = GetPolyline();
-  if ( !aPolyline.IsNull() )
+bool HYDROData_Zone::IsSubmersible() const
+{
+  HYDROData_SequenceOfObjects anObjects = GetObjects();
+  HYDROData_SequenceOfObjects::Iterator anObjsIter( anObjects );
+  for ( ; anObjsIter.More(); anObjsIter.Next() )
   {
-    aPath = aPolyline->painterPath();
-  }
+    Handle(HYDROData_Object) aRefGeomObj =
+      Handle(HYDROData_Object)::DownCast( anObjsIter.Value() );
+    if ( aRefGeomObj.IsNull() )
+      continue;
 
-  return aPath;
+    if( !aRefGeomObj->IsSubmersible() )
+      return false; //if one of geometry objects is not submersible the zone is considered as not submersible
+  }
+  return true;
 }