refs #1330: basic implementation of the not zoomable polyline arrows

[modules/hydro.git] / src / HYDROData / HYDROData_Document.cxx

diff --git a/src/HYDROData/HYDROData_Document.cxx b/src/HYDROData/HYDROData_Document.cxx
index 6a8f7376402874535a0698799b981a0a1c76ccc6..b59211f6e3cf6e0703b456cdca13852c54645f9b 100644 (file)
--- a/src/HYDROData/HYDROData_Document.cxx
+++ b/src/HYDROData/HYDROData_Document.cxx
@@ -49,6 +49,8 @@ static const int TAG_OBJECTS = 2; // tag of the objects sub-tree
 static const int TAG_HISTORY = 3; // tag of the history sub-tree (Root for History)
 static const int TAG_LOCAL_CS = 4; // tag of local coordinate system information
 static const int TAG_DEF_STRICKLER_COEFF = 5; // tag of default strickler coefficient
+static const int TAG_COUNT_QUADTREE = 6; // tag of number of quadtrees created so far
+static const int TAG_COUNT_DELAUNAY = 7; // tag of number of Delaunay triangulations created so far
 static const gp_Pnt2d DEFAULT_LOCAL_CS( 0, 0 );
 
 using namespace std;
@@ -231,6 +233,56 @@ void HYDROData_Document::SetDefaultStricklerCoefficient( double theCoeff ) const
     }
 }
 
+int HYDROData_Document::GetCountQuadtree() const
+{
+  int nbQuad = 0;
+  TDF_Label aLabel = myDoc->Main().FindChild(TAG_COUNT_QUADTREE, Standard_False);
+  if ( !aLabel.IsNull() )
+  {
+      Handle(TDataStd_Integer) anAttr;
+      if ( aLabel.FindAttribute( TDataStd_Integer::GetID(), anAttr ) )
+          nbQuad = anAttr->Get();
+  }
+  return nbQuad;
+}
+
+void HYDROData_Document::SetCountQuadtree( int nbQuad) const
+{
+  TDF_Label aLabel = myDoc->Main().FindChild(TAG_COUNT_QUADTREE);
+  if ( !aLabel.IsNull() )
+  {
+      Handle(TDataStd_Integer) anAttr;
+      if ( !aLabel.FindAttribute( TDataStd_Integer::GetID(), anAttr ) )
+          aLabel.AddAttribute( anAttr = new TDataStd_Integer() );
+      anAttr->Set( nbQuad );
+  }
+}
+
+int HYDROData_Document::GetCountDelaunay() const
+{
+  int nbDelaunay = 0;
+  TDF_Label aLabel = myDoc->Main().FindChild(TAG_COUNT_DELAUNAY, Standard_False);
+  if ( !aLabel.IsNull() )
+  {
+      Handle(TDataStd_Integer) anAttr;
+      if ( aLabel.FindAttribute( TDataStd_Integer::GetID(), anAttr ) )
+        nbDelaunay = anAttr->Get();
+  }
+  return nbDelaunay;
+}
+
+void HYDROData_Document::SetCountDelaunay( int nbDelaunay) const
+{
+  TDF_Label aLabel = myDoc->Main().FindChild(TAG_COUNT_DELAUNAY);
+  if ( !aLabel.IsNull() )
+  {
+      Handle(TDataStd_Integer) anAttr;
+      if ( !aLabel.FindAttribute( TDataStd_Integer::GetID(), anAttr ) )
+          aLabel.AddAttribute( anAttr = new TDataStd_Integer() );
+      anAttr->Set( nbDelaunay );
+  }
+}
+
 bool HYDROData_Document::DumpToPython( const QString& thePyScriptPath,
                                        const bool     theIsMultiFile ) const
 {
@@ -313,8 +365,8 @@ QStringList HYDROData_Document::DumpToPython( const QString& thePyScriptPath,
   QStringList aResScript;
 
   aResScript << QString( "from HYDROPy import *" );
-  aResScript << QString( "from PyQt4.QtCore import *" );
-  aResScript << QString( "from PyQt4.QtGui import *" );
+  aResScript << QString( "from PyQt5.QtCore import *" );
+  aResScript << QString( "from PyQt5.QtGui import *" );
 
   if ( theIsMultiFile )
   {