src/CALCULATOR/CALCULATOR_TEST.py

   1 # Copyright (C) 2007-2021  CEA/DEN, EDF R&D, OPEN CASCADE
   2 #
   3 # Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
   4 # CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
   5 #
   6 # This library is free software; you can redistribute it and/or
   7 # modify it under the terms of the GNU Lesser General Public
   8 # License as published by the Free Software Foundation; either
   9 # version 2.1 of the License, or (at your option) any later version.
  10 #
  11 # This library is distributed in the hope that it will be useful,
  12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14 # Lesser General Public License for more details.
  15 #
  16 # You should have received a copy of the GNU Lesser General Public
  17 # License along with this library; if not, write to the Free Software
  18 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  19 #
  20 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  21 #
  22
  23 import os
  24
  25 import MED_ORB
  26 import CALCULATOR_ORB
  27
  28 import salome
  29
  30 from MEDCoupling import *
  31 from MEDLoader import *
  32 from MEDCouplingCorba import *
  33 from MEDCouplingClient import *
  34
  35 # This test program is based on the field named fieldcelldoublevector in
  36 # med file ${DATA_DIR}/MedFiles/pointe.med
  37 medFile = os.path.join(os.environ["DATA_DIR"], "MedFiles", "pointe.med")
  38 meshName = "maa1"
  39 fieldName = "fieldcelldoublevector"
  40
  41 # init SALOME session
  42 salome.salome_init()
  43
  44 # Get MED component
  45 print("[CALC] Get reference to MED component ...")
  46 med = salome.lcc.FindOrLoadComponent("FactoryServer", "MED")
  47 print("[CALC] ---")
  48
  49 # Get CALCULATOR component
  50 print("[CALC] Get reference to CALCULATOR component ...")
  51 calculator = salome.lcc.FindOrLoadComponent("FactoryServer", "CALCULATOR")
  52 print("[CALC] ---")
  53
  54 # Get a CORBA field proxy on the distant field (located in the med server)
  55 print("[CALC] Read field %s ..." % fieldName)
  56 f = ReadFieldCell(medFile, meshName, 0, fieldName, -1, -1)
  57 fieldcelldouble = MEDCouplingFieldDoubleServant._this(f)
  58
  59 print("[CALC] -> fieldcelldouble is:")
  60 print(f)
  61 print(f.getName())
  62 print(f.getDescription())
  63 print(f.getNumberOfComponents())
  64 print("[CALC] ---")
  65
  66 print("[CALC] Clone field: fieldcelldoublevector -> f1,f2,f3,f4 ...")
  67 (f1,f2,f3,f4) = calculator.cloneField(fieldcelldouble)
  68 print("[CALC] -> f1 is:")
  69 calculator.printField(f1)
  70 print("[CALC] ---")
  71
  72 print("[CALC] Add fields f2+f3 ...")
  73 f_add = calculator.add(f2, f3)
  74 print("[CALC] -> f_add is:")
  75 calculator.printField(f_add)
  76 print("[CALC] ---")
  77
  78 print("[CALC] Apply linear function to f4 ...")
  79 f_lin = calculator.applyLin(f4, 2.0, 1.0)
  80 print("[CALC] -> f_add is:")
  81 calculator.printField(f_lin)
  82 print("[CALC] ---")
  83
  84 print("[CALC] Apply Norm Max to f_lin ...")
  85 norm = calculator.normMax(f_lin)
  86 print("[CALC] -> norm is ", norm)
  87 print("[CALC] ---")
  88
  89 print("[CALC] Clone fields created by Calculator via client classes ...")
  90 f_addLocal = MEDCouplingFieldDoubleClient.New(f_add)
  91 f_addLocal.setName(f_addLocal.getName() + "add")
  92 f_linLocal = MEDCouplingFieldDoubleClient.New(f_lin)
  93 f_linLocal.setName(f_linLocal.getName() + "lin")
  94 print("[CALC] -> f_addLocal is ", f_addLocal)
  95 print("[CALC] -> f_linLocal is ", f_linLocal)
  96 print("[CALC] ---")
  97
  98 print("[CALC] Get information from the local copy of the distant mesh")
  99 meshLocal = f_addLocal.getMesh()
 100 print("[CALC] -> meshLocal is", meshLocal)
 101 print("[CALC] ---")
 102
 103 print("[CALC] Write mesh and fields to MED file ...")
 104 import tempfile
 105 outfile = tempfile.NamedTemporaryFile(prefix="Calculator_pointe_", suffix=".med")
 106 outfile.close()
 107 WriteUMesh(outfile.name, meshLocal, True)
 108 WriteFieldUsingAlreadyWrittenMesh(outfile.name, f_addLocal)
 109 WriteFieldUsingAlreadyWrittenMesh(outfile.name, f_linLocal)
 110 os.remove(outfile.name)
 111 print("[CALC] ---")
 112
 113 print("[CALC] End of Calculator Test!")