src/tests/daSalome/test_aster_zzzz159a_functions.py

   1 import numpy
   2 import pickle
   3 import sys
   4
   5 sys.path.insert(0, init_data['SOURCES_ROOT'])
   6 import test_aster_zzzz159a_aster_functions as Code_Aster
   7
   8 # Configuration du module
   9 Code_Aster.debug = init_data["debug"]
  10 Code_Aster.ASTER_ROOT = init_data["ASTER_ROOT"]
  11 Code_Aster.SOURCES_ROOT = init_data['SOURCES_ROOT']
  12 Code_Aster.export = init_data["export"]
  13 Code_Aster.calcul = init_data["calcul"]
  14 Code_Aster.parametres = init_data["parametres"]
  15 Code_Aster.python_version = init_data["python_version"]
  16
  17 print computation
  18 method = ""
  19 for param in computation["specificParameters"]:
  20   if param["name"] == "method":
  21     method = param["value"]
  22
  23 # Extraction des données et remise en forme (normalement à faire
  24 # dans le code
  25 # On sait qu'on a trois variables
  26 input_data = []
  27 for i in range(3):
  28   input_data.append(computation["inputValues"][0][0][i][0])
  29
  30 if method == "Adjoint":
  31   input_data = (input_data, [])
  32   for i in range(22):
  33     if i < 11:
  34       input_data[1].append(computation["inputValues"][0][0][3][i])
  35     else:
  36       input_data[1].append(computation["inputValues"][0][0][4][i-11])
  37
  38 result = {}
  39 result["specificOutputInfos"] = []
  40 result["returnCode"] = 0
  41 result["errorMessage"] = ""
  42
  43 outputValues = [[[[]]]]
  44 if method == "Direct":
  45   output_data = Code_Aster.Calcul_Aster_Ponctuel(input_data)
  46   outputValues[0][0] = [[],[]]
  47   for i in range(22):
  48     if i < 11:
  49       outputValues[0][0][0].append(output_data[i])
  50     else:
  51       outputValues[0][0][1].append(output_data[i])
  52
  53 if method == "Tangent":
  54   output_data = Code_Aster.Calcul_Aster_Ponctuel(input_data)
  55   outputValues[0][0] = [[],[]]
  56   for i in range(22):
  57     if i < 11:
  58       outputValues[0][0][0].append(output_data[i])
  59     else:
  60       outputValues[0][0][1].append(output_data[i])
  61
  62 if method == "Adjoint":
  63   output_data = Code_Aster.Calcul_Aster_Adjoint(input_data)
  64   outputValues[0][0] = [[],[],[]]
  65   outputValues[0][0][0].append(output_data[0])
  66   outputValues[0][0][1].append(output_data[1])
  67   outputValues[0][0][2].append(output_data[2])
  68
  69 result["outputValues"] = outputValues
  70
  71 print "Computation end"