src/tests/Test/tutorial_2.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 tutorial_2 associe au tutorial 2
  23 """
  24 __revision__ = "V4.02"
  25
  26 #========================================================================
  27 TEST_NAME = "tutorial_2"
  28 DEBUG = False
  29 N_ITER_TEST_FILE = 2
  30 #========================================================================
  31 import os
  32 import tempfile
  33 import sys
  34 import HOMARD
  35 import salome
  36 #
  37 # ==================================
  38 PATH_HOMARD = os.getenv('HOMARD_ROOT_DIR')
  39 # Repertoire des scripts utilitaires
  40 REP_PYTHON = os.path.join(PATH_HOMARD, "bin", "salome", "test", "HOMARD")
  41 REP_PYTHON = os.path.normpath(REP_PYTHON)
  42 sys.path.append(REP_PYTHON)
  43 from test_util import remove_dir
  44 from test_util import test_results
  45 # Repertoire des donnees du test
  46 REP_DATA = os.path.join(PATH_HOMARD, "share", "salome", "homardsamples")
  47 REP_DATA = os.path.normpath(REP_DATA)
  48 # Repertoire des resultats
  49 if DEBUG :
  50   DIRCASE = os.path.join("/tmp", TEST_NAME)
  51   if ( os.path.isdir(DIRCASE) ) :
  52     remove_dir(DIRCASE)
  53   os.mkdir(DIRCASE)
  54 else :
  55   DIRCASE = tempfile.mkdtemp(prefix=TEST_NAME)
  56 # Repertoire des donnees du tutorial
  57 DATA_TUTORIAL = os.path.join(PATH_HOMARD, "share", "doc", "salome", "gui", "HOMARD", "fr", "_downloads")
  58 DATA_TUTORIAL = os.path.normpath(DATA_TUTORIAL)
  59 sys.path.append(DATA_TUTORIAL)
  60 from tutorial_util import gzip_gunzip
  61 # ==================================
  62 gzip_gunzip(DATA_TUTORIAL, 2, -1)
  63 # ==================================
  64
  65 salome.salome_init()
  66 import iparameters
  67 IPAR = iparameters.IParameters(salome.myStudy.GetCommonParameters("Interface Applicative", 1))
  68 IPAR.append("AP_MODULES_LIST", "Homard")
  69 #
  70 #
  71 #========================= Debut de la fonction ==================================
  72 #
  73 def homard_exec(nom, ficmed, verbose=False):
  74   """
  75 Python script for HOMARD
  76   """
  77   erreur = 0
  78   message = ""
  79   #
  80   while not erreur :
  81     #
  82     HOMARD.SetCurrentStudy(salome.myStudy)
  83     #
  84     # Creation des zones
  85     # ==================
  86     if verbose :
  87       print(". Zones")
  88     # Box "Zone_12_0"
  89     zone_12_0 = HOMARD.CreateZoneBox ('Zone_12_0', -0.1, 1.1, -0.1, 1.1, 0.9, 1.1)
  90     #
  91     # Sphere "Zone_12_1"
  92     zone_12_1 = HOMARD.CreateZoneSphere ('Zone_12_1', 0., 0., 0., 1.05)
  93     #
  94     # Box "Zone_12_2"
  95     zone_12_2 = HOMARD.CreateZoneBox ('Zone_12_2', -0.1, 0.51, -0.1, 0.51, -0.1, 0.51)
  96     #
  97     # Hypotheses
  98     # ==========
  99     if verbose :
 100       print(". Hypothèses")
 101     # Hypothese "hypo_2"
 102     # ==================
 103     hypo_2 = HOMARD.CreateHypothesis('hypo_2')
 104     hypo_2.AddZone('Zone_12_1', 1)
 105     hypo_2.AddZone('Zone_12_0', 1)
 106     #
 107     # Hypothese "hypo_2_bis"
 108     # ======================
 109     hypo_2_bis = HOMARD.CreateHypothesis('hypo_2_bis')
 110     hypo_2_bis.AddZone('Zone_12_0', -1)
 111     hypo_2_bis.AddZone('Zone_12_2', 1)
 112     #
 113     # Cas
 114     # ===
 115     if verbose :
 116       print(". Cas")
 117     le_cas = HOMARD.CreateCase('case_2', nom, ficmed)
 118     le_cas.SetDirName(DIRCASE)
 119     #
 120     # Itérations
 121     # ==========
 122     if verbose :
 123       option = 2
 124     else :
 125       option = 1
 126     if verbose :
 127       print(". Itérations")
 128     #
 129     # Iteration "iter_2_1"
 130     # ====================
 131     iter_2_1 = le_cas.NextIteration('iter_2_1')
 132     iter_2_1.SetMeshName('M_1')
 133     iter_2_1.SetMeshFile(os.path.join(DIRCASE, "maill.01.med"))
 134     iter_2_1.AssociateHypo('hypo_2')
 135     erreur = iter_2_1.Compute(1, option)
 136     if erreur :
 137       break
 138     #
 139     # Iteration "iter_2_2"
 140     # ====================
 141     iter_2_2 = iter_2_1.NextIteration('iter_2_2')
 142     iter_2_2.SetMeshName('M_2')
 143     iter_2_2.SetMeshFile(os.path.join(DIRCASE, "maill.01.med"))
 144     iter_2_2.AssociateHypo('hypo_2_bis')
 145     erreur = iter_2_2.Compute(1, option)
 146     if erreur :
 147       break
 148   #
 149     break
 150   #
 151   if erreur :
 152     message += "Erreur au calcul de l'itération %d" % erreur
 153   #
 154   return erreur, message
 155 #
 156 #==========================  Fin de la fonction ==================================
 157 #
 158 ERREUR = 0
 159 MESSAGE = ""
 160 while not ERREUR :
 161   #
 162   # A. Exec of HOMARD-SALOME
 163   #
 164   HOMARD = salome.lcc.FindOrLoadComponent('FactoryServer', 'HOMARD')
 165   assert HOMARD is not None, "Impossible to load homard engine"
 166   HOMARD.SetLanguageShort("fr")
 167 #
 168   FICMED = os.path.join(DATA_TUTORIAL, TEST_NAME+".00.med")
 169   try:
 170     ERREUR, MESSAGE = homard_exec("MZERO", FICMED, DEBUG)
 171   except Exception, eee:
 172     ERREUR = 2
 173     MESSAGE = eee.message
 174   #
 175   if ERREUR :
 176     MESSAGE += "Pb in homard_exec"
 177     break
 178   #
 179   # B. Test of the results
 180   #
 181   N_REP_TEST_FILE = N_ITER_TEST_FILE
 182   DESTROY_DIR = not DEBUG
 183   test_results(REP_DATA, TEST_NAME, DIRCASE, N_ITER_TEST_FILE, N_REP_TEST_FILE, DESTROY_DIR)
 184   #
 185   break
 186 #
 187 if ERREUR:
 188   raise Exception(MESSAGE)
 189 #
 190 # ==================================
 191 gzip_gunzip(DATA_TUTORIAL, 2, 1)
 192 # ==================================
 193 #
 194 if salome.sg.hasDesktop():
 195   salome.sg.updateObjBrowser(True)
 196   iparameters.getSession().restoreVisualState(1)
 197