src/tests/Test/tutorial_1.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_1 associe au tutorial 1
  23 """
  24 __revision__ = "V4.02"
  25
  26 #========================================================================
  27 TEST_NAME = "tutorial_1"
  28 DEBUG = False
  29 N_ITER_TEST_FILE = 3
  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, 1, -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     # Hypotheses
  85     # ==========
  86     if verbose :
  87       print(". Hypothèses")
  88     hypo_1 = HOMARD.CreateHypothesis('hypo_1')
  89     hypo_1.SetUnifRefinUnRef(1)
  90     #
  91     # Cas
  92     # ===
  93     if verbose :
  94       print(". Cas")
  95     le_cas = HOMARD.CreateCase('case_1', nom, ficmed)
  96     le_cas.SetDirName(DIRCASE)
  97     #
  98     # Itérations
  99     # ==========
 100     if verbose :
 101       option = 2
 102     else :
 103       option = 1
 104     if verbose :
 105       print(". Itérations")
 106     # Iteration "iter_1_1"
 107     iter_1_1 = le_cas.NextIteration('iter_1_1')
 108     iter_1_1.SetMeshName('MESH')
 109     iter_1_1.SetMeshFile(os.path.join(DIRCASE, "maill.01.med"))
 110     iter_1_1.AssociateHypo('hypo_1')
 111     erreur = iter_1_1.Compute(1, option)
 112     if erreur :
 113       break
 114
 115     # Iteration "iter_1_2"
 116     iter_1_2 = iter_1_1.NextIteration('iter_1_2')
 117     iter_1_2.SetMeshName('MESH')
 118     iter_1_2.SetMeshFile(os.path.join(DIRCASE, "maill.01.med"))
 119     iter_1_2.AssociateHypo('hypo_1')
 120     erreur = iter_1_2.Compute(1, option)
 121     if erreur :
 122       break
 123
 124     # Iteration "iter_1_3"
 125     iter_1_3 = iter_1_2.NextIteration('iter_1_3')
 126     iter_1_3.SetMeshName('MESH')
 127     iter_1_3.SetMeshFile(os.path.join(DIRCASE, "maill.03.med"))
 128     iter_1_3.AssociateHypo('hypo_1')
 129     erreur = iter_1_3.Compute(1, option)
 130     if erreur :
 131       break
 132   #
 133     break
 134   #
 135   if erreur :
 136     message += "Erreur au calcul de l'itération %d" % erreur
 137   #
 138   return erreur, message
 139 #
 140 #==========================  Fin de la fonction ==================================
 141 #
 142 ERREUR = 0
 143 MESSAGE = ""
 144 while not ERREUR :
 145   #
 146   # A. Exec of HOMARD-SALOME
 147   #
 148   HOMARD = salome.lcc.FindOrLoadComponent('FactoryServer', 'HOMARD')
 149   assert HOMARD is not None, "Impossible to load homard engine"
 150   HOMARD.SetLanguageShort("fr")
 151 #
 152   FICMED = os.path.join(DATA_TUTORIAL, TEST_NAME+".00.med")
 153   try:
 154     ERREUR, MESSAGE = homard_exec("MAILL", FICMED, DEBUG)
 155   except Exception, eee:
 156     ERREUR = 2
 157     MESSAGE = eee.message
 158   #
 159   if ERREUR :
 160     MESSAGE += "Pb in homard_exec"
 161     break
 162   #
 163   # B. Test of the results
 164   #
 165   N_REP_TEST_FILE = N_ITER_TEST_FILE
 166   DESTROY_DIR = not DEBUG
 167   test_results(REP_DATA, TEST_NAME, DIRCASE, N_ITER_TEST_FILE, N_REP_TEST_FILE, DESTROY_DIR)
 168   #
 169   break
 170 #
 171 if ERREUR:
 172   raise Exception(MESSAGE)
 173 #
 174 # ==================================
 175 gzip_gunzip(DATA_TUTORIAL, 1, 1)
 176 # ==================================
 177 #
 178 if salome.sg.hasDesktop():
 179   salome.sg.updateObjBrowser(True)
 180   iparameters.getSession().restoreVisualState(1)
 181