import { RootNode } from './root-node.js';
import { lastChar } from './util.js';
import { InternalNode } from './internal-node.js';
import { LeafNode } from './leaf-node.js';
import type { TruthTable, NonLeafNode, BooleanString } from './types.js';

export function createBddFromTruthTable(
    truthTable: TruthTable
): RootNode {
    const root = new RootNode();

    const firstKey = truthTable.keys().next().value;
    const keyLength = firstKey.length;
    const mustBeSize = Math.pow(2, keyLength);
    if (truthTable.size !== mustBeSize) {
        throw new Error('truth table has missing entries');
    }


    for (const [stateSet, value] of truthTable) {
        let lastNode: NonLeafNode = root;

        // itterate over each char of the state
        for (let i = 0; i < (stateSet.length - 1); i++) {
            const level = i + 1;
            const state: BooleanString = stateSet.charAt(i) as BooleanString;

            // if node for this state-char not exists, add new one
            if (!lastNode.branches.getBranch(state)) {
                lastNode.branches.setBranch(
                    state,
                    new InternalNode(
                        level,
                        root,
                        lastNode,
                    )
                );
            }
            lastNode = lastNode.branches.getBranch(state) as NonLeafNode;
        }

        // last node is leaf-node
        const lastState = lastChar(stateSet) as BooleanString;
        if (lastNode.branches.getBranch(lastState)) {
            throw new Error('leafNode already exists, this should not happen');
        }
        lastNode.branches.setBranch(
            lastState,
            new LeafNode(
                stateSet.length,
                root,
                value,
                lastNode,
            )
        );
    }

    return root;
}