# -*- coding: utf-8 -*-
-# Copyright (C) 2011-2016 CEA/DEN, EDF R&D
+# Copyright (C) 2011-2021 CEA/DEN, EDF R&D
#
# This library is free software; you can redistribute it and/or
# modify it under the terms of the GNU Lesser General Public
Specific conditions for Code_Saturne
Test test_5
"""
-__revision__ = "V2.01"
+__revision__ = "V2.05"
-#========================================================================
-TEST_NAME = "test_5"
-DEBUG = False
-VERBOSE = True
-N_ITER_TEST_FILE = 3
-NBCELL_X = 10
-NBCELL_Y = 10
-NBCELL_Z = 10
-LG_X = 360.
-LG_Y = 240.
-LG_Z = 160.
-MESH_NAME = "MESH"
-#========================================================================
import os
-import tempfile
import sys
import numpy as np
+
import salome
import HOMARD
-import MEDCoupling as mc
+import medcoupling as mc
import MEDLoader as ml
-#
+
# ==================================
PATH_HOMARD = os.getenv('HOMARD_ROOT_DIR')
# Repertoire des scripts utilitaires
REP_PYTHON = os.path.join(PATH_HOMARD, "bin", "salome", "test", "HOMARD")
REP_PYTHON = os.path.normpath(REP_PYTHON)
sys.path.append(REP_PYTHON)
-from test_util import remove_dir
+from test_util import get_dir
from test_util import test_results
-# Repertoire des donnees du test
-REP_DATA = os.path.join(PATH_HOMARD, "share", "salome", "homardsamples")
-REP_DATA = os.path.normpath(REP_DATA)
-# Repertoire des resultats
-if DEBUG :
- DIRCASE = os.path.join("/tmp", TEST_NAME)
- if ( os.path.isdir(DIRCASE) ) :
- remove_dir(DIRCASE)
- os.mkdir(DIRCASE)
-else :
- DIRCASE = tempfile.mkdtemp()
# ==================================
+#========================================================================
+TEST_NAME = "test_5"
+DEBUG = False
+VERBOSE = False
+N_ITER_TEST_FILE = 3
+NBCELL_X = 10
+NBCELL_Y = 10
+NBCELL_Z = 10
+LG_X = 360.
+LG_Y = 240.
+LG_Z = 160.
+MESH_NAME = "MESH"
+# Répertoires pour ce test
+REP_DATA, DIRCASE = get_dir(PATH_HOMARD, TEST_NAME, DEBUG)
+#========================================================================
+
salome.salome_init()
-#
-from MEDCouplingRemapper import MEDCouplingRemapper
-import iparameters
-IPAR = iparameters.IParameters(salome.myStudy.GetCommonParameters("Interface Applicative", 1))
-IPAR.append("AP_MODULES_LIST", "Homard")
-#
-#========================================================================
#========================================================================
def mesh_exec():
- """
-Python script for MED
- """
+ """Python script for MED"""
error = 0
-#
+
while not error :
- #
- # Creation of the mesh
- # ====================
+
+# Creation of the mesh
+# ====================
maillage_3d = ml.MEDCouplingUMesh(MESH_NAME, 2)
maillage_3d.setMeshDimension(3)
- #
- # Creation of the nodes
- # ====================
- #
+
+# Creation of the nodes
+# ====================
+
nbno_x = NBCELL_X + 1
nbno_y = NBCELL_Y + 1
nbno_z = NBCELL_Z + 1
-#
+
delta_x = LG_X / float(NBCELL_X)
delta_y = LG_Y / float(NBCELL_Y)
delta_z = LG_Z / float(NBCELL_Z)
-#
+
coordinates = list()
coo_z = -0.5*LG_Z
- for kaux in range(nbno_z) :
+ for _ in range(nbno_z) :
coo_y = -0.5*LG_Y
- for jaux in range(nbno_y) :
+ for _ in range(nbno_y) :
coo_x = -0.5*LG_X
- for iaux in range(nbno_x) :
+ for _ in range(nbno_x) :
coordinates.append(coo_x)
coordinates.append(coo_y)
coordinates.append(coo_z)
coo_x += delta_x
coo_y += delta_y
coo_z += delta_z
-#
- nbr_nodes = nbno_x*nbno_y*nbno_z
- les_coords = ml.DataArrayDouble(coordinates, nbr_nodes, 3)
+
+ nbno = nbno_x*nbno_y*nbno_z
+ les_coords = ml.DataArrayDouble(coordinates, nbno, 3)
maillage_3d.setCoords(les_coords)
- #
- # Creation of the cells
- # =====================
- #
+
+# Creation of the cells
+# =====================
+
nbr_cell_3d = NBCELL_X*NBCELL_Y*NBCELL_Z
maillage_3d.allocateCells(nbr_cell_3d)
-#
+
decala_z = nbno_x*nbno_y
# kaux = numero de la tranche en z
for kaux in range(1, nbno_z) :
-#
+
#print ". Tranche en z numero %d" % kaux
decala = decala_z*(kaux-1)
# jaux = numero de la tranche en y
for jaux in range(1, nbno_y) :
-#
+
#print ". Tranche en y numero %d" % jaux
# iaux = numero de la tranche en x
for iaux in range(1, nbno_x) :
-#
+
#print ". Tranche en x numero %d" % iaux
nref = decala+iaux-1
laux = [nref, nref+nbno_x, nref+1+nbno_x, nref+1, nref+decala_z, nref+nbno_x+decala_z, nref+1+nbno_x+decala_z, nref+1+decala_z]
- #if self.verbose_max :
- #if ( ( iaux==1 and jaux==1 and kaux==1 ) or ( iaux==(nbr_nodes_x-1) and jaux==(nbr_nodes_y-1) and kaux==(nbr_nodes_z-1) ) ) :
- #print ". Maille %d : " % (iaux*jaux*kaux), laux
+ if VERBOSE:
+ if ( ( iaux==1 and jaux==1 and kaux==1 ) or ( iaux==(nbno_x-1) and jaux==(nbno_y-1) and kaux==(nbno_z-1) ) ) :
+ print (". Maille {} : {}".format((iaux*jaux*kaux),laux))
maillage_3d.insertNextCell(ml.NORM_HEXA8, 8, laux)
-#
+
decala += nbno_x
-#
+
maillage_3d.finishInsertingCells()
- #
- # Agregation into a structure of MEDLoader
- # ========================================
- #
+
+# Agregation into a structure of MEDLoader
+# ========================================
+
meshMEDFile3D = ml.MEDFileUMesh()
meshMEDFile3D.setName(MESH_NAME)
-#
+
meshMEDFile3D.setMeshAtLevel(0, maillage_3d)
-#
+
meshMEDFile3D.rearrangeFamilies()
- #
- # MED exportation
- # ===============
- #
+
+# MED exportation
+# ===============
try:
ficmed = os.path.join(DIRCASE, 'maill.00.med')
#print "Ecriture du maillage dans le fichier", ficmed
meshMEDFile3D.write(ficmed, 2)
- except Exception as eee:
+ except IOError as eee:
error = 2
raise Exception('MEDFileUMesh.write() failed. ' + str(eee))
- #
+
break
- #
+
return error
#========================================================================
-#
+
#========================================================================
def field_exec(niter):
"""
Python script for MEDCoupling
"""
error = 0
-#
+
while not error :
- #
- # The mesh
- # ========
+
+# The mesh
+# ========
ficmed = os.path.join(DIRCASE, 'maill.%02d.med' % niter)
meshMEDFileRead = ml.MEDFileMesh.New(ficmed)
mesh_read0 = meshMEDFileRead.getMeshAtLevel(0)
- # Barycenter of the cells
- # =======================
+# Barycenter of the cells
+# =======================
cg_hexa_ml = mesh_read0.computeIsoBarycenterOfNodesPerCell()
cg_hexa = cg_hexa_ml.toNumPyArray()
- # Target
- # ======
+# Target
+# ======
xyz_p = np.zeros(3, dtype=np.float)
xyz_p[0] = -0.20*float(1-niter) * LG_X
xyz_p[1] = -0.15*float(1-niter) * LG_Y
xyz_p[2] = -0.10*float(1-niter) * LG_Z
- # Values of the field
- # ===================
+# Values of the field
+# ===================
nbr_cell_3d = mesh_read0.getNumberOfCells()
valeur = mc.DataArrayDouble(nbr_cell_3d)
for num_mail in range(nbr_cell_3d) :
valeur[num_mail] = 1.e0 / max ( 1.e-5, distance)
#print ". valeur", valeur
nparr = valeur.toNumPyArray()
- print(". mini/maxi", nparr.min(), nparr.max())
- #
- # Creation of the field
- # =====================
+ print(". mini/maxi {}/{}".format(nparr.min(),nparr.max()))
+
+# Creation of the field
+# =====================
field = ml.MEDCouplingFieldDouble(ml.ON_CELLS, ml.ONE_TIME)
field.setArray(valeur)
field.setMesh(mesh_read0)
field.setName("DISTANCE")
- #
+
fMEDFile_ch = ml.MEDFileField1TS()
fMEDFile_ch.setFieldNoProfileSBT(field) # No profile desired on the field, Sort By Type
fMEDFile_ch.write(ficmed, 0) # 0 to indicate that we *append* (and no overwrite) to the MED file
- #
+
break
- #
+
return error, ficmed
#========================================================================
+
#========================================================================
def homard_exec():
- """
-Python script for HOMARD
- """
+ """Python script for HOMARD"""
error = 0
-#
+
while not error :
- #
- HOMARD.UpdateStudy()
- #
- # Creation of the hypothese DISTANCE INVERSE
- # ==========================================
+
+# Creation of the hypothese DISTANCE INVERSE
+# ==========================================
hyponame = "DISTANCE INVERSE"
print("-------- Creation of the hypothesis", hyponame)
hypo_5 = HOMARD.CreateHypothesis(hyponame)
hypo_5.SetUnRefThr(1, 0.015)
print(hyponame, " : champ utilisé :", hypo_5.GetFieldName())
print(".. caractéristiques de l'adaptation :", hypo_5.GetField())
- #
- # Creation of the cases
- # =====================
- # Creation of the case
+
+# Creation of the cases
+# =====================
print("-------- Creation of the case", TEST_NAME)
mesh_file = os.path.join(DIRCASE, 'maill.00.med')
case_test_5 = HOMARD.CreateCase(TEST_NAME, 'MESH', mesh_file)
case_test_5.SetDirName(DIRCASE)
case_test_5.SetConfType(1)
case_test_5.SetExtType(1)
- #
- # Creation of the iterations
- # ==========================
- #
+
+# Creation of the iterations
+# ==========================
+
for niter in range(N_ITER_TEST_FILE) :
- #
+
s_niterp1 = "%02d" % (niter + 1)
- #
+
# Creation of the indicator
#
error, ficmed_indic = field_exec(niter)
if error :
error = 10
break
- #
+
# Creation of the iteration
- #
+
iter_name = "I_" + TEST_NAME + "_" + s_niterp1
print("-------- Creation of the iteration", iter_name)
if ( niter == 0 ) :
if error :
error = 20
break
- #
+
break
- #
+
return error
#========================================================================
-#
+
# Geometry and Mesh
-#
+
try :
ERROR = mesh_exec()
if ERROR :
raise Exception('Pb in mesh_exec')
-except Exception as eee:
+except RuntimeError as eee:
raise Exception('Pb in mesh_exec: '+str(eee.message))
HOMARD = salome.lcc.FindOrLoadComponent('FactoryServer', 'HOMARD')
assert HOMARD is not None, "Impossible to load homard engine"
HOMARD.SetLanguageShort("fr")
-#
+
# Exec of HOMARD-SALOME
-#
+
try :
ERROR = homard_exec()
if ERROR :
raise Exception('Pb in homard_exec at iteration %d' %ERROR )
-except Exception as eee:
+except RuntimeError as eee:
raise Exception('Pb in homard_exec: '+str(eee.message))
-#
+
# Test of the results
-#
+
N_REP_TEST_FILE = N_ITER_TEST_FILE
DESTROY_DIR = not DEBUG
test_results(REP_DATA, TEST_NAME, DIRCASE, N_ITER_TEST_FILE, N_REP_TEST_FILE, DESTROY_DIR)
-#
+
if salome.sg.hasDesktop():
salome.sg.updateObjBrowser()
- iparameters.getSession().restoreVisualState(1)
-
