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/HYDRO")
   5
   6 import sys
   7 import salome
   8
   9 salome.salome_init()
  10 theStudy = salome.myStudy
  11
  12 #----------------------
  13 # --- HYDRO
  14 #----------------------
  15
  16 from HYDROPy import *
  17 from PyQt5.QtCore import *
  18 from PyQt5.QtGui import *
  19
  20 hydro_doc = HYDROData_Document.Document( theStudy._get_StudyId() )
  21
  22 hydro_doc.SetLocalCS( 430000, 6.35e+06 )
  23
  24 garonne_ign_01 = hydro_doc.CreateObject( KIND_IMAGE )
  25 garonne_ign_01.SetName( "garonne_ign_01" )
  26
  27 garonne_ign_01.SetZLevel( 0 )
  28
  29
  30 if not(garonne_ign_01.LoadImage(os.path.join(HYDRO_SAMPLES, "garonne_ign_01.png" ))):
  31   raise ValueError('problem while loading image')
  32
  33 garonne_ign_01.SetLocalPoints( QPoint( 40, 817 ),
  34                                QPoint( 1325, 85 ) )
  35
  36 garonne_ign_01.SetGlobalPoints( 1,
  37                                 QPointF( 471562, 6.36775e+06 ),
  38                                 QPointF( 489400, 6.37702e+06 ) )
  39
  40 garonne_ign_01.Update()
  41
  42 garonne_ign_02 = hydro_doc.CreateObject( KIND_IMAGE )
  43 garonne_ign_02.SetName( "garonne_ign_02" )
  44
  45 garonne_ign_02.SetZLevel( 1 )
  46
  47
  48 if not(garonne_ign_02.LoadImage( os.path.join(HYDRO_SAMPLES, "garonne_ign_02.png" ))):
  49   raise ValueError('problem while loading image')
  50
  51 garonne_ign_02.SetLocalPoints( QPoint( 1389, 447 ),
  52                                QPoint( 784, 481 ) )
  53
  54 garonne_ign_02.SetGlobalPoints( 3,
  55                                 QPointF( 631, 95 ),
  56                                 QPointF( 26, 129 ) )
  57 garonne_ign_02.SetTrsfReferenceImage( garonne_ign_01 )
  58
  59 garonne_ign_02.Update()
  60
  61 garonne = hydro_doc.CreateObject( KIND_POLYLINEXY )
  62 garonne.SetName( "garonne" )
  63
  64 garonne.SetZLevel( 2 )
  65
  66 garonne.AddSection( "Section_1", 1, 1 )
  67 garonne.AddPoint( 0, gp_XY( 38191.48, 33184.55 ) )
  68 garonne.AddPoint( 0, gp_XY( 38553.78, 33013.53 ) )
  69 garonne.AddPoint( 0, gp_XY( 38993.90, 32767.64 ) )
  70 garonne.AddPoint( 0, gp_XY( 39678.58, 32533.58 ) )
  71 garonne.AddPoint( 0, gp_XY( 40240.26, 32305.27 ) )
  72 garonne.AddPoint( 0, gp_XY( 40717.94, 32193.70 ) )
  73 garonne.AddPoint( 0, gp_XY( 41361.00, 32464.18 ) )
  74 garonne.AddPoint( 0, gp_XY( 41831.41, 32626.69 ) )
  75 garonne.AddPoint( 0, gp_XY( 41950.14, 32618.24 ) )
  76 garonne.AddPoint( 0, gp_XY( 42524.97, 32419.61 ) )
  77 garonne.AddPoint( 0, gp_XY( 42909.80, 32291.33 ) )
  78 garonne.AddPoint( 0, gp_XY( 43177.03, 32269.95 ) )
  79 garonne.AddPoint( 0, gp_XY( 43465.65, 32344.78 ) )
  80 garonne.AddPoint( 0, gp_XY( 43754.27, 32152.37 ) )
  81 garonne.AddPoint( 0, gp_XY( 44128.40, 31842.37 ) )
  82 garonne.AddPoint( 0, gp_XY( 44003.90, 31269.93 ) )
  83 garonne.AddPoint( 0, gp_XY( 44010.82, 30869.63 ) )
  84 garonne.AddPoint( 0, gp_XY( 44064.26, 30655.84 ) )
  85 garonne.AddPoint( 0, gp_XY( 44434.73, 30366.18 ) )
  86 garonne.AddPoint( 0, gp_XY( 45208.04, 30484.81 ) )
  87 garonne.AddPoint( 0, gp_XY( 45705.01, 30287.34 ) )
  88 garonne.AddPoint( 0, gp_XY( 45933.89, 30046.87 ) )
  89 garonne.AddPoint( 0, gp_XY( 46127.34, 29490.68 ) )
  90 garonne.AddPoint( 0, gp_XY( 46111.60, 29226.08 ) )
  91 garonne.AddPoint( 0, gp_XY( 46228.62, 28784.05 ) )
  92 garonne.AddPoint( 0, gp_XY( 46247.64, 28449.75 ) )
  93 garonne.AddPoint( 0, gp_XY( 46108.48, 28086.94 ) )
  94 garonne.AddPoint( 0, gp_XY( 45618.52, 27553.74 ) )
  95 garonne.AddPoint( 0, gp_XY( 45276.77, 27042.78 ) )
  96 garonne.AddPoint( 0, gp_XY( 45001.89, 26273.15 ) )
  97 garonne.AddPoint( 0, gp_XY( 45167.76, 25975.34 ) )
  98 garonne.AddPoint( 0, gp_XY( 46018.15, 25848.84 ) )
  99 garonne.AddPoint( 0, gp_XY( 46378.94, 25872.21 ) )
 100 garonne.AddPoint( 0, gp_XY( 46885.57, 25937.64 ) )
 101 garonne.AddPoint( 0, gp_XY( 47278.36, 25782.88 ) )
 102 garonne.AddPoint( 0, gp_XY( 47153.86, 25728.04 ) )
 103 garonne.AddPoint( 0, gp_XY( 46987.51, 25749.97 ) )
 104 garonne.AddPoint( 0, gp_XY( 46658.26, 25715.71 ) )
 105 garonne.AddPoint( 0, gp_XY( 45650.76, 25634.46 ) )
 106 garonne.AddPoint( 0, gp_XY( 45122.63, 25748.21 ) )
 107 garonne.AddPoint( 0, gp_XY( 44797.63, 26325.09 ) )
 108 garonne.AddPoint( 0, gp_XY( 45114.51, 27168.75 ) )
 109 garonne.AddPoint( 0, gp_XY( 45508.28, 27713.21 ) )
 110 garonne.AddPoint( 0, gp_XY( 45994.19, 28382.53 ) )
 111 garonne.AddPoint( 0, gp_XY( 45927.01, 29201.34 ) )
 112 garonne.AddPoint( 0, gp_XY( 45756.39, 29859.47 ) )
 113 garonne.AddPoint( 0, gp_XY( 45390.76, 30151.97 ) )
 114 garonne.AddPoint( 0, gp_XY( 45039.85, 30120.84 ) )
 115 garonne.AddPoint( 0, gp_XY( 44513.26, 30084.28 ) )
 116 garonne.AddPoint( 0, gp_XY( 43923.42, 30426.28 ) )
 117 garonne.AddPoint( 0, gp_XY( 43768.81, 30934.29 ) )
 118 garonne.AddPoint( 0, gp_XY( 43820.10, 31425.20 ) )
 119 garonne.AddPoint( 0, gp_XY( 43854.78, 31801.12 ) )
 120 garonne.AddPoint( 0, gp_XY( 43479.64, 32053.54 ) )
 121 garonne.AddPoint( 0, gp_XY( 42923.47, 32051.85 ) )
 122 garonne.AddPoint( 0, gp_XY( 42493.64, 32187.97 ) )
 123 garonne.AddPoint( 0, gp_XY( 42074.27, 32302.91 ) )
 124 garonne.AddPoint( 0, gp_XY( 41624.75, 32330.93 ) )
 125 garonne.AddPoint( 0, gp_XY( 41325.93, 32187.97 ) )
 126 garonne.AddPoint( 0, gp_XY( 40879.52, 31987.29 ) )
 127 garonne.AddPoint( 0, gp_XY( 40308.78, 32018.46 ) )
 128 garonne.AddPoint( 0, gp_XY( 39782.86, 32207.17 ) )
 129 garonne.AddPoint( 0, gp_XY( 38814.10, 32560.43 ) )
 130 garonne.AddPoint( 0, gp_XY( 38302.17, 32903.70 ) )
 131
 132 garonne.Update()
 133
 134
 135 domaine = hydro_doc.CreateObject( KIND_POLYLINEXY )
 136 domaine.SetName( "domaine" )
 137
 138 domaine.SetZLevel( 3 )
 139
 140 domaine.AddSection( "Section_1", 0, 1 )
 141 domaine.AddPoint( 0, gp_XY( 39901.15, 34762.30 ) );
 142 domaine.AddPoint( 0, gp_XY( 38086.50, 29964.69 ) );
 143 domaine.AddPoint( 0, gp_XY( 44835.83, 24264.97 ) );
 144 domaine.AddPoint( 0, gp_XY( 46758.56, 25021.86 ) );
 145 domaine.AddPoint( 0, gp_XY( 46757.20, 26601.21 ) );
 146 domaine.AddPoint( 0, gp_XY( 48566.45, 30152.81 ) );
 147 domaine.AddPoint( 0, gp_XY( 45532.36, 33239.83 ) );
 148 domaine.AddPoint( 0, gp_XY( 43548.77, 34048.39 ) );
 149
 150 domaine.Update()
 151
 152
 153 lit_majeur = hydro_doc.CreateObject( KIND_POLYLINEXY )
 154 lit_majeur.SetName( "lit_majeur" )
 155
 156 lit_majeur.SetZLevel( 7 )
 157
 158 lit_majeur.AddSection( "Section_1", 1, 1 )
 159 lit_majeur.AddPoint( 0, gp_XY( 41225.22, 34118.46 ) )
 160 lit_majeur.AddPoint( 0, gp_XY( 41716.41, 33705.87 ) )
 161 lit_majeur.AddPoint( 0, gp_XY( 42384.43, 33293.27 ) )
 162 lit_majeur.AddPoint( 0, gp_XY( 42561.26, 32684.19 ) )
 163 lit_majeur.AddPoint( 0, gp_XY( 42875.62, 32369.83 ) )
 164 lit_majeur.AddPoint( 0, gp_XY( 43700.82, 32350.18 ) )
 165 lit_majeur.AddPoint( 0, gp_XY( 44526.02, 31662.51 ) )
 166 lit_majeur.AddPoint( 0, gp_XY( 45370.87, 31819.69 ) )
 167 lit_majeur.AddPoint( 0, gp_XY( 46274.66, 31682.16 ) )
 168 lit_majeur.AddPoint( 0, gp_XY( 46981.97, 31328.50 ) )
 169 lit_majeur.AddPoint( 0, gp_XY( 47021.27, 30719.43 ) )
 170 lit_majeur.AddPoint( 0, gp_XY( 47217.74, 30051.41 ) )
 171 lit_majeur.AddPoint( 0, gp_XY( 47846.46, 29560.22 ) )
 172 lit_majeur.AddPoint( 0, gp_XY( 48730.61, 28735.02 ) )
 173 lit_majeur.AddPoint( 0, gp_XY( 48809.20, 27732.99 ) )
 174 lit_majeur.AddPoint( 0, gp_XY( 49143.21, 27163.21 ) )
 175 lit_majeur.AddPoint( 0, gp_XY( 49654.05, 26829.20 ) )
 176 lit_majeur.AddPoint( 0, gp_XY( 49693.34, 25355.63 ) )
 177 lit_majeur.AddPoint( 0, gp_XY( 48612.72, 23214.04 ) )
 178 lit_majeur.AddPoint( 0, gp_XY( 47453.51, 23115.80 ) )
 179 lit_majeur.AddPoint( 0, gp_XY( 46471.13, 24176.77 ) )
 180 lit_majeur.AddPoint( 0, gp_XY( 45960.29, 25257.39 ) )
 181 lit_majeur.AddPoint( 0, gp_XY( 44899.32, 25591.40 ) )
 182 lit_majeur.AddPoint( 0, gp_XY( 44270.60, 26573.78 ) )
 183 lit_majeur.AddPoint( 0, gp_XY( 43248.92, 27909.82 ) )
 184 lit_majeur.AddPoint( 0, gp_XY( 42757.73, 28243.83 ) )
 185 lit_majeur.AddPoint( 0, gp_XY( 42266.54, 28342.07 ) )
 186 lit_majeur.AddPoint( 0, gp_XY( 41736.06, 28106.30 ) )
 187 lit_majeur.AddPoint( 0, gp_XY( 41402.05, 28833.26 ) )
 188 lit_majeur.AddPoint( 0, gp_XY( 40910.86, 28911.85 ) )
 189 lit_majeur.AddPoint( 0, gp_XY( 40576.85, 29324.45 ) )
 190 lit_majeur.AddPoint( 0, gp_XY( 40144.60, 29599.51 ) )
 191 lit_majeur.AddPoint( 0, gp_XY( 39024.69, 30130.00 ) )
 192 lit_majeur.AddPoint( 0, gp_XY( 38612.09, 30365.77 ) )
 193 lit_majeur.AddPoint( 0, gp_XY( 38828.21, 33391.50 ) )
 194
 195 lit_majeur.Update()
 196
 197 axeDigue = hydro_doc.CreateObject( KIND_POLYLINEXY );
 198 axeDigue.SetName( "axeDigue" );
 199
 200 axeDigue.SetZLevel( 8 );
 201
 202 axeDigue.AddSection( "Section_1", 1, 0 );
 203 axeDigue.AddPoint( 0, gp_XY( 40961.39, 31800.95 ) );
 204 axeDigue.AddPoint( 0, gp_XY( 41296.95, 31992.70 ) );
 205 axeDigue.AddPoint( 0, gp_XY( 41656.48, 31872.85 ) );
 206 axeDigue.AddPoint( 0, gp_XY( 42004.02, 31441.42 ) );
 207 axeDigue.AddPoint( 0, gp_XY( 42123.86, 31021.97 ) );
 208 axeDigue.AddPoint( 0, gp_XY( 42579.27, 30926.10 ) );
 209 axeDigue.AddPoint( 0, gp_XY( 43082.60, 30638.48 ) );
 210 axeDigue.AddPoint( 0, gp_XY( 43130.54, 30075.22 ) );
 211 axeDigue.AddPoint( 0, gp_XY( 43538.00, 29763.63 ) );
 212 axeDigue.AddPoint( 0, gp_XY( 44053.33, 29787.60 ) );
 213 axeDigue.AddPoint( 0, gp_XY( 44736.43, 29847.52 ) );
 214
 215 axeDigue.Update();
 216
 217
 218 Cloud_02 = hydro_doc.CreateObject( KIND_BATHYMETRY )
 219 Cloud_02.SetName( "Cloud_02" )
 220
 221 Cloud_02.SetAltitudesInverted( 0 )
 222 if not(Cloud_02.ImportFromFile( os.path.join(HYDRO_SAMPLES, "Cloud_02.xyz" ))):
 223   raise ValueError('problem while loading bathymetry')
 224
 225 Cloud_02.Update()
 226
 227
 228 garonne_point_L93 = hydro_doc.CreateObject( KIND_BATHYMETRY )
 229 garonne_point_L93.SetName( "garonne_point_L93" )
 230
 231 garonne_point_L93.SetAltitudesInverted( 0 )
 232 if not(garonne_point_L93.ImportFromFile( os.path.join(HYDRO_SAMPLES, "garonne_point_L93.xyz" ))):
 233   raise ValueError('problem while loading bathymetry')
 234
 235 garonne_point_L93.Update()
 236
 237
 238 altitudeDigue = hydro_doc.CreateObject( KIND_PROFILE );
 239 altitudeDigue.SetName( "altitudeDigue" );
 240
 241 profile_points = [ gp_XY( 1, 23 ),
 242                    gp_XY( 5000, 25.5 ) ];
 243 altitudeDigue.SetParametricPoints( profile_points );
 244
 245 altitudeDigue.Update();
 246
 247
 248 sectionDigue = hydro_doc.CreateObject( KIND_PROFILE );
 249 sectionDigue.SetName( "sectionDigue" );
 250
 251 profile_points = [ gp_XY( -10, -10 ),
 252                    gp_XY( -8, -1 ),
 253                    gp_XY( 0, 0 ),
 254                    gp_XY( 8, -1 ),
 255                    gp_XY( 10, -10 ) ];
 256 sectionDigue.SetParametricPoints( profile_points );
 257
 258 sectionDigue.Update();
 259
 260
 261 axe3DDigue = hydro_doc.CreateObject( KIND_POLYLINE );
 262 axe3DDigue.SetName( "axe3DDigue" );
 263
 264 axe3DDigue.SetPolylineXY( axeDigue );
 265 axe3DDigue.SetProfileUZ( altitudeDigue.GetProfileUZ() );
 266
 267 axe3DDigue.Update();
 268
 269
 270 litMineur = hydro_doc.CreateObject( KIND_IMMERSIBLE_ZONE )
 271 litMineur.SetName( "litMineur" )
 272
 273 litMineur.SetZLevel( 6 )
 274
 275 litMineur.SetAltitudeObject( garonne_point_L93 )
 276 litMineur.SetPolyline( garonne )
 277
 278 litMineur.Update()
 279
 280
 281 litMajeur = hydro_doc.CreateObject( KIND_IMMERSIBLE_ZONE )
 282 litMajeur.SetName( "litMajeur" )
 283
 284 litMajeur.SetZLevel( 5 )
 285
 286 litMajeur.SetFillingColor( QColor( 0, 170, 127, 255 ) )
 287
 288 litMajeur.SetAltitudeObject( Cloud_02 )
 289 litMajeur.SetPolyline( lit_majeur )
 290
 291 litMajeur.Update()
 292
 293
 294 domaineEtendu = hydro_doc.CreateObject( KIND_IMMERSIBLE_ZONE )
 295 domaineEtendu.SetName( "domaineEtendu" )
 296
 297 domaineEtendu.SetZLevel( 4 )
 298
 299 domaineEtendu.SetFillingColor( QColor( 201, 203, 55, 255 ) )
 300
 301 domaineEtendu.SetAltitudeObject( Cloud_02 )
 302 domaineEtendu.SetPolyline( domaine )
 303
 304 domaineEtendu.Update()
 305
 306
 307 digue = hydro_doc.CreateObject( KIND_DIGUE );
 308 digue.SetName( "digue" );
 309
 310 digue.SetZLevel( 9 );
 311
 312 digue.SetGuideLine( axe3DDigue );
 313 digue.SetProfile( sectionDigue );
 314 digue.SetEquiDistance( 50 );
 315
 316 digue.Update();
 317
 318
 319 # Calculation case
 320 garonne_1 = hydro_doc.CreateObject( KIND_CALCULATION )
 321 garonne_1.SetName( "garonne_1" )
 322
 323 garonne_1.SetAssignmentMode( HYDROData_CalculationCase.MANUAL )
 324 garonne_1.AddGeometryObject( litMineur )
 325 garonne_1.AddGeometryObject( domaineEtendu )
 326 garonne_1.AddGeometryObject( litMajeur )
 327 garonne_1.AddGeometryObject( digue );
 328
 329 case_geom_group = domaineEtendu.GetGroup( 0 );
 330 garonne_1.AddGeometryGroup( case_geom_group );
 331 case_geom_group = litMineur.GetGroup( 0 );
 332 garonne_1.AddGeometryGroup( case_geom_group );
 333 case_geom_group = litMajeur.GetGroup( 0 );
 334 garonne_1.AddGeometryGroup( case_geom_group );
 335 case_geom_group = digue.GetGroup( 2 );
 336 garonne_1.AddGeometryGroup( case_geom_group );
 337 case_geom_group = digue.GetGroup( 0 );
 338 garonne_1.AddGeometryGroup( case_geom_group );
 339 case_geom_group = digue.GetGroup( 3 );
 340 garonne_1.AddGeometryGroup( case_geom_group );
 341 case_geom_group = digue.GetGroup( 1 );
 342 garonne_1.AddGeometryGroup( case_geom_group );
 343 garonne_1.SetBoundaryPolyline( domaine );
 344
 345 # Start the algorithm of the partition and assignment
 346 garonne_1.Update();
 347
 348 garonne_1_litMineur = hydro_doc.FindObjectByName( "garonne_1_Reg_1" )
 349 garonne_1_Zone_1 = hydro_doc.FindObjectByName( "garonne_1_Zone_1" )
 350 garonne_1_Zone_1.SetMergeType( HYDROData_Zone.Merge_ZMIN )
 351 garonne_1_Zone_1.SetColor( QColor( 192, 137, 64 ))
 352 garonne_1_litMineur.AddZone( garonne_1_Zone_1 )
 353
 354 garonne_1_riveDroite = hydro_doc.FindObjectByName( "garonne_1_Reg_2" )
 355 garonne_1_Zone_2 = hydro_doc.FindObjectByName( "garonne_1_Zone_2" )
 356 garonne_1_Zone_2.SetColor( QColor( 64, 192, 98 ))
 357 garonne_1_riveDroite.AddZone( garonne_1_Zone_2 )
 358
 359 garonne_1_Zone_3 = hydro_doc.FindObjectByName( "garonne_1_Zone_3" )
 360 garonne_1_Zone_3.SetMergeType( HYDROData_Zone.Merge_Object )
 361 Cloud_02 = hydro_doc.FindObjectByName( "Cloud_02" )
 362 garonne_1_Zone_3.SetMergeObject( Cloud_02 )
 363 garonne_1_Zone_3.SetColor( QColor( 64, 66, 192 ))
 364 garonne_1_riveDroite.AddZone( garonne_1_Zone_3 )
 365
 366 garonne_1_riveGauche = hydro_doc.FindObjectByName( "garonne_1_Reg_3" )
 367 garonne_1_Zone_4 = hydro_doc.FindObjectByName( "garonne_1_Zone_4" )
 368 garonne_1_Zone_4.SetMergeType( HYDROData_Zone.Merge_Object )
 369 Cloud_02 = hydro_doc.FindObjectByName( "Cloud_02" )
 370 garonne_1_Zone_4.SetMergeObject( Cloud_02 )
 371 garonne_1_Zone_4.SetColor( QColor( 64, 192, 164 ))
 372 garonne_1_riveGauche.AddZone( garonne_1_Zone_4 )
 373
 374 garonne_1_Zone_5 = hydro_doc.FindObjectByName( "garonne_1_Zone_5" )
 375 garonne_1_Zone_5.SetColor( QColor( 192, 149, 64 ))
 376 garonne_1_riveGauche.AddZone( garonne_1_Zone_5 )
 377
 378 garonne_1_digue = hydro_doc.FindObjectByName( "garonne_1_Reg_4" )
 379 garonne_1_Zone_6 = hydro_doc.FindObjectByName( "garonne_1_Zone_6" )
 380 garonne_1_Zone_6.SetMergeType( HYDROData_Zone.Merge_ZMAX )
 381 garonne_1_Zone_6.SetColor( QColor( 64, 192, 177 ))
 382 garonne_1_digue.AddZone( garonne_1_Zone_6 )
 383
 384 garonne_1_litMineur.SetName("garonne_1_litMineur")
 385 garonne_1_riveDroite.SetName("garonne_1_riveDroite")
 386 garonne_1_riveGauche.SetName("garonne_1_riveGauche")
 387 garonne_1_digue.SetName("garonne_1_digue")
 388
 389 # Export of the calculation case
 390 garonne_1_entry = garonne_1.Export( theStudy._get_StudyId() )
 391
 392 #----------------------
 393 # --- Geometry
 394 #----------------------
 395
 396 # Get geometry shape and print debug information
 397 import GEOM
 398 from salome.geom import geomBuilder
 399 import math
 400 import SALOMEDS
 401 from salome.hydrotools.controls import controlGeomProps
 402
 403 geompy = geomBuilder.New(theStudy)
 404
 405 print "Entry:", garonne_1_entry
 406 HYDRO_garonne_1 = salome.IDToObject( str( garonne_1_entry ) )
 407 print "Geom shape:", HYDRO_garonne_1
 408 print "Geom shape name:", HYDRO_garonne_1.GetName()
 409
 410
 411 [garonne_riveGauche,garonne_digue,garonne_litMineur,garonne_riveDroite] = geompy.ExtractShapes(HYDRO_garonne_1, geompy.ShapeType["FACE"], True)
 412
 413 controlGeomProps(geompy, garonne_riveGauche,  39490.835288, 35845737.590926)
 414 controlGeomProps(geompy, garonne_digue,       10341.466108,   103090.760662)
 415 controlGeomProps(geompy, garonne_litMineur,   30337.548492,  3488480.304388)
 416 controlGeomProps(geompy, garonne_riveDroite,  32012.343241, 25998769.23615)
 417
 418 # --- manual identification of all useful edge groups (boundary conditions)
 419
 420 allEdgesIds = geompy.SubShapeAllIDs(HYDRO_garonne_1, geompy.ShapeType["EDGE"])
 421 print "allEdgesIds", allEdgesIds
 422
 423 (isDone, ClosedFreeBoundary, OpenFreeBoundary) = geompy.GetFreeBoundary(HYDRO_garonne_1)
 424 geompy.addToStudyInFather(HYDRO_garonne_1, ClosedFreeBoundary[0], "ClosedFreeBoundary")
 425
 426 freeBoundary = geompy.ExtractShapes(ClosedFreeBoundary[0], geompy.ShapeType["EDGE"], True)
 427 freeBoundaryIds = [ geompy.GetSubShapeID(HYDRO_garonne_1, freeBoundary[i]) for i in range(len(freeBoundary)) ]
 428 print "freeBoundaryIds", freeBoundaryIds
 429
 430 [litMineur_droite] = geompy.GetSharedShapesMulti([garonne_riveDroite, garonne_litMineur], geompy.ShapeType["EDGE"], True)
 431 [litMineur_gauche] = geompy.GetSharedShapesMulti([garonne_riveGauche, garonne_litMineur], geompy.ShapeType["EDGE"], True)
 432 geompy.addToStudyInFather(HYDRO_garonne_1, litMineur_droite, "litMineur_droite")
 433 geompy.addToStudyInFather(HYDRO_garonne_1, litMineur_gauche, "litMineur_gauche")
 434 rives = [litMineur_droite, litMineur_gauche]
 435 rivesIds = [ geompy.GetSubShapeID(HYDRO_garonne_1, rives[i]) for i in range(len(rives)) ]
 436 print "rivesIds", rivesIds
 437
 438 edges_litMineur = geompy.GetSharedShapesMulti([HYDRO_garonne_1, garonne_litMineur], geompy.ShapeType["EDGE"], True)
 439 edges_riveGauche = geompy.GetSharedShapesMulti([HYDRO_garonne_1, garonne_riveGauche], geompy.ShapeType["EDGE"], True)
 440 edges_riveDroite = geompy.GetSharedShapesMulti([HYDRO_garonne_1, garonne_riveDroite], geompy.ShapeType["EDGE"], True)
 441 edges_litMineurIds = [ geompy.GetSubShapeID(HYDRO_garonne_1, edges_litMineur[i]) for i in range(len(edges_litMineur)) ]
 442 edges_riveGaucheIds = [ geompy.GetSubShapeID(HYDRO_garonne_1, edges_riveGauche[i]) for i in range(len(edges_riveGauche)) ]
 443 edges_riveDroiteIds = [ geompy.GetSubShapeID(HYDRO_garonne_1, edges_riveDroite[i]) for i in range(len(edges_riveDroite)) ]
 444
 445 print "edges_litMineurIds", edges_litMineurIds
 446 print "edges_riveGaucheIds", edges_riveGaucheIds
 447 print "edges_riveDroiteIds", edges_riveDroiteIds
 448
 449 sectionsIds = [Id for Id in edges_litMineurIds if Id not in rivesIds]
 450 print "sectionsIds", sectionsIds
 451 SectionsGaronne = geompy.CreateGroup(HYDRO_garonne_1, geompy.ShapeType["EDGE"])
 452 geompy.UnionIDs(SectionsGaronne, sectionsIds)
 453 geompy.addToStudyInFather(HYDRO_garonne_1, SectionsGaronne, "SectionsGaronne")
 454
 455 bordGaucheDomaineIds = [Id for Id in freeBoundaryIds if Id in edges_riveGaucheIds]
 456 bordDroiteDomaineIds = [Id for Id in freeBoundaryIds if Id in edges_riveDroiteIds]
 457 print "bordGaucheDomaineIds", bordGaucheDomaineIds
 458 print "bordDroiteDomaineIds", bordDroiteDomaineIds
 459 bordGaucheDomaine = geompy.CreateGroup(HYDRO_garonne_1, geompy.ShapeType["EDGE"])
 460 geompy.UnionIDs(bordGaucheDomaine, bordGaucheDomaineIds)
 461 geompy.addToStudyInFather(HYDRO_garonne_1, bordGaucheDomaine, "bordGaucheDomaine")
 462 bordDroiteDomaine = geompy.CreateGroup(HYDRO_garonne_1, geompy.ShapeType["EDGE"])
 463 geompy.UnionIDs(bordDroiteDomaine, bordDroiteDomaineIds)
 464 geompy.addToStudyInFather(HYDRO_garonne_1, bordDroiteDomaine, "bordDroiteDomaine")
 465
 466 amont = geompy.GetEdgeNearPoint(HYDRO_garonne_1, geompy.MakeVertex(46757.861314, 25833.234752, 0))
 467 aval = geompy.GetEdgeNearPoint(HYDRO_garonne_1, geompy.MakeVertex(39078.979127, 32588.627279, 0))
 468 geompy.addToStudyInFather(HYDRO_garonne_1, amont, "amont")
 469 geompy.addToStudyInFather(HYDRO_garonne_1, aval, "aval")
 470
 471 amontDigue = geompy.GetEdgeNearPoint(HYDRO_garonne_1, geompy.MakeVertex(44736.43, 29847.52, 0))
 472 avalDigue = geompy.GetEdgeNearPoint(HYDRO_garonne_1, geompy.MakeVertex(40961.39, 31800.95, 0))
 473 geompy.addToStudyInFather(HYDRO_garonne_1, amontDigue, "amontDigue")
 474 geompy.addToStudyInFather(HYDRO_garonne_1, avalDigue, "avalDigue")
 475
 476 SectionsDigue = geompy.CreateGroup(HYDRO_garonne_1, geompy.ShapeType["EDGE"])
 477 geompy.UnionIDs(SectionsDigue, [geompy.GetSubShapeID(HYDRO_garonne_1, amontDigue), geompy.GetSubShapeID(HYDRO_garonne_1, avalDigue)])
 478 geompy.addToStudyInFather( HYDRO_garonne_1, SectionsDigue, 'SectionsDigue' )
 479
 480 #----------------------
 481 # --- Meshing
 482 #----------------------
 483
 484 import  SMESH, SALOMEDS
 485 from salome.smesh import smeshBuilder
 486 from salome.hydrotools.controls import controlMeshStats, controlSubMeshStats
 487 import tempfile
 488
 489 smesh = smeshBuilder.New(theStudy)
 490
 491 # --- algorithms and hypothesis
 492 garonne_1 = smesh.Mesh(HYDRO_garonne_1)
 493
 494 NETGEN_2D = garonne_1.Triangle(algo=smeshBuilder.NETGEN_1D2D)
 495 NETGEN_2D_Parameters = NETGEN_2D.Parameters()
 496 NETGEN_2D_Parameters.SetMaxSize( 200 )
 497 NETGEN_2D_Parameters.SetSecondOrder( 0 )
 498 NETGEN_2D_Parameters.SetOptimize( 1 )
 499 NETGEN_2D_Parameters.SetFineness( 4 )
 500 NETGEN_2D_Parameters.SetMinSize( 50 )
 501 NETGEN_2D_Parameters.SetUseSurfaceCurvature( 1 )
 502 NETGEN_2D_Parameters.SetFuseEdges( 1 )
 503 NETGEN_2D_Parameters.SetQuadAllowed( 0 )
 504
 505 algo2D_litMineur = garonne_1.Quadrangle(algo=smeshBuilder.QUAD_MA_PROJ,geom=garonne_litMineur)
 506 algo1D_litMineur = garonne_1.Segment(geom=garonne_litMineur)
 507 hypo1D_litMineur = algo1D_litMineur.LocalLength(100,None,1e-07)
 508 subMesh_litMineur = algo1D_litMineur.GetSubMesh()
 509 smesh.SetName(subMesh_litMineur, "litMineur")
 510
 511 algo1D_SectionsGaronne = garonne_1.Segment(geom=SectionsGaronne)
 512 hypo1D_SectionsGaronne = algo1D_SectionsGaronne.NumberOfSegments(8)
 513 hypo1D_SectionsGaronne.SetDistrType( 0 )
 514 subMesh_SectionsGaronne = algo1D_SectionsGaronne.GetSubMesh()
 515 smesh.SetName(subMesh_SectionsGaronne, "SectionsGaronne")
 516
 517 algo2D_digue = garonne_1.Quadrangle(algo=smeshBuilder.QUAD_MA_PROJ,geom=garonne_digue)
 518 algo1D_digue = garonne_1.Segment(geom=garonne_digue)
 519 hypo1D_digue = algo1D_digue.LocalLength(50,None,1e-07)
 520 subMesh_digue = algo1D_digue.GetSubMesh()
 521 smesh.SetName(subMesh_digue, "digue")
 522
 523 algo1D_SectionsDigue = garonne_1.Segment(geom=SectionsDigue)
 524 hypo1D_SectionsDigue = algo1D_SectionsDigue.NumberOfSegments(4)
 525 hypo1D_SectionsDigue.SetDistrType( 0 )
 526 subMesh_SectionsDigue = algo1D_SectionsDigue.GetSubMesh()
 527 smesh.SetName(subMesh_SectionsDigue, "SectionsDigue")
 528
 529 isDone = garonne_1.SetMeshOrder( [ [ subMesh_SectionsGaronne, subMesh_litMineur ], [ subMesh_SectionsDigue, subMesh_digue ]])
 530
 531 # --- compute mesh
 532 isDone = garonne_1.Compute()
 533 isDone = garonne_1.SplitQuadObject( garonne_1, 1 )
 534 isDone = garonne_1.ReorientObject( garonne_1 )
 535
 536 # --- geometrical groups of faces
 537 riveGauche_1 = garonne_1.GroupOnGeom(garonne_riveGauche,'riveGauche',SMESH.FACE)
 538 litMineur_1 = garonne_1.GroupOnGeom(garonne_litMineur,'litMineur',SMESH.FACE)
 539 riveDroite_1 = garonne_1.GroupOnGeom(garonne_riveDroite,'riveDroite',SMESH.FACE)
 540 digue_1 = garonne_1.GroupOnGeom(garonne_digue,'digue',SMESH.FACE)
 541
 542 # --- geometrical groups of edges
 543 ClosedFreeBoundary_1 = garonne_1.GroupOnGeom(ClosedFreeBoundary[0],'ClosedFreeBoundary',SMESH.EDGE)
 544 litMineur_droite_1 = garonne_1.GroupOnGeom(litMineur_droite,'litMineur_droite',SMESH.EDGE)
 545 litMineur_gauche_1 = garonne_1.GroupOnGeom(litMineur_gauche,'litMineur_gauche',SMESH.EDGE)
 546 SectionsGaronne_1 = garonne_1.GroupOnGeom(SectionsGaronne,'SectionsGaronne',SMESH.EDGE)
 547 bordGaucheDomaine_1 = garonne_1.GroupOnGeom(bordGaucheDomaine,'bordGaucheDomaine',SMESH.EDGE)
 548 bordDroiteDomaine_1 = garonne_1.GroupOnGeom(bordDroiteDomaine,'bordDroiteDomaine',SMESH.EDGE)
 549 amont_1 = garonne_1.GroupOnGeom(amont,'amont',SMESH.EDGE)
 550 aval_1 = garonne_1.GroupOnGeom(aval,'aval',SMESH.EDGE)
 551 SectionsDigue_1 = garonne_1.GroupOnGeom(SectionsDigue,'SectionsDigue',SMESH.EDGE)
 552
 553 # --- geometrical groups of nodes
 554 garonne_1_litMineur_2 = garonne_1.GroupOnGeom(garonne_litMineur,'garonne_1_litMineur',SMESH.NODE)
 555 garonne_1_riveDroite_2 = garonne_1.GroupOnGeom(garonne_riveDroite,'garonne_1_riveDroite',SMESH.NODE)
 556 garonne_1_riveGauche_2 = garonne_1.GroupOnGeom(garonne_riveGauche,'garonne_1_riveGauche',SMESH.NODE)
 557 garonne_1_digue_2 = garonne_1.GroupOnGeom(garonne_digue,'garonne_1_digue',SMESH.NODE)
 558 ClosedFreeBoundary_2 = garonne_1.GroupOnGeom(ClosedFreeBoundary[0],'ClosedFreeBoundary',SMESH.NODE)
 559 litMineur_droite_2 = garonne_1.GroupOnGeom(litMineur_droite,'litMineur_droite',SMESH.NODE)
 560 litMineur_gauche_2 = garonne_1.GroupOnGeom(litMineur_gauche,'litMineur_gauche',SMESH.NODE)
 561 SectionsGaronne_2 = garonne_1.GroupOnGeom(SectionsGaronne,'SectionsGaronne',SMESH.NODE)
 562 bordGaucheDomaine_2 = garonne_1.GroupOnGeom(bordGaucheDomaine,'bordGaucheDomaine',SMESH.NODE)
 563 bordDroiteDomaine_2 = garonne_1.GroupOnGeom(bordDroiteDomaine,'bordDroiteDomaine',SMESH.NODE)
 564 amont_2 = garonne_1.GroupOnGeom(amont,'amont',SMESH.NODE)
 565 aval_2 = garonne_1.GroupOnGeom(aval,'aval',SMESH.NODE)
 566 SectionsDigue_2 = garonne_1.GroupOnGeom(SectionsDigue,'SectionsDigue',SMESH.NODE)
 567
 568 garonne_1.SetAutoColor( 1 )
 569 tmpdir = tempfile.mkdtemp()
 570 print "tmpdir=",tmpdir
 571 fichierMaillage = os.path.join(tmpdir, 'garonne_1.med')
 572 garonne_1.ExportMED(fichierMaillage, 0, SMESH.MED_V2_2, 1, None ,1)
 573
 574 mesures = garonne_1.GetMeshInfo()
 575 d= {}
 576 for key, value in mesures.iteritems():
 577   d[str(key)] = value
 578 nbTriangles = d['Entity_Triangle']
 579 nbNodes = d['Entity_Node']
 580
 581 controlMeshStats(garonne_1, 6190, 691, 12201)
 582 controlSubMeshStats(litMineur_1, 2384)
 583 controlSubMeshStats(riveDroite_1, 2400)
 584 controlSubMeshStats(riveGauche_1, 6585)
 585 controlSubMeshStats(digue_1, 832)
 586
 587 if salome.sg.hasDesktop():
 588   salome.sg.updateObjBrowser(1)
 589 #----------------------
 590 # --- Z interpolation with HYDRO
 591 #----------------------
 592
 593 from salome.hydrotools.interpolZ import interpolZ, createZfield2
 594 from salome.hydrotools.controls import controlStatZ
 595
 596 # --- case name in HYDRO
 597 nomCas = 'garonne_1'
 598
 599 # --- dictionary [med group name] = region name
 600 dicoGroupeRegion= dict(litMineur  = 'garonne_1_litMineur',
 601                        riveDroite = 'garonne_1_riveDroite',
 602                        riveGauche = 'garonne_1_riveGauche',
 603                        digue      = 'garonne_1_digue',
 604                        )
 605 # --- value to use for Z when the node is not in a region (used to detect problems)
 606 zUndef = 90
 607
 608 # --- Z interpolation on the bathymety/altimetry on the mesh nodes
 609 statz = interpolZ(nomCas, fichierMaillage, dicoGroupeRegion, zUndef)
 610 #print statz
 611 refstatz = {'riveDroite': (10.88, 32.61, 24.09, 5.13, 17.57, 31.33, 0.2),
 612             'riveGauche': (7.72, 72.40, 21.59, 8.37, 16.71, 35.71, 0.2),
 613             'litMineur': (2.06, 25.41, 13.93, 4.33, 8.47, 21.78),
 614             'digue': (14.10, 27.09, 22.01, 2.56, 18.15, 25.02)}
 615 controlStatZ(statz, refstatz)
 616
 617 # --- add a field on nodes of type double with z values, named "BOTTOM"
 618 #createZfield2(fichierMaillage)