from thinc.typedefs cimport atom_t
from thinc.linear.avgtron cimport AveragedPerceptron
from thinc.structs cimport FeatureC

from .stateclass cimport StateClass
from .arc_eager cimport TransitionSystem
from ..vocab cimport Vocab
from ..tokens.doc cimport Doc
from ..structs cimport TokenC
from ._state cimport StateC


cdef class ParserModel(AveragedPerceptron):
    cdef int set_featuresC(self, atom_t* context, FeatureC* features,
            const StateC* state) nogil
 

cdef class Parser:
    cdef readonly Vocab vocab
    cdef public object model
    cdef readonly ParserModel _model
    cdef readonly TransitionSystem moves
    cdef readonly object cfg
    cdef public object _multitasks

    cdef int parseC(self, StateC* state, TokenC* tokens, int length, int nr_feat) nogil