convert/convert_openturns_study.py

   1 # Copyright (C) 2007-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 import re
  20 from convert_python import PythonParser
  21
  22 def entryPoint():
  23    """
  24    Return a dictionary containing the description needed to load the plugin
  25    """
  26    return {
  27           'name' : 'openturns_study',
  28           'factory' : OTStudyParser
  29           }
  30
  31 class OTStudyParser(PythonParser):
  32    """
  33    This converter works like PythonParser, except that it also initializes all
  34    model variables to None in order to avoid Python syntax errors when loading
  35    a file with a different or inexistent definition of variables.
  36    """
  37    # We look for pattern "ModelVariable=NOMVAR,"
  38    pattern_model_variable = re.compile(r'ModelVariable\s*=\s*(\w+)\s*,')
  39
  40    def convert(self, outformat, appli=None):
  41       text = PythonParser.convert(self, outformat, appli)
  42       varnames = self.pattern_model_variable.findall(text)
  43       newtext = ""
  44       for var in varnames:
  45          newtext += "%s = None\n" % var
  46       newtext += text
  47       return newtext