154 lines
4 KiB
Cython
154 lines
4 KiB
Cython
#cython: language_level=3
|
|
#cython: c_string_type=str
|
|
#cython: c_string_encoding=ascii
|
|
#cython: boundscheck=False
|
|
#cython: wrapparound=False
|
|
#cython: cdivision=True
|
|
#cython: embedsignature=True
|
|
|
|
from libcpp.memory cimport shared_ptr, make_shared, dynamic_pointer_cast
|
|
from libcpp.string cimport string
|
|
from libcpp cimport bool
|
|
from range cimport cpp_RangeBase
|
|
from range_factory cimport cpp_RangeFactoryBase, cpp_CRangeFactory
|
|
from array cimport cpp_CArrayBase, cpp_MArray
|
|
from index cimport cpp_DIndex
|
|
|
|
|
|
## ============
|
|
## Range
|
|
## ============
|
|
|
|
cdef class Range:
|
|
cdef shared_ptr[cpp_RangeBase] cpp_range
|
|
|
|
def size(self):
|
|
return self.cpp_range.get().size()
|
|
|
|
def dim(self):
|
|
return self.cpp_range.get().dim()
|
|
|
|
def sub(self):
|
|
return getSubRange(self)
|
|
|
|
def index(self):
|
|
return getRangeIndex(self)
|
|
|
|
|
|
cdef class RangeFactory:
|
|
cdef shared_ptr[cpp_RangeFactoryBase] cpp_rfactory
|
|
|
|
def __cinit__(self,rangetype,**kvargs):
|
|
cdef size_t size = 0
|
|
if rangetype == 'C':
|
|
size = kvargs['size']
|
|
self.cpp_rfactory = dynamic_pointer_cast[cpp_RangeFactoryBase,cpp_CRangeFactory](
|
|
make_shared[cpp_CRangeFactory](size))
|
|
else:
|
|
raise Exception('unknown range type:'+ rangetype)
|
|
|
|
def create(self):
|
|
r = Range()
|
|
r.cpp_range = self.cpp_rfactory.get().create()
|
|
return r
|
|
|
|
|
|
## ===========
|
|
## Index
|
|
## ===========
|
|
|
|
cdef class Index:
|
|
|
|
def lex(self):
|
|
return 0
|
|
|
|
def dim(self):
|
|
return 0
|
|
|
|
cdef class DIndex (Index):
|
|
cdef shared_ptr[cpp_DIndex] cpp_index
|
|
cdef bool itercall
|
|
|
|
def __cinit__(self,_range,_lexpos=0):
|
|
cdef Range r = _range
|
|
cdef size_t l = _lexpos
|
|
self.cpp_index = make_shared[cpp_DIndex] (r.cpp_range,l)
|
|
self.itercall = False
|
|
|
|
def __iter__(self):
|
|
self.cpp_index.get().setlpos(0)
|
|
self.itercall = True # otherwise first (i.e. zeroth) will be excluded
|
|
return self
|
|
|
|
def __next__(self):
|
|
cdef DIndex ret = self
|
|
if self.itercall:
|
|
ret.itercall = False
|
|
return ret
|
|
if self.lex() < self.cpp_index.get().range().get().size()-1:
|
|
ret.cpp_index = make_shared[cpp_DIndex] (self.cpp_index.get().plus(1))
|
|
return ret
|
|
else:
|
|
raise StopIteration
|
|
|
|
def lex(self):
|
|
return self.cpp_index.get().lex()
|
|
|
|
def dim(self):
|
|
return self.cpp_index.get().dim()
|
|
|
|
def stringMeta(self):
|
|
return self.cpp_index.get().stringMeta()
|
|
|
|
def getRangeIndex(_range):
|
|
return DIndex(_range)
|
|
|
|
## ===========
|
|
## Array
|
|
## ===========
|
|
|
|
cdef class Array:
|
|
|
|
def size(self):
|
|
return 0
|
|
|
|
def range(self):
|
|
cdef Range r = Range()
|
|
return r
|
|
|
|
|
|
cdef class Array_Double (Array):
|
|
cdef shared_ptr[cpp_CArrayBase[double]] cpp_array
|
|
|
|
def __cinit__(self,_range):
|
|
cdef Range r = _range
|
|
self.cpp_array = dynamic_pointer_cast[cpp_CArrayBase[double],cpp_MArray[double]] (make_shared[cpp_MArray[double]] (r.cpp_range) )
|
|
|
|
def size(self):
|
|
return self.cpp_array.get().size()
|
|
|
|
def range(self):
|
|
cdef Range r = Range()
|
|
r.cpp_range = self.cpp_array.get().range()
|
|
return r
|
|
|
|
cdef class Array_Range (Array):
|
|
cdef shared_ptr[cpp_CArrayBase[shared_ptr[cpp_RangeBase]]] cpp_array
|
|
|
|
def __cinit__(self,_range):
|
|
cdef Range r = _range
|
|
self.cpp_array = dynamic_pointer_cast[cpp_CArrayBase[shared_ptr[cpp_RangeBase]],cpp_MArray[shared_ptr[cpp_RangeBase]]] (make_shared[cpp_MArray[shared_ptr[cpp_RangeBase]]] (r.cpp_range) )
|
|
|
|
def size(self):
|
|
return self.cpp_array.get().size()
|
|
|
|
def range(self):
|
|
cdef Range r = Range()
|
|
r.cpp_range = self.cpp_array.get().range()
|
|
return r
|
|
|
|
def getSubRange(_range):
|
|
cdef Range r = _range
|
|
cdef Array_Range a = Array_Range()
|
|
a.cpp_array = dynamic_pointer_cast[cpp_CArrayBase[shared_ptr[cpp_RangeBase]],cpp_MArray[shared_ptr[cpp_RangeBase]]] (make_shared[cpp_MArray[shared_ptr[cpp_RangeBase]]] (r.cpp_range.get().sub()) )
|
|
return a
|