From 3653033315922155cc0ff2b2103f3e7eea490b56 Mon Sep 17 00:00:00 2001 From: Paul RASCLE Date: Thu, 1 Oct 2020 10:41:42 +0200 Subject: [PATCH] test h033 complete case with modifed regions --- doc/salome/examples/CMakeLists.txt | 1 + .../examples/CTestTestfileInstall.cmake | 1 + doc/salome/examples/constraint2.shp | Bin 236 -> 380 bytes doc/salome/examples/constraint2.shx | Bin 108 -> 108 bytes .../h033_completeCaseModifiedRegions.py | 309 ++++++++++++++++++ 5 files changed, 311 insertions(+) create mode 100644 doc/salome/examples/h033_completeCaseModifiedRegions.py diff --git a/doc/salome/examples/CMakeLists.txt b/doc/salome/examples/CMakeLists.txt index 22e97cdd..b07a6757 100644 --- a/doc/salome/examples/CMakeLists.txt +++ b/doc/salome/examples/CMakeLists.txt @@ -55,6 +55,7 @@ SET(EXAMPLES_TESTS h030_editNonHydroMesh.py h031_interpolationLineaireStream.py h032_completeCaseNoRegions.py + h033_completeCaseModifiedRegions.py ) SET(HYDRO_SAMPLES diff --git a/doc/salome/examples/CTestTestfileInstall.cmake b/doc/salome/examples/CTestTestfileInstall.cmake index 4a2527db..1eabdb2b 100644 --- a/doc/salome/examples/CTestTestfileInstall.cmake +++ b/doc/salome/examples/CTestTestfileInstall.cmake @@ -57,6 +57,7 @@ SET(EXAMPLES_TESTS h030_editNonHydroMesh h031_interpolationLineaireStream h032_completeCaseNoRegions + h033_completeCaseModifiedRegions ) FOREACH(tfile ${EXAMPLES_TESTS}) diff --git a/doc/salome/examples/constraint2.shp b/doc/salome/examples/constraint2.shp index 837d929a9dae7b92ceb9a91675c4b481abccaaf4..545b6e7e0edb06b6c9280712d3f4e0489ecdc1a1 100644 GIT binary patch literal 380 zcmZQzQ0HR64)(oZW?%p!23NEB%X*f}IL5s_v*UhPgyW^+)jM+U$~aaye4C*d5#fkk zJBlbHkk>(sSs;acP_5N1I&Dwa$vD23=xh21w0GkF@+zg;`R3j>XYxoq@@(Z54dWGb#2eQ#mm4NaWR``5y0Q&QD<*M5)Kz^KPg!_7+ewe>g n?3(iTuaq5B5_r_qDr literal 236 zcmZQzQ0HR64$59IGcW)VLv@Qz+tYP2j_)P*@%MK+(NjHetyUhT}d-&R;b(sQf0ZQUF}ctG};Ay`;Jkx>?rUbe)Xjdx^fLe_;`h=H51E@=wS(Zc0pcO%9K6oERuJF@b$z0voFl I0|SEt06AR}+yDRo diff --git a/doc/salome/examples/h033_completeCaseModifiedRegions.py b/doc/salome/examples/h033_completeCaseModifiedRegions.py new file mode 100644 index 00000000..651f9cb4 --- /dev/null +++ b/doc/salome/examples/h033_completeCaseModifiedRegions.py @@ -0,0 +1,309 @@ +# -*- coding: utf-8 -*- + +import os +HYDRO_SAMPLES = os.path.join( os.environ["HYDRO_ROOT_DIR"], "bin/salome/test/HYDRO") +HYDRO_TEST_RESOURCES = "/home/paul/testsHydro" #os.environ("HYDRO_TEST_RESOURCES") +if not os.path.isdir(HYDRO_TEST_RESOURCES): + os.mkdir(HYDRO_TEST_RESOURCES) + +import tempfile +tmpdir = tempfile.mkdtemp() +print("tmpdir=",tmpdir) + +import sys +import salome + +salome.salome_init() + +#---------------------- +# --- HYDRO +#---------------------- + +from HYDROPy import * +from PyQt5.QtCore import * +from PyQt5.QtGui import * + +from salome.hydrotools.hydroGeoMeshUtils import importPolylines, importBathymetry, createImmersibleZone, mergePolylines + +hydro_doc = HYDROData_Document.Document() + +offsetX = 430000. +offsetY = 6350000. +hydro_doc.SetLocalCS( offsetX, offsetY ) + +garonne = importPolylines(hydro_doc, os.path.join(HYDRO_SAMPLES, "garonne.shp"), 'garonne_PolyXY', True, 2) +lit_majeur = importPolylines(hydro_doc, os.path.join(HYDRO_SAMPLES, "lit_majeur.shp"), 'lit_majeur_PolyXY', True, 3) +ile = importPolylines(hydro_doc, os.path.join(HYDRO_SAMPLES, "ile.shp"), 'ile_PolyXY', True, 4) +domaine = importPolylines(hydro_doc, os.path.join(HYDRO_SAMPLES, "domaine.shp"), 'domaine_PolyXY', False, 5) + +Cloud_02 = importBathymetry(hydro_doc, "Cloud_02", HYDRO_SAMPLES) +garonne_point_L93 = importBathymetry(hydro_doc, "garonne_point_L93", HYDRO_SAMPLES) + +litMineur = createImmersibleZone(hydro_doc, "litMineur", garonne[0], Cloud_02, True, 6) +litMajeur = createImmersibleZone(hydro_doc, "litMajeur", lit_majeur[0], garonne_point_L93, True, 5) +ilot = createImmersibleZone(hydro_doc, "ilot", ile[0], Cloud_02, False, 9) +domaineEtendu = createImmersibleZone(hydro_doc, "domaineEtendu", domaine[0], Cloud_02, True, 9) + +# Calculation case +casGaronne = hydro_doc.CreateObject( KIND_CALCULATION ) +casGaronne.SetName( "casGaronne" ) + +casGaronne.SetAssignmentMode( HYDROData_CalculationCase.MANUAL ) +casGaronne.AddGeometryObject( litMineur ) +casGaronne.AddGeometryObject( domaineEtendu ) +casGaronne.AddGeometryObject( litMajeur ) +casGaronne.AddGeometryObject( ilot ) + +case_geom_group = domaineEtendu.GetGroup( 0 ) +casGaronne.AddGeometryGroup( case_geom_group ) +case_geom_group = litMineur.GetGroup( 0 ) +casGaronne.AddGeometryGroup( case_geom_group ) +case_geom_group = litMajeur.GetGroup( 0 ) +casGaronne.AddGeometryGroup( case_geom_group ) +casGaronne.SetBoundaryPolyline( domaine[0] ) + +# Start the algorithm of the partition and assignment +casGaronne.Update() +casGaronne_litMineur = hydro_doc.FindObjectByName( "casGaronne_Reg_1" ) +casGaronne_Zone_1 = hydro_doc.FindObjectByName( "casGaronne_Zone_1" ) +casGaronne_Zone_1.SetMergeType( HYDROData_Zone.Merge_ZMIN ) +casGaronne_Zone_1.SetColor( QColor( 192, 113, 64 )) +casGaronne_litMineur.AddZone( casGaronne_Zone_1 ) + +casGaronne_riveDroite = hydro_doc.FindObjectByName( "casGaronne_Reg_2" ) +casGaronne_Zone_2 = hydro_doc.FindObjectByName( "casGaronne_Zone_2" ) +casGaronne_Zone_2.SetMergeType( HYDROData_Zone.Merge_Object ) +Cloud_02 = hydro_doc.FindObjectByName( "Cloud_02" ) +casGaronne_Zone_2.SetMergeObject( Cloud_02 ) +casGaronne_Zone_2.SetColor( QColor( 141, 192, 64 )) +casGaronne_riveDroite.AddZone( casGaronne_Zone_2 ) + +casGaronne_Zone_3 = hydro_doc.FindObjectByName( "casGaronne_Zone_3" ) +casGaronne_Zone_3.SetMergeType( HYDROData_Zone.Merge_Object ) +Cloud_02 = hydro_doc.FindObjectByName( "Cloud_02" ) +casGaronne_Zone_3.SetMergeObject( Cloud_02 ) +casGaronne_Zone_3.SetColor( QColor( 64, 192, 77 )) +casGaronne_riveDroite.AddZone( casGaronne_Zone_3 ) + +casGaronne_riveGauche = hydro_doc.FindObjectByName( "casGaronne_Reg_3" ) +casGaronne_Zone_4 = hydro_doc.FindObjectByName( "casGaronne_Zone_4" ) +casGaronne_Zone_4.SetMergeType( HYDROData_Zone.Merge_Object ) +Cloud_02 = hydro_doc.FindObjectByName( "Cloud_02" ) +casGaronne_Zone_4.SetMergeObject( Cloud_02 ) +casGaronne_Zone_4.SetColor( QColor( 64, 75, 192 )) +casGaronne_riveGauche.AddZone( casGaronne_Zone_4 ) + +casGaronne_Zone_5 = hydro_doc.FindObjectByName( "casGaronne_Zone_5" ) +casGaronne_Zone_5.SetMergeType( HYDROData_Zone.Merge_Object ) +Cloud_02 = hydro_doc.FindObjectByName( "Cloud_02" ) +casGaronne_Zone_5.SetMergeObject( Cloud_02 ) +casGaronne_Zone_5.SetColor( QColor( 64, 192, 77 )) +casGaronne_riveGauche.AddZone( casGaronne_Zone_5 ) + +casGaronne_ile = hydro_doc.FindObjectByName( "casGaronne_Reg_4" ) +casGaronne_Zone_6 = hydro_doc.FindObjectByName( "casGaronne_Zone_6" ) +casGaronne_Zone_6.SetMergeType( HYDROData_Zone.Merge_ZMAX ) +casGaronne_Zone_6.SetColor( QColor( 228, 145, 28 )) +casGaronne_ile.AddZone( casGaronne_Zone_6 ) + +casGaronne_litMineur.SetName("casGaronne_litMineur") +casGaronne_riveDroite.SetName("casGaronne_riveDroite") +casGaronne_riveGauche.SetName("casGaronne_riveGauche") +casGaronne_ile.SetName( "casGaronne_ile" ) + +# Export of the calculation case +casGaronne_entry = casGaronne.Export() + +#---------------------- +# --- Geometry +#---------------------- + +# Get geometry shape and print debug information +import GEOM +from salome.geom import geomBuilder +import math +import SALOMEDS +from salome.hydrotools.controls import controlGeomProps + +geompy = geomBuilder.New() + +print("Entry:", casGaronne_entry) +HYDRO_casGaronne = salome.IDToObject( str( casGaronne_entry ) ) +print("Geom shape:", HYDRO_casGaronne) +print("Geom shape name:", HYDRO_casGaronne.GetName()) + +# --- manual definition: geometrical faces + +[garonne_litMineur,garonne_riveDroite,garonne_riveGauche] = geompy.SubShapeAll(HYDRO_casGaronne, geompy.ShapeType["FACE"]) + +controlGeomProps(geompy, garonne_riveGauche, 33378.8047, 34595667.2876) +controlGeomProps(geompy, garonne_litMineur, 30337.548492, 3488480.304388) +controlGeomProps(geompy, garonne_riveDroite, 32012.343241, 25998769.23615) + +# --- manual identification of all useful edge groups (boundary conditions) + +allEdgesIds = geompy.SubShapeAllIDs(HYDRO_casGaronne, geompy.ShapeType["EDGE"]) +print("allEdgesIds", allEdgesIds) + +(isDone, ClosedFreeBoundary, OpenFreeBoundary) = geompy.GetFreeBoundary(HYDRO_casGaronne) +geompy.addToStudyInFather(HYDRO_casGaronne, ClosedFreeBoundary[0], "ClosedFreeBoundary") + +freeBoundary = geompy.ExtractShapes(ClosedFreeBoundary[0], geompy.ShapeType["EDGE"], True) +freeBoundaryIds = [ geompy.GetSubShapeID(HYDRO_casGaronne, freeBoundary[i]) for i in range(len(freeBoundary)) ] +print("freeBoundaryIds", freeBoundaryIds) + +[litMineur_droite] = geompy.GetSharedShapesMulti([garonne_riveDroite, garonne_litMineur], geompy.ShapeType["EDGE"], True) +[litMineur_gauche] = geompy.GetSharedShapesMulti([garonne_riveGauche, garonne_litMineur], geompy.ShapeType["EDGE"], True) +geompy.addToStudyInFather(HYDRO_casGaronne, litMineur_droite, "litMineur_droite") +geompy.addToStudyInFather(HYDRO_casGaronne, litMineur_gauche, "litMineur_gauche") +rives = [litMineur_droite, litMineur_gauche] +rivesIds = [ geompy.GetSubShapeID(HYDRO_casGaronne, rives[i]) for i in range(len(rives)) ] +print("rivesIds", rivesIds) + +edges_litMineur = geompy.GetSharedShapesMulti([HYDRO_casGaronne, garonne_litMineur], geompy.ShapeType["EDGE"], True) +edges_riveGauche = geompy.GetSharedShapesMulti([HYDRO_casGaronne, garonne_riveGauche], geompy.ShapeType["EDGE"], True) +edges_riveDroite = geompy.GetSharedShapesMulti([HYDRO_casGaronne, garonne_riveDroite], geompy.ShapeType["EDGE"], True) +edges_litMineurIds = [ geompy.GetSubShapeID(HYDRO_casGaronne, edges_litMineur[i]) for i in range(len(edges_litMineur)) ] +edges_riveGaucheIds = [ geompy.GetSubShapeID(HYDRO_casGaronne, edges_riveGauche[i]) for i in range(len(edges_riveGauche)) ] +edges_riveDroiteIds = [ geompy.GetSubShapeID(HYDRO_casGaronne, edges_riveDroite[i]) for i in range(len(edges_riveDroite)) ] + +print("edges_litMineurIds", edges_litMineurIds) +print("edges_riveGaucheIds", edges_riveGaucheIds) +print("edges_riveDroiteIds", edges_riveDroiteIds) + +sectionsIds = [Id for Id in edges_litMineurIds if Id not in rivesIds] +print("sectionsIds", sectionsIds) +SectionsGaronne = geompy.CreateGroup(HYDRO_casGaronne, geompy.ShapeType["EDGE"]) +geompy.UnionIDs(SectionsGaronne, sectionsIds) +geompy.addToStudyInFather(HYDRO_casGaronne, SectionsGaronne, "SectionsGaronne") + +bordGaucheDomaineIds = [Id for Id in freeBoundaryIds if Id in edges_riveGaucheIds] +bordDroiteDomaineIds = [Id for Id in freeBoundaryIds if Id in edges_riveDroiteIds] +print("bordGaucheDomaineIds", bordGaucheDomaineIds) +print("bordDroiteDomaineIds", bordDroiteDomaineIds) +bordGaucheDomaine = geompy.CreateGroup(HYDRO_casGaronne, geompy.ShapeType["EDGE"]) +geompy.UnionIDs(bordGaucheDomaine, bordGaucheDomaineIds) +geompy.addToStudyInFather(HYDRO_casGaronne, bordGaucheDomaine, "bordGaucheDomaine") +bordDroiteDomaine = geompy.CreateGroup(HYDRO_casGaronne, geompy.ShapeType["EDGE"]) +geompy.UnionIDs(bordDroiteDomaine, bordDroiteDomaineIds) +geompy.addToStudyInFather(HYDRO_casGaronne, bordDroiteDomaine, "bordDroiteDomaine") + +amont = geompy.GetEdgeNearPoint(HYDRO_casGaronne, geompy.MakeVertex(46757.861314, 25833.234752, 0)) +aval = geompy.GetEdgeNearPoint(HYDRO_casGaronne, geompy.MakeVertex(39078.979127, 32588.627279, 0)) +geompy.addToStudyInFather(HYDRO_casGaronne, amont, "amont") +geompy.addToStudyInFather(HYDRO_casGaronne, aval, "aval") + +#---------------------- +# --- Meshing +#---------------------- + +import SMESH, SALOMEDS +from salome.smesh import smeshBuilder +from salome.hydrotools.controls import controlMeshStats, controlSubMeshStats +import tempfile + +smesh = smeshBuilder.New() + +# --- algorithms and hypothesis +garonne_2 = smesh.Mesh(HYDRO_casGaronne) + +NETGEN_2D = garonne_2.Triangle(algo=smeshBuilder.NETGEN_1D2D) +NETGEN_2D_Parameters = NETGEN_2D.Parameters() +NETGEN_2D_Parameters.SetMaxSize( 200 ) +NETGEN_2D_Parameters.SetSecondOrder( 0 ) +NETGEN_2D_Parameters.SetOptimize( 1 ) +NETGEN_2D_Parameters.SetFineness( 4 ) +NETGEN_2D_Parameters.SetMinSize( 50 ) +NETGEN_2D_Parameters.SetUseSurfaceCurvature( 1 ) +NETGEN_2D_Parameters.SetFuseEdges( 1 ) +NETGEN_2D_Parameters.SetQuadAllowed( 0 ) + +algo2D_litMineur = garonne_2.Quadrangle(algo=smeshBuilder.QUAD_MA_PROJ,geom=garonne_litMineur) +algo1D_litMineur = garonne_2.Segment(geom=garonne_litMineur) +hypo1D_litMineur = algo1D_litMineur.LocalLength(100,None,1e-07) +subMesh_litMineur = algo1D_litMineur.GetSubMesh() +smesh.SetName(subMesh_litMineur, "litMineur") + +algo1D_SectionsGaronne = garonne_2.Segment(geom=SectionsGaronne) +hypo1D_SectionsGaronne = algo1D_SectionsGaronne.NumberOfSegments(8) +hypo1D_SectionsGaronne.SetDistrType( 0 ) +subMesh_SectionsGaronne = algo1D_SectionsGaronne.GetSubMesh() +smesh.SetName(subMesh_SectionsGaronne, "SectionsGaronne") + +isDone = garonne_2.SetMeshOrder( [ [ subMesh_SectionsGaronne, subMesh_litMineur ] ]) + +# --- compute mesh +isDone = garonne_2.Compute() +isDone = garonne_2.SplitQuadObject( garonne_2, 1 ) +isDone = garonne_2.ReorientObject( garonne_2 ) + +# --- geometrical groups of faces +riveGauche_1 = garonne_2.GroupOnGeom(garonne_riveGauche,'riveGauche',SMESH.FACE) +litMineur_1 = garonne_2.GroupOnGeom(garonne_litMineur,'litMineur',SMESH.FACE) +riveDroite_1 = garonne_2.GroupOnGeom(garonne_riveDroite,'riveDroite',SMESH.FACE) + +# --- geometrical groups of edges + +ClosedFreeBoundary_1 = garonne_2.GroupOnGeom(ClosedFreeBoundary[0],'ClosedFreeBoundary',SMESH.EDGE) +litMineur_droite_1 = garonne_2.GroupOnGeom(litMineur_droite,'litMineur_droite',SMESH.EDGE) +litMineur_gauche_1 = garonne_2.GroupOnGeom(litMineur_gauche,'litMineur_gauche',SMESH.EDGE) +Sectionsgaronne_2 = garonne_2.GroupOnGeom(SectionsGaronne,'SectionsGaronne',SMESH.EDGE) +bordGaucheDomaine_1 = garonne_2.GroupOnGeom(bordGaucheDomaine,'bordGaucheDomaine',SMESH.EDGE) +bordDroiteDomaine_1 = garonne_2.GroupOnGeom(bordDroiteDomaine,'bordDroiteDomaine',SMESH.EDGE) +amont_1 = garonne_2.GroupOnGeom(amont,'amont',SMESH.EDGE) +aval_1 = garonne_2.GroupOnGeom(aval,'aval',SMESH.EDGE) + +# --- geometrical groups of nodes + +garonne_2_litMineur_2 = garonne_2.GroupOnGeom(garonne_litMineur,'garonne_2_litMineur',SMESH.NODE) +garonne_2_riveDroite_2 = garonne_2.GroupOnGeom(garonne_riveDroite,'garonne_2_riveDroite',SMESH.NODE) +garonne_2_riveGauche_2 = garonne_2.GroupOnGeom(garonne_riveGauche,'garonne_2_riveGauche',SMESH.NODE) +ClosedFreeBoundary_2 = garonne_2.GroupOnGeom(ClosedFreeBoundary[0],'ClosedFreeBoundary',SMESH.NODE) +litMineur_droite_2 = garonne_2.GroupOnGeom(litMineur_droite,'litMineur_droite',SMESH.NODE) +litMineur_gauche_2 = garonne_2.GroupOnGeom(litMineur_gauche,'litMineur_gauche',SMESH.NODE) +SectionsGaronne_2 = garonne_2.GroupOnGeom(SectionsGaronne,'SectionsGaronne',SMESH.NODE) +bordGaucheDomaine_2 = garonne_2.GroupOnGeom(bordGaucheDomaine,'bordGaucheDomaine',SMESH.NODE) +bordDroiteDomaine_2 = garonne_2.GroupOnGeom(bordDroiteDomaine,'bordDroiteDomaine',SMESH.NODE) +amont_2 = garonne_2.GroupOnGeom(amont,'amont',SMESH.NODE) +aval_2 = garonne_2.GroupOnGeom(aval,'aval',SMESH.NODE) + +garonne_2.SetAutoColor( 1 ) +fichierMaillage = os.path.join(HYDRO_TEST_RESOURCES, 'garonne_2.med') +garonne_2.ExportMED(fichierMaillage, 0, SMESH.MED_V2_2, 1, None ,1) + +controlMeshStats(garonne_2, 3952, 512, 7690) +controlSubMeshStats(litMineur_1, 2384) +controlSubMeshStats(riveDroite_1, 2321) +controlSubMeshStats(riveGauche_1, 2985) + +if salome.sg.hasDesktop(): + salome.sg.updateObjBrowser() + +#---------------------- +# --- HYDRO complete case +#---------------------- + +constraint2 = importPolylines(hydro_doc, os.path.join(HYDRO_SAMPLES, "constraint2.shp"), 'constraint2_PolyXY', True, 5) +res = HYDROData_CompleteCalcCase.AddObjects(hydro_doc, casGaronne, constraint2, True) +if not res[0]: + raise ValueError("complete calculation case unsuccessful") +if res[1]: + raise ValueError("complete calculation case should not have created new regions") + +regions = casGaronne.GetRegions() +for r in regions: + print("region: %s" %r.GetName()) + zones = r.GetZones() + for z in zones: + print(" zone: %s mergeType: %s" %(z.GetName(), z.GetMergeType())) + if z.GetMergeType() == 0: + Cloud_02 = hydro_doc.FindObjectByName( "Cloud_02" ) + z.SetMergeType( HYDROData_Zone.Merge_Object ) + z.SetMergeObject( Cloud_02 ) + r.AddZone(z) +# casGaronne.Update() # Do not update! Breaks regions! + +casGaronne2_entry = casGaronne.Export() + +# TODO: test something to check the final result + -- 2.39.2