[python-arrayfire] 51/58: Cleaning up Index

[Ghislain Vaillant]

This is an automated email from the git hooks/post-receive script.

ghisvail-guest pushed a commit to branch master
in repository python-arrayfire.

commit 64fb62ef1a610e0ab5ca2130f480353769a9871d
Author: Pavan Yalamanchili <contact at pavanky.com>
Date:   Sat Sep 24 20:38:38 2016 -0700

    Cleaning up Index
---
 arrayfire/array.py |  5 +----
 arrayfire/index.py | 22 ++++++++++++++++++----
 2 files changed, 19 insertions(+), 8 deletions(-)

diff --git a/arrayfire/array.py b/arrayfire/array.py
index d9f78bf..17c17f2 100644
--- a/arrayfire/array.py
+++ b/arrayfire/array.py
@@ -178,10 +178,7 @@ def _get_info(dims, buf_len):
 
 
 def _get_indices(key):
-
-    S = Index(slice(None))
-    inds = _Index4(S, S, S, S)
-
+    inds = _Index4()
     if isinstance(key, tuple):
         n_idx = len(key)
         for n in range(n_idx):
diff --git a/arrayfire/index.py b/arrayfire/index.py
index 18c68dc..093a883 100644
--- a/arrayfire/index.py
+++ b/arrayfire/index.py
@@ -59,7 +59,20 @@ class Seq(ct.Structure):
             if (S.start is not None):
                 self.begin = c_double_t(S.start)
             if (S.stop is not None):
-                self.end = c_double_t(S.stop - math.copysign(1, self.step))
+                self.end = c_double_t(S.stop)
+
+            # handle special cases
+            if self.begin >= 0 and self.end >=0 and self.end <= self.begin and self.step >= 0:
+                self.begin = 1
+                self.end   = 1
+                self.step  = 1
+            elif self.begin < 0 and self.end < 0 and self.end >= self.begin and self.step <= 0:
+                self.begin = -2
+                self.end   = -2
+                self.step  = -1
+
+            if (S.stop is not None):
+                self.end = self.end - math.copysign(1, self.step)
         else:
             raise IndexError("Invalid type while indexing arrayfire.array")
 
@@ -217,14 +230,15 @@ class Index(ct.Structure):
             arr = c_void_ptr_t(self.idx.arr)
             backend.get().af_release_array(arr)
 
+_span = Index(slice(None))
 class _Index4(object):
-    def __init__(self, idx0, idx1, idx2, idx3):
+    def __init__(self):
         index_vec = Index * 4
-        self.array = index_vec(idx0, idx1, idx2, idx3)
+        self.array = index_vec(_span, _span, _span, _span)
         # Do not lose those idx as self.array keeps
         # no reference to them. Otherwise the destructor
         # is prematurely called
-        self.idxs = [idx0,idx1,idx2,idx3]
+        self.idxs = [_span, _span, _span, _span]
     @property
     def pointer(self):
         return c_pointer(self.array)

-- 
Alioth's /usr/local/bin/git-commit-notice on /srv/git.debian.org/git/debian-science/packages/python-arrayfire.git