summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--python/astra/plugin.py7
-rw-r--r--src/PluginAlgorithm.cpp2
2 files changed, 8 insertions, 1 deletions
diff --git a/python/astra/plugin.py b/python/astra/plugin.py
index 891f6c9..bbbc450 100644
--- a/python/astra/plugin.py
+++ b/python/astra/plugin.py
@@ -65,6 +65,13 @@ class base(object):
log.error(str(e))
raise
+ def astra_run(self, its):
+ try:
+ self.run(its)
+ except Exception as e:
+ log.error(str(e))
+ raise
+
def register(name, className):
"""Register plugin with ASTRA.
diff --git a/src/PluginAlgorithm.cpp b/src/PluginAlgorithm.cpp
index a27ce2c..7dcaf68 100644
--- a/src/PluginAlgorithm.cpp
+++ b/src/PluginAlgorithm.cpp
@@ -62,7 +62,7 @@ bool CPluginAlgorithm::initialize(const Config& _cfg){
void CPluginAlgorithm::run(int _iNrIterations){
if(instance==NULL) return;
- PyObject *retVal = PyObject_CallMethod(instance, "run", "i",_iNrIterations);
+ PyObject *retVal = PyObject_CallMethod(instance, "astra_run", "i",_iNrIterations);
if(retVal==NULL) return;
Py_DECREF(retVal);
}