Timo/graphviz-builder
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

allow directional / undirectional

Browse Source
This commit is contained in:
Timo Hocker 2020-04-28 08:39:57 +02:00
parent 6bdbda9e30
commit 373285ec46
3 changed files with 18 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
lib/Edge.ts
View File

@ -6,10 +6,12 @@ export class Edge {
public target: string; public target: string;
public style?: EdgeStyles; public style?: EdgeStyles;
public color?: Color; public color?: Color;
private _directional: boolean;
public constructor (origin: string, target: string) { public constructor (origin: string, target: string, directional: boolean) {
this.origin = origin; this.origin = origin;
this.target = target; this.target = target;
this._directional = directional;
} }
// eslint-disable-next-line @typescript-eslint/naming-convention // eslint-disable-next-line @typescript-eslint/naming-convention
@ -23,7 +25,9 @@ export class Edge {
const attr_string = ` [${attributes.map ((v) => `${v.name}="${v.value}"`) const attr_string = ` [${attributes.map ((v) => `${v.name}="${v.value}"`)
.join (',')}]`; .join (',')}]`;
return `${this.origin} -> ${this.target}${attributes.length > 0 return `${this.origin} -${
this._directional ? '>' : '-'
} ${this.target}${attributes.length > 0
? attr_string ? attr_string
: ''}`; : ''}`;
} }

5
lib/Graph.ts
View File

@ -18,12 +18,13 @@ export class Graph extends Element {
public edges: Array<Edge> = []; public edges: Array<Edge> = [];
public style?: GraphStyles; public style?: GraphStyles;
public color?: Color; public color?: Color;
public directional = true;
// eslint-disable-next-line @typescript-eslint/naming-convention // eslint-disable-next-line @typescript-eslint/naming-convention
public toString (): string { public toString (): string {
const header = this.parent const header = this.parent
? `subgraph cluster_${this.full_name}` ? `subgraph cluster_${this.full_name}`
: `digraph ${this.full_name}`; : `${this.directional ? 'di' : ''}graph ${this.full_name}`;
const attributes = []; const attributes = [];
if (this.color) if (this.color)
attributes.push ({ name: 'color', value: this.color.toString () }); attributes.push ({ name: 'color', value: this.color.toString () });
@ -85,6 +86,6 @@ export class Graph extends Element {
} }
public add_edge (origin: string, target: string): void { public add_edge (origin: string, target: string): void {
this.edges.push (new Edge (origin, target)); this.edges.push (new Edge (origin, target, this.directional));
} }
} }

10
test/Edge.ts
View File

@ -2,7 +2,15 @@ import test from 'ava';
import { Edge, Color, EdgeStyles } from '../lib'; import { Edge, Color, EdgeStyles } from '../lib';
test ('serialize', (t) => { test ('serialize', (t) => {
const e = new Edge ('foo', 'bar'); const e = new Edge ('foo', 'bar', false);
e.color = Color.white;
e.style = EdgeStyles.dashed;
const serialized = e.toString ();
t.is (serialized, 'foo -- bar [style="dashed",color="#ffffff"]');
});
test ('serialize directional', (t) => {
const e = new Edge ('foo', 'bar', true);
e.color = Color.white; e.color = Color.white;
e.style = EdgeStyles.dashed; e.style = EdgeStyles.dashed;
const serialized = e.toString (); const serialized = e.toString ();