from libc.stdint cimport uint32_t
from libc.stdint cimport uint64_t
from spacy.word cimport Lexeme
from spacy.tokens cimport Tokens
from spacy.lexeme cimport LexemeC
from preshed.maps cimport PreshMap

from cymem.cymem cimport Pool

from libcpp.utility cimport pair
from libcpp.vector cimport vector
from libc.stdint cimport uint64_t, int64_t


cdef extern from "Python.h":
    cdef bint Py_UNICODE_ISSPACE(Py_UNICODE ch)
    cdef bint Py_UNICODE_ISALNUM(Py_UNICODE ch)


cdef struct String:
    Py_UNICODE* chars
    size_t n
    uint64_t key


cdef class Lexicon:
    cdef Pool _mem
    cpdef readonly size_t size

    cdef vector[LexemeC*] lexemes

    cpdef Lexeme lookup(self, unicode string)
    cdef LexemeC* get(self, String* s) except NULL
    
    cdef PreshMap _dict
    
    cdef list _string_features
    cdef list _flag_features


cdef class Language:
    cdef Pool _mem
    cdef unicode name
    cdef PreshMap cache
    cdef PreshMap specials
    cpdef readonly Lexicon lexicon

    cdef object prefix_re
    cdef object suffix_re

    cpdef Tokens tokenize(self, unicode text)
    cpdef Lexeme lookup(self, unicode text)

    cdef int _tokenize(self, vector[LexemeC*] *tokens_v, String* string) except -1

    cdef int _split_body_token(self, vector[LexemeC*] *tokens, String* string) except -1
    cdef int _find_prefix(self, Py_UNICODE* characters, size_t length)
    cdef int _find_suffix(self, Py_UNICODE* characters, size_t length)
    
    cdef int _attach_tokens(self, vector[LexemeC*] *tokens, String* string,
                            vector[LexemeC*] *prefixes,
                            vector[LexemeC*] *suffixes) except -1

    cdef int _save_cached(self, vector[LexemeC*] *tokens, uint64_t key, size_t n) except -1