Salome HOME
aabc11a1a16993801696bb96019359c443f0697e
[modules/homard.git] / tests / test_1.py
1 # -*- coding: utf-8 -*-
2 # Copyright (C) 2011-2014  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 Copyright EDF-R&D 2010, 2014
23 Test test_1
24 """
25 __revision__ = "V2.4"
26
27 #========================================================================
28 Test_Name = "test_1"
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 pathHomard = os.getenv('HOMARD_ROOT_DIR')
39 # Repertoire des donnees du test
40 Rep_Test = os.path.join(pathHomard, "share", "salome", "resources", "homard")
41 Rep_Test = os.path.normpath(Rep_Test)
42 sys.path.append(Rep_Test)
43 from test_util import test_results
44 # Repertoire des resultats
45 dircase = tempfile.mkdtemp()
46 # ==================================
47
48 salome.salome_init()
49 import iparameters
50 ipar = iparameters.IParameters(salome.myStudy.GetCommonParameters("Interface Applicative", 1))
51 ipar.append("AP_MODULES_LIST", "Homard")
52 #
53 #========================================================================
54 #========================================================================
55 def homard_exec(theStudy):
56   """
57 Python script for HOMARD
58   """
59   error = 0
60 #
61   while not error :
62   #
63     homard.SetCurrentStudy(theStudy)
64   #
65   # Creation of the zones
66   # =====================
67   # Creation of the box Zone_1_1
68     Zone_1_1 = homard.CreateZoneBox('Zone_1_1', -0.01, 1.01, -0.01, 0.4, -0.01, 0.6)
69
70   # Creation of the sphere Zone_1_2
71     Zone_1_2 = homard.CreateZoneSphere('Zone_1_2', 0.5, 0.6, 0.7, 0.75)
72   #
73   # Creation of the hypotheses
74   # ==========================
75   # Creation of the hypothesis a10_1pc_de_mailles_a_raffiner_sur_ERRE_ELEM_SIGM
76     HypoName_1 = "a10_1pc_de_mailles_a_raffiner_sur_ERRE_ELEM_SIGM"
77     print "-------- Creation of the hypothesis", HypoName_1
78     Hypo_1_1 = homard.CreateHypothesis(HypoName_1)
79     Hypo_1_1.SetField('RESU____ERRE_ELEM_SIGM__________')
80     Hypo_1_1.SetUseComp(0)
81     Hypo_1_1.AddComp('ERREST')
82     Hypo_1_1.SetRefinThr(3, 10.1)
83     Hypo_1_1.AddFieldInterp('RESU____DEPL____________________')
84     Hypo_1_1.AddFieldInterp('RESU____ERRE_ELEM_SIGM__________')
85     print HypoName_1, " : zones utilisées :", Hypo_1_1.GetZones()
86     print HypoName_1, " : champ utilisé :", Hypo_1_1.GetFieldName()
87     print HypoName_1, " : composantes utilisées :", Hypo_1_1.GetComps()
88     if ( len (Hypo_1_1.GetFieldName()) > 0 ) :
89       print ".. caractéristiques de l'adaptation :", Hypo_1_1.GetField()
90     print HypoName_1, " : champs interpolés :", Hypo_1_1.GetFieldInterps()
91   # Creation of the hypothesis Zones_1_et_2
92     HypoName_2 = "Zones_1_et_2"
93     print "-------- Creation of the hypothesis", HypoName_2
94     Zones_1_et_2 = homard.CreateHypothesis(HypoName_2)
95     Zones_1_et_2.AddZone('Zone_1_1', 1)
96     Zones_1_et_2.AddZone('Zone_1_2', 1)
97     print HypoName_2, " : zones utilisées :", Zones_1_et_2.GetZones()
98     print HypoName_2, " : champ utilisé :", Zones_1_et_2.GetFieldName()
99     if ( len (Zones_1_et_2.GetFieldName()) > 0 ) :
100       print ".. caractéristiques de l'adaptation :", Zones_1_et_2.GetField()
101     print HypoName_2, " : champs interpolés :", Zones_1_et_2.GetFieldInterps()
102   #
103   # Creation of the cases
104   # =====================
105     # Creation of the case
106     CaseName = "Case_" + Test_Name
107     print "-------- Creation of the case", CaseName
108     MeshFile = os.path.join(Rep_Test, Test_Name + '.00.med')
109     Case_test_1 = homard.CreateCase(CaseName, 'MAILL', MeshFile)
110     Case_test_1.SetDirName(dircase)
111     Case_test_1.SetConfType(1)
112   #
113   # Creation of the iterations
114   # ==========================
115   # Creation of the iteration 1
116     IterName = "I_" + Test_Name + "_1"
117     print "-------- Creation of the iteration", IterName
118     Iter_test_1_1 = Case_test_1.NextIteration(IterName)
119     Iter_test_1_1.AssociateHypo(HypoName_1)
120     print ". Hypothese :", HypoName_1
121     Iter_test_1_1.SetMeshName('M1')
122     Iter_test_1_1.SetMeshFile(os.path.join(dircase, 'maill.01.med'))
123     Iter_test_1_1.SetFieldFile(os.path.join(Rep_Test, Test_Name + '.00.med'))
124     Iter_test_1_1.SetTimeStepRank(1, 1)
125     Iter_test_1_1.SetFieldInterpTimeStep('RESU____DEPL____________________', 1)
126     Iter_test_1_1.SetFieldInterpTimeStepRank('RESU____ERRE_ELEM_SIGM__________', 1, 1)
127     print ". Instants d'interpolation :", Iter_test_1_1.GetFieldInterpsTimeStepRank()
128     error = Iter_test_1_1.Compute(1, 1)
129     if error :
130       error = 1
131       break
132
133   # Creation of the iteration 2
134     IterName = "I_" + Test_Name + "_2"
135     print "-------- Creation of the iteration", IterName
136     Iter_test_1_2 = Iter_test_1_1.NextIteration(IterName)
137     Iter_test_1_2.AssociateHypo(HypoName_1)
138     print ". Hypothese :", HypoName_1
139     Iter_test_1_2.SetMeshName('M2')
140     Iter_test_1_2.SetMeshFile(os.path.join(dircase, 'maill.02.med'))
141     Iter_test_1_2.SetFieldFile(os.path.join(Rep_Test, Test_Name + '.01.med'))
142     Iter_test_1_2.SetTimeStepRank(1, 1)
143     Iter_test_1_2.SetFieldInterpTimeStep('RESU____DEPL____________________', 1)
144     Iter_test_1_2.SetFieldInterpTimeStepRank('RESU____ERRE_ELEM_SIGM__________', 1, 1)
145     print ". Instants d'interpolation :", Iter_test_1_2.GetFieldInterpsTimeStepRank()
146     error = Iter_test_1_2.Compute(1, 1)
147     if error :
148       error = 2
149       break
150
151   # Creation of the iteration 3
152     IterName = "I_" + Test_Name + "_3"
153     print "-------- Creation of the iteration", IterName
154     Iter_test_1_3 = Iter_test_1_2.NextIteration(IterName)
155     Iter_test_1_3.AssociateHypo(HypoName_2)
156     print ". Hypothese :", HypoName_2
157     Iter_test_1_3.SetMeshName('M3')
158     Iter_test_1_3.SetMeshFile(os.path.join(dircase, 'maill.03.med'))
159     Iter_test_1_2.SetFieldFile(os.path.join(Rep_Test, Test_Name + '.02.med'))
160     print ". Instants d'interpolation :", Iter_test_1_3.GetFieldInterpsTimeStepRank()
161     error = Iter_test_1_3.Compute(1, 1)
162     if error :
163       error = 3
164       break
165   #
166   # Creation of the schema YACS
167   # ===========================
168     ScriptFile = os.path.join(pathHomard, "share", "doc", "salome", "gui", "HOMARD", "en", "_downloads", "yacs_script_test.py")
169     ScriptFile = os.path.normpath(ScriptFile)
170     DirName = dircase
171     YACS_test_1 = Case_test_1.CreateYACSSchema("YACS_test_1", ScriptFile, DirName, MeshFile)
172     error = YACS_test_1.Write()
173     if error :
174       error = 4
175       break
176   #
177     break
178   #
179   return error
180
181 #========================================================================
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 # Exec of HOMARD-SALOME
188 #
189 try :
190   error_main = homard_exec(salome.myStudy)
191   if error_main :
192     raise Exception('Pb in homard_exec at iteration %d' %error_main )
193 except Exception, e:
194   raise Exception('Pb in homard_exec: '+e.message)
195 #
196 # Test of the results
197 #
198 n_rep_test_file = n_iter_test_file
199 test_results(Rep_Test, Test_Name, dircase, n_iter_test_file, n_rep_test_file)
200 #
201 if salome.sg.hasDesktop():
202   salome.sg.updateObjBrowser(1)
203   iparameters.getSession().restoreVisualState(1)
204