import { RootNode } from '../root-node.js';
import { AbstractNode } from '../abstract-node.js';
import { getCharOfLevel, getCharOfValue, getNextCharId } from './string-format.js';
import {
    FIRST_CHAR_CODE_FOR_ID
} from './string-format.js';


export function bddToMinimalString(bdd: RootNode): string {
    let ret = '';
    let currentCharCode = FIRST_CHAR_CODE_FOR_ID;

    // add leaf node count
    const leafNodeAmount = bdd.getLeafNodes().length;
    if (leafNodeAmount > 99) {
        throw new Error('cannot build string with too many leaf nodes');
    }

    ret += leafNodeAmount.toString().padStart(2, '0');

    const levelsHighestFirst = bdd.levels.slice().reverse();
    const idByNode: Map<AbstractNode, string> = new Map();
    levelsHighestFirst.forEach(level => {
        const nodes = bdd.getNodesOfLevel(level);
        nodes.forEach(node => {
            const stringRep = nodeToString(node, idByNode, currentCharCode);
            currentCharCode = stringRep.nextCode;
            idByNode.set(node, stringRep.id);
            ret += stringRep.str;
        });
    });
    return ret;
}

export function nodeToString(
    node: AbstractNode,
    idByNode: Map<AbstractNode, string>,
    lastCode: number
): {
    id: string,
    str: string,
    nextCode: number
} {
    const nextId = getNextCharId(lastCode);
    switch (node.type) {
        case 'LeafNode':
            const valueChar = getCharOfValue(node.asLeafNode().value);
            return {
                id: nextId.char,
                nextCode: nextId.nextCode,
                str: nextId.char + valueChar
            };
        case 'InternalNode':
            const branch0Id = idByNode.get(node.asInternalNode().branches.getBranch('0'));
            const branch1Id = idByNode.get(node.asInternalNode().branches.getBranch('1'));
            return {
                id: nextId.char,
                nextCode: nextId.nextCode,
                str: nextId.char + branch0Id + branch1Id + getCharOfLevel(node.level)
            };
        case 'RootNode':
            const branch0IdRoot = idByNode.get(node.asRootNode().branches.getBranch('0'));
            const branch1IdRoot = idByNode.get(node.asRootNode().branches.getBranch('1'));
            return {
                id: nextId.char,
                nextCode: nextId.nextCode,
                str: '' + branch0IdRoot + branch1IdRoot + getCharOfLevel(node.level)
            };
        default:
            throw new Error('unknown node type');
    }
}