unipept-visualizations/src/visualizations/treeview/TreeviewNode.ts

The Unipept visualisation library

import DataNode from "./../../DataNode";

export default class TreeviewNode extends DataNode {
    public previousPosition: { x: number, y: number } = { x: 0, y: 0 };

    private selected: boolean = false;
    private collapsed: boolean = false;
    private color: string = "";

    public isCollapsed(): boolean {
        return this.collapsed;
    }

    public setCollapsed(value: boolean): void {
        this.collapsed = value;
    }

    public isSelected(): boolean {
        return this.selected;
    }

    public getColor(): string {
        return this.color;
    }

    /**
     * Mark this node and all of its children as (de)selected.
     *
     * @param value True if the node should be marked as selected, false otherwise.
     */
    public setSelected(value: boolean) {
        this.selected = value;
        for (const child of this.children) {
            (child as TreeviewNode).setSelected(value);
        }
    }

    /**
     * Recursively collapse all children of this node.
     */
    public collapseAll(): void {
        for (const child of (this.children as TreeviewNode[])) {
            child.setCollapsed(true);
            child.collapseAll();
        }
    }

    /**
     * Collapse this node.
     */
    public collapse(): void {
        for (const child of (this.children as TreeviewNode[])) {
            child.setCollapsed(true);
        }
    }

    /**
     * Expand this node and all of its children recursively.
     */
    public expandAll(): void {
        this.expand(100);
    }

    /**
     * Expand this node and all children that are maximum i levels deeper than the current node.
     *
     * @param i Maximum amount of levels deeper at which nodes will be expanded.
     */
    public expand(i: number): void {
        if (i > 0) {
            if (this.children.length > 0) {
                for (const child of (this.children as TreeviewNode[])) {
                    child.setCollapsed(false);
                    child.expand(i - 1);
                }
            }
        }
    }

    /**
     * Recursively sets the color of this node and all of it's children to the provided value.
     *
     * @param color HTML hex string that represents a valid color.
     */
    public setColor(color: string): void {
        this.color = color;
        for (const child of (this.children as TreeviewNode[])) {
            child.setColor(color);
        }
    }
}