doc/salome/examples/h014_caseDigueManualInterpolZ.py

   1 # -*- coding: utf-8 -*-
   2
   3 import os
   4 HYDRO_SAMPLES = os.path.join( os.environ["HYDRO_ROOT_DIR"], "bin/salome/test")
   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 from salome.hydrotools.hydroGeoMeshUtils import createAxis3DDEmbankmentAltiProfile, createAxis3DDEmbankmentBathy
  29 from salome.hydrotools.hydroGeoMeshUtils import createEmbankmentSectionA, createEmbankmentSectionB
  30
  31 hydro_doc = HYDROData_Document.Document()
  32
  33 offsetX = 430000.
  34 offsetY = 6350000.
  35 hydro_doc.SetLocalCS( offsetX, offsetY )
  36
  37 garonne_ign_01 = loadImage(hydro_doc, os.path.join(HYDRO_SAMPLES, "garonne_ign_01.png"), "garonne_ign_01", 0)
  38 GeolocaliseImageCoords(garonne_ign_01,
  39                        QPoint(40, 817), QPoint(1325, 85),
  40                        QPointF(471562, 6.36775e+06), QPointF(489400, 6.37702e+06))
  41
  42 garonne_ign_02 = loadImage(hydro_doc, os.path.join(HYDRO_SAMPLES, "garonne_ign_02.png"), "garonne_ign_02", 1)
  43 GeolocaliseImageReference(garonne_ign_02, garonne_ign_01,
  44                           QPoint(1389, 447), QPoint(784, 481),
  45                           QPoint(631, 95), QPoint(26, 129))
  46
  47 garonnes = importPolylines(hydro_doc, os.path.join(HYDRO_SAMPLES, "garonne.shp"), True, 6)
  48 limites_domaine = importPolylines(hydro_doc, os.path.join(HYDRO_SAMPLES, "domaine.shp"), False, 7)
  49 lits_majeur = importPolylines(hydro_doc, os.path.join(HYDRO_SAMPLES, "lit_majeur.shp"), True, 8)
  50 axesDigue = importPolylines(hydro_doc, os.path.join(HYDRO_SAMPLES, "axeDigue.shp"), True, 9)
  51
  52 Cloud_02 = importBathymetry(hydro_doc, os.path.join(HYDRO_SAMPLES, "Cloud_02.xyz"))
  53 garonne_point_L93 = importBathymetry(hydro_doc, os.path.join(HYDRO_SAMPLES, "garonne_point_L93.xyz"))
  54
  55 domaineEtendu = createImmersibleZone(hydro_doc, "domaineEtendu", limites_domaine[0], Cloud_02, True, 2)
  56 litMajeur = createImmersibleZone(hydro_doc, "litMajeur", lits_majeur[0], garonne_point_L93, True, 3)
  57 litMineur = createImmersibleZone(hydro_doc, "litMineur", garonnes[0], garonne_point_L93, True, 4)
  58
  59 altiDiguePts = [(1., 23.0),
  60                 ( 5000., 25.5 )]
  61 altiProfile, axe3D = createAxis3DDEmbankmentAltiProfile(hydro_doc, "axe3DDigue", axesDigue[0], altiDiguePts)
  62 sectionPoints = [(0, 0), (8, -1), (10,-10)]
  63 section, digue = createEmbankmentSectionA(hydro_doc, "digue", axe3D, sectionPoints, 50., 5)
  64
  65 # Calculation case
  66 garonne_1 = hydro_doc.CreateObject( KIND_CALCULATION )
  67 garonne_1.SetName( "garonne_1" )
  68
  69 garonne_1.SetAssignmentMode( HYDROData_CalculationCase.MANUAL )
  70 garonne_1.AddGeometryObject( litMineur )
  71 garonne_1.AddGeometryObject( domaineEtendu )
  72 garonne_1.AddGeometryObject( litMajeur )
  73 garonne_1.AddGeometryObject( digue );
  74
  75 case_geom_group = domaineEtendu.GetGroup( 0 );
  76 garonne_1.AddGeometryGroup( case_geom_group );
  77 case_geom_group = litMineur.GetGroup( 0 );
  78 garonne_1.AddGeometryGroup( case_geom_group );
  79 case_geom_group = litMajeur.GetGroup( 0 );
  80 garonne_1.AddGeometryGroup( case_geom_group );
  81 case_geom_group = digue.GetGroup( 2 );
  82 garonne_1.AddGeometryGroup( case_geom_group );
  83 case_geom_group = digue.GetGroup( 0 );
  84 garonne_1.AddGeometryGroup( case_geom_group );
  85 case_geom_group = digue.GetGroup( 3 );
  86 garonne_1.AddGeometryGroup( case_geom_group );
  87 case_geom_group = digue.GetGroup( 1 );
  88 garonne_1.AddGeometryGroup( case_geom_group );
  89 garonne_1.SetBoundaryPolyline( limites_domaine[0] );
  90
  91 # Start the algorithm of the partition and assignment
  92 garonne_1.Update()
  93
  94 garonne_1_litMineur = hydro_doc.FindObjectByName( "garonne_1_Reg_1" )
  95 garonne_1_Zone_1 = hydro_doc.FindObjectByName( "garonne_1_Zone_1" )
  96 garonne_1_Zone_1.SetMergeType( HYDROData_Zone.Merge_ZMIN )
  97 garonne_1_Zone_1.SetColor( QColor( 192, 137, 64 ))
  98 garonne_1_litMineur.AddZone( garonne_1_Zone_1 )
  99
 100 garonne_1_riveDroite = hydro_doc.FindObjectByName( "garonne_1_Reg_2" )
 101 garonne_1_Zone_2 = hydro_doc.FindObjectByName( "garonne_1_Zone_2" )
 102 garonne_1_Zone_2.SetColor( QColor( 64, 192, 98 ))
 103 garonne_1_riveDroite.AddZone( garonne_1_Zone_2 )
 104
 105 garonne_1_Zone_3 = hydro_doc.FindObjectByName( "garonne_1_Zone_3" )
 106 garonne_1_Zone_3.SetMergeType( HYDROData_Zone.Merge_Object )
 107 Cloud_02 = hydro_doc.FindObjectByName( "Cloud_02" )
 108 garonne_1_Zone_3.SetMergeObject( Cloud_02 )
 109 garonne_1_Zone_3.SetColor( QColor( 64, 66, 192 ))
 110 garonne_1_riveDroite.AddZone( garonne_1_Zone_3 )
 111
 112 garonne_1_riveGauche = hydro_doc.FindObjectByName( "garonne_1_Reg_3" )
 113 garonne_1_Zone_4 = hydro_doc.FindObjectByName( "garonne_1_Zone_4" )
 114 garonne_1_Zone_4.SetMergeType( HYDROData_Zone.Merge_Object )
 115 Cloud_02 = hydro_doc.FindObjectByName( "Cloud_02" )
 116 garonne_1_Zone_4.SetMergeObject( Cloud_02 )
 117 garonne_1_Zone_4.SetColor( QColor( 64, 192, 164 ))
 118 garonne_1_riveGauche.AddZone( garonne_1_Zone_4 )
 119
 120 garonne_1_Zone_5 = hydro_doc.FindObjectByName( "garonne_1_Zone_5" )
 121 garonne_1_Zone_5.SetColor( QColor( 192, 149, 64 ))
 122 garonne_1_riveGauche.AddZone( garonne_1_Zone_5 )
 123
 124 garonne_1_digue = hydro_doc.FindObjectByName( "garonne_1_Reg_4" )
 125 garonne_1_Zone_6 = hydro_doc.FindObjectByName( "garonne_1_Zone_6" )
 126 garonne_1_Zone_6.SetMergeType( HYDROData_Zone.Merge_ZMAX )
 127 garonne_1_Zone_6.SetColor( QColor( 64, 192, 177 ))
 128 garonne_1_digue.AddZone( garonne_1_Zone_6 )
 129
 130 garonne_1_litMineur.SetName("garonne_1_litMineur")
 131 garonne_1_riveDroite.SetName("garonne_1_riveDroite")
 132 garonne_1_riveGauche.SetName("garonne_1_riveGauche")
 133 garonne_1_digue.SetName("garonne_1_digue")
 134
 135 # Export of the calculation case
 136 garonne_1_entry = garonne_1.Export()
 137
 138 #----------------------
 139 # --- Geometry
 140 #----------------------
 141
 142 # Get geometry shape and print debug information
 143 import GEOM
 144 from salome.geom import geomBuilder
 145 import math
 146 import SALOMEDS
 147 from salome.hydrotools.controls import controlGeomProps
 148
 149 geompy = geomBuilder.New()
 150
 151 print("Entry:", garonne_1_entry)
 152 HYDRO_garonne_1 = salome.IDToObject( str( garonne_1_entry ) )
 153 print("Geom shape:", HYDRO_garonne_1)
 154 print("Geom shape name:", HYDRO_garonne_1.GetName())
 155
 156
 157 children = getChildrenInStudy(HYDRO_garonne_1)
 158 garonne_riveGauche = children["garonne_1_riveGauche"]
 159 garonne_litMineur = children["garonne_1_litMineur"]
 160 garonne_riveDroite = children["garonne_1_riveDroite"]
 161 garonne_digue = children["garonne_1_digue"]
 162
 163 controlGeomProps(geompy, garonne_riveGauche,  39490.835288, 35845737.590926)
 164 controlGeomProps(geompy, garonne_digue,       10341.466108,   103090.760662)
 165 controlGeomProps(geompy, garonne_litMineur,   30337.548492,  3488480.304388)
 166 controlGeomProps(geompy, garonne_riveDroite,  32012.343241, 25998769.23615)
 167
 168 # --- manual identification of all useful edge groups (boundary conditions)
 169
 170 allEdgesIds = geompy.SubShapeAllIDs(HYDRO_garonne_1, geompy.ShapeType["EDGE"])
 171 print("allEdgesIds", allEdgesIds)
 172
 173 (isDone, ClosedFreeBoundary, OpenFreeBoundary) = geompy.GetFreeBoundary(HYDRO_garonne_1)
 174 geompy.addToStudyInFather(HYDRO_garonne_1, ClosedFreeBoundary[0], "ClosedFreeBoundary")
 175
 176 freeBoundary = geompy.ExtractShapes(ClosedFreeBoundary[0], geompy.ShapeType["EDGE"], True)
 177 freeBoundaryIds = [ geompy.GetSubShapeID(HYDRO_garonne_1, freeBoundary[i]) for i in range(len(freeBoundary)) ]
 178 print("freeBoundaryIds", freeBoundaryIds)
 179
 180 [litMineur_droite] = geompy.GetSharedShapesMulti([garonne_riveDroite, garonne_litMineur], geompy.ShapeType["EDGE"], True)
 181 [litMineur_gauche] = geompy.GetSharedShapesMulti([garonne_riveGauche, garonne_litMineur], geompy.ShapeType["EDGE"], True)
 182 geompy.addToStudyInFather(HYDRO_garonne_1, litMineur_droite, "litMineur_droite")
 183 geompy.addToStudyInFather(HYDRO_garonne_1, litMineur_gauche, "litMineur_gauche")
 184 rives = [litMineur_droite, litMineur_gauche]
 185 rivesIds = [ geompy.GetSubShapeID(HYDRO_garonne_1, rives[i]) for i in range(len(rives)) ]
 186 print("rivesIds", rivesIds)
 187
 188 edges_litMineur = geompy.GetSharedShapesMulti([HYDRO_garonne_1, garonne_litMineur], geompy.ShapeType["EDGE"], True)
 189 edges_riveGauche = geompy.GetSharedShapesMulti([HYDRO_garonne_1, garonne_riveGauche], geompy.ShapeType["EDGE"], True)
 190 edges_riveDroite = geompy.GetSharedShapesMulti([HYDRO_garonne_1, garonne_riveDroite], geompy.ShapeType["EDGE"], True)
 191 edges_litMineurIds = [ geompy.GetSubShapeID(HYDRO_garonne_1, edges_litMineur[i]) for i in range(len(edges_litMineur)) ]
 192 edges_riveGaucheIds = [ geompy.GetSubShapeID(HYDRO_garonne_1, edges_riveGauche[i]) for i in range(len(edges_riveGauche)) ]
 193 edges_riveDroiteIds = [ geompy.GetSubShapeID(HYDRO_garonne_1, edges_riveDroite[i]) for i in range(len(edges_riveDroite)) ]
 194
 195 print("edges_litMineurIds", edges_litMineurIds)
 196 print("edges_riveGaucheIds", edges_riveGaucheIds)
 197 print("edges_riveDroiteIds", edges_riveDroiteIds)
 198
 199 sectionsIds = [Id for Id in edges_litMineurIds if Id not in rivesIds]
 200 print("sectionsIds", sectionsIds)
 201 SectionsGaronne = geompy.CreateGroup(HYDRO_garonne_1, geompy.ShapeType["EDGE"])
 202 geompy.UnionIDs(SectionsGaronne, sectionsIds)
 203 geompy.addToStudyInFather(HYDRO_garonne_1, SectionsGaronne, "SectionsGaronne")
 204
 205 bordGaucheDomaineIds = [Id for Id in freeBoundaryIds if Id in edges_riveGaucheIds]
 206 bordDroiteDomaineIds = [Id for Id in freeBoundaryIds if Id in edges_riveDroiteIds]
 207 print("bordGaucheDomaineIds", bordGaucheDomaineIds)
 208 print("bordDroiteDomaineIds", bordDroiteDomaineIds)
 209 bordGaucheDomaine = geompy.CreateGroup(HYDRO_garonne_1, geompy.ShapeType["EDGE"])
 210 geompy.UnionIDs(bordGaucheDomaine, bordGaucheDomaineIds)
 211 geompy.addToStudyInFather(HYDRO_garonne_1, bordGaucheDomaine, "bordGaucheDomaine")
 212 bordDroiteDomaine = geompy.CreateGroup(HYDRO_garonne_1, geompy.ShapeType["EDGE"])
 213 geompy.UnionIDs(bordDroiteDomaine, bordDroiteDomaineIds)
 214 geompy.addToStudyInFather(HYDRO_garonne_1, bordDroiteDomaine, "bordDroiteDomaine")
 215
 216 amont = geompy.GetEdgeNearPoint(HYDRO_garonne_1, geompy.MakeVertex(46757.861314, 25833.234752, 0))
 217 aval = geompy.GetEdgeNearPoint(HYDRO_garonne_1, geompy.MakeVertex(39078.979127, 32588.627279, 0))
 218 geompy.addToStudyInFather(HYDRO_garonne_1, amont, "amont")
 219 geompy.addToStudyInFather(HYDRO_garonne_1, aval, "aval")
 220
 221 amontDigue = geompy.GetEdgeNearPoint(HYDRO_garonne_1, geompy.MakeVertex(44736.43, 29847.52, 0))
 222 avalDigue = geompy.GetEdgeNearPoint(HYDRO_garonne_1, geompy.MakeVertex(40961.39, 31800.95, 0))
 223 geompy.addToStudyInFather(HYDRO_garonne_1, amontDigue, "amontDigue")
 224 geompy.addToStudyInFather(HYDRO_garonne_1, avalDigue, "avalDigue")
 225
 226 SectionsDigue = geompy.CreateGroup(HYDRO_garonne_1, geompy.ShapeType["EDGE"])
 227 geompy.UnionIDs(SectionsDigue, [geompy.GetSubShapeID(HYDRO_garonne_1, amontDigue), geompy.GetSubShapeID(HYDRO_garonne_1, avalDigue)])
 228 geompy.addToStudyInFather( HYDRO_garonne_1, SectionsDigue, 'SectionsDigue' )
 229
 230 #----------------------
 231 # --- Meshing
 232 #----------------------
 233
 234 import  SMESH, SALOMEDS
 235 from salome.smesh import smeshBuilder
 236 from salome.hydrotools.controls import controlMeshStats, controlSubMeshStats
 237 import tempfile
 238
 239 smesh = smeshBuilder.New()
 240
 241 # --- algorithms and hypothesis
 242 garonne_1 = smesh.Mesh(HYDRO_garonne_1)
 243
 244 NETGEN_2D = garonne_1.Triangle(algo=smeshBuilder.NETGEN_1D2D)
 245 NETGEN_2D_Parameters = NETGEN_2D.Parameters()
 246 NETGEN_2D_Parameters.SetMaxSize( 200 )
 247 NETGEN_2D_Parameters.SetSecondOrder( 0 )
 248 NETGEN_2D_Parameters.SetOptimize( 1 )
 249 NETGEN_2D_Parameters.SetFineness( 4 )
 250 NETGEN_2D_Parameters.SetMinSize( 50 )
 251 NETGEN_2D_Parameters.SetUseSurfaceCurvature( 1 )
 252 NETGEN_2D_Parameters.SetFuseEdges( 1 )
 253 NETGEN_2D_Parameters.SetQuadAllowed( 0 )
 254
 255 algo2D_litMineur = garonne_1.Quadrangle(algo=smeshBuilder.QUAD_MA_PROJ,geom=garonne_litMineur)
 256 algo1D_litMineur = garonne_1.Segment(geom=garonne_litMineur)
 257 hypo1D_litMineur = algo1D_litMineur.LocalLength(100,None,1e-07)
 258 subMesh_litMineur = algo1D_litMineur.GetSubMesh()
 259 smesh.SetName(subMesh_litMineur, "litMineur")
 260
 261 algo1D_SectionsGaronne = garonne_1.Segment(geom=SectionsGaronne)
 262 hypo1D_SectionsGaronne = algo1D_SectionsGaronne.NumberOfSegments(8)
 263 hypo1D_SectionsGaronne.SetDistrType( 0 )
 264 subMesh_SectionsGaronne = algo1D_SectionsGaronne.GetSubMesh()
 265 smesh.SetName(subMesh_SectionsGaronne, "SectionsGaronne")
 266
 267 algo2D_digue = garonne_1.Quadrangle(algo=smeshBuilder.QUAD_MA_PROJ,geom=garonne_digue)
 268 algo1D_digue = garonne_1.Segment(geom=garonne_digue)
 269 hypo1D_digue = algo1D_digue.LocalLength(50,None,1e-07)
 270 subMesh_digue = algo1D_digue.GetSubMesh()
 271 smesh.SetName(subMesh_digue, "digue")
 272
 273 algo1D_SectionsDigue = garonne_1.Segment(geom=SectionsDigue)
 274 hypo1D_SectionsDigue = algo1D_SectionsDigue.NumberOfSegments(4)
 275 hypo1D_SectionsDigue.SetDistrType( 0 )
 276 subMesh_SectionsDigue = algo1D_SectionsDigue.GetSubMesh()
 277 smesh.SetName(subMesh_SectionsDigue, "SectionsDigue")
 278
 279 isDone = garonne_1.SetMeshOrder( [ [ subMesh_SectionsGaronne, subMesh_litMineur ], [ subMesh_SectionsDigue, subMesh_digue ]])
 280
 281 # --- compute mesh
 282 isDone = garonne_1.Compute()
 283 isDone = garonne_1.SplitQuadObject( garonne_1, 1 )
 284 isDone = garonne_1.ReorientObject( garonne_1 )
 285
 286 # --- geometrical groups of faces
 287 riveGauche_1 = garonne_1.GroupOnGeom(garonne_riveGauche,'riveGauche',SMESH.FACE)
 288 litMineur_1 = garonne_1.GroupOnGeom(garonne_litMineur,'litMineur',SMESH.FACE)
 289 riveDroite_1 = garonne_1.GroupOnGeom(garonne_riveDroite,'riveDroite',SMESH.FACE)
 290 digue_1 = garonne_1.GroupOnGeom(garonne_digue,'digue',SMESH.FACE)
 291
 292 # --- geometrical groups of edges
 293 ClosedFreeBoundary_1 = garonne_1.GroupOnGeom(ClosedFreeBoundary[0],'ClosedFreeBoundary',SMESH.EDGE)
 294 litMineur_droite_1 = garonne_1.GroupOnGeom(litMineur_droite,'litMineur_droite',SMESH.EDGE)
 295 litMineur_gauche_1 = garonne_1.GroupOnGeom(litMineur_gauche,'litMineur_gauche',SMESH.EDGE)
 296 SectionsGaronne_1 = garonne_1.GroupOnGeom(SectionsGaronne,'SectionsGaronne',SMESH.EDGE)
 297 bordGaucheDomaine_1 = garonne_1.GroupOnGeom(bordGaucheDomaine,'bordGaucheDomaine',SMESH.EDGE)
 298 bordDroiteDomaine_1 = garonne_1.GroupOnGeom(bordDroiteDomaine,'bordDroiteDomaine',SMESH.EDGE)
 299 amont_1 = garonne_1.GroupOnGeom(amont,'amont',SMESH.EDGE)
 300 aval_1 = garonne_1.GroupOnGeom(aval,'aval',SMESH.EDGE)
 301 SectionsDigue_1 = garonne_1.GroupOnGeom(SectionsDigue,'SectionsDigue',SMESH.EDGE)
 302
 303 # --- geometrical groups of nodes
 304 garonne_1_litMineur_2 = garonne_1.GroupOnGeom(garonne_litMineur,'garonne_1_litMineur',SMESH.NODE)
 305 garonne_1_riveDroite_2 = garonne_1.GroupOnGeom(garonne_riveDroite,'garonne_1_riveDroite',SMESH.NODE)
 306 garonne_1_riveGauche_2 = garonne_1.GroupOnGeom(garonne_riveGauche,'garonne_1_riveGauche',SMESH.NODE)
 307 garonne_1_digue_2 = garonne_1.GroupOnGeom(garonne_digue,'garonne_1_digue',SMESH.NODE)
 308 ClosedFreeBoundary_2 = garonne_1.GroupOnGeom(ClosedFreeBoundary[0],'ClosedFreeBoundary',SMESH.NODE)
 309 litMineur_droite_2 = garonne_1.GroupOnGeom(litMineur_droite,'litMineur_droite',SMESH.NODE)
 310 litMineur_gauche_2 = garonne_1.GroupOnGeom(litMineur_gauche,'litMineur_gauche',SMESH.NODE)
 311 SectionsGaronne_2 = garonne_1.GroupOnGeom(SectionsGaronne,'SectionsGaronne',SMESH.NODE)
 312 bordGaucheDomaine_2 = garonne_1.GroupOnGeom(bordGaucheDomaine,'bordGaucheDomaine',SMESH.NODE)
 313 bordDroiteDomaine_2 = garonne_1.GroupOnGeom(bordDroiteDomaine,'bordDroiteDomaine',SMESH.NODE)
 314 amont_2 = garonne_1.GroupOnGeom(amont,'amont',SMESH.NODE)
 315 aval_2 = garonne_1.GroupOnGeom(aval,'aval',SMESH.NODE)
 316 SectionsDigue_2 = garonne_1.GroupOnGeom(SectionsDigue,'SectionsDigue',SMESH.NODE)
 317
 318 garonne_1.SetAutoColor( 1 )
 319 tmpdir = tempfile.mkdtemp()
 320 print("tmpdir=",tmpdir)
 321 fichierMaillage = os.path.join(tmpdir, 'garonne_1.med')
 322 garonne_1.ExportMED(fichierMaillage, 0, SMESH.MED_V2_2, 1, None ,1)
 323
 324 mesures = garonne_1.GetMeshInfo()
 325 d= {}
 326 for key, value in mesures.items():
 327   d[str(key)] = value
 328 nbTriangles = d['Entity_Triangle']
 329 nbNodes = d['Entity_Node']
 330
 331 controlMeshStats(garonne_1, 6190, 691, 12201)
 332 controlSubMeshStats(litMineur_1, 2384)
 333 controlSubMeshStats(riveDroite_1, 2400)
 334 controlSubMeshStats(riveGauche_1, 6585)
 335 controlSubMeshStats(digue_1, 832)
 336
 337 if salome.sg.hasDesktop():
 338   salome.sg.updateObjBrowser()
 339 #----------------------
 340 # --- Z interpolation with HYDRO
 341 #----------------------
 342
 343 from salome.hydrotools.interpolZ import interpolZ
 344 from salome.hydrotools.controls import controlStatZ
 345
 346 # --- case name in HYDRO
 347 nomCas = 'garonne_1'
 348
 349 # --- dictionary [med group name] = region name
 350 dicoGroupeRegion= dict(litMineur  = 'garonne_1_litMineur',
 351                        riveDroite = 'garonne_1_riveDroite',
 352                        riveGauche = 'garonne_1_riveGauche',
 353                        digue      = 'garonne_1_digue',
 354                        )
 355 # --- value to use for Z when the node is not in a region (used to detect problems)
 356 zUndef = 90
 357 # --- interpolation Method: 0 = nearest point on bathymetry (default), 1 = linear interpolation
 358 interpolMethod = 0
 359 # --- produce a 3D mesh (Z set to its value instead of 0
 360 m3d = True
 361
 362 # --- Z interpolation on the bathymety/altimetry on the mesh nodes
 363 statz = interpolZ(nomCas, fichierMaillage, dicoGroupeRegion, zUndef, interpolMethod, m3d)
 364 #print statz
 365 refstatz = {'riveDroite': (10.88, 32.61, 24.09, 5.13, 17.57, 31.33, 0.2),
 366             'riveGauche': (7.72, 72.40, 21.59, 8.37, 16.71, 35.71, 0.4),
 367             'litMineur': (2.06, 25.41, 13.93, 4.33, 8.47, 21.78),
 368             'digue': (14.10, 27.09, 22.01, 2.56, 18.15, 25.02)}
 369 controlStatZ(statz, refstatz)
 370
 371 # --- add a field on nodes of type double with z values, named "BOTTOM"
 372 #createZfield2(fichierMaillage)