Python calculate_slice_strides Examples

Example #1

File: iter.py Project: sota/pypy-old

 def apply(self, space, orig_arr):
     arr = orig_arr.implementation
     shape = self.extend_shape(arr.shape)
     r = calculate_slice_strides(arr.shape, arr.start, arr.get_strides(),
                                 arr.get_backstrides(), self.l)
     _, start, strides, backstrides = r
     return W_NDimArray.new_slice(space, start, strides[:], backstrides[:],
                                  shape[:], arr, orig_arr)

Example #2

File: iter.py Project: sota/pypy

 def apply(self, space, orig_arr):
     arr = orig_arr.implementation
     shape = self.extend_shape(arr.shape)
     r = calculate_slice_strides(arr.shape, arr.start, arr.get_strides(),
                                 arr.get_backstrides(), self.l)
     _, start, strides, backstrides = r
     return W_NDimArray.new_slice(space, start, strides[:], backstrides[:],
                                  shape[:], arr, orig_arr)

Example #3

File: interp_iter.py Project: Debug-Orz/Sypy

 def transform(self, arr, t):
     if isinstance(t, BroadcastTransform):
         r = calculate_broadcast_strides(self.strides, self.backstrides,
                                         self.res_shape, t.res_shape)
         return ViewIterator(self.offset, r[0], r[1], t.res_shape)
     elif isinstance(t, ViewTransform):
         r = calculate_slice_strides(self.res_shape, self.offset,
                                     self.strides,
                                     self.backstrides, t.chunks.l)
         return ViewIterator(r[1], r[2], r[3], r[0])

Example #4

 def transform(self, arr, t):
     if isinstance(t, BroadcastTransform):
         r = calculate_broadcast_strides(self.strides, self.backstrides,
                                         self.res_shape, t.res_shape)
         return ViewIterator(self.offset, r[0], r[1], t.res_shape)
     elif isinstance(t, ViewTransform):
         r = calculate_slice_strides(self.res_shape, self.offset,
                                     self.strides, self.backstrides,
                                     t.chunks.l)
         return ViewIterator(r[1], r[2], r[3], r[0])

Example #5

    def apply(self, arr):
        from pypy.module.micronumpy.interp_numarray import W_NDimSlice,\
             VirtualSlice, ConcreteArray

        shape = self.extend_shape(arr.shape)
        if not isinstance(arr, ConcreteArray):
            return VirtualSlice(arr, self, shape)
        r = calculate_slice_strides(arr.shape, arr.start, arr.strides,
                                    arr.backstrides, self.l)
        _, start, strides, backstrides = r
        return W_NDimSlice(start, strides[:], backstrides[:], shape[:], arr)

Example #6

File: interp_iter.py Project: Debug-Orz/Sypy

    def apply(self, arr):
        from pypy.module.micronumpy.interp_numarray import W_NDimSlice,\
             VirtualSlice, ConcreteArray

        shape = self.extend_shape(arr.shape)
        if not isinstance(arr, ConcreteArray):
            return VirtualSlice(arr, self, shape)
        r = calculate_slice_strides(arr.shape, arr.start, arr.strides,
                                    arr.backstrides, self.l)
        _, start, strides, backstrides = r
        return W_NDimSlice(start, strides[:], backstrides[:],
                           shape[:], arr)

Example #7

File: interp_numarray.py Project: purepython/pypy

 def create_slice(self, chunks):
     shape = []
     i = -1
     for i, (start_, stop, step, lgt) in enumerate(chunks):
         if step != 0:
             shape.append(lgt)
     s = i + 1
     assert s >= 0
     shape += self.shape[s:]
     if not isinstance(self, ConcreteArray):
         return VirtualSlice(self, chunks, shape)
     r = calculate_slice_strides(self.shape, self.start, self.strides,
                                 self.backstrides, chunks)
     _, start, strides, backstrides = r
     return W_NDimSlice(start, strides[:], backstrides[:], shape[:], self)

Example #8

 def create_slice(self, chunks):
     shape = []
     i = -1
     for i, (start_, stop, step, lgt) in enumerate(chunks):
         if step != 0:
             shape.append(lgt)
     s = i + 1
     assert s >= 0
     shape += self.shape[s:]
     if not isinstance(self, ConcreteArray):
         return VirtualSlice(self, chunks, shape)
     r = calculate_slice_strides(self.shape, self.start, self.strides,
                                 self.backstrides, chunks)
     _, start, strides, backstrides = r
     return W_NDimSlice(start, strides[:], backstrides[:],
                        shape[:], self)