Date: 2013-07-02T14:39:29+00:00

Actual source code: pythonsys.c
petsc-3.4.2 2013-07-02

      1: #include <petsc-private/petscimpl.h>       /*I "petscsys.h" I*/

      3: /* ---------------------------------------------------------------- */

      5: #if !defined(PETSC_PYTHON_EXE)
      6: #define PETSC_PYTHON_EXE "python"
      7: #endif

     11: static PetscErrorCode PetscPythonFindExecutable(char pythonexe[PETSC_MAX_PATH_LEN])
     12: {
     13:   PetscBool      flag;

     17:   /* get the path for the Python interpreter executable */
     18:   PetscStrncpy(pythonexe,PETSC_PYTHON_EXE,PETSC_MAX_PATH_LEN);
     19:   PetscOptionsGetString(NULL,"-python",pythonexe,PETSC_MAX_PATH_LEN,&flag);
     20:   if (!flag || pythonexe[0]==0) {
     21:     PetscStrncpy(pythonexe,PETSC_PYTHON_EXE,PETSC_MAX_PATH_LEN);
     22:   }
     23:   return(0);
     24: }

     28: static PetscErrorCode PetscPythonFindLibrary(char pythonexe[PETSC_MAX_PATH_LEN],char pythonlib[PETSC_MAX_PATH_LEN])
     29: {
     30:   const char     cmdline[] = "-c 'import sys; print(sys.exec_prefix); print(sys.version[:3])'";
     31:   char           command[PETSC_MAX_PATH_LEN+1+sizeof(cmdline)+1];
     32:   char           prefix[PETSC_MAX_PATH_LEN],version[8],sep[2]={PETSC_DIR_SEPARATOR, 0},*eol;
     33:   FILE           *fp = NULL;
     34:   char           path[PETSC_MAX_PATH_LEN+1];
     35:   PetscBool      found = PETSC_FALSE;

     39: #if defined(PETSC_PYTHON_LIB)
     40:   PetscStrcpy(pythonlib,PETSC_PYTHON_LIB);
     41:   return(0);
     42: #endif

     44:   /* call Python to find out the name of the Python dynamic library */
     45:   PetscStrncpy(command,pythonexe,PETSC_MAX_PATH_LEN);
     46:   PetscStrcat(command," ");
     47:   PetscStrcat(command,cmdline);
     48: #if defined(PETSC_HAVE_POPEN)
     49:   PetscPOpen(PETSC_COMM_SELF,NULL,command,"r",&fp);
     50:   if (!fgets(prefix,sizeof(prefix),fp)) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Python: bad output from executable: %s",pythonexe);
     51:   if (!fgets(version,sizeof(version),fp)) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Python: bad output from executable: %s",pythonexe);
     52:   PetscPClose(PETSC_COMM_SELF,fp,NULL);
     53: #else
     54:   SETERRQ(PETSC_COMM_SELF,1,"Python: Aborted due to missing popen()");
     55: #endif
     56:   /* remove newlines */
     57:   PetscStrchr(prefix,'\n',&eol);
     58:   if (eol) eol[0] = 0;
     59:   PetscStrchr(version,'\n',&eol);
     60:   if (eol) eol[0] = 0;

     62:   /* test for $prefix/lib64/libpythonX.X[.so]*/
     63:   PetscStrcpy(pythonlib,prefix);
     64:   PetscStrcat(pythonlib,sep);
     65:   PetscStrcat(pythonlib,"lib64");
     66:   PetscTestDirectory(pythonlib,'r',&found);
     67:   if (found) {
     68:     PetscStrcat(pythonlib,sep);
     69:     PetscStrcat(pythonlib,"libpython");
     70:     PetscStrcat(pythonlib,version);
     71:     PetscDLLibraryRetrieve(PETSC_COMM_SELF,pythonlib,path,PETSC_MAX_PATH_LEN,&found);
     72:     if (found) return(0);
     73:   }

     75:   /* test for $prefix/lib/libpythonX.X[.so]*/
     76:   PetscStrcpy(pythonlib,prefix);
     77:   PetscStrcat(pythonlib,sep);
     78:   PetscStrcat(pythonlib,"lib");
     79:   PetscTestDirectory(pythonlib,'r',&found);
     80:   if (found) {
     81:     PetscStrcat(pythonlib,sep);
     82:     PetscStrcat(pythonlib,"libpython");
     83:     PetscStrcat(pythonlib,version);
     84:     PetscDLLibraryRetrieve(PETSC_COMM_SELF,pythonlib,path,PETSC_MAX_PATH_LEN,&found);
     85:     if (found) return(0);
     86:   }

     88:   /* nothing good found */
     89:   PetscMemzero(pythonlib,PETSC_MAX_PATH_LEN);
     90:   PetscInfo(0,"Python dynamic library not found\n");
     91:   return(0);
     92: }

     94: /* ---------------------------------------------------------------- */

     96: typedef struct _Py_object_t PyObject; /* fake definition */

     98: static PyObject* Py_None = 0;

    100: static const char* (*Py_GetVersion)(void);

    102: static int       (*Py_IsInitialized)(void);
    103: static void      (*Py_InitializeEx)(int);
    104: static void      (*Py_Finalize)(void);

    106: static void      (*PySys_SetArgv)(int,char**);
    107: static PyObject* (*PySys_GetObject)(const char*);
    108: static PyObject* (*PyObject_CallMethod)(PyObject*,const char*, const char*, ...);
    109: static PyObject* (*PyImport_ImportModule)(const char*);

    111: static void      (*Py_IncRef)(PyObject*);
    112: static void      (*Py_DecRef)(PyObject*);

    114: static void      (*PyErr_Clear)(void);
    115: static PyObject* (*PyErr_Occurred)(void);
    116: static void      (*PyErr_Fetch)(PyObject**,PyObject**,PyObject**);
    117: static void      (*PyErr_NormalizeException)(PyObject**,PyObject**, PyObject**);
    118: static void      (*PyErr_Display)(PyObject*,PyObject*,PyObject*);
    119: static void      (*PyErr_Restore)(PyObject*,PyObject*,PyObject*);

    122: #define PetscDLPyLibOpen(libname) \
    123:   PetscDLLibraryAppend(PETSC_COMM_SELF,&PetscDLLibrariesLoaded,libname)
    124: #define PetscDLPyLibSym(symbol, value) \
    125:   PetscDLLibrarySym(PETSC_COMM_SELF,&PetscDLLibrariesLoaded,NULL,symbol,(void**)value)
    126: #define PetscDLPyLibClose(comm) \
    127:   do { } while (0)

    131: static PetscErrorCode PetscPythonLoadLibrary(const char pythonlib[])
    132: {

    136:   /* open the Python dynamic library */
    137:   PetscDLPyLibOpen(pythonlib);
    138:   PetscInfo1(0,"Python: loaded dynamic library %s\n", pythonlib);
    139:   /* look required symbols from the Python C-API */
    140:   PetscDLPyLibSym("_Py_NoneStruct"        , &Py_None               );
    141:   PetscDLPyLibSym("Py_GetVersion"         , &Py_GetVersion         );
    142:   PetscDLPyLibSym("Py_IsInitialized"      , &Py_IsInitialized      );
    143:   PetscDLPyLibSym("Py_InitializeEx"       , &Py_InitializeEx       );
    144:   PetscDLPyLibSym("Py_Finalize"           , &Py_Finalize           );
    145:   PetscDLPyLibSym("PySys_GetObject"       , &PySys_GetObject       );
    146:   PetscDLPyLibSym("PySys_SetArgv"         , &PySys_SetArgv         );
    147:   PetscDLPyLibSym("PyObject_CallMethod"   , &PyObject_CallMethod   );
    148:   PetscDLPyLibSym("PyImport_ImportModule" , &PyImport_ImportModule );
    149:   PetscDLPyLibSym("Py_IncRef"             , &Py_IncRef             );
    150:   PetscDLPyLibSym("Py_DecRef"             , &Py_DecRef             );
    151:   PetscDLPyLibSym("PyErr_Clear"           , &PyErr_Clear           );
    152:   PetscDLPyLibSym("PyErr_Occurred"        , &PyErr_Occurred        );
    153:   PetscDLPyLibSym("PyErr_Fetch"             , &PyErr_Fetch             );
    154:   PetscDLPyLibSym("PyErr_NormalizeException", &PyErr_NormalizeException);
    155:   PetscDLPyLibSym("PyErr_Display",            &PyErr_Display           );
    156:   PetscDLPyLibSym("PyErr_Restore",            &PyErr_Restore           );
    157:   /* XXX TODO: check that ALL symbols were there !!! */
    158:   if (!Py_None)          SETERRQ(PETSC_COMM_SELF,1,"Python: failed to load symbols from dynamic library");
    159:   if (!Py_GetVersion)    SETERRQ(PETSC_COMM_SELF,1,"Python: failed to load symbols from dynamic library");
    160:   if (!Py_IsInitialized) SETERRQ(PETSC_COMM_SELF,1,"Python: failed to load symbols from dynamic library");
    161:   if (!Py_InitializeEx)  SETERRQ(PETSC_COMM_SELF,1,"Python: failed to load symbols from dynamic library");
    162:   if (!Py_Finalize)      SETERRQ(PETSC_COMM_SELF,1,"Python: failed to load symbols from dynamic library");
    163:   PetscInfo(0,"Python: all required symbols loaded from Python dynamic library\n");
    164:   return(0);
    165: }

    167: /* ---------------------------------------------------------------- */

    169: static char      PetscPythonExe[PETSC_MAX_PATH_LEN] = { 0 };
    170: static char      PetscPythonLib[PETSC_MAX_PATH_LEN] = { 0 };
    171: static PetscBool PetscBeganPython = PETSC_FALSE;

    175: /*@C
    176:   PetscPythonFinalize - Finalize Python.

    178:   Level: intermediate

    180: .keywords: Python
    181: @*/
    182: PetscErrorCode  PetscPythonFinalize(void)
    183: {
    185:   if (PetscBeganPython) { if (Py_IsInitialized()) Py_Finalize(); }
    186:   PetscBeganPython = PETSC_FALSE;
    187:   return(0);
    188: }

    192: /*@C
    193:   PetscPythonInitialize - Initialize Python and import petsc4py.

    195:    Input Parameter:
    196: +  pyexe - path to the Python interpreter executable, or NULL.
    197: -  pylib - full path to the Python dynamic library, or NULL.

    199:   Level: intermediate

    201: .keywords: Python

    203: @*/
    204: PetscErrorCode  PetscPythonInitialize(const char pyexe[],const char pylib[])
    205: {
    206:   PyObject       *module = 0;

    210:   if (PetscBeganPython) return(0);
    211:   /* Python executable */
    212:   if (pyexe && pyexe[0] != 0) {
    213:     PetscStrncpy(PetscPythonExe,pyexe,sizeof(PetscPythonExe));
    214:   } else {
    215:     PetscPythonFindExecutable(PetscPythonExe);
    216:   }
    217:   /* Python dynamic library */
    218:   if (pylib && pylib[0] != 0) {
    219:     PetscStrncpy(PetscPythonLib,pylib,sizeof(PetscPythonLib));
    220:   } else {
    221:     PetscPythonFindLibrary(PetscPythonExe,PetscPythonLib);
    222:   }
    223:   /* dynamically load Python library */
    224:   PetscPythonLoadLibrary(PetscPythonLib);
    225:   /* initialize Python */
    226:   PetscBeganPython = PETSC_FALSE;
    227:   if (!Py_IsInitialized()) {
    228:     static PetscBool registered = PETSC_FALSE;
    229:     const char       *py_version;
    230:     PyObject         *sys_path;
    231:     char             path[PETSC_MAX_PATH_LEN] = { 0 };

    233:     /* initialize Python */
    234:     Py_InitializeEx(0); /* 0: do not install signal handlers */
    235:     /*  build 'sys.argv' list */
    236:     py_version = Py_GetVersion();
    237:     if (py_version[0] == '2') {
    238:       int argc = 0; char **argv = 0;
    239:       PetscGetArgs(&argc,&argv);
    240:       PySys_SetArgv(argc,argv);
    241:     }
    242:     if (py_version[0] == '3') {
    243:       /* XXX 'argv' is type 'wchar_t**' */
    244:       PySys_SetArgv(0,NULL);
    245:     }
    246:     /* add PETSC_LIB_DIR in front of 'sys.path' */
    247:     sys_path = PySys_GetObject("path");
    248:     if (sys_path) {
    249:       PetscStrreplace(PETSC_COMM_SELF,"${PETSC_LIB_DIR}",path,sizeof(path));
    250:       Py_DecRef(PyObject_CallMethod(sys_path,"insert","is",(int)0,(char*)path));
    251:     }
    252:     /* register finalizer */
    253:     if (!registered) {
    254:       PetscRegisterFinalize(PetscPythonFinalize);

    256:       registered = PETSC_TRUE;
    257:     }
    258:     PetscBeganPython = PETSC_TRUE;
    259:   }
    260:   /* import 'petsc4py.PETSc' module */
    261:   module = PyImport_ImportModule("petsc4py.PETSc");
    262:   if (module) {
    263:     PetscInfo(0,"Python: successfully imported  module 'petsc4py.PETSc'\n");

    265:     Py_DecRef(module); module = 0;
    266:   } else {
    267:     PetscPythonPrintError();
    268:     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Python: could not import module 'petsc4py.PETSc', perhaps your PYTHONPATH does not contain it\n");
    269:   }
    270:   return(0);
    271: }

    275: /*@C
    276:   PetscPythonPrintError - Print Python errors.

    278:   Level: developer

    280: .keywords: Python

    282: @*/
    283: PetscErrorCode  PetscPythonPrintError(void)
    284: {
    285:   PyObject *exc=0, *val=0, *tb=0;

    288:   if (!PetscBeganPython) return(0);
    289:   if (!PyErr_Occurred()) return(0);
    290:   PyErr_Fetch(&exc,&val,&tb);
    291:   PyErr_NormalizeException(&exc,&val,&tb);
    292:   PyErr_Display(exc ? exc : Py_None,
    293:                 val ? val : Py_None,
    294:                 tb  ? tb  : Py_None);
    295:   PyErr_Restore(exc,val,tb);
    296:   return(0);
    297: }

    299: /* ---------------------------------------------------------------- */

    301: PETSC_EXTERN PetscErrorCode (*PetscPythonMonitorSet_C)(PetscObject,const char[]);
    302: PetscErrorCode (*PetscPythonMonitorSet_C)(PetscObject,const char[]) = NULL;

    306: /*@C
    307:   PetscPythonMonitorSet - Set Python monitor

    309:   Level: developer

    311: .keywords: Python

    313: @*/
    314: PetscErrorCode PetscPythonMonitorSet(PetscObject obj, const char url[])
    315: {

    321:   if (PetscPythonMonitorSet_C == NULL) {
    322:     PetscPythonInitialize(NULL,NULL);
    323:     if (PetscPythonMonitorSet_C == NULL) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"Couldn't initialize Python support for monitors");
    324:   }
    325:   PetscPythonMonitorSet_C(obj,url);
    326:   return(0);
    327: }

    329: /* ---------------------------------------------------------------- */