src/tests/Test/test_6.py

   1 # -*- coding: utf-8 -*-
   2 # Copyright (C) 2011-2016  CEA/DEN, EDF R&D
   3 #
   4 # This library is free software; you can redistribute it and/or
   5 # modify it under the terms of the GNU Lesser General Public
   6 # License as published by the Free Software Foundation; either
   7 # version 2.1 of the License, or (at your option) any later version.
   8 #
   9 # This library is distributed in the hope that it will be useful,
  10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12 # Lesser General Public License for more details.
  13 #
  14 # You should have received a copy of the GNU Lesser General Public
  15 # License along with this library; if not, write to the Free Software
  16 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  17 #
  18 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  19 #
  20 """
  21 Python script for HOMARD
  22 Test test_6
  23 """
  24 __revision__ = "V1.04"
  25
  26 #========================================================================
  27 TEST_NAME = "test_6"
  28 DEBUG = False
  29 N_ITER_TEST_FILE = 3
  30 #
  31 TAILLE = 10.
  32 LG_ARETE = TAILLE*2.5
  33 #========================================================================
  34 import os
  35 import sys
  36 import HOMARD
  37 import salome
  38 #
  39 # ==================================
  40 PATH_HOMARD = os.getenv('HOMARD_ROOT_DIR')
  41 # Repertoire des scripts utilitaires
  42 REP_PYTHON = os.path.join(PATH_HOMARD, "bin", "salome", "test", "HOMARD")
  43 REP_PYTHON = os.path.normpath(REP_PYTHON)
  44 sys.path.append(REP_PYTHON)
  45 from test_util import saveGeometry
  46 from test_util import get_dir
  47 from test_util import test_results
  48 # ==================================
  49 # Répertoires pour ce test
  50 REP_DATA, DIRCASE = get_dir(PATH_HOMARD, TEST_NAME, DEBUG)
  51 # ==================================
  52
  53 salome.salome_init()
  54 import iparameters
  55 IPAR = iparameters.IParameters(salome.myStudy.GetCommonParameters("Interface Applicative", 1))
  56 IPAR.append("AP_MODULES_LIST", "Homard")
  57 #
  58 #========================= Debut de la fonction ==================================
  59 #
  60 def create_geom(nom_obj, taille, verbose=False) :
  61   """
  62 Création de la géométrie
  63   """
  64 #
  65   erreur = 0
  66 #
  67   if verbose :
  68     texte = "Geometrie '%s'\n" % nom_obj
  69     texte += "Taille de base = %f" % taille
  70     print (texte)
  71 #
  72   from salome.geom import geomBuilder
  73   geompy = geomBuilder.New()
  74 #
  75 # 1. Les sommets et la première ligne
  76 #
  77   vertex_1 = geompy.MakeVertex( 0.*taille,  0.*taille, 0.*taille, theName = "V1")
  78   vertex_2 = geompy.MakeVertex( 5.*taille,  2.*taille, 0.*taille, theName = "V2")
  79   vertex_3 = geompy.MakeVertex(10.*taille,  1.*taille, 0.*taille, theName = "V3")
  80   vertex_4 = geompy.MakeVertex(16.*taille,  4.*taille, 0.*taille, theName = "V4")
  81   vertex_5 = geompy.MakeVertex(16.*taille, 10.*taille, 0.*taille, theName = "V5")
  82 #
  83   courbe_0 = geompy.MakeInterpol([vertex_1, vertex_2, vertex_3, vertex_4, vertex_5], False, False, theName="courbe_0")
  84 #
  85 # 2. Les sommets et la seconde ligne
  86 #
  87   sommet_1 = geompy.MakeVertex( 0.*taille,  0.*taille, 20.*taille, theName = "S1")
  88   sommet_2 = geompy.MakeVertex( 6.*taille, -5.*taille, 20.*taille, theName = "S2")
  89   sommet_3 = geompy.MakeVertex(11.*taille, -2.*taille, 20.*taille, theName = "S3")
  90   sommet_4 = geompy.MakeVertex(12.*taille,  3.*taille, 20.*taille, theName = "S4")
  91   sommet_5 = geompy.MakeVertex(16.*taille, 10.*taille, 20.*taille, theName = "S5")
  92 #
  93   courbe_1 = geompy.MakeInterpol([sommet_1, sommet_2, sommet_3, sommet_4, sommet_5], False, False, theName="courbe_1")
  94 #
  95 # 3. La face de base
  96 #
  97   structure_g = geompy.MakeFilling([courbe_0, courbe_1], theName=nom_obj)
  98 #
  99 # 4. Groupes : on cherche les entites par des proximités avec des shapes bien choisies
 100 #
 101   l_groupes_g = list()
 102 #
 103   shape = geompy.GetFaceNearPoint (structure_g, vertex_2)
 104   nom = "Voile"
 105   groupe_g = geompy.CreateGroup(structure_g, geompy.ShapeType["FACE"], nom)
 106   geompy.UnionList ( groupe_g, [shape] )
 107   l_groupes_g.append( (nom, groupe_g, 2) )
 108 #
 109   shape = geompy.GetEdgeNearPoint (structure_g, vertex_2)
 110   nom = "C_0"
 111   groupe_g = geompy.CreateGroup(structure_g, geompy.ShapeType["EDGE"], nom)
 112   geompy.UnionList ( groupe_g, [shape] )
 113   l_groupes_g.append( (nom, groupe_g, 1) )
 114 #
 115   shape = geompy.GetEdgeNearPoint (structure_g, sommet_2)
 116   nom = "C_1"
 117   groupe_g = geompy.CreateGroup(structure_g, geompy.ShapeType["EDGE"], nom)
 118   geompy.UnionList ( groupe_g, [shape] )
 119   l_groupes_g.append( (nom, groupe_g, 1) )
 120 #
 121   shape = geompy.GetEdge (structure_g, vertex_1, sommet_1)
 122   nom = "D_0"
 123   groupe_g = geompy.CreateGroup(structure_g, geompy.ShapeType["EDGE"], nom)
 124   geompy.UnionList ( groupe_g, [shape] )
 125   l_groupes_g.append( (nom, groupe_g, 1) )
 126 #
 127   shape = geompy.GetEdge (structure_g, vertex_5, sommet_5)
 128   nom = "D_1"
 129   groupe_g = geompy.CreateGroup(structure_g, geompy.ShapeType["EDGE"], nom)
 130   geompy.UnionList ( groupe_g, [shape] )
 131   l_groupes_g.append( (nom, groupe_g, 1) )
 132 #
 133   return erreur, structure_g, l_groupes_g
 134 #
 135 #==========================  Fin de la fonction ==================================
 136 #
 137 #========================= Debut de la fonction ==================================
 138 #
 139 def create_mail(lg_arete, structure_g, l_groupes_g, rep_mail, verbose=False) :
 140   """
 141 Création du maillage
 142   """
 143 #
 144   erreur = 0
 145   message = ""
 146   ficmed = ""
 147 #
 148   while not erreur :
 149 #
 150     nom = structure_g.GetName()
 151     if verbose :
 152       texte = "Maillage de '%s'\n" % nom
 153       texte += "lg_arete = %f\n" % lg_arete
 154       texte += "rep_mail = '%s'" % rep_mail
 155       print (texte)
 156 #
 157     from salome.smesh import smeshBuilder
 158     smesh = smeshBuilder.New()
 159 #
 160 # 2. Maillage de calcul
 161 #
 162     maill_00 = smesh.Mesh(structure_g)
 163     smesh.SetName(maill_00.GetMesh(), nom)
 164 #
 165     MG_CADSurf = maill_00.Triangle(algo=smeshBuilder.MG_CADSurf)
 166     smesh.SetName(MG_CADSurf.GetAlgorithm(), 'MG_CADSurf')
 167 #
 168     MG_CADSurf_Parameters = MG_CADSurf.Parameters()
 169     smesh.SetName(MG_CADSurf_Parameters, 'MG_CADSurf Triangles')
 170     MG_CADSurf_Parameters.SetPhySize( lg_arete )
 171     MG_CADSurf_Parameters.SetMinSize( lg_arete/20. )
 172     MG_CADSurf_Parameters.SetMaxSize( lg_arete*5. )
 173     MG_CADSurf_Parameters.SetChordalError( lg_arete )
 174     MG_CADSurf_Parameters.SetAngleMesh( 12. )
 175 #
 176 # 3. Les groupes issus de la géométrie
 177 #
 178     for taux in l_groupes_g :
 179       groupe_m = maill_00.Group(taux[1])
 180       smesh.SetName(groupe_m, taux[0])
 181 #
 182 # 4. Calcul
 183 #
 184     isDone = maill_00.Compute()
 185     if not isDone :
 186       message += "Probleme dans le maillage de la surface."
 187       erreur = 13
 188       break
 189 #
 190 # 5. Export MED
 191 #
 192     ficmed = os.path.join(rep_mail,'maill.00.med')
 193     texte = "Ecriture du fichier '%s'" % ficmed
 194     if verbose :
 195       print (texte)
 196     try:
 197       maill_00.ExportMED(ficmed)
 198     except IOError as eee:
 199       error = 2
 200       raise Exception('ExportMED() failed. ' + str(eee))
 201 #
 202     break
 203 #
 204   return erreur, message, ficmed
 205 #
 206 #==========================  Fin de la fonction ==================================
 207 #
 208 #========================= Debut de la fonction ==================================
 209 #
 210 def homard_exec(nom, ficmed, xao_file, verbose=False):
 211   """
 212 Python script for HOMARD
 213   """
 214   erreur = 0
 215   message = ""
 216 #
 217   while not erreur :
 218     #
 219     HOMARD.UpdateStudy()
 220     #
 221     # Frontière
 222     # =========
 223     if verbose :
 224       print(". Frontière")
 225     cao_name = "CAO_" + nom
 226     la_frontiere = HOMARD.CreateBoundaryCAO(cao_name, xao_file)
 227     #
 228     # Hypotheses
 229     # ==========
 230     if verbose :
 231       print(". Hypothèses")
 232     hyponame = "hypo_" + nom
 233     l_hypothese = HOMARD.CreateHypothesis(hyponame)
 234     l_hypothese.SetUnifRefinUnRef(1)
 235     #
 236     # Cas
 237     # ===
 238     if verbose :
 239       print(". Cas")
 240     le_cas = HOMARD.CreateCase('case_'+nom, nom, ficmed)
 241     le_cas.SetDirName(DIRCASE)
 242     le_cas.AddBoundary(cao_name)
 243     #
 244     # Creation of the iterations
 245     # ==========================
 246     if verbose :
 247       option = 2
 248     else :
 249       option = 1
 250     #
 251     for niter in range(1, N_ITER_TEST_FILE+1):
 252       if verbose :
 253         print(". Itération numéro %d" % niter)
 254       iter_name = "I_" + nom + "_%02d" % niter
 255       if ( niter == 1 ) :
 256         l_iteration = le_cas.NextIteration(iter_name)
 257       else :
 258         l_iteration = l_iteration.NextIteration(iter_name)
 259       l_iteration.SetMeshName(nom)
 260       mesh_file = os.path.join(DIRCASE, "maill.%02d.med" % niter)
 261       l_iteration.SetMeshFile(mesh_file)
 262       l_iteration.AssociateHypo(hyponame)
 263       erreur = l_iteration.Compute(1, option)
 264       if erreur :
 265         erreur = niter
 266         break
 267     #
 268     break
 269   #
 270   if erreur :
 271     message += "Erreur au calcul de l'itération %d" % erreur
 272   #
 273   return erreur, message
 274 #
 275 #==========================  Fin de la fonction ==================================
 276 #
 277 #
 278 ERREUR = 0
 279 MESSAGE = ""
 280 while not ERREUR :
 281   #
 282   VERBOSE = DEBUG
 283   #
 284   # A. Geometry
 285   #
 286   ERREUR, STRUCTURE_G, L_GROUPES_G = create_geom(TEST_NAME, TAILLE, VERBOSE)
 287   if ERREUR :
 288     MESSAGE = "The construction of the geometry failed."
 289     break
 290   #
 291   # B. Save the geometry
 292   #
 293   XAO_FILE = os.path.join(DIRCASE, TEST_NAME+".xao")
 294   try :
 295     ERREUR = saveGeometry(XAO_FILE, TEST_NAME, "test_salome_"+TEST_NAME)
 296   except IOError as eee:
 297     ERREUR = os.error
 298     MESSAGE = str(eee.message)
 299   #
 300   if ERREUR :
 301     MESSAGE += "Pb in saveGeometry"
 302     break
 303   #
 304   # C. Mesh
 305   #
 306   ERREUR, MESSAGE, FICMED = create_mail(LG_ARETE, STRUCTURE_G, L_GROUPES_G, DIRCASE, VERBOSE)
 307   if ERREUR :
 308     break
 309   #
 310   # D. Exec of HOMARD-SALOME
 311   #
 312   HOMARD = salome.lcc.FindOrLoadComponent('FactoryServer', 'HOMARD')
 313   assert HOMARD is not None, "Impossible to load homard engine"
 314   HOMARD.SetLanguageShort("fr")
 315 #
 316   try:
 317     ERREUR, MESSAGE = homard_exec(TEST_NAME, FICMED, XAO_FILE, VERBOSE)
 318   except RuntimeError as eee:
 319     ERREUR = os.error
 320     MESSAGE = str(eee.message)
 321   #
 322   if ERREUR :
 323     MESSAGE += "Pb in homard_exec"
 324     break
 325   #
 326   # E. Test of the results
 327   #
 328   N_REP_TEST_FILE = N_ITER_TEST_FILE
 329   DESTROY_DIR = not DEBUG
 330   test_results(REP_DATA, TEST_NAME, DIRCASE, N_ITER_TEST_FILE, N_REP_TEST_FILE, DESTROY_DIR)
 331   #
 332   break
 333 #
 334 if ERREUR:
 335   MESSAGE = "\nErreur numéro %d\n" % ERREUR + MESSAGE
 336   raise Exception(MESSAGE)
 337 #
 338 if salome.sg.hasDesktop():
 339   salome.sg.updateObjBrowser()
 340   iparameters.getSession().restoreVisualState(1)
 341