doc/salome/examples/h009_normalCaseManual.py

   1 # -*- coding: utf-8 -*-
   2
   3 import os
   4 HYDRO_SAMPLES = os.path.join( os.environ["HYDRO_ROOT_DIR"], "bin/salome/test/HYDRO")
   5 HYDRO_TEST_RESOURCES = os.path.join(os.environ["HYDRO_DIR"], "bin", "salome", "test", "tmp_test")
   6 if not os.path.isdir(HYDRO_TEST_RESOURCES):
   7     os.mkdir(HYDRO_TEST_RESOURCES)
   8
   9 import tempfile
  10 tmpdir = tempfile.mkdtemp()
  11 print("tmpdir=",tmpdir)
  12
  13 import sys
  14 import salome
  15
  16 salome.salome_init()
  17
  18 #----------------------
  19 # --- HYDRO
  20 #----------------------
  21
  22 from HYDROPy import *
  23 from PyQt5.QtCore import *
  24 from PyQt5.QtGui import *
  25
  26 from salome.hydrotools.hydroGeoMeshUtils import loadImage, GeolocaliseImageCoords, GeolocaliseImageReference
  27 from salome.hydrotools.hydroGeoMeshUtils import importPolylines, importBathymetry, createImmersibleZone, mergePolylines, getChildrenInStudy
  28
  29 hydro_doc = HYDROData_Document.Document()
  30
  31 offsetX = 430000.
  32 offsetY = 6350000.
  33 hydro_doc.SetLocalCS( offsetX, offsetY )
  34
  35 garonne_ign_01 = loadImage(hydro_doc, os.path.join(HYDRO_SAMPLES, "garonne_ign_01.png"), "garonne_ign_01", 0)
  36 GeolocaliseImageCoords(garonne_ign_01,
  37                        QPoint(40, 817), QPoint(1325, 85),
  38                        QPointF(471562, 6.36775e+06), QPointF(489400, 6.37702e+06))
  39
  40 garonne_ign_02 = loadImage(hydro_doc, os.path.join(HYDRO_SAMPLES, "garonne_ign_02.png"), "garonne_ign_02", 1)
  41 GeolocaliseImageReference(garonne_ign_02, garonne_ign_01,
  42                           QPoint(1389, 447), QPoint(784, 481),
  43                           QPoint(631, 95), QPoint(26, 129))
  44
  45 garonnes = importPolylines(hydro_doc, os.path.join(HYDRO_SAMPLES, "garonne.shp"), "garonne_PolyXY", True, 5)
  46 limites_domaine = importPolylines(hydro_doc, os.path.join(HYDRO_SAMPLES, "domaine.shp"), "domaine_PolyXY", False, 6)
  47 lits_majeur = importPolylines(hydro_doc, os.path.join(HYDRO_SAMPLES, "lit_majeur.shp"), "lit_majeur_PolyXY", True, 7)
  48
  49 Cloud_02 = importBathymetry(hydro_doc, os.path.join(HYDRO_SAMPLES, "Cloud_02.xyz"))
  50 garonne_point_L93 = importBathymetry(hydro_doc, os.path.join(HYDRO_SAMPLES, "garonne_point_L93.xyz"))
  51
  52 domaineEtendu = createImmersibleZone(hydro_doc, "domaineEtendu", limites_domaine[0], Cloud_02, True, 2)
  53 litMajeur = createImmersibleZone(hydro_doc, "litMajeur", lits_majeur[0], garonne_point_L93, True, 3)
  54 litMineur = createImmersibleZone(hydro_doc, "litMineur", garonnes[0], garonne_point_L93, True, 4)
  55
  56 # Calculation case
  57 garonne_1 = hydro_doc.CreateObject( KIND_CALCULATION )
  58 garonne_1.SetName( "garonne_1" )
  59
  60 garonne_1.SetAssignmentMode( HYDROData_CalculationCase.MANUAL )
  61 garonne_1.AddGeometryObject( litMineur )
  62 garonne_1.AddGeometryObject( domaineEtendu )
  63 garonne_1.AddGeometryObject( litMajeur )
  64
  65 case_geom_group = domaineEtendu.GetGroup( 0 )
  66 garonne_1.AddGeometryGroup( case_geom_group )
  67 case_geom_group = litMineur.GetGroup( 0 )
  68 garonne_1.AddGeometryGroup( case_geom_group )
  69 case_geom_group = litMajeur.GetGroup( 0 )
  70 garonne_1.AddGeometryGroup( case_geom_group )
  71 garonne_1.SetBoundaryPolyline( limites_domaine[0] )
  72
  73 # Start the algorithm of the partition and assignment
  74 garonne_1.Update()
  75 garonne_1_litMineur = hydro_doc.FindObjectByName( "garonne_1_Reg_1" )
  76 garonne_1_Zone_1 = hydro_doc.FindObjectByName( "garonne_1_Zone_1" )
  77 garonne_1_Zone_1.SetMergeType( HYDROData_Zone.Merge_ZMIN )
  78 garonne_1_Zone_1.SetColor( QColor( 192, 113, 64 ))
  79 garonne_1_litMineur.AddZone( garonne_1_Zone_1 )
  80
  81 garonne_1_riveDroite = hydro_doc.FindObjectByName( "garonne_1_Reg_2" )
  82 garonne_1_Zone_2 = hydro_doc.FindObjectByName( "garonne_1_Zone_2" )
  83 garonne_1_Zone_2.SetColor( QColor( 141, 192, 64 ))
  84 garonne_1_riveDroite.AddZone( garonne_1_Zone_2 )
  85
  86 garonne_1_Zone_3 = hydro_doc.FindObjectByName( "garonne_1_Zone_3" )
  87 garonne_1_Zone_3.SetMergeType( HYDROData_Zone.Merge_Object )
  88 Cloud_02 = hydro_doc.FindObjectByName( "Cloud_02" )
  89 garonne_1_Zone_3.SetMergeObject( Cloud_02 )
  90 garonne_1_Zone_3.SetColor( QColor( 64, 192, 77 ))
  91 garonne_1_riveDroite.AddZone( garonne_1_Zone_3 )
  92
  93 garonne_1_riveGauche = hydro_doc.FindObjectByName( "garonne_1_Reg_3" )
  94 garonne_1_Zone_4 = hydro_doc.FindObjectByName( "garonne_1_Zone_4" )
  95 garonne_1_Zone_4.SetMergeType( HYDROData_Zone.Merge_Object )
  96 Cloud_02 = hydro_doc.FindObjectByName( "Cloud_02" )
  97 garonne_1_Zone_4.SetMergeObject( Cloud_02 )
  98 garonne_1_Zone_4.SetColor( QColor( 64, 75, 192 ))
  99 garonne_1_riveGauche.AddZone( garonne_1_Zone_4 )
 100
 101 garonne_1_Zone_5 = hydro_doc.FindObjectByName( "garonne_1_Zone_5" )
 102 garonne_1_Zone_5.SetColor( QColor( 64, 192, 77 ))
 103 garonne_1_riveGauche.AddZone( garonne_1_Zone_5 )
 104
 105 garonne_1_litMineur.SetName( "garonne_1_litMineur" )
 106 garonne_1_riveDroite.SetName( "garonne_1_riveDroite" )
 107 garonne_1_riveGauche.SetName( "garonne_1_riveGauche" )
 108
 109 # Export of the calculation case
 110 garonne_1_entry = garonne_1.Export()
 111
 112 #----------------------
 113 # --- Geometry
 114 #----------------------
 115
 116 # Get geometry shape and print debug information
 117 import GEOM
 118 from salome.geom import geomBuilder
 119 import math
 120 import SALOMEDS
 121 from salome.hydrotools.controls import controlGeomProps
 122
 123 geompy = geomBuilder.New()
 124
 125 print("Entry:", garonne_1_entry)
 126 HYDRO_garonne_1 = salome.IDToObject( str( garonne_1_entry ) )
 127 print("Geom shape:", HYDRO_garonne_1)
 128 print("Geom shape name:", HYDRO_garonne_1.GetName())
 129
 130
 131 children = getChildrenInStudy(HYDRO_garonne_1)
 132 garonne_riveGauche = children["garonne_1_riveGauche"]
 133 garonne_litMineur = children["garonne_1_litMineur"]
 134 garonne_riveDroite = children["garonne_1_riveDroite"]
 135 #[garonne_litMineur,garonne_riveDroite,garonne_riveGauche] = geompy.SubShapeAll(HYDRO_garonne_1, geompy.ShapeType["FACE"])
 136
 137 controlGeomProps(geompy, garonne_riveGauche,  29149.36918,  35948828.352061)
 138 controlGeomProps(geompy, garonne_litMineur,   30337.548492,  3488480.304388)
 139 controlGeomProps(geompy, garonne_riveDroite,  32012.343241, 25998769.23615)
 140
 141 # --- manual identification of all useful edge groups (boundary conditions)
 142
 143 allEdgesIds = geompy.SubShapeAllIDs(HYDRO_garonne_1, geompy.ShapeType["EDGE"])
 144 print("allEdgesIds", allEdgesIds)
 145
 146 (isDone, ClosedFreeBoundary, OpenFreeBoundary) = geompy.GetFreeBoundary(HYDRO_garonne_1)
 147 geompy.addToStudyInFather(HYDRO_garonne_1, ClosedFreeBoundary[0], "ClosedFreeBoundary")
 148
 149 freeBoundary = geompy.ExtractShapes(ClosedFreeBoundary[0], geompy.ShapeType["EDGE"], True)
 150 freeBoundaryIds = [ geompy.GetSubShapeID(HYDRO_garonne_1, freeBoundary[i]) for i in range(len(freeBoundary)) ]
 151 print("freeBoundaryIds", freeBoundaryIds)
 152
 153 [litMineur_droite] = geompy.GetSharedShapesMulti([garonne_riveDroite, garonne_litMineur], geompy.ShapeType["EDGE"], True)
 154 [litMineur_gauche] = geompy.GetSharedShapesMulti([garonne_riveGauche, garonne_litMineur], geompy.ShapeType["EDGE"], True)
 155 geompy.addToStudyInFather(HYDRO_garonne_1, litMineur_droite, "litMineur_droite")
 156 geompy.addToStudyInFather(HYDRO_garonne_1, litMineur_gauche, "litMineur_gauche")
 157 rives = [litMineur_droite, litMineur_gauche]
 158 rivesIds = [ geompy.GetSubShapeID(HYDRO_garonne_1, rives[i]) for i in range(len(rives)) ]
 159 print("rivesIds", rivesIds)
 160
 161 edges_litMineur = geompy.GetSharedShapesMulti([HYDRO_garonne_1, garonne_litMineur], geompy.ShapeType["EDGE"], True)
 162 edges_riveGauche = geompy.GetSharedShapesMulti([HYDRO_garonne_1, garonne_riveGauche], geompy.ShapeType["EDGE"], True)
 163 edges_riveDroite = geompy.GetSharedShapesMulti([HYDRO_garonne_1, garonne_riveDroite], geompy.ShapeType["EDGE"], True)
 164 edges_litMineurIds = [ geompy.GetSubShapeID(HYDRO_garonne_1, edges_litMineur[i]) for i in range(len(edges_litMineur)) ]
 165 edges_riveGaucheIds = [ geompy.GetSubShapeID(HYDRO_garonne_1, edges_riveGauche[i]) for i in range(len(edges_riveGauche)) ]
 166 edges_riveDroiteIds = [ geompy.GetSubShapeID(HYDRO_garonne_1, edges_riveDroite[i]) for i in range(len(edges_riveDroite)) ]
 167
 168 print("edges_litMineurIds", edges_litMineurIds)
 169 print("edges_riveGaucheIds", edges_riveGaucheIds)
 170 print("edges_riveDroiteIds", edges_riveDroiteIds)
 171
 172 sectionsIds = [Id for Id in edges_litMineurIds if Id not in rivesIds]
 173 print("sectionsIds", sectionsIds)
 174 SectionsGaronne = geompy.CreateGroup(HYDRO_garonne_1, geompy.ShapeType["EDGE"])
 175 geompy.UnionIDs(SectionsGaronne, sectionsIds)
 176 geompy.addToStudyInFather(HYDRO_garonne_1, SectionsGaronne, "SectionsGaronne")
 177
 178 bordGaucheDomaineIds = [Id for Id in freeBoundaryIds if Id in edges_riveGaucheIds]
 179 bordDroiteDomaineIds = [Id for Id in freeBoundaryIds if Id in edges_riveDroiteIds]
 180 print("bordGaucheDomaineIds", bordGaucheDomaineIds)
 181 print("bordDroiteDomaineIds", bordDroiteDomaineIds)
 182 bordGaucheDomaine = geompy.CreateGroup(HYDRO_garonne_1, geompy.ShapeType["EDGE"])
 183 geompy.UnionIDs(bordGaucheDomaine, bordGaucheDomaineIds)
 184 geompy.addToStudyInFather(HYDRO_garonne_1, bordGaucheDomaine, "bordGaucheDomaine")
 185 bordDroiteDomaine = geompy.CreateGroup(HYDRO_garonne_1, geompy.ShapeType["EDGE"])
 186 geompy.UnionIDs(bordDroiteDomaine, bordDroiteDomaineIds)
 187 geompy.addToStudyInFather(HYDRO_garonne_1, bordDroiteDomaine, "bordDroiteDomaine")
 188
 189 amont = geompy.GetEdgeNearPoint(HYDRO_garonne_1, geompy.MakeVertex(46757.861314, 25833.234752, 0))
 190 aval = geompy.GetEdgeNearPoint(HYDRO_garonne_1, geompy.MakeVertex(39078.979127, 32588.627279, 0))
 191 geompy.addToStudyInFather(HYDRO_garonne_1, amont, "amont")
 192 geompy.addToStudyInFather(HYDRO_garonne_1, aval, "aval")
 193
 194 if salome.sg.hasDesktop():
 195   salome.sg.updateObjBrowser()