0eaa5aa784
elements per line in a more efficient way + now elements_names are passed as a list + new function to recover only the header of a column
105 lines
3.7 KiB
Cython
105 lines
3.7 KiB
Cython
#cython: language_level=3
|
|
|
|
from .capi.obidmscolumn cimport obi_close_column,\
|
|
obi_truncate_and_close_column, \
|
|
obi_column_get_obiint_with_elt_name, \
|
|
obi_column_get_obiint_with_elt_idx, \
|
|
obi_column_set_obiint_with_elt_name, \
|
|
obi_column_set_obiint_with_elt_idx
|
|
from .capi.obierrno cimport obi_errno
|
|
from .capi.obitypes cimport OBIInt_NA
|
|
|
|
from obitools3.utils cimport str2bytes
|
|
|
|
from cpython.int cimport PyInt_FromLong
|
|
|
|
|
|
cdef class OBIDMS_column_int(OBIDMS_column):
|
|
|
|
cpdef object get_line(self, size_t line_nb):
|
|
cdef obiint_t value
|
|
cdef object result
|
|
value = obi_column_get_obiint_with_elt_idx(self.pointer, 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, size_t line_nb, object value):
|
|
raise Exception("Column is read-only")
|
|
|
|
cpdef close(self):
|
|
if obi_close_column(self.pointer) < 0 :
|
|
raise Exception("Problem closing a column")
|
|
|
|
|
|
cdef class OBIDMS_column_int_writable(OBIDMS_column_int):
|
|
|
|
cpdef set_line(self, size_t line_nb, object value):
|
|
if obi_column_set_obiint_with_elt_idx(self.pointer, line_nb, 0, <obiint_t> value) < 0:
|
|
raise Exception("Problem setting a value in a column")
|
|
|
|
cpdef close(self):
|
|
if obi_truncate_and_close_column(self.pointer) < 0 :
|
|
raise Exception("Problem closing a column")
|
|
|
|
|
|
cdef class OBIDMS_column_int_multi_elts(OBIDMS_column_int):
|
|
|
|
cpdef object get_item(self, size_t line_nb, str element_name):
|
|
cdef obiint_t value
|
|
cdef object result
|
|
value = obi_column_get_obiint_with_elt_name(self.pointer, 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, size_t line_nb) :
|
|
cdef obiint_t value
|
|
cdef object result
|
|
cdef size_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(self.pointer, line_nb, i)
|
|
if obi_errno > 0 :
|
|
raise IndexError(line_nb)
|
|
result[self.elements_names[i]] = PyInt_FromLong(value)
|
|
if all_NA and (value != OBIInt_NA) :
|
|
all_NA = False
|
|
if all_NA :
|
|
result = None
|
|
return result
|
|
|
|
cpdef set_item(self, size_t line_nb, str element_name, obiint_t value):
|
|
raise Exception("Column is read-only")
|
|
|
|
cpdef set_line(self, size_t line_nb, object values):
|
|
raise Exception("Column is read-only")
|
|
|
|
|
|
cdef class OBIDMS_column_int_multi_elts_writable(OBIDMS_column_int_multi_elts):
|
|
|
|
cpdef set_item(self, size_t line_nb, str element_name, obiint_t value):
|
|
if obi_column_set_obiint_with_elt_name(self.pointer, line_nb, str2bytes(element_name), value) < 0:
|
|
raise Exception("Problem setting a value in a column")
|
|
|
|
cpdef set_line(self, size_t line_nb, object values):
|
|
cdef obiint_t value
|
|
for element_name in values :
|
|
value = <obiint_t> values[element_name]
|
|
self.set_item(line_nb, element_name, value)
|
|
|
|
cpdef close(self):
|
|
if obi_truncate_and_close_column(self.pointer) < 0 :
|
|
raise Exception("Problem closing a column")
|
|
|