src/tests/Test/tutorial_3.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_3 associe au tutorial 3
  23 """
  24 __revision__ = "V4.04"
  25
  26 #========================================================================
  27 TEST_NAME = "tutorial_3"
  28 DEBUG = False
  29 N_ITER_TEST_FILE = 2
  30 #========================================================================
  31 import os
  32 import sys
  33 import HOMARD
  34 import salome
  35 #
  36 # ==================================
  37 PATH_HOMARD = os.getenv('HOMARD_ROOT_DIR')
  38 # Repertoire des scripts utilitaires
  39 REP_PYTHON = os.path.join(PATH_HOMARD, "bin", "salome", "test", "HOMARD")
  40 REP_PYTHON = os.path.normpath(REP_PYTHON)
  41 sys.path.append(REP_PYTHON)
  42 from test_util import get_dir
  43 from test_util import get_dir_tutorial
  44 from test_util import test_results
  45 # ==================================
  46 # Répertoires pour ce test
  47 REP_DATA, DIRCASE = get_dir(PATH_HOMARD, TEST_NAME, DEBUG)
  48 DATA_TUTORIAL = get_dir_tutorial(PATH_HOMARD)
  49 # ==================================
  50 sys.path.append(DATA_TUTORIAL)
  51 from tutorial_util import gzip_gunzip
  52 # ==================================
  53 gzip_gunzip(DATA_TUTORIAL, 3, -1)
  54 # ==================================
  55
  56 salome.salome_init()
  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.SetCurrentStudy(salome.myStudy)
  74     #
  75     # Hypotheses
  76     # ==========
  77     if verbose :
  78       print(". Hypothèses")
  79     # Hypothese "hypo_0vers1"
  80     # =======================
  81     hypo_0vers1 = HOMARD.CreateHypothesis('hypo_0vers1')
  82     # Characterization of the field
  83     hypo_0vers1.SetField('SOLU_0__QIRE_ELEM_SIGM__________')
  84     hypo_0vers1.SetUseComp(0)
  85     hypo_0vers1.AddComp('ERREST          ')
  86     hypo_0vers1.SetRefinThr(3, 1.0)
  87     hypo_0vers1.SetTypeFieldInterp(2)
  88     hypo_0vers1.AddFieldInterp('SOLU_0__DEPL____________________')
  89     hypo_0vers1.AddFieldInterp('SOLU_0__ERRE_ELEM_SIGM__________')
  90     #
  91     # Hypothese "hypo_1vers2"
  92     # =======================
  93     hypo_1vers2 = HOMARD.CreateHypothesis('hypo_1vers2')
  94     # Characterization of the field
  95     hypo_1vers2.SetField('SOLU_1__QIRE_ELEM_SIGM__________')
  96     hypo_1vers2.SetUseComp(0)
  97     hypo_1vers2.AddComp('ERREST          ')
  98     hypo_1vers2.SetRefinThr(3, 1.5)
  99     hypo_1vers2.SetUnRefThr(3, 6.)
 100     hypo_1vers2.SetTypeFieldInterp(2)
 101     hypo_1vers2.AddFieldInterp('SOLU_1__DEPL____________________')
 102     hypo_1vers2.AddFieldInterp('SOLU_1__QIRE_ELEM_SIGM__________')
 103     #
 104     # Hypothese "hypo_1vers2_bis"
 105     # ===========================
 106     hypo_1vers2_bis = HOMARD.CreateHypothesis('hypo_1vers2_bis')
 107     # Characterization of the field
 108     hypo_1vers2_bis.SetField('SOLU_1__DEPL____________________')
 109     hypo_1vers2_bis.SetUseComp(1)
 110     hypo_1vers2_bis.AddComp('DX')
 111     hypo_1vers2_bis.AddComp('DY')
 112     hypo_1vers2_bis.AddComp('DZ')
 113     hypo_1vers2_bis.SetRefinThr(1, 0.0001)
 114     hypo_1vers2_bis.SetUnRefThr(1, 0.000001)
 115     hypo_1vers2_bis.SetTypeFieldInterp(0)
 116     #
 117     # Cas
 118     # ===
 119     if verbose :
 120       print(". Cas")
 121     le_cas = HOMARD.CreateCase('case_3', nom, ficmed)
 122     le_cas.SetDirName(DIRCASE)
 123     #
 124     # Itérations
 125     # ==========
 126     if verbose :
 127       option = 2
 128     else :
 129       option = 1
 130     if verbose :
 131       print(". Itérations")
 132     # Iteration "iter_3_1"
 133     # ====================
 134     iter_3_1 = le_cas.NextIteration('iter_3_1')
 135     iter_3_1.SetMeshName('H_1')
 136     iter_3_1.SetMeshFile(os.path.join(DIRCASE, "maill.01.med"))
 137     iter_3_1.SetFieldFile(os.path.join(DATA_TUTORIAL, "tutorial_3.00.med"))
 138     iter_3_1.SetTimeStepRank( 1, 1)
 139     iter_3_1.AssociateHypo('hypo_0vers1')
 140     erreur = iter_3_1.Compute(1, option)
 141     if erreur :
 142       break
 143     #
 144     # Iteration "iter_3_2"
 145     # ====================
 146     iter_3_2 = iter_3_1.NextIteration('iter_3_2')
 147     iter_3_2.SetMeshName('H_2')
 148     iter_3_2.SetMeshFile(os.path.join(DIRCASE, "maill.02.med"))
 149     iter_3_2.SetFieldFile(os.path.join(DATA_TUTORIAL, "tutorial_3.01.med"))
 150     iter_3_2.SetTimeStepRank(1, 1)
 151     iter_3_2.AssociateHypo('hypo_1vers2')
 152     erreur = iter_3_2.Compute(1, option)
 153     if erreur :
 154       break
 155     #
 156     # Iteration "iter_3_2_bis"
 157     # ========================
 158     iter_3_2_bis = iter_3_1.NextIteration('iter_3_2_bis')
 159     iter_3_2_bis.SetMeshName('H_2_bis')
 160     iter_3_2_bis.SetMeshFile(os.path.join(DIRCASE, "maill.02.bis.med"))
 161     iter_3_2_bis.SetFieldFile(os.path.join(DATA_TUTORIAL, "tutorial_3.01.med"))
 162     iter_3_2_bis.SetTimeStepRank(1, 1)
 163     iter_3_2_bis.AssociateHypo('hypo_1vers2_bis')
 164     erreur = iter_3_2_bis.Compute(1, option)
 165     if erreur :
 166       break
 167   #
 168     break
 169   #
 170   if erreur :
 171     message += "Erreur au calcul de l'itération %d" % erreur
 172   #
 173   return erreur, message
 174 #
 175 #==========================  Fin de la fonction ==================================
 176 #
 177 ERREUR = 0
 178 MESSAGE = ""
 179 while not ERREUR :
 180   #
 181   # A. Exec of HOMARD-SALOME
 182   #
 183   HOMARD = salome.lcc.FindOrLoadComponent('FactoryServer', 'HOMARD')
 184   assert HOMARD is not None, "Impossible to load homard engine"
 185   HOMARD.SetLanguageShort("fr")
 186 #
 187   FICMED = os.path.join(DATA_TUTORIAL, TEST_NAME+".00.med")
 188   try:
 189     ERREUR, MESSAGE = homard_exec("G_0", FICMED, DEBUG)
 190   except RuntimeError as eee:
 191     ERREUR = 2
 192     MESSAGE = str(eee.message)
 193   #
 194   if ERREUR :
 195     MESSAGE += "Pb in homard_exec"
 196     break
 197   #
 198   # B. Test of the results
 199   #
 200   N_REP_TEST_FILE = 3
 201   DESTROY_DIR = not DEBUG
 202   test_results(REP_DATA, TEST_NAME, DIRCASE, N_ITER_TEST_FILE, N_REP_TEST_FILE, DESTROY_DIR)
 203   #
 204   break
 205 #
 206 if ERREUR:
 207   raise Exception(MESSAGE)
 208 #
 209 # ==================================
 210 gzip_gunzip(DATA_TUTORIAL, 3, 1)
 211 # ==================================
 212 #
 213 if salome.sg.hasDesktop():
 214   salome.sg.updateObjBrowser(True)
 215   iparameters.getSession().restoreVisualState(1)
 216