mirror of https://github.com/explosion/spaCy.git
65 lines
1.7 KiB
Cython
65 lines
1.7 KiB
Cython
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 _check_cache(self, vector[LexemeC*] *tokens, String* string) except -1
|
|
cdef int _tokenize(self, vector[LexemeC*] *tokens_v, 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
|
|
|