src/evalyfx/YACSEvalSession.cxx

   1 // Copyright (C) 2012-2015  CEA/DEN, 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, or (at your option) any later version.
   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 : Anthony Geay (EDF R&D)
  20
  21 #include "YACSEvalSession.hxx"
  22 #include "YACSEvalSessionInternal.hxx"
  23
  24 #include "Exception.hxx"
  25
  26 #include <Python.h>
  27
  28 const char YACSEvalSession::KERNEL_ROOT_DIR[]="KERNEL_ROOT_DIR";
  29
  30 const char YACSEvalSession::CORBA_CONFIG_ENV_VAR_NAME[]="OMNIORB_CONFIG";
  31
  32 YACSEvalSession::YACSEvalSession():_isLaunched(false),_port(-1),_salomeInstanceModule(0),_salomeInstance(0),_internal(new YACSEvalSessionInternal)
  33 {
  34   if(!Py_IsInitialized())
  35     Py_Initialize();
  36   //
  37   _salomeInstanceModule=PyImport_ImportModule(const_cast<char *>("salome_instance"));
  38 }
  39
  40 YACSEvalSession::~YACSEvalSession()
  41 {
  42   delete _internal;
  43   if(isLaunched())
  44     {
  45       PyObject *terminateSession(PyObject_GetAttrString(_salomeInstance,const_cast<char *>("stop")));//new
  46       PyObject *res(PyObject_CallObject(terminateSession,0));
  47       Py_XDECREF(res);
  48       Py_XDECREF(terminateSession);
  49     }
  50   Py_XDECREF(_salomeInstance);
  51   Py_XDECREF(_salomeInstanceModule);
  52 }
  53
  54 void YACSEvalSession::launch()
  55 {
  56   if(isLaunched())
  57     return ;
  58   PyObject *salomeInstance(PyObject_GetAttrString(_salomeInstanceModule,const_cast<char *>("SalomeInstance")));//new
  59   PyObject *startMeth(PyObject_GetAttrString(salomeInstance,const_cast<char *>("start")));
  60   Py_XDECREF(salomeInstance);
  61   PyObject *myArgs(PyTuple_New(0));//new
  62   PyObject *myKWArgs(PyDict_New());//new
  63   PyDict_SetItemString(myKWArgs,"shutdown_servers",Py_True);//Py_True ref not stolen
  64   _salomeInstance=PyObject_Call(startMeth,myArgs,myKWArgs);//new
  65   PyObject *getPortMeth(PyObject_GetAttrString(_salomeInstance,const_cast<char *>("get_port")));//new
  66   PyObject *portPy(PyObject_CallObject(getPortMeth,0));//new
  67   _port=PyInt_AsLong(portPy);
  68   Py_XDECREF(portPy);
  69   Py_XDECREF(getPortMeth);
  70   Py_XDECREF(myKWArgs);
  71   Py_XDECREF(myArgs);
  72   //
  73   PyObject *osPy(PyImport_ImportModule(const_cast<char *>("os")));//new
  74   PyObject *environPy(PyObject_GetAttrString(osPy,const_cast<char *>("environ")));//new
  75   PyObject *corbaConfigStr(PyString_FromString(const_cast<char *>(CORBA_CONFIG_ENV_VAR_NAME)));//new
  76   PyObject *corbaConfigFileNamePy(PyObject_GetItem(environPy,corbaConfigStr));//new
  77   _corbaConfigFileName=PyString_AsString(corbaConfigFileNamePy);
  78   Py_XDECREF(corbaConfigFileNamePy);
  79   Py_XDECREF(corbaConfigStr);
  80   Py_XDECREF(environPy);
  81   Py_XDECREF(osPy);
  82   _isLaunched=true;
  83 }
  84
  85 void YACSEvalSession::checkLaunched() const
  86 {
  87   if(!isLaunched())
  88     throw YACS::Exception("YACSEvalSession::checkLaunched : not launched !");
  89 }
  90
  91 int YACSEvalSession::getPort() const
  92 {
  93   checkLaunched();
  94   return _port;
  95 }
  96
  97 std::string YACSEvalSession::getCorbaConfigFileName() const
  98 {
  99   checkLaunched();
 100   return _corbaConfigFileName;
 101 }
 102
 103 std::string YACSEvalSession::GetPathToAdd()
 104 {
 105   std::string ret;
 106   PyObject *osPy(PyImport_ImportModule(const_cast<char *>("os")));//new
 107   PyObject *kernelRootDir(0);// os.environ["KERNEL_ROOT_DIR"]
 108   {
 109     PyObject *environPy(PyObject_GetAttrString(osPy,const_cast<char *>("environ")));//new
 110     PyObject *kernelRootDirStr(PyString_FromString(const_cast<char *>(KERNEL_ROOT_DIR)));//new
 111     kernelRootDir=PyObject_GetItem(environPy,kernelRootDirStr);//new
 112     Py_XDECREF(kernelRootDirStr);
 113     Py_XDECREF(environPy);
 114   }
 115   {
 116     PyObject *pathPy(PyObject_GetAttrString(osPy,const_cast<char *>("path")));//new
 117     PyObject *joinPy(PyObject_GetAttrString(pathPy,const_cast<char *>("join")));//new
 118     PyObject *myArgs(PyTuple_New(4));
 119     Py_XINCREF(kernelRootDir); PyTuple_SetItem(myArgs,0,kernelRootDir);
 120     PyTuple_SetItem(myArgs,1,PyString_FromString(const_cast<char *>("bin")));
 121     PyTuple_SetItem(myArgs,2,PyString_FromString(const_cast<char *>("salome")));
 122     PyTuple_SetItem(myArgs,3,PyString_FromString(const_cast<char *>("appliskel")));
 123     PyObject *res(PyObject_CallObject(joinPy,myArgs));
 124     ret=PyString_AsString(res);
 125     Py_XDECREF(res);
 126     Py_XDECREF(myArgs);
 127     Py_XDECREF(joinPy);
 128     Py_XDECREF(pathPy);
 129   }
 130   Py_XDECREF(kernelRootDir);
 131   Py_XDECREF(osPy);
 132   return ret;
 133 }