summaryrefslogtreecommitdiffstats
path: root/python/astra/astra_c.pyx
diff options
context:
space:
mode:
authorWillem Jan Palenstijn <wjp@usecode.org>2015-02-26 16:30:43 +0100
committerWillem Jan Palenstijn <wjp@usecode.org>2015-02-26 16:30:43 +0100
commit9e2bb413a937aefe57f4fcf343413543ae57258a (patch)
treef4d87d40ae17775e4e3c744476d31d56b5dba64b /python/astra/astra_c.pyx
parent0ca00f4c671d6d583ae77838d3e0d4fcd411f077 (diff)
parente0aca18f687e9f49223ffb24b9be354bed4b150a (diff)
downloadastra-9e2bb413a937aefe57f4fcf343413543ae57258a.tar.gz
astra-9e2bb413a937aefe57f4fcf343413543ae57258a.tar.bz2
astra-9e2bb413a937aefe57f4fcf343413543ae57258a.tar.xz
astra-9e2bb413a937aefe57f4fcf343413543ae57258a.zip
Merge pull request #16 from dmpelt/python-interface
Add Python interface
Diffstat (limited to 'python/astra/astra_c.pyx')
-rw-r--r--python/astra/astra_c.pyx79
1 files changed, 79 insertions, 0 deletions
diff --git a/python/astra/astra_c.pyx b/python/astra/astra_c.pyx
new file mode 100644
index 0000000..342a214
--- /dev/null
+++ b/python/astra/astra_c.pyx
@@ -0,0 +1,79 @@
+#-----------------------------------------------------------------------
+#Copyright 2013 Centrum Wiskunde & Informatica, Amsterdam
+#
+#Author: Daniel M. Pelt
+#Contact: D.M.Pelt@cwi.nl
+#Website: http://dmpelt.github.io/pyastratoolbox/
+#
+#
+#This file is part of the Python interface to the
+#All Scale Tomographic Reconstruction Antwerp Toolbox ("ASTRA Toolbox").
+#
+#The Python interface to the ASTRA Toolbox is free software: you can redistribute it and/or modify
+#it under the terms of the GNU General Public License as published by
+#the Free Software Foundation, either version 3 of the License, or
+#(at your option) any later version.
+#
+#The Python interface to the ASTRA Toolbox is distributed in the hope that it will be useful,
+#but WITHOUT ANY WARRANTY; without even the implied warranty of
+#MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+#GNU General Public License for more details.
+#
+#You should have received a copy of the GNU General Public License
+#along with the Python interface to the ASTRA Toolbox. If not, see <http://www.gnu.org/licenses/>.
+#
+#-----------------------------------------------------------------------
+# distutils: language = c++
+# distutils: libraries = astra
+
+include "config.pxi"
+import six
+from .utils import wrap_from_bytes
+
+from libcpp.string cimport string
+from libcpp cimport bool
+cdef extern from "astra/Globals.h" namespace "astra":
+ int getVersion()
+ string getVersionString()
+ bool cudaEnabled()
+
+IF HAVE_CUDA==True:
+ cdef extern from "../cuda/2d/darthelper.h" namespace "astraCUDA":
+ bool setGPUIndex(int)
+ELSE:
+ def setGPUIndex():
+ pass
+
+def credits():
+ six.print_("""
+All Scale Tomographic Reconstruction Antwerp Toolbox (ASTRA-Toolbox)
+was developed at the University of Antwerp by
+ * Prof. dr. Joost Batenburg
+ * Andrei Dabravolski
+ * Gert Merckx
+ * Willem Jan Palenstijn
+ * Tom Roelandts
+ * Prof. dr. Jan Sijbers
+ * dr. Wim van Aarle
+ * Sander van der Maar
+ * dr. Gert Van Gompel
+
+Python interface written by
+ * Daniel M. Pelt (CWI, Amsterdam)""")
+
+
+def use_cuda():
+ return cudaEnabled()
+
+
+def version(printToScreen=False):
+ if printToScreen:
+ six.print_(wrap_from_bytes(getVersionString()))
+ else:
+ return getVersion()
+
+def set_gpu_index(idx):
+ if use_cuda()==True:
+ ret = setGPUIndex(idx)
+ if not ret:
+ six.print_("Failed to set GPU " + str(idx))