script.cpp

#ifdef SPROXEL_USE_PYTHON
#include <QtDebug>
#include <QFileInfo>
#include <QDir>
#include <QMessageBox>
#include "MainWindow.h"
#include "script.h"
#include "pyConsole.h"
#include "pyBindings.h"


extern void init_sproxel_bindings();


#define GLUE(a, b) a##b


//ZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZ//


QDir exe_dir;

static PyObject *glue=NULL;

PyObject *py_save_project=NULL, *py_load_project=NULL, *py_init_plugin_pathes=NULL,
  *py_scan_plugins=NULL, *py_register_plugins=NULL, *py_unregister_plugins=NULL;


//ZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZ//


#define TOPYT(cls) \
  static PyObject * GLUE(cls, _toPy_py) = NULL; \
  PyObject * GLUE(cls, _toPy) (class cls *p) \
  { \
    if (!p) { Py_RETURN_NONE; } \
    if (!GLUE(cls, _toPy_py)) { PyErr_SetString(PyExc_StandardError, "No " #cls "_toPy in SproxelGlue"); return NULL; } \
    PyObject *c=PyCapsule_New(p, NULL, NULL); if (!c) return NULL; \
    PyObject *w=PyObject_CallFunctionObjArgs(GLUE(cls, _toPy_py), c, NULL); \
    Py_DECREF(c); \
    return w; \
  }


#define TOCPP(cls) \
  static PyObject * GLUE(cls, _toCpp_py) = NULL; \
  class cls * GLUE(cls, _toCpp) (PyObject *w) \
  { \
    if (!w) { PyErr_SetString(PyExc_StandardError, "NULL PyObject in " #cls "_toCpp"); return NULL; } \
    if (w==Py_None) return NULL; \
    if (!GLUE(cls, _toCpp_py)) { PyErr_SetString(PyExc_StandardError, "No " #cls "_toCpp in SproxelGlue"); return NULL; } \
    PyObject *c=PyObject_CallFunctionObjArgs(GLUE(cls, _toCpp_py), w, NULL); if (!c) return NULL; \
    void *p=PyCapsule_GetPointer(c, NULL); \
    Py_DECREF(c); \
    return (class cls*)p; \
  }


#include "glue/classGlue.h"


//ZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZ//


void init_script(const char *exe_path)
{
  exe_dir=QFileInfo(exe_path).absoluteDir();

  qDebug() << "prog name:" << exe_path;
  Py_SetProgramName((char*)exe_path);

  qDebug() << "init py...";
  Py_Initialize();
  qDebug() << "init py ok";

  init_python_console();
  init_sproxel_bindings();

  pycon("exe path: %s", exe_path);
  pycon("exe dir: %s", exe_dir.absolutePath().toLocal8Bit().data());

  #ifndef _WIN32
    QString code="import sys\nsys.path.insert(0, \"";
    code.append(exe_dir.absolutePath());
    code.append("\")\nprint 'sys.path:', sys.path\n");
    PyRun_SimpleString(code.toLocal8Bit().data());
  #endif

  pycon("Importing PySide.SproxelGlue...");
  glue=PyImport_ImportModule("PySide.SproxelGlue");
  if (!glue)
  {
    pycon("Failed to import PySide.SproxelGlue");
    #ifdef _WIN32
      QMessageBox::critical(NULL, "Sproxel Error", "Failed to import PySide.SproxelGlue");
    #endif
  }
  else
  {
    pycon("Imported PySide.SproxelGlue, getting methods...");

    bool gotErrors=false;

    #define TOPYT(cls) \
      GLUE(cls, _toPy_py ) = PyObject_GetAttrString(glue, #cls "_toPy" ); \
      if (PyErr_Occurred()) { PyErr_Print(); gotErrors=true; }

    #define TOCPP(cls) \
      GLUE(cls, _toCpp_py) = PyObject_GetAttrString(glue, #cls "_toCpp"); \
      if (PyErr_Occurred()) { PyErr_Print(); gotErrors=true; }

    #include "glue/classGlue.h"

    pycon("Glue methods: %s", gotErrors ? "some missing" : "all OK");
    if (gotErrors) QMessageBox::critical(NULL, "Sproxel Error", "Some PySide.SproxelGlue methods are missing.");
  }

  PyObject *mod=PyImport_ImportModule("sproxel_utils");
  if (!mod)
  {
    PyErr_Print();
    QMessageBox::critical(NULL, "Sproxel Error", "Failed to import sproxel_utils");
  }
  else
  {
    // check required methods
    bool gotErrors=false;

    py_save_project=PyObject_GetAttrString(mod, "save_project");
    if (PyErr_Occurred()) { PyErr_Print(); gotErrors=true; }

    py_load_project=PyObject_GetAttrString(mod, "load_project");
    if (PyErr_Occurred()) { PyErr_Print(); gotErrors=true; }

    py_init_plugin_pathes=PyObject_GetAttrString(mod, "init_plugin_pathes");
    if (PyErr_Occurred()) { PyErr_Print(); gotErrors=true; }

    py_scan_plugins=PyObject_GetAttrString(mod, "scan_plugins");
    if (PyErr_Occurred()) { PyErr_Print(); gotErrors=true; }

    py_register_plugins=PyObject_GetAttrString(mod, "register_plugins");
    if (PyErr_Occurred()) { PyErr_Print(); gotErrors=true; }

    py_unregister_plugins=PyObject_GetAttrString(mod, "unregister_plugins");
    if (PyErr_Occurred()) { PyErr_Print(); gotErrors=true; }

    pycon("Scripted methods: %s", gotErrors ? "some missing" : "all OK");
    if (gotErrors) QMessageBox::critical(NULL, "Sproxel Error", "Some scripted methods are missing.");

    Py_DECREF(mod); mod=NULL;
  }
}


void close_script()
{
  close_python_console();

  Py_XDECREF(py_unregister_plugins); py_unregister_plugins=NULL;
  Py_XDECREF(py_register_plugins); py_register_plugins=NULL;
  Py_XDECREF(py_scan_plugins); py_scan_plugins=NULL;
  Py_XDECREF(py_init_plugin_pathes); py_init_plugin_pathes=NULL;
  Py_XDECREF(py_save_project); py_save_project=NULL;
  Py_XDECREF(py_load_project); py_load_project=NULL;

  #define TOPYT(cls) Py_XDECREF(GLUE(cls, _toPy_py )); GLUE(cls, _toPy_py ) = NULL;
  #define TOCPP(cls) Py_XDECREF(GLUE(cls, _toCpp_py)); GLUE(cls, _toCpp_py) = NULL;
  #include "glue/classGlue.h"

  Py_XDECREF(glue); glue=NULL;

  qDebug() << "fin py...";
  Py_Finalize();
  qDebug() << "fin py ok";
}


//ZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZ//


void script_set_main_window(MainWindow *mwin)
{
  pycon("Adding Sproxel objects to script...");

  PyObject *mod=PyImport_AddModule("sproxel");
  PyObject *o;
  bool gotErrors=false;

  o=QMainWindow_toPy(mwin);
  if (!o) { gotErrors=true; PyErr_Print(); o=Py_None; Py_INCREF(o); }
  PyModule_AddObject(mod, "main_window", o);

  o=PyList_New(1);
  PyList_SetItem(o, 0, qstr_to_py(exe_dir.absoluteFilePath("plugins")));
  PyModule_AddObject(mod, "plugin_pathes", o);

  PyModule_AddObject(mod, "undo", undo_manager_to_py(mwin->undoManager()));

  //== TODO: add more components to script

  // init plugin pathes
  if (py_init_plugin_pathes)
  {
    PyObject *res=PyObject_CallFunctionObjArgs(py_init_plugin_pathes, NULL);
    if (!res) PyErr_Print();
    else Py_DECREF(res);
  }

  pycon("Sproxel objects: %s", gotErrors ? "FAILED to add to script" : "all added");
}


//ZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZ//


void scan_plugins()
{
  if (py_scan_plugins)
  {
    PyObject *res=PyObject_CallFunctionObjArgs(py_scan_plugins, NULL);
    if (!res) PyErr_Print();
    else Py_DECREF(res);
  }
}


void register_plugins()
{
  if (py_register_plugins)
  {
    PyObject *res=PyObject_CallFunctionObjArgs(py_register_plugins, NULL);
    if (!res) PyErr_Print();
    else Py_DECREF(res);
  }
}


void unregister_plugins()
{
  if (py_unregister_plugins)
  {
    PyObject *res=PyObject_CallFunctionObjArgs(py_unregister_plugins, NULL);
    if (!res) PyErr_Print();
    else Py_DECREF(res);
  }
}


//ZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZ//


bool run_script(const QString &fn)
{
  QString filename=exe_dir.filePath(fn);
  #ifdef _WIN32
    FILE *file=_wfopen(filename.utf16(), L"r");
  #else
    FILE *file=fopen(filename.toUtf8(), "r");
  #endif
  if (file)
  {
    pycon("Starting script %S", filename.unicode());

    PyObject *mod=PyImport_AddModule("__main__");
    PyObject *modDict=PyModule_GetDict(mod);

    PyObject *o=PyRun_File(file, qPrintable(filename), Py_file_input, modDict, modDict);
    fclose(file);
    Py_XDECREF(o);

    if (PyErr_Occurred())
    {
      PyErr_Print();
      return false;
    }

    return true;
  }
  else
  {
    pycon("Failed to open script file %S", filename.unicode());
    return false;
  }
}
#endif