import rr from "./railroad.js";

export function railroadDiagram(builder, elementID, elementClass) {
    const diagram = builder(rr);
    const diagramSVG = diagram.toSVG();
    if (elementClass) {
        diagramSVG.classList.add(elementClass);
    }

    const figure = document.getElementById(elementID);
    figure.appendChild(diagramSVG);
}