# cython: language_level=3
from .common cimport *
from .transducer_header cimport TransducerHeader
from .transducer_alphabet cimport TransducerAlphabet

cdef class Transducer:
    """
    A class representing a finite state transducer for morphological analysis.
    """

    def __init__(self, file, TransducerHeader h, TransducerAlphabet a, bint is_weighted=True):
        """
        Initializes the Transducer instance.

        :param file: A file containing the transducer data.
        :param h: A TransducerHeader instance representing the header of the transducer.
        :param a: A TransducerAlphabet instance representing the alphabet of the transducer.
        :param is_weighted: A boolean indicating if the transducer is weighted. Defaults to True.
        """
        self.header = h
        self.alphabet = a
        self.is_weighted = is_weighted
        self.operations = self.alphabet.operations
        self.symbol_map = {}
        self.construct_symbol_map()
        self.index_table = IndexTable(file, h.get_index_table_size())
        self.transition_table = TransitionTable(
            file, h.get_target_table_size(), is_weighted=self.is_weighted)

    cpdef void construct_symbol_map(self):
        for i in range(self.header.get_input_symbol_count()):
            _w = self.alphabet.keyTable[i]
            if len(_w) <= 1:
                # Check if there's already a nested structure for multi-char symbols
                if _w not in self.symbol_map:
                    self.symbol_map[_w] = {None: i}
                else:
                    # Preserve existing nested structure, just add the single-char mapping
                    self.symbol_map[_w][None] = i
            else:
                if _w[0] not in self.symbol_map:
                    self.symbol_map[_w[0]] = {}
                _o = self.symbol_map[_w[0]]
                for j in range(1, len(_w)):
                    _c = _w[j]
                    if _c not in _o:
                        _o[_c] = {}
                    _o = _o[_c]
                if None in _o:
                    raise ValueError("Duplicate symbol in symbol map")
                _o[None] = i