79 lines
2.9 KiB
Cython
79 lines
2.9 KiB
Cython
#cython: language_level=3
|
|
|
|
from .capi.obiview cimport obi_column_get_obiint_with_elt_name_in_view, \
|
|
obi_column_get_obiint_with_elt_idx_in_view, \
|
|
obi_column_set_obiint_with_elt_name_in_view, \
|
|
obi_column_set_obiint_with_elt_idx_in_view
|
|
from .capi.obierrno cimport obi_errno
|
|
from .capi.obitypes cimport OBIInt_NA, obiint_t
|
|
|
|
from obitools3.utils cimport str2bytes
|
|
|
|
from cpython.int cimport PyInt_FromLong
|
|
|
|
|
|
cdef class OBIDMS_column_int(OBIDMS_column):
|
|
|
|
cpdef object get_line(self, index_t line_nb):
|
|
cdef obiint_t value
|
|
cdef object result
|
|
value = obi_column_get_obiint_with_elt_idx_in_view(self.view, (self.pointer)[0], line_nb, 0)
|
|
if obi_errno > 0 :
|
|
raise IndexError(line_nb)
|
|
if value == OBIInt_NA :
|
|
result = None
|
|
else :
|
|
result = PyInt_FromLong(value)
|
|
return result
|
|
|
|
cpdef set_line(self, index_t line_nb, object value):
|
|
if value is None :
|
|
value = OBIInt_NA
|
|
if obi_column_set_obiint_with_elt_idx_in_view(self.view, (self.pointer)[0], line_nb, 0, <obiint_t> value) < 0:
|
|
raise Exception("Problem setting a value in a column")
|
|
|
|
|
|
cdef class OBIDMS_column_multi_elts_int(OBIDMS_column_multi_elts):
|
|
|
|
cpdef object get_item(self, index_t line_nb, str element_name):
|
|
cdef obiint_t value
|
|
cdef object result
|
|
value = obi_column_get_obiint_with_elt_name_in_view(self.view, (self.pointer)[0], line_nb, str2bytes(element_name))
|
|
if obi_errno > 0 :
|
|
raise IndexError(line_nb, element_name)
|
|
if value == OBIInt_NA :
|
|
result = None
|
|
else :
|
|
result = PyInt_FromLong(value)
|
|
return result
|
|
|
|
cpdef object get_line(self, index_t line_nb) :
|
|
cdef obiint_t value
|
|
cdef object value_in_result
|
|
cdef dict result
|
|
cdef index_t i
|
|
cdef bint all_NA
|
|
result = {}
|
|
all_NA = True
|
|
for i in range(self.nb_elements_per_line) :
|
|
value = obi_column_get_obiint_with_elt_idx_in_view(self.view, (self.pointer)[0], line_nb, i)
|
|
if obi_errno > 0 :
|
|
raise IndexError(line_nb)
|
|
if value == OBIInt_NA :
|
|
value_in_result = None
|
|
else :
|
|
value_in_result = PyInt_FromLong(value)
|
|
result[self.elements_names[i]] = value_in_result
|
|
if all_NA and (value_in_result is not None) :
|
|
all_NA = False
|
|
if all_NA :
|
|
result = None # TODO discuss
|
|
return result
|
|
|
|
cpdef set_item(self, index_t line_nb, str element_name, object value):
|
|
if value is None :
|
|
value = OBIInt_NA
|
|
if obi_column_set_obiint_with_elt_name_in_view(self.view, (self.pointer)[0], line_nb, str2bytes(element_name), <obiint_t> value) < 0:
|
|
raise Exception("Problem setting a value in a column")
|
|
|