Salome HOME
SetCurrentStudy --> UpdateStudy
[modules/homard.git] / src / tests / Test / tutorial_6.py
1 # -*- coding: utf-8 -*-
2
3 # Copyright (C) 2011-2016  CEA/DEN, EDF R&D
4 #
5 # This library is free software; you can redistribute it and/or
6 # modify it under the terms of the GNU Lesser General Public
7 # License as published by the Free Software Foundation; either
8 # version 2.1 of the License, or (at your option) any later version.
9 #
10 # This library is distributed in the hope that it will be useful,
11 # but WITHOUT ANY WARRANTY; without even the implied warranty of
12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13 # Lesser General Public License for more details.
14 #
15 # You should have received a copy of the GNU Lesser General Public
16 # License along with this library; if not, write to the Free Software
17 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
18 #
19 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
20 #
21
22 """
23 Python script for HOMARD
24 Test tutorial_6 associe au tutorial 6
25 """
26 __revision__ = "V4.05"
27
28 #========================================================================
29 TEST_NAME = "tutorial_6"
30 DEBUG = False
31 N_ITER_TEST_FILE = 3
32 #========================================================================
33 import os
34 import sys
35 import HOMARD
36 import salome
37 #
38 # ==================================
39 PATH_HOMARD = os.getenv('HOMARD_ROOT_DIR')
40 # Repertoire des scripts utilitaires
41 REP_PYTHON = os.path.join(PATH_HOMARD, "bin", "salome", "test", "HOMARD")
42 REP_PYTHON = os.path.normpath(REP_PYTHON)
43 sys.path.append(REP_PYTHON)
44 from test_util import get_dir
45 from test_util import get_dir_tutorial
46 from test_util import test_results
47 # ==================================
48 # Répertoires pour ce test
49 REP_DATA, DIRCASE = get_dir(PATH_HOMARD, TEST_NAME, DEBUG)
50 DATA_TUTORIAL = get_dir_tutorial(PATH_HOMARD)
51 # ==================================
52 sys.path.append(DATA_TUTORIAL)
53 from tutorial_util import gzip_gunzip
54 # ==================================
55 gzip_gunzip(DATA_TUTORIAL, 4, -1)
56 gzip_gunzip(DATA_TUTORIAL, 6, -1)
57 # ==================================
58 #
59 import salome
60 salome.salome_init()
61 import HOMARD
62 #
63 import iparameters
64 IPAR = iparameters.IParameters(salome.myStudy.GetCommonParameters("Interface Applicative", 1))
65 IPAR.append("AP_MODULES_LIST", "Homard")
66 #
67 #
68 #========================= Debut de la fonction ==================================
69 #
70 def homard_exec(nom, ficmed, nomfr, ficfrmed, verbose=False):
71   """
72 Python script for HOMARD
73   """
74   erreur = 0
75   message = ""
76 #
77   while not erreur :
78     #
79   #  HOMARD.UpdateStudy()
80     #
81     # Frontières
82     # ==========
83     if verbose :
84       print(". Frontières")
85     boun_6_1 = HOMARD.CreateBoundaryDi('intersection', nomfr, ficfrmed)
86     #
87     boun_6_2 = HOMARD.CreateBoundaryCylinder('cyl_1_ext', 0.0, 25., -25., 25., 50., 75., 100.)
88     #
89     boun_6_3 = HOMARD.CreateBoundaryCylinder('cyl_2_ext', 17.5, -2.5, -12.5, -100., -75., -25., 50.)
90     #
91     boun_6_6 = HOMARD.CreateBoundaryCylinder('cyl_1_int', 0.0, 25., -25., 25., 50., 75., 75.)
92     #
93     boun_6_5 = HOMARD.CreateBoundaryCylinder('cyl_2_int', 17.5, -2.5, -12.5, -100., -75., -25., 25.)
94     #
95     # Hypotheses
96     # ==========
97     if verbose :
98       print(". Hypothèses")
99     # Creation of the hypothesis hypo_0_1
100     l_hypothese_0_1 = HOMARD.CreateHypothesis('hypo_6_0_1')
101     l_hypothese_0_1.SetUnifRefinUnRef(1)
102     l_hypothese_0_1.AddGroup('IN1')
103     l_hypothese_0_1.AddGroup('IN2')
104     l_hypothese_0_1.AddGroup('T1_INT_I')
105     l_hypothese_0_1.AddGroup('T1_INT_O')
106     l_hypothese_0_1.AddGroup('T2_INT')
107     # Creation of the hypothesis hypo_1_2
108     l_hypothese_1_2 = HOMARD.CreateHypothesis('hypo_6_1_2')
109     l_hypothese_1_2.SetUnifRefinUnRef(1)
110     l_hypothese_1_2.AddGroup('T1_EXT_I')
111     l_hypothese_1_2.AddGroup('T1_EXT_O')
112     l_hypothese_1_2.AddGroup('T2_EXT')
113     # Creation of the hypothesis hypo_2_3
114     l_hypothese_2_3 = HOMARD.CreateHypothesis('hypo_6_2_3')
115     l_hypothese_2_3.SetUnifRefinUnRef(1)
116     l_hypothese_2_3.AddGroup('INT_I')
117     l_hypothese_2_3.AddGroup('INT_E')
118     l_hypothese_2_3.AddGroup('IN1')
119     l_hypothese_2_3.AddGroup('IN2')
120     #
121     # Cas
122     # ===
123     if verbose :
124       print(". Cas")
125     le_cas = HOMARD.CreateCase('case_'+nom, nom, ficmed)
126     le_cas.SetDirName(DIRCASE)
127     le_cas.AddBoundary( 'intersection' )
128     le_cas.AddBoundaryGroup( 'cyl_1_int', 'T1_INT_I' )
129     le_cas.AddBoundaryGroup( 'cyl_1_ext', 'T1_EXT_I' )
130     le_cas.AddBoundaryGroup( 'cyl_1_int', 'T1_INT_O' )
131     le_cas.AddBoundaryGroup( 'cyl_1_ext', 'T1_EXT_O' )
132     le_cas.AddBoundaryGroup( 'cyl_2_int', 'T2_INT' )
133     le_cas.AddBoundaryGroup( 'cyl_2_ext', 'T2_EXT' )
134     #
135     # Itérations
136     # ==========
137     if verbose :
138       option = 2
139     else :
140       option = 1
141     if verbose :
142       print(". Itérations")
143     # Iteration iter_6_1 : raffinement selon les faces internes
144     iter_6_1 = le_cas.NextIteration('iter_6_1')
145     iter_6_1.SetMeshName('PIQUAGE_1')
146     iter_6_1.SetMeshFile(os.path.join(DIRCASE, "maill.01.med"))
147     iter_6_1.AssociateHypo('hypo_6_0_1')
148     erreur = iter_6_1.Compute(1, option)
149     print ("erreur = %d" % erreur)
150     if erreur :
151       break
152     # Iteration iter_6_2 : raffinement selon les faces externes
153     iter_6_2 = iter_6_1.NextIteration('iter_6_2')
154     iter_6_2.SetMeshName('PIQUAGE_2')
155     iter_6_2.SetMeshFile(os.path.join(DIRCASE, "maill.02.med"))
156     iter_6_2.AssociateHypo('hypo_6_1_2')
157     erreur = iter_6_2.Compute(1, option)
158     if erreur :
159       break
160     # Iteration iter_6_3 : second raffinement selon les faces externes
161     iter_6_3 = iter_6_2.NextIteration('iter_6_3')
162     iter_6_3.SetMeshName('PIQUAGE_3')
163     iter_6_3.SetMeshFile(os.path.join(DIRCASE, "maill.03.med"))
164     iter_6_3.AssociateHypo('hypo_6_2_3')
165     erreur = iter_6_3.Compute(1, option)
166     if erreur :
167       break
168   #
169     break
170   #
171   if erreur :
172     message += "Erreur au calcul de l'itération %d" % erreur
173   #
174   return erreur, message
175 #
176 #==========================  Fin de la fonction ==================================
177 #
178 ERREUR = 0
179 MESSAGE = ""
180 while not ERREUR :
181   #
182   # A. Exec of HOMARD-SALOME
183   #
184   HOMARD = salome.lcc.FindOrLoadComponent('FactoryServer', 'HOMARD')
185   assert HOMARD is not None, "Impossible to load homard engine"
186   HOMARD.SetLanguageShort("fr")
187 #
188   FICMED = os.path.join(DATA_TUTORIAL, "tutorial_4.00.med")
189   FICFRMED = os.path.join(DATA_TUTORIAL, TEST_NAME+".fr.med")
190   try:
191     ERREUR, MESSAGE = homard_exec("PIQUAGE", FICMED, "COURBES", FICFRMED, DEBUG)
192   except RuntimeError as eee:
193     ERREUR = 2
194     MESSAGE = str(eee.message)
195   #
196   if ERREUR :
197     MESSAGE += "Pb in homard_exec"
198     break
199   #
200   # B. Test of the results
201   #
202   N_REP_TEST_FILE = N_ITER_TEST_FILE
203   DESTROY_DIR = not DEBUG
204   test_results(REP_DATA, TEST_NAME, DIRCASE, N_ITER_TEST_FILE, N_REP_TEST_FILE, DESTROY_DIR)
205   #
206   break
207 #
208 if ERREUR:
209   raise Exception(MESSAGE)
210 #
211 # ==================================
212 gzip_gunzip(DATA_TUTORIAL, 4, 1)
213 gzip_gunzip(DATA_TUTORIAL, 6, 1)
214 # ==================================
215
216 if salome.sg.hasDesktop():
217   salome.sg.updateObjBrowser()
218   iparameters.getSession().restoreVisualState(1)
219