/*@C
PetscPythonInitialize - Initialize Python and import petsc4py.
Input Parameter:
+ pyexe - path to the Python interpreter executable, or PETSC_NULL.
- pylib - full path to the Python dynamic library, or PETSC_NULL.
Level: intermediate
.keywords: Python
@*/
PetscErrorCode PETSC_DLLEXPORT PetscPythonInitialize(const char pyexe[],const char pylib[])
{
int argc = 0;
char **argv = 0;
PyObject *module = 0;
static PetscTruth registered = PETSC_FALSE;
PetscErrorCode ierr;
PetscFunctionBegin;
if (PetscBeganPython) PetscFunctionReturn(0);
/* Python executable */
if (pyexe && pyexe[0] != 0) {
ierr = PetscStrncpy(PetscPythonExe,pyexe,sizeof(PetscPythonExe));CHKERRQ(ierr);
} else {
ierr = PetscPythonFindExecutable(PetscPythonExe);CHKERRQ(ierr);
}
/* Python dynamic library */
if (pylib && pylib[0] != 0) {
ierr = PetscStrncpy(PetscPythonLib,pylib,sizeof(PetscPythonLib));CHKERRQ(ierr);
} else {
ierr = PetscPythonFindLibrary(PetscPythonExe,PetscPythonLib);CHKERRQ(ierr);
}
/* dynamically load Python library */
ierr = PetscPythonLoadLibrary(PetscPythonLib);CHKERRQ(ierr);
/* initialize Python */
PetscBeganPython = PETSC_FALSE;
if (!Py_IsInitialized()) {
/* call below does not install signal handlers */
Py_InitializeEx(0);
/* call below required to build 'sys.argv' list */
ierr = PetscGetArgs(&argc,&argv);CHKERRQ(ierr);
if (argc && argv && argv[0]) PySys_SetArgv(argc,argv);
/* register finalizer */
if (!registered) {
ierr = PetscRegisterFinalize(PetscPythonFinalize);CHKERRQ(ierr);
registered = PETSC_TRUE;
}
PetscBeganPython = PETSC_TRUE;
}
/* import 'petsc4py.PETSc' module */
module = PyImport_ImportModule("petsc4py.PETSc");
if (module) {
ierr = PetscInfo(0,"Python: successfully imported module 'petsc4py.PETSc'\n");CHKERRQ(ierr);
Py_DecRef(module); module = 0;
} else {
SETERRQ(PETSC_ERR_PLIB,"Python: could not import module 'petsc4py.PETSc', perhaps your PYTHONPATH does not contain it\n");
}
PetscFunctionReturn(0);
}
/*@C
PetscPythonInitialize - Initialize Python and import petsc4py.
Input Parameter:
+ pyexe - path to the Python interpreter executable, or NULL.
- pylib - full path to the Python dynamic library, or NULL.
Level: intermediate
.keywords: Python
@*/
PetscErrorCode PetscPythonInitialize(const char pyexe[],const char pylib[])
{
PyObject *module = 0;
PetscErrorCode ierr;
PetscFunctionBegin;
if (PetscBeganPython) PetscFunctionReturn(0);
/* Python executable */
if (pyexe && pyexe[0] != 0) {
ierr = PetscStrncpy(PetscPythonExe,pyexe,sizeof(PetscPythonExe));CHKERRQ(ierr);
} else {
ierr = PetscPythonFindExecutable(PetscPythonExe);CHKERRQ(ierr);
}
/* Python dynamic library */
if (pylib && pylib[0] != 0) {
ierr = PetscStrncpy(PetscPythonLib,pylib,sizeof(PetscPythonLib));CHKERRQ(ierr);
} else {
ierr = PetscPythonFindLibrary(PetscPythonExe,PetscPythonLib);CHKERRQ(ierr);
}
/* dynamically load Python library */
ierr = PetscPythonLoadLibrary(PetscPythonLib);CHKERRQ(ierr);
/* initialize Python */
PetscBeganPython = PETSC_FALSE;
if (!Py_IsInitialized()) {
static PetscBool registered = PETSC_FALSE;
const char *py_version;
PyObject *sys_path;
char path[PETSC_MAX_PATH_LEN] = { 0 };
/* initialize Python */
Py_InitializeEx(0); /* 0: do not install signal handlers */
/* build 'sys.argv' list */
py_version = Py_GetVersion();
if (py_version[0] == '2') {
int argc = 0; char **argv = 0;
ierr = PetscGetArgs(&argc,&argv);CHKERRQ(ierr);
PySys_SetArgv(argc,argv);
}
if (py_version[0] == '3') {
/* XXX 'argv' is type 'wchar_t**' */
PySys_SetArgv(0,NULL);
}
/* add PETSC_LIB_DIR in front of 'sys.path' */
sys_path = PySys_GetObject("path");
if (sys_path) {
ierr = PetscStrreplace(PETSC_COMM_SELF,"${PETSC_LIB_DIR}",path,sizeof(path));CHKERRQ(ierr);
Py_DecRef(PyObject_CallMethod(sys_path,"insert","is",(int)0,(char*)path));
}
/* register finalizer */
if (!registered) {
ierr = PetscRegisterFinalize(PetscPythonFinalize);CHKERRQ(ierr);
registered = PETSC_TRUE;
}
PetscBeganPython = PETSC_TRUE;
}
/* import 'petsc4py.PETSc' module */
module = PyImport_ImportModule("petsc4py.PETSc");
if (module) {
ierr = PetscInfo(0,"Python: successfully imported module 'petsc4py.PETSc'\n");CHKERRQ(ierr);
Py_DecRef(module); module = 0;
} else {
PetscPythonPrintError();
SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Python: could not import module 'petsc4py.PETSc', perhaps your PYTHONPATH does not contain it\n");
}
PetscFunctionReturn(0);
}
