--- /dev/null
+# -*- coding: utf-8 -*-
+
+import os
+HYDRO_SAMPLES = os.path.join( os.environ["HYDRO_ROOT_DIR"], "bin/salome/test/HYDRO")
+
+import sys
+import salome
+
+salome.salome_init()
+theStudy = salome.myStudy
+
+from HYDROPy import *
+from PyQt4.QtCore import *
+from PyQt4.QtGui import *
+
+# -------------------------------------
+
+def controlGeomProps(geomShape, refLength, refArea):
+ props = geompy.BasicProperties(geomShape)
+ print " Wires length: ", props[0]
+ print " Surface area: ", props[1]
+ print " Volume : ", props[2]
+ deltaLength = 2.0*abs((props[0] - refLength)/(props[0] + refLength))
+ deltaArea = 2.0*abs((props[1] - refArea)/(props[1] + refArea))
+ if deltaLength > 1e-7 or deltaArea > 1e-7 or props[2] != 0:
+ print "While must be:"
+ print " Wires length: ", refLength
+ print " Surface area: ", refArea
+ print " Volume : ", 0.
+ raise ValueError("Bad length or area")
+
+# -------------------------------------
+
+hydro_doc = HYDROData_Document.Document( theStudy._get_StudyId() )
+
+hydro_doc.SetLocalCS( 430000, 6.35e+06 )
+
+garonne_ign_01 = hydro_doc.CreateObject( KIND_IMAGE )
+garonne_ign_01.SetName( "garonne_ign_01" )
+
+garonne_ign_01.SetZLevel( 0 )
+
+
+garonne_ign_01.LoadImage(os.path.join(HYDRO_SAMPLES, "garonne_ign_01.png" ))
+
+garonne_ign_01.SetLocalPoints( QPoint( 40, 817 ),
+ QPoint( 1325, 85 ) )
+
+garonne_ign_01.SetGlobalPoints( 1,
+ QPointF( 471562, 6.36775e+06 ),
+ QPointF( 489400, 6.37702e+06 ) )
+
+garonne_ign_01.Update()
+
+garonne_ign_02 = hydro_doc.CreateObject( KIND_IMAGE )
+garonne_ign_02.SetName( "garonne_ign_02" )
+
+garonne_ign_02.SetZLevel( 1 )
+
+
+garonne_ign_02.LoadImage( os.path.join(HYDRO_SAMPLES, "garonne_ign_02.png" ))
+
+garonne_ign_02.SetLocalPoints( QPoint( 1389, 447 ),
+ QPoint( 784, 481 ) )
+
+garonne_ign_02.SetGlobalPoints( 3,
+ QPointF( 631, 95 ),
+ QPointF( 26, 129 ) )
+garonne_ign_02.SetTrsfReferenceImage( garonne_ign_01 )
+
+garonne_ign_02.Update()
+
+garonne = hydro_doc.CreateObject( KIND_POLYLINEXY )
+garonne.SetName( "garonne" )
+
+garonne.SetZLevel( 2 )
+
+garonne.AddSection( "Section_1", 1, 1 )
+garonne.AddPoint( 0, gp_XY( 40717.94, 32193.70 ) )
+garonne.AddPoint( 0, gp_XY( 41361.00, 32464.18 ) )
+garonne.AddPoint( 0, gp_XY( 41831.41, 32626.69 ) )
+garonne.AddPoint( 0, gp_XY( 41950.14, 32618.24 ) )
+garonne.AddPoint( 0, gp_XY( 42524.97, 32419.61 ) )
+garonne.AddPoint( 0, gp_XY( 42909.80, 32291.33 ) )
+garonne.AddPoint( 0, gp_XY( 43177.03, 32269.95 ) )
+garonne.AddPoint( 0, gp_XY( 43465.65, 32344.78 ) )
+garonne.AddPoint( 0, gp_XY( 43754.27, 32152.37 ) )
+garonne.AddPoint( 0, gp_XY( 44128.40, 31842.37 ) )
+garonne.AddPoint( 0, gp_XY( 44003.90, 31269.93 ) )
+garonne.AddPoint( 0, gp_XY( 44010.82, 30869.63 ) )
+garonne.AddPoint( 0, gp_XY( 44064.26, 30655.84 ) )
+garonne.AddPoint( 0, gp_XY( 44434.73, 30366.18 ) )
+garonne.AddPoint( 0, gp_XY( 45208.04, 30484.81 ) )
+garonne.AddPoint( 0, gp_XY( 45705.01, 30287.34 ) )
+garonne.AddPoint( 0, gp_XY( 45933.89, 30046.87 ) )
+garonne.AddPoint( 0, gp_XY( 46127.34, 29490.68 ) )
+garonne.AddPoint( 0, gp_XY( 46111.60, 29226.08 ) )
+garonne.AddPoint( 0, gp_XY( 46228.62, 28784.05 ) )
+garonne.AddPoint( 0, gp_XY( 46247.64, 28449.75 ) )
+garonne.AddPoint( 0, gp_XY( 46108.48, 28086.94 ) )
+garonne.AddPoint( 0, gp_XY( 45618.52, 27553.74 ) )
+garonne.AddPoint( 0, gp_XY( 45276.77, 27042.78 ) )
+garonne.AddPoint( 0, gp_XY( 45001.89, 26273.15 ) )
+garonne.AddPoint( 0, gp_XY( 45167.76, 25975.34 ) )
+garonne.AddPoint( 0, gp_XY( 46018.15, 25848.84 ) )
+garonne.AddPoint( 0, gp_XY( 46378.94, 25872.21 ) )
+garonne.AddPoint( 0, gp_XY( 46885.57, 25937.64 ) )
+garonne.AddPoint( 0, gp_XY( 47278.36, 25782.88 ) )
+garonne.AddPoint( 0, gp_XY( 47153.86, 25728.04 ) )
+garonne.AddPoint( 0, gp_XY( 46987.51, 25749.97 ) )
+garonne.AddPoint( 0, gp_XY( 46658.26, 25715.71 ) )
+garonne.AddPoint( 0, gp_XY( 45650.76, 25634.46 ) )
+garonne.AddPoint( 0, gp_XY( 45122.63, 25748.21 ) )
+garonne.AddPoint( 0, gp_XY( 44797.63, 26325.09 ) )
+garonne.AddPoint( 0, gp_XY( 45114.51, 27168.75 ) )
+garonne.AddPoint( 0, gp_XY( 45508.28, 27713.21 ) )
+garonne.AddPoint( 0, gp_XY( 45994.19, 28382.53 ) )
+garonne.AddPoint( 0, gp_XY( 45927.01, 29201.34 ) )
+garonne.AddPoint( 0, gp_XY( 45756.39, 29859.47 ) )
+garonne.AddPoint( 0, gp_XY( 45390.76, 30151.97 ) )
+garonne.AddPoint( 0, gp_XY( 45039.85, 30120.84 ) )
+garonne.AddPoint( 0, gp_XY( 44513.26, 30084.28 ) )
+garonne.AddPoint( 0, gp_XY( 43923.42, 30426.28 ) )
+garonne.AddPoint( 0, gp_XY( 43768.81, 30934.29 ) )
+garonne.AddPoint( 0, gp_XY( 43820.10, 31425.20 ) )
+garonne.AddPoint( 0, gp_XY( 43854.78, 31801.12 ) )
+garonne.AddPoint( 0, gp_XY( 43479.64, 32053.54 ) )
+garonne.AddPoint( 0, gp_XY( 42923.47, 32051.85 ) )
+garonne.AddPoint( 0, gp_XY( 42493.64, 32187.97 ) )
+garonne.AddPoint( 0, gp_XY( 42074.27, 32302.91 ) )
+garonne.AddPoint( 0, gp_XY( 41624.75, 32330.93 ) )
+garonne.AddPoint( 0, gp_XY( 41325.93, 32187.97 ) )
+garonne.AddPoint( 0, gp_XY( 40879.52, 31987.29 ) )
+garonne.AddPoint( 0, gp_XY( 40308.78, 32018.46 ) )
+garonne.AddPoint( 0, gp_XY( 39782.86, 32207.17 ) )
+garonne.AddPoint( 0, gp_XY( 38814.10, 32560.43 ) )
+garonne.AddPoint( 0, gp_XY( 38302.17, 32903.70 ) )
+garonne.AddPoint( 0, gp_XY( 38191.48, 33184.55 ) )
+garonne.AddPoint( 0, gp_XY( 38553.78, 33013.53 ) )
+garonne.AddPoint( 0, gp_XY( 38993.90, 32767.64 ) )
+garonne.AddPoint( 0, gp_XY( 39678.58, 32533.58 ) )
+garonne.AddPoint( 0, gp_XY( 40240.26, 32305.27 ) )
+
+garonne.Update()
+
+
+domaine = hydro_doc.CreateObject( KIND_POLYLINEXY )
+domaine.SetName( "domaine" )
+
+domaine.SetZLevel( 3 )
+
+domaine.AddSection( "Section_1", 0, 1 )
+domaine.AddPoint( 0, gp_XY( 40325.30, 34641.11 ) )
+domaine.AddPoint( 0, gp_XY( 37581.56, 30348.45 ) )
+domaine.AddPoint( 0, gp_XY( 44835.83, 24264.97 ) )
+domaine.AddPoint( 0, gp_XY( 46006.47, 24575.11 ) )
+domaine.AddPoint( 0, gp_XY( 48787.77, 30167.56 ) )
+domaine.AddPoint( 0, gp_XY( 45532.36, 33239.83 ) )
+domaine.AddPoint( 0, gp_XY( 43548.77, 34048.39 ) )
+
+domaine.Update()
+
+
+lit_majeur = hydro_doc.CreateObject( KIND_POLYLINEXY )
+lit_majeur.SetName( "lit_majeur" )
+
+lit_majeur.SetZLevel( 7 )
+
+lit_majeur.AddSection( "Section_1", 1, 1 )
+lit_majeur.AddPoint( 0, gp_XY( 41225.22, 34118.46 ) )
+lit_majeur.AddPoint( 0, gp_XY( 41716.41, 33705.87 ) )
+lit_majeur.AddPoint( 0, gp_XY( 42384.43, 33293.27 ) )
+lit_majeur.AddPoint( 0, gp_XY( 42561.26, 32684.19 ) )
+lit_majeur.AddPoint( 0, gp_XY( 42875.62, 32369.83 ) )
+lit_majeur.AddPoint( 0, gp_XY( 43700.82, 32350.18 ) )
+lit_majeur.AddPoint( 0, gp_XY( 44526.02, 31662.51 ) )
+lit_majeur.AddPoint( 0, gp_XY( 45370.87, 31819.69 ) )
+lit_majeur.AddPoint( 0, gp_XY( 46274.66, 31682.16 ) )
+lit_majeur.AddPoint( 0, gp_XY( 46981.97, 31328.50 ) )
+lit_majeur.AddPoint( 0, gp_XY( 47021.27, 30719.43 ) )
+lit_majeur.AddPoint( 0, gp_XY( 47217.74, 30051.41 ) )
+lit_majeur.AddPoint( 0, gp_XY( 47846.46, 29560.22 ) )
+lit_majeur.AddPoint( 0, gp_XY( 48730.61, 28735.02 ) )
+lit_majeur.AddPoint( 0, gp_XY( 48809.20, 27732.99 ) )
+lit_majeur.AddPoint( 0, gp_XY( 49143.21, 27163.21 ) )
+lit_majeur.AddPoint( 0, gp_XY( 49654.05, 26829.20 ) )
+lit_majeur.AddPoint( 0, gp_XY( 49693.34, 25355.63 ) )
+lit_majeur.AddPoint( 0, gp_XY( 48612.72, 23214.04 ) )
+lit_majeur.AddPoint( 0, gp_XY( 47453.51, 23115.80 ) )
+lit_majeur.AddPoint( 0, gp_XY( 46471.13, 24176.77 ) )
+lit_majeur.AddPoint( 0, gp_XY( 45960.29, 25257.39 ) )
+lit_majeur.AddPoint( 0, gp_XY( 44899.32, 25591.40 ) )
+lit_majeur.AddPoint( 0, gp_XY( 44270.60, 26573.78 ) )
+lit_majeur.AddPoint( 0, gp_XY( 43248.92, 27909.82 ) )
+lit_majeur.AddPoint( 0, gp_XY( 42757.73, 28243.83 ) )
+lit_majeur.AddPoint( 0, gp_XY( 42266.54, 28342.07 ) )
+lit_majeur.AddPoint( 0, gp_XY( 41736.06, 28106.30 ) )
+lit_majeur.AddPoint( 0, gp_XY( 41402.05, 28833.26 ) )
+lit_majeur.AddPoint( 0, gp_XY( 40910.86, 28911.85 ) )
+lit_majeur.AddPoint( 0, gp_XY( 40576.85, 29324.45 ) )
+lit_majeur.AddPoint( 0, gp_XY( 40144.60, 29599.51 ) )
+lit_majeur.AddPoint( 0, gp_XY( 39024.69, 30130.00 ) )
+lit_majeur.AddPoint( 0, gp_XY( 38612.09, 30365.77 ) )
+lit_majeur.AddPoint( 0, gp_XY( 38828.21, 33391.50 ) )
+
+lit_majeur.Update()
+
+
+Cloud_02 = hydro_doc.CreateObject( KIND_BATHYMETRY )
+Cloud_02.SetName( "Cloud_02" )
+
+Cloud_02.SetAltitudesInverted( 0 )
+Cloud_02.ImportFromFile( os.path.join(HYDRO_SAMPLES, "Cloud_02.xyz" ))
+
+Cloud_02.Update()
+
+
+garonne_point_L93 = hydro_doc.CreateObject( KIND_BATHYMETRY )
+garonne_point_L93.SetName( "garonne_point_L93" )
+
+garonne_point_L93.SetAltitudesInverted( 0 )
+garonne_point_L93.ImportFromFile( os.path.join(HYDRO_SAMPLES, "garonne_point_L93.xyz" ))
+
+garonne_point_L93.Update()
+
+
+litMineur = hydro_doc.CreateObject( KIND_IMMERSIBLE_ZONE )
+litMineur.SetName( "litMineur" )
+
+litMineur.SetZLevel( 6 )
+
+litMineur.SetAltitudeObject( garonne_point_L93 )
+litMineur.SetPolyline( garonne )
+
+litMineur.Update()
+
+
+litMajeur = hydro_doc.CreateObject( KIND_IMMERSIBLE_ZONE )
+litMajeur.SetName( "litMajeur" )
+
+litMajeur.SetZLevel( 5 )
+
+litMajeur.SetFillingColor( QColor( 0, 170, 127, 255 ) )
+
+litMajeur.SetAltitudeObject( Cloud_02 )
+litMajeur.SetPolyline( lit_majeur )
+
+litMajeur.Update()
+
+
+domaineEtendu = hydro_doc.CreateObject( KIND_IMMERSIBLE_ZONE )
+domaineEtendu.SetName( "domaineEtendu" )
+
+domaineEtendu.SetZLevel( 4 )
+
+domaineEtendu.SetFillingColor( QColor( 201, 203, 55, 255 ) )
+
+domaineEtendu.SetAltitudeObject( Cloud_02 )
+domaineEtendu.SetPolyline( domaine )
+
+domaineEtendu.Update()
+
+
+# Calculation case
+garonne_1 = hydro_doc.CreateObject( KIND_CALCULATION )
+garonne_1.SetName( "garonne_1" )
+
+garonne_1.SetAssignmentMode( HYDROData_CalculationCase.MANUAL )
+garonne_1.SetAssignmentLandCoverMode( HYDROData_CalculationCase.AUTOMATIC )
+garonne_1.AddGeometryObject( litMineur )
+garonne_1.AddGeometryObject( domaineEtendu )
+garonne_1.AddGeometryObject( litMajeur )
+
+case_geom_group = domaineEtendu.GetGroup( 0 )
+garonne_1.AddGeometryGroup( case_geom_group )
+case_geom_group = litMineur.GetGroup( 0 )
+garonne_1.AddGeometryGroup( case_geom_group )
+case_geom_group = litMajeur.GetGroup( 0 )
+garonne_1.AddGeometryGroup( case_geom_group )
+garonne_1.SetBoundaryPolyline( domaine )
+
+# Start the algorithm of the partition and assignment
+garonne_1.Update()
+garonne_1_litMineur = hydro_doc.FindObjectByName( "garonne_1_Reg_1" )
+garonne_1_Zone_1 = hydro_doc.FindObjectByName( "garonne_1_Zone_1" )
+garonne_1_Zone_1.SetMergeType( HYDROData_Zone.Merge_ZMIN )
+garonne_1_litMineur.AddZone( garonne_1_Zone_1 )
+garonne_1_riveDroite = hydro_doc.FindObjectByName( "garonne_1_Reg_2" )
+garonne_1_Zone_2 = hydro_doc.FindObjectByName( "garonne_1_Zone_2" )
+garonne_1_riveDroite.AddZone( garonne_1_Zone_2 )
+garonne_1_Zone_3 = hydro_doc.FindObjectByName( "garonne_1_Zone_3" )
+garonne_1_riveDroite.AddZone( garonne_1_Zone_3 )
+garonne_1_riveGauche = hydro_doc.FindObjectByName( "garonne_1_Reg_3" )
+garonne_1_Zone_4 = hydro_doc.FindObjectByName( "garonne_1_Zone_4" )
+garonne_1_riveGauche.AddZone( garonne_1_Zone_4 )
+garonne_1_Zone_5 = hydro_doc.FindObjectByName( "garonne_1_Zone_5" )
+garonne_1_riveGauche.AddZone( garonne_1_Zone_5 )
+
+garonne_1_litMineur.SetName("garonne_1_litMineur")
+garonne_1_riveDroite.SetName("garonne_1_riveDroite")
+garonne_1_riveGauche.SetName("garonne_1_riveGauche")
+
+# Export of the calculation case
+garonne_1_entry = garonne_1.Export( theStudy._get_StudyId() )
+
+# Get geometry shape and print debug information
+import GEOM
+from salome.geom import geomBuilder
+import math
+import SALOMEDS
+
+geompy = geomBuilder.New(theStudy)
+
+print "Entry:", garonne_1_entry
+HYDRO_garonne_1 = salome.IDToObject( str( garonne_1_entry ) )
+print "Geom shape:", HYDRO_garonne_1
+print "Geom shape name:", HYDRO_garonne_1.GetName()
+
+[riveGauche,litMineur,riveDroite] = geompy.ExtractShapes(HYDRO_garonne_1, geompy.ShapeType["FACE"], True)
+# --- groups already created at HYDRO export
+#garonne_1_lit_mineur_Outer = geompy.CreateGroup(HYDRO_garonne_1, geompy.ShapeType["EDGE"])
+#geompy.UnionIDs(garonne_1_lit_mineur_Outer, [7, 4, 11])
+#garonne_1_domaineEtendu_Outer = geompy.CreateGroup(HYDRO_garonne_1, geompy.ShapeType["EDGE"])
+#geompy.UnionIDs(garonne_1_domaineEtendu_Outer, [13, 9, 25, 16, 19, 21, 23, 27, 28, 31, 33, 41, 39, 37, 35])
+# --- publication already done at HYDRO export
+#geompy.addToStudy( HYDRO_garonne_1, 'HYDRO_garonne_1' )
+#geompy.addToStudyInFather( HYDRO_garonne_1, garonne_1_lit_mineur_Outer, 'garonne_1_lit_mineur_Outer' )
+#geompy.addToStudyInFather( HYDRO_garonne_1, garonne_1_domaineEtendu_Outer, 'garonne_1_domaineEtendu_Outer' )
+geompy.addToStudyInFather( HYDRO_garonne_1, riveGauche, 'riveGauche' )
+geompy.addToStudyInFather( HYDRO_garonne_1, litMineur, 'litMineur' )
+geompy.addToStudyInFather( HYDRO_garonne_1, riveDroite, 'riveDroite' )
+
+controlGeomProps(riveGauche, 29255.966464, 36865800.781542)
+controlGeomProps(litMineur, 30220.240241, 3309295.161572)
+controlGeomProps(riveDroite, 31841.799764, 26782824.517334)
+
+if salome.sg.hasDesktop():
+ salome.sg.updateObjBrowser(1)
