src/Container/Container_init_python.cxx

   1 // Copyright (C) 2007-2016  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 #include <time.h>
  24 #ifndef WIN32
  25   #include <sys/time.h>
  26 #endif
  27 #include <string>
  28
  29 #include "utilities.h"
  30 #include "Container_init_python.hxx"
  31
  32 #if PY_VERSION_HEX < 0x03050000
  33 static wchar_t*
  34 Py_DecodeLocale(const char *arg, size_t *size)
  35 {
  36     wchar_t *res;
  37     unsigned char *in;
  38     wchar_t *out;
  39     size_t argsize = strlen(arg) + 1;
  40
  41     if (argsize > PY_SSIZE_T_MAX/sizeof(wchar_t))
  42         return NULL;
  43     res = (wchar_t*) PyMem_RawMalloc(argsize*sizeof(wchar_t));
  44     if (!res)
  45         return NULL;
  46
  47     in = (unsigned char*)arg;
  48     out = res;
  49     while(*in)
  50         if(*in < 128)
  51             *out++ = *in++;
  52         else
  53             *out++ = 0xdc00 + *in++;
  54     *out = 0;
  55     if (size != NULL)
  56         *size = out - res;
  57     return res;
  58 }
  59 #endif
  60
  61 void KERNEL_PYTHON::init_python(int argc, char **argv)
  62 {
  63   if (Py_IsInitialized())
  64     {
  65       MESSAGE("Python already initialized");
  66       return;
  67     }
  68   MESSAGE("=================================================================");
  69   MESSAGE("Python Initialization...");
  70   MESSAGE("=================================================================");
  71   // set stdout to line buffering (aka C++ std::cout)
  72   setvbuf(stdout, (char *)NULL, _IOLBF, BUFSIZ);
  73   wchar_t* salome_python;
  74   char* env_python=getenv("SALOME_PYTHON");
  75   if(env_python != 0)
  76     {
  77       wchar_t* salome_python = Py_DecodeLocale(env_python, NULL);
  78       Py_SetProgramName(salome_python);
  79     }
  80   Py_Initialize(); // Initialize the interpreter
  81   if (Py_IsInitialized())
  82     {
  83       MESSAGE("Python initialized eh eh eh");
  84     }
  85   wchar_t **changed_argv = new wchar_t*[argc]; // Setting arguments
  86   for (int i = 0; i < argc; i++)
  87   {
  88     changed_argv[i] = Py_DecodeLocale(argv[i], NULL);
  89   }
  90   PySys_SetArgv(argc, changed_argv);
  91
  92   PyRun_SimpleString("import threading\n");
  93   // VSR (22/09/2016): This is a workaround to prevent invoking qFatal() from PyQt5
  94   // causing application aborting
  95   std::string script;
  96   script += "def _custom_except_hook(exc_type, exc_value, exc_traceback):\n";
  97   script += "  import sys\n";
  98   script += "  sys.__excepthook__(exc_type, exc_value, exc_traceback)\n";
  99   script += "  pass\n";
 100   script += "\n";
 101   script += "import sys\n";
 102   script += "sys.excepthook = _custom_except_hook\n";
 103   script += "del _custom_except_hook, sys\n";
 104   int res = PyRun_SimpleString(script.c_str());
 105   // VSR (22/09/2016): end of workaround
 106   PyEval_InitThreads(); // Create (and acquire) the interpreter lock
 107   PyThreadState *pts = PyGILState_GetThisThreadState();
 108   PyEval_ReleaseThread(pts);
 109   //delete[] changed_argv;
 110 }
 111