Merge pull request #16 from dmpelt/python-interface

Add Python interface

1 files changed, 58 insertions, 0 deletions

diff --git a/python/astra/astra.py b/python/astra/astra.py
new file mode 100644
index 0000000..26b1ff0
--- /dev/null
+++ b/python/astra/astra.py
@@ -0,0 +1,58 @@
+#-----------------------------------------------------------------------
+#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/>.
+#
+#-----------------------------------------------------------------------
+
+from . import astra_c as a
+
+def credits():
+    """Print credits of the ASTRA Toolbox."""
+    return a.credits()
+
+
+def use_cuda():
+    """Test if CUDA is enabled.
+    
+    :returns: :class:`bool` -- ``True`` if CUDA is enabled.
+    """
+    return a.use_cuda()
+
+
+def version(printToScreen=False):
+    """Check version of the ASTRA Toolbox.
+    
+    :param printToScreen: If ``True``, print version string. If ``False``, return version integer.
+    :type printToScreen: :class:`bool`
+    :returns: :class:`string` or :class:`int` -- The version string or integer.
+    
+    """
+    return a.version(printToScreen)
+
+def set_gpu_index(idx):
+    """Set default GPU index to use.
+    
+    :param idx: GPU index
+    :type idx: :class:`int`
+    """
+    a.set_gpu_index(idx)