Merge branch 'master'

4 files changed, 33 insertions, 11 deletions

diff --git a/python/astra/functions.py b/python/astra/functions.py
index b826b86..e38b5bc 100644
--- a/python/astra/functions.py
+++ b/python/astra/functions.py
@@ -32,7 +32,11 @@
 
 from . import creators as ac
 import numpy as np
-from six.moves import range
+try:
+    from six.moves import range
+except ImportError:
+    # six 1.3.0
+    from six.moves import xrange as range
 
 from . import data2d
 from . import data3d
diff --git a/python/astra/log_c.pyx b/python/astra/log_c.pyx
index 969cc06..f16329f 100644
--- a/python/astra/log_c.pyx
+++ b/python/astra/log_c.pyx
@@ -52,16 +52,20 @@ cdef extern from "astra/Logging.h" namespace "astra::CLogger":
     void setFormatScreen(const char *fmt)
 
 def log_debug(sfile, sline, message):
-    debug(six.b(sfile),sline,six.b(message))
+    cstr = list(map(six.b,(sfile,message)))
+    debug(cstr[0],sline,cstr[1])
 
 def log_info(sfile, sline, message):
-    info(six.b(sfile),sline,six.b(message))
+    cstr = list(map(six.b,(sfile,message)))
+    info(cstr[0],sline,cstr[1])
 
 def log_warn(sfile, sline, message):
-    warn(six.b(sfile),sline,six.b(message))
+    cstr = list(map(six.b,(sfile,message)))
+    warn(cstr[0],sline,cstr[1])
 
 def log_error(sfile, sline, message):
-    error(six.b(sfile),sline,six.b(message))
+    cstr = list(map(six.b,(sfile,message)))
+    error(cstr[0],sline,cstr[1])
 
 def log_enable():
     enable()
@@ -82,10 +86,12 @@ def log_disableFile():
     disableFile()
 
 def log_setFormatFile(fmt):
-    setFormatFile(six.b(fmt))
+    cstr = six.b(fmt)
+    setFormatFile(cstr)
 
 def log_setFormatScreen(fmt):
-    setFormatScreen(six.b(fmt))
+    cstr = six.b(fmt)
+    setFormatScreen(cstr)
 
 enumList = [LOG_DEBUG,LOG_INFO,LOG_WARN,LOG_ERROR]
 
@@ -93,4 +99,5 @@ def log_setOutputScreen(fd, level):
     setOutputScreen(fd, enumList[level])
 
 def log_setOutputFile(filename, level):
-    setOutputFile(six.b(filename), enumList[level])
\ No newline at end of file
+    cstr = six.b(filename)
+    setOutputFile(cstr, enumList[level])
diff --git a/python/astra/matrix_c.pyx b/python/astra/matrix_c.pyx
index b0d8bc4..d099a75 100644
--- a/python/astra/matrix_c.pyx
+++ b/python/astra/matrix_c.pyx
@@ -27,7 +27,12 @@
 # distutils: libraries = astra
 
 import six
-from six.moves import range
+try:
+    from six.moves import range
+except ImportError:
+    # six 1.3.0
+    from six.moves import xrange as range
+
 import numpy as np
 import scipy.sparse as ss
 
diff --git a/python/astra/optomo.py b/python/astra/optomo.py
index 0c37353..0108674 100644
--- a/python/astra/optomo.py
+++ b/python/astra/optomo.py
@@ -32,7 +32,13 @@ from . import creators
 from . import algorithm
 from . import functions
 import numpy as np
-from six.moves import range, reduce
+from six.moves import reduce
+try:
+    from six.moves import range
+except ImportError:
+    # six 1.3.0
+    from six.moves import xrange as range
+
 import operator
 import scipy.sparse.linalg
 
@@ -158,7 +164,7 @@ class OpTomo(scipy.sparse.linalg.LinearOperator):
         :param extraOptions: Extra options to use during reconstruction (i.e. for cfg['option']).
         :type extraOptions: :class:`dict`
         """
-        self.__checkArray(s, self.sshape)
+        s = self.__checkArray(s, self.sshape)
         sid = self.data_mod.link('-sino',self.pg,s)
         v = np.zeros(self.vshape,dtype=np.float32)
         vid = self.data_mod.link('-vol',self.vg,v)