Salome HOME
Copyright update 2021
[modules/homard.git] / src / tests / Test / test_5.py
index 569666d7aa93f033e535929b86de74ce6d460c72..32cbf63510c7a63c9a4ebd459c6154caac558b44 100755 (executable)
@@ -1,5 +1,5 @@
 # -*- coding: utf-8 -*-
-# Copyright (C) 2011-2019  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
@@ -22,29 +22,17 @@ Python script for HOMARD
 Specific conditions for Code_Saturne
 Test test_5
 """
-__revision__ = "V2.04"
+__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 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
@@ -54,44 +42,48 @@ sys.path.append(REP_PYTHON)
 from test_util import get_dir
 from test_util import test_results
 # ==================================
+
+#========================================================================
+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 _ in range(nbno_z) :
@@ -105,55 +97,54 @@ Python script for MED
           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
@@ -161,39 +152,39 @@ Python script for MED
     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) :
@@ -205,37 +196,34 @@ Python script for MEDCoupling
       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)
@@ -245,33 +233,32 @@ Python script for HOMARD
     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 ) :
@@ -286,15 +273,15 @@ Python script for HOMARD
       if error :
         error = 20
         break
-  #
+
     break
-  #
+
   return error
 
 #========================================================================
-#
+
 # Geometry and Mesh
-#
+
 try :
   ERROR = mesh_exec()
   if ERROR :
@@ -305,23 +292,21 @@ except RuntimeError as eee:
 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 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)
-