src/tests/daSalome/test017_3DVAR_function_script.py

   1 # Copyright (C) 2010-2013 EDF R&D
   2 #
   3 # This library is free software; you can redistribute it and/or
   4 # modify it under the terms of the GNU Lesser General Public
   5 # License as published by the Free Software Foundation; either
   6 # version 2.1 of the License.
   7 #
   8 # This library is distributed in the hope that it will be useful,
   9 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11 # Lesser General Public License for more details.
  12 #
  13 # You should have received a copy of the GNU Lesser General Public
  14 # License along with this library; if not, write to the Free Software
  15 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  16 #
  17 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  18 #
  19 # Author: André Ribes, andre.ribes@edf.fr, EDF R&D
  20
  21 import numpy
  22
  23 print computation
  24 method = ""
  25 for param in computation["specificParameters"]:
  26   if param["name"] == "method":
  27     method = param["value"]
  28 print "Method found is", method
  29
  30 dimension = 300
  31 H  = numpy.matrix(numpy.core.identity(dimension))
  32
  33 def FunctionH( X ):
  34     return H * X
  35
  36 def AdjointH( (X, Y) ):
  37     return H.T * Y
  38
  39 print computation["inputValues"][0][0][0]
  40 print computation["inputValues"][0][0][0][0]
  41
  42 if method == "Direct":
  43   data = FunctionH(numpy.matrix(computation["inputValues"][0][0][0]).T)
  44
  45 if method == "Tangent":
  46   data = FunctionH(numpy.matrix(computation["inputValues"][0][0][0]).T)
  47
  48 if method == "Adjoint":
  49   data = AdjointH((numpy.matrix(computation["inputValues"][0][0][0]).T, numpy.matrix(computation["inputValues"][0][0][1]).T))
  50
  51
  52 outputValues = [[[[]]]]
  53 it = data.flat
  54 for val in it:
  55   outputValues[0][0][0].append(val)
  56
  57 print outputValues
  58
  59 result = {}
  60 result["outputValues"] = outputValues
  61 result["specificOutputInfos"] = []
  62 result["returnCode"] = 0
  63 result["errorMessage"] = ""
  64
  65 print result
  66 print "Computation end"