src/vsscript/vsscript.cpp

   1 #include "VSScript.h"
   2 #include "cython/vapoursynth_api.h"
   3
   4 struct VSScript : public VPYScriptExport {
   5 };
   6
   7 int preInitialized = 0;
   8 int initializationCount = 0;
   9 int scriptId = 1000;
  10 PyThreadState *ts = NULL;
  11 PyGILState_STATE s;
  12
  13 VS_API(int) vseval_init(void) {
  14         if (initializationCount == 0) {
  15                 preInitialized = Py_IsInitialized();
  16                 if (!preInitialized)
  17                         Py_InitializeEx(0);
  18                 PyGILState_STATE s = PyGILState_Ensure();
  19                 int result = import_vapoursynth();
  20                 if (result)
  21                         return 0;
  22                 vpy_initVSScript();
  23                 ts = PyEval_SaveThread();
  24         }
  25         initializationCount++;
  26     return initializationCount;
  27 }
  28
  29 VS_API(int) vseval_finalize(void) {
  30         initializationCount--;
  31     if (initializationCount)
  32         return initializationCount;
  33         //PyEval_RestoreThread(ts);
  34         //PyGILState_Release(s);
  35         //if (!preInitialized)
  36         //      Py_Finalize();
  37     return 0;
  38 }
  39
  40 VS_API(int) vseval_evaluateScript(VSScript **handle, const char *script, const char *errorFilename) {
  41     if (*handle == NULL) {
  42         *handle = new VSScript();
  43         (*handle)->pyenvdict = NULL;
  44         (*handle)->errstr = NULL;
  45                 (*handle)->id = scriptId++;
  46     }
  47     return vpy_evaluateScript(*handle, script, errorFilename);
  48 }
  49
  50 VS_API(void) vseval_freeScript(VSScript *handle) {
  51         if (handle) {
  52                 vpy_freeScript(handle);
  53                 delete handle;
  54         }
  55 }
  56
  57 VS_API(const char *) vseval_getError(VSScript *handle) {
  58     return vpy_getError(handle);
  59 }
  60
  61 VS_API(VSNodeRef *) vseval_getOutput(VSScript *handle, int index) {
  62         return vpy_getOutput(handle, index);
  63 }
  64
  65 VS_API(void) vseval_clearOutput(VSScript *handle, int index) {
  66         vpy_clearOutput(handle, index);
  67 }
  68
  69 VS_API(VSCore *) vseval_getCore(VSScript *handle) {
  70     return vpy_getCore(handle);
  71 }
  72
  73 VS_API(const VSAPI *) vseval_getVSApi(void) {
  74     return vpy_getVSApi();
  75 }
  76
  77 VS_API(int) vseval_getVariable(VSScript *handle, const char *name, VSMap *dst) {
  78         return vpy_getVariable(handle, name, dst);
  79 }
  80
  81 VS_API(void) vseval_setVariable(VSScript *handle, const VSMap *vars) {
  82         vpy_setVariable(handle, (VSMap *)vars);
  83 }
  84
  85 VS_API(int) vseval_clearVariable(VSScript *handle, const char *name) {
  86         return vpy_clearVariable(handle, name);
  87 }
  88
  89 VS_API(void) vseval_clearEnvironment(VSScript *handle) {
  90     vpy_clearEnvironment(handle);
  91 }