55 lines
1.4 KiB
TypeScript
55 lines
1.4 KiB
TypeScript
import { Element } from './Element';
|
|
import { Edge } from './Edge';
|
|
import { Node } from './Node';
|
|
|
|
export class Graph extends Element {
|
|
public children: Array<Graph> = [];
|
|
public nodes: Array<Node> = [];
|
|
public is_root = false;
|
|
public edges: Array<Edge> = [];
|
|
|
|
// eslint-disable-next-line @typescript-eslint/naming-convention
|
|
public toString (level = 0): string {
|
|
return `subgraph cluster_${this.full_name} {
|
|
${this.children.map ((c) => c.toString (level + 1))
|
|
.join ('\n ')}
|
|
|
|
${this.nodes.map ((c) => c.toString ())
|
|
.join ('\n ')}
|
|
|
|
${this.edges.map ((c) => c.toString ())
|
|
.join ('\n ')}
|
|
}`.replace (/\n/gu, `\n${' '.repeat (level)}`)
|
|
.replace (/^\s+$/gmu, '');
|
|
}
|
|
|
|
public add_node (constructor: () => Node): void {
|
|
const node = constructor ();
|
|
node.parent = this.full_name;
|
|
this.nodes.push (node);
|
|
}
|
|
|
|
public add_graph (constructor: () => Graph): void {
|
|
const graph = constructor ();
|
|
graph.parent = this.full_name;
|
|
graph.update_parent ();
|
|
this.children.push (graph);
|
|
}
|
|
|
|
public update_parent (): void {
|
|
for (const node of this.nodes)
|
|
node.parent = this.full_name;
|
|
|
|
for (const graph of this.children) {
|
|
graph.parent = this.full_name;
|
|
graph.update_parent ();
|
|
}
|
|
}
|
|
|
|
public add_edge (origin: string, target: string): void {
|
|
this.edges.push (
|
|
new Edge (`${this.full_name}_${origin}`, `${this.full_name}_${target}`)
|
|
);
|
|
}
|
|
}
|