gemrb/plugins/GUIScript/PythonHelpers.cpp

   1 /* GemRB - Infinity Engine Emulator
   2  * Copyright (C) 2003 The GemRB Project
   3  *
   4  * This program is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU General Public License
   6  * as published by the Free Software Foundation; either version 2
   7  * of the License, or (at your option) any later version.
   8
   9  * This program is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  * GNU General Public License for more details.
  13
  14  * You should have received a copy of the GNU General Public License
  15  * along with this program; if not, write to the Free Software
  16  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  17  *
  18  *
  19  */
  20
  21 #include "PythonHelpers.h"
  22
  23 static bool CallPython(PyObject *Function, PyObject *args = NULL)
  24 {
  25         if (!Function) {
  26                 return false;
  27         }
  28         PyObject *ret = PyObject_CallObject(Function, args);
  29         Py_XDECREF( args );
  30         if (ret == NULL) {
  31                 if (PyErr_Occurred()) {
  32                         PyErr_Print();
  33                 }
  34                 return false;
  35         }
  36         Py_DECREF(ret);
  37         return true;
  38 }
  39
  40 static bool CallPython(PyObject *Function, int arg)
  41 {
  42         if (!Function) {
  43                 return false;
  44         }
  45         PyObject *args = Py_BuildValue("(i)", arg);
  46         return CallPython(Function, args);
  47 }
  48
  49 StringCallback::StringCallback(const char *str)
  50         : Name(PyString_FromString(const_cast<char*>(str)))
  51 {
  52 }
  53
  54 StringCallback::StringCallback(PyObject *Name)
  55         : Name(Name)
  56 {
  57         Py_XINCREF(Name);
  58 }
  59
  60 StringCallback::~StringCallback()
  61 {
  62         Py_XDECREF(Name);
  63 }
  64
  65 PyObject *StringCallback::GetFunction()
  66 {
  67         if (!Name || !Py_IsInitialized()) {
  68                 return NULL;
  69         }
  70         /* Borrowed reference */
  71         PyObject *Function = PyDict_GetItem(gs->pDict, Name);
  72         if (!Function || !PyCallable_Check(Function)) {
  73                 printMessage("GUIScript", "Missing callback function:", LIGHT_RED);
  74                 printf("%s\n", PyString_AsString(Name));
  75                 return NULL;
  76         }
  77         return Function;
  78 }
  79
  80 bool StringCallback::call()
  81 {
  82         return CallPython(GetFunction());
  83 }
  84
  85 bool StringCallback::call(int arg)
  86 {
  87         return CallPython(GetFunction(), arg);
  88 }
  89
  90 PythonCallback::PythonCallback(PyObject *Function)
  91         : Function(Function)
  92 {
  93         if (Function && PyCallable_Check(Function)) {
  94                 Py_INCREF(Function);
  95         } else {
  96                 Function = NULL;
  97         }
  98 }
  99
 100 PythonCallback::~PythonCallback()
 101 {
 102         if (Py_IsInitialized()) {
 103                 Py_XDECREF(Function);
 104         }
 105 }
 106
 107 bool PythonCallback::call ()
 108 {
 109         if (!Function || !Py_IsInitialized()) {
 110                 return false;
 111         }
 112         return CallPython(Function);
 113 }
 114
 115 bool PythonCallback::call (int arg)
 116 {
 117         if (!Function || !Py_IsInitialized()) {
 118                 return false;
 119         }
 120         return CallPython(Function, arg);
 121 }