tests/test_3.py

   1 # -*- coding: iso-8859-1 -*-
   2 # Copyright (C) 2011-2013  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.
   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 Copyright EDF-R&D 2011, 2013
  23 Test test_3
  24 """
  25 __revision__ = "V1.10"
  26
  27 #========================================================================
  28 Test_Name = "test_3"
  29 n_boucle = 2
  30 n_iter_test_file = 2
  31 #========================================================================
  32 import os
  33 import tempfile
  34 import HOMARD
  35 import salome
  36 #
  37 pathHomard = os.getenv('HOMARD_ROOT_DIR')
  38 Rep_Test = os.path.join(pathHomard, "share", "salome", "resources", "homard")
  39 Rep_Test = os.path.normpath(Rep_Test)
  40 Rep_Test_Resu = tempfile.mktemp()
  41 os.mkdir(Rep_Test_Resu)
  42
  43 salome.salome_init()
  44 import iparameters
  45 ipar = iparameters.IParameters(salome.myStudy.GetCommonParameters("Interface Applicative", 1))
  46 ipar.append("AP_MODULES_LIST", "Homard")
  47 #========================================================================
  48 #========================================================================
  49 def remove_dir(directory) :
  50   """
  51 Empties, then removes a directory.
  52 Copyright EDF-R&D 2013
  53   """
  54 #
  55   l_aux = os.listdir(directory)
  56   for fic in l_aux :
  57     fic_a = os.path.join(directory, fic)
  58     if os.path.isdir(fic_a) :
  59       remove_dir(fic_a)
  60     else :
  61       os.remove(fic_a)
  62   os.rmdir(directory)
  63 #
  64   return
  65 #
  66 #========================================================================
  67 #========================================================================
  68 def homard_exec(theStudy):
  69   """
  70 Python script for HOMARD
  71 Copyright EDF-R&D 2010, 2013
  72   """
  73   error = 0
  74 #
  75   while not error :
  76 #
  77     homard.SetCurrentStudy(theStudy)
  78 #
  79 # Creation of the boundaries
  80 # ==========================
  81 # Creation of the discrete boundary
  82     Boundary_3_1 = homard.CreateBoundaryDi('courbes', 'COURBES', os.path.join(Rep_Test, Test_Name + '.fr.med'))
  83 #
  84 # Creation of the external cylinder
  85     Boundary_3_2 = homard.CreateBoundaryCylinder('cyl_ext', 50.0, 25., -25., 1., 0., 0., 100.)
  86 #
  87 # Creation of the internal cylinder
  88     Boundary_3_3 = homard.CreateBoundaryCylinder('cyl_int', 50.0, 25., -25., 1., 0., 0., 50.)
  89 #
  90 # Creation of the first sphere
  91     Boundary_3_4 = homard.CreateBoundarySphere('sphere_1', 50.0, 25., -25., 100.)
  92 #
  93 # Creation of the second sphere
  94     Boundary_3_5 = homard.CreateBoundarySphere('sphere_2', 450.0, 25., -25., 100.)
  95 #
  96 # Creation of the hypotheses
  97 # ==========================
  98 # Uniform refinement
  99     HypoName = "Hypo_3"
 100     print "-------- Creation of the hypothesis", HypoName
 101     Hypo_3 = homard.CreateHypothesis(HypoName)
 102     Hypo_3.SetAdapRefinUnRef(-1, 1, 0)
 103 #
 104     for num in range (n_boucle+1) :
 105 #
 106       print "-------- num =", num, "--------"
 107 #
 108 # Creation of the case Case_3
 109 # ===========================
 110       if ( num <= 1 ) :
 111         CaseName = "Case_3"
 112         print "-------- Creation of the case", CaseName
 113         MeshFile = os.path.join(Rep_Test, Test_Name + '.00.med')
 114         Case_3 = homard.CreateCase(CaseName, 'MOYEU', MeshFile)
 115         Case_3.SetDirName(Rep_Test_Resu)
 116         Case_3.SetConfType(1)
 117         Case_3.AddBoundaryGroup('courbes', '')
 118         Case_3.AddBoundaryGroup('cyl_ext', 'EXT')
 119         Case_3.AddBoundaryGroup('cyl_int', 'INT')
 120         Case_3.AddBoundaryGroup('sphere_1', 'END_1')
 121         Case_3.AddBoundaryGroup('sphere_2', 'END_2')
 122 #
 123 # Creation of the iterations
 124 # ==========================
 125   # Creation of the iteration Iter_3_1
 126       IterName = "Iter_3_1"
 127       print "-------- Creation of the iteration", IterName
 128       Iter_3_1 = Case_3.NextIteration(IterName)
 129       Iter_3_1.SetMeshName('MOYEU_1')
 130       Iter_3_1.SetMeshFile(os.path.join(Rep_Test_Resu, 'maill.01.med'))
 131       Iter_3_1.AssociateHypo('Hypo_3')
 132       error = Iter_3_1.Compute(1, 1)
 133       if error :
 134         error = 10*num + 1
 135         break
 136
 137   # Creation of the iteration Iter_3_2Iter_3_1
 138       IterName = "Iter_3_2"
 139       print "-------- Creation of the iteration", IterName
 140       Iter_3_2 = Iter_3_1.NextIteration(IterName)
 141       Iter_3_2.SetMeshName('MOYEU_2')
 142       Iter_3_2.SetMeshFile(os.path.join(Rep_Test_Resu, 'maill.02.med'))
 143       Iter_3_2.AssociateHypo('Hypo_3')
 144       error = Iter_3_2.Compute(1, 1)
 145       if error :
 146         error = 10*num + 2
 147         break
 148   #
 149   # Creation of the schema YACS
 150   # ===========================
 151       ScriptFile = os.path.join(pathHomard, "share", "doc", "salome", "gui", "HOMARD", "en", "_downloads", "yacs_script.py")
 152       ScriptFile = os.path.normpath(ScriptFile)
 153       DirName = Rep_Test_Resu
 154       YACSName = "YACS_3"
 155       print "-------- Creation of the schema", YACSName
 156       YACS_3 = Case_3.CreateYACSSchema(YACSName, ScriptFile, DirName, MeshFile)
 157       YACS_3.SetType(2)
 158       error = YACS_3.Write()
 159       if error :
 160         error = 10*num + 5
 161         break
 162
 163   # Destructions
 164   # ============
 165   # Destruction of the schema, sauf a la fin
 166       if ( num < n_boucle ) :
 167         print "-------- Destruction of the schema", YACS_3.GetName()
 168         error = YACS_3.Delete(1)
 169         if error :
 170           error = 10*num + 6
 171           break
 172   # After the first loop, the case is deleted, except the final mesh files
 173   # All the iterations are deleted
 174       if ( num == 0 ) :
 175         print "-------- Destruction of the case", Case_3.GetName()
 176         error = Case_3.Delete(0)
 177         if error :
 178           break
 179   # After the second loop, the iterations are deleted, with the final mesh files
 180       elif ( num == 1 ) :
 181   # Recursive destruction of the iterations
 182         print "-------- Recursive destruction of the iteration", Iter_3_1.GetName()
 183         error = Iter_3_1.Delete(1)
 184         if error :
 185           error = 10*num + 3
 186           break
 187   # Destruction and creation of the hypothese
 188         if ( num == 1 ) :
 189           print "-------- Destruction of the hypothese", Hypo_3.GetName()
 190           error = Hypo_3.Delete()
 191           if error :
 192             error = 10*num + 4
 193             break
 194           HypoName = "Hypo_3"
 195           print "-------- Creation of the hypothesis", HypoName
 196           Hypo_3 = homard.CreateHypothesis(HypoName)
 197           Hypo_3.SetAdapRefinUnRef(-1, 1, 0)
 198 #
 199     break
 200 #
 201   return error
 202
 203 #========================================================================
 204
 205 homard = salome.lcc.FindOrLoadComponent('FactoryServer', 'HOMARD')
 206 assert homard is not None, "Impossible to load homard engine"
 207 #
 208 # Exec of HOMARD-SALOME
 209 #
 210 try :
 211   error_main = homard_exec(salome.myStudy)
 212   if error_main :
 213     raise Exception('Pb in homard_exec at iteration %d' %error_main )
 214 except Exception, e:
 215   raise Exception('Pb in homard_exec: '+e.message)
 216 #
 217 # Test of the result
 218 #
 219 test_file_suff = "apad.%02d.bilan" % n_iter_test_file
 220 rep_test_file = "I%02d" % (n_iter_test_file*n_boucle)
 221 #
 222 test_file = os.path.join(Rep_Test, Test_Name + "." + test_file_suff)
 223 mess_error_ref = "\nReference file: " + test_file
 224 try :
 225   file = open (test_file, "r")
 226   mess_ref = file.readlines()
 227   file.close()
 228 except :
 229   mess_error = mess_error_ref + "\nThis file does not exist.\n"
 230   raise Exception(mess_error)
 231 #
 232 test_file = os.path.join(Rep_Test_Resu, rep_test_file, test_file_suff)
 233 if os.path.isfile (test_file) :
 234   file = open (test_file, "r")
 235   mess = file.readlines()
 236   file.close()
 237 else :
 238   mess_error  = "\nResult file: " + test_file
 239   mess_error += "\nThis file does not exist.\n"
 240   raise Exception(mess_error)
 241
 242 nblign = len(mess_ref)
 243 if ( len(mess) != nblign ):
 244   mess_error = mess_error_ref +  "\nResult file: " + test_file
 245   mess_error += "\nThe number of lines of the files are not the same.\n"
 246   raise Exception(mess_error)
 247
 248 for num in range(nblign) :
 249   if (( "creation" not in mess_ref[num] ) and ( mess_ref[num] != mess[num])) :
 250     message_erreur = "\nRefe : " + mess_ref[num]
 251     message_erreur += "Test : " + mess[num][:-1]
 252     message_erreur += "\nThe test is different from the reference."
 253     raise Exception(message_erreur)
 254 #
 255 remove_dir(Rep_Test_Resu)
 256 #
 257 if salome.sg.hasDesktop():
 258   salome.sg.updateObjBrowser(1)
 259   iparameters.getSession().restoreVisualState(1)
 260