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.05"
  25
  26 #========================================================================
  27 TEST_NAME = "tutorial_1"
  28 DEBUG = False
  29 N_ITER_TEST_FILE = 3
  30 #========================================================================
  31 import os
  32 import sys
  33 #
  34 # ==================================
  35 PATH_HOMARD = os.getenv("HOMARD_ROOT_DIR")
  36 # Repertoire des scripts utilitaires
  37 REP_PYTHON = os.path.join(PATH_HOMARD, "bin", "salome", "test", "HOMARD")
  38 REP_PYTHON = os.path.normpath(REP_PYTHON)
  39 sys.path.append(REP_PYTHON)
  40 from test_util import get_dir
  41 from test_util import get_dir_tutorial
  42 from test_util import test_results
  43 # ==================================
  44 # Répertoires pour ce test
  45 REP_DATA, DIRCASE = get_dir(PATH_HOMARD, TEST_NAME, DEBUG)
  46 DATA_TUTORIAL = get_dir_tutorial(PATH_HOMARD)
  47 # ==================================
  48 sys.path.append(DATA_TUTORIAL)
  49 from tutorial_util import gzip_gunzip
  50 # ==================================
  51 gzip_gunzip(DATA_TUTORIAL, 1, -1)
  52 # ==================================
  53 #
  54 import salome
  55 salome.salome_init()
  56 import HOMARD
  57 import iparameters
  58 IPAR = iparameters.IParameters(salome.myStudy.GetCommonParameters("Interface Applicative", 1))
  59 IPAR.append("AP_MODULES_LIST", "Homard")
  60 #
  61 #
  62 #========================= Debut de la fonction ==================================
  63 #
  64 def homard_exec(nom, ficmed, verbose=False):
  65   """
  66 Python script for HOMARD
  67   """
  68   erreur = 0
  69   message = ""
  70   #
  71   while not erreur :
  72   #
  73   #  HOMARD.UpdateStudy()
  74     #
  75     # Hypotheses
  76     # ==========
  77     if verbose :
  78       print(". Hypothèses")
  79     hypo_1 = HOMARD.CreateHypothesis('hypo_1')
  80     hypo_1.SetUnifRefinUnRef(1)
  81     #
  82     # Cas
  83     # ===
  84     if verbose :
  85       print(". Cas")
  86     le_cas = HOMARD.CreateCase('case_1', nom, ficmed)
  87     le_cas.SetDirName(DIRCASE)
  88     #
  89     # Itérations
  90     # ==========
  91     if verbose :
  92       option = 2
  93     else :
  94       option = 1
  95     if verbose :
  96       print(". Itérations")
  97     # Iteration "iter_1_1"
  98     iter_1_1 = le_cas.NextIteration('iter_1_1')
  99     iter_1_1.SetMeshName('MESH')
 100     iter_1_1.SetMeshFile(os.path.join(DIRCASE, "maill.01.med"))
 101     iter_1_1.AssociateHypo('hypo_1')
 102     erreur = iter_1_1.Compute(1, option)
 103     if erreur :
 104       break
 105
 106     # Iteration "iter_1_2"
 107     iter_1_2 = iter_1_1.NextIteration('iter_1_2')
 108     iter_1_2.SetMeshName('MESH')
 109     iter_1_2.SetMeshFile(os.path.join(DIRCASE, "maill.01.med"))
 110     iter_1_2.AssociateHypo('hypo_1')
 111     erreur = iter_1_2.Compute(1, option)
 112     if erreur :
 113       break
 114
 115     # Iteration "iter_1_3"
 116     iter_1_3 = iter_1_2.NextIteration('iter_1_3')
 117     iter_1_3.SetMeshName('MESH')
 118     iter_1_3.SetMeshFile(os.path.join(DIRCASE, "maill.03.med"))
 119     iter_1_3.AssociateHypo('hypo_1')
 120     erreur = iter_1_3.Compute(1, option)
 121     if erreur :
 122       break
 123   #
 124     break
 125   #
 126   if erreur :
 127     message += "Erreur au calcul de l'itération %d" % erreur
 128   #
 129   return erreur, message
 130 #
 131 #==========================  Fin de la fonction ==================================
 132 #
 133 ERREUR = 0
 134 MESSAGE = ""
 135 while not ERREUR :
 136   #
 137   # A. Exec of HOMARD-SALOME
 138   #
 139   HOMARD = salome.lcc.FindOrLoadComponent('FactoryServer', 'HOMARD')
 140   assert HOMARD is not None, "Impossible to load homard engine"
 141   HOMARD.SetLanguageShort("fr")
 142 #
 143   FICMED = os.path.join(DATA_TUTORIAL, TEST_NAME+".00.med")
 144   try:
 145     ERREUR, MESSAGE = homard_exec("MAILL", FICMED, DEBUG)
 146   except RuntimeError as eee:
 147     ERREUR = 2
 148     MESSAGE = str(eee.message)
 149   #
 150   if ERREUR :
 151     MESSAGE += "Pb in homard_exec"
 152     break
 153   #
 154   # B. Test of the results
 155   #
 156   N_REP_TEST_FILE = N_ITER_TEST_FILE
 157   DESTROY_DIR = not DEBUG
 158   test_results(REP_DATA, TEST_NAME, DIRCASE, N_ITER_TEST_FILE, N_REP_TEST_FILE, DESTROY_DIR)
 159   #
 160   break
 161 #
 162 if ERREUR:
 163   raise Exception(MESSAGE)
 164 #
 165 # ==================================
 166 gzip_gunzip(DATA_TUTORIAL, 1, 1)
 167 # ==================================
 168 #
 169 if salome.sg.hasDesktop():
 170   salome.sg.updateObjBrowser()
 171   iparameters.getSession().restoreVisualState(1)
 172