obitools3/python/obitools3/obidms/_obidmscolumn_char.pyx

#cython: language_level=3
 
from .capi.obidmscolumn cimport obi_truncate_and_close_column, \
                                obi_column_get_obichar_with_elt_name_in_view, \
                                obi_column_get_obichar_with_elt_idx_in_view, \
                                obi_column_set_obichar_with_elt_name_in_view, \
                                obi_column_set_obichar_with_elt_idx_in_view
from .capi.obierrno     cimport obi_errno
from .capi.obitypes     cimport OBIChar_NA, obichar_t

from obitools3.utils cimport str2bytes, bytes2str


cdef class OBIDMS_column_char(OBIDMS_column):

    cpdef object get_line(self, index_t line_nb):
        cdef obichar_t value
        cdef object result
        value = obi_column_get_obichar_with_elt_idx_in_view(self.view, (self.pointer)[0], line_nb, 0)
        if obi_errno > 0 :
            raise IndexError(line_nb)
        if value == OBIChar_NA :
            result = None
        else :
            result = bytes2str(value)
        return result

    cpdef set_line(self, index_t line_nb, object value):
        if value is None :
            value = OBIChar_NA
        if obi_column_set_obichar_with_elt_idx_in_view(self.view, (self.pointer)[0], line_nb, 0, str2bytes(value)[0]) < 0:
            raise Exception("Problem setting a value in a column")


cdef class OBIDMS_column_multi_elts_char(OBIDMS_column_multi_elts):

    cpdef object get_item(self, index_t line_nb, str element_name):
        cdef obichar_t value
        cdef object result
        value = obi_column_get_obichar_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 == OBIChar_NA :
            result = None
        else :
            result = bytes2str(value)
        return result

    cpdef object get_line(self, index_t line_nb) :
        cdef obichar_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_obichar_with_elt_idx_in_view(self.view, (self.pointer)[0], line_nb, i)
            if obi_errno > 0 :
                raise IndexError(line_nb)
            if value == OBIChar_NA :
                value_in_result = None
            else :
                value_in_result = bytes2str(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
        return result

    cpdef set_item(self, index_t line_nb, str element_name, object value):
        if value is None :
            value = OBIChar_NA
        if obi_column_set_obichar_with_elt_name_in_view(self.view, (self.pointer)[0], line_nb, str2bytes(element_name), str2bytes(value)[0]) < 0:
            raise Exception("Problem setting a value in a column")