create-gojs-kit
Version:
A CLI for downloading GoJS samples, extensions, and docs
255 lines (253 loc) • 9.49 kB
JavaScript
/*
* Copyright 1998-2025 by Northwoods Software Corporation. All Rights Reserved.
*/
/*
* This is an extension and not part of the main GoJS library.
* The source code for this is at extensionsJSM/SerpentineLayout.ts.
* Note that the API for this class may change with any version, even point releases.
* If you intend to use an extension in production, you should copy the code to your own source directory.
* Extensions can be found in the GoJS kit under the extensions or extensionsJSM folders.
* See the Extensions intro page (https://gojs.net/latest/intro/extensions.html) for more information.
*/
/**
* A custom {@link go.Layout} that lays out a chain of nodes in a snake-like fashion.
*
* This layout assumes the graph is a chain of Nodes,
* positioning nodes in horizontal rows back and forth, alternating between left-to-right
* and right-to-left within the {@link wrap} limit.
* {@link spacing} controls the distance between nodes.
* {@link leftSpot} and {@link rightSpot} determine the Spots to use for the {@link go.Link.fromSpot} and {@link go.Link.toSpot}.
*
* When this layout is the Diagram.layout, it is automatically invalidated when the viewport changes size.
*
* If you want to experiment with this extension, try the <a href="../../samples/Serpentine.html">Serpentine Layout</a> sample.
* @category Layout Extension
*/
class SerpentineLayout extends go.Layout {
/**
* Constructs a SerpentineLayout and sets the {@link isViewportSized} property to true.
*/
constructor(init) {
super();
this.isViewportSized = true;
this._spacing = new go.Size(30, 30);
this._wrap = NaN;
this._root = null;
this._leftSpot = go.Spot.Left;
this._rightSpot = go.Spot.Right;
if (init)
Object.assign(this, init);
}
/**
* Gets or sets the {@link go.Size} whose width specifies the horizontal space between nodes
* and whose height specifies the minimum vertical space between nodes.
*
* The default value is 30x30.
*/
get spacing() {
return this._spacing;
}
set spacing(val) {
if (!this._spacing.equals(val)) {
if (!(val instanceof go.Size))
throw new Error('new value for SerpentineLayout.spacing must be a Size, not: ' + val);
this._spacing = val;
this.invalidateLayout();
}
}
/**
* Gets or sets the total width of the layout.
*
* The default value is NaN, which for {@link go.Diagram.layout}s means that it uses
* the {@link go.Diagram.viewportBounds}.
*/
get wrap() {
return this._wrap;
}
set wrap(val) {
if (this._wrap !== val) {
if (typeof val !== 'number')
throw new Error('SerpentineLayout.wrap must be a number');
this._wrap = val;
this.invalidateLayout();
}
}
/**
* Gets or sets the starting node of the sequence.
*
* The default value is null, which causes the layout to look for a node without any incoming links.
*/
get root() {
return this._root;
}
set root(val) {
if (this._root !== val) {
if (val !== null && !(val instanceof go.Node))
throw new Error('SerpentinelLayout.root must be a go.Node');
this._root = val;
this.invalidateLayout();
}
}
/**
* Gets or sets the Spot to use on the left side of a Node.
*
* The default value is {@link go.Spot.Left}.
*/
get leftSpot() {
return this._leftSpot;
}
set leftSpot(val) {
if (!this._leftSpot.equals(val)) {
if (!(val instanceof go.Spot))
throw new Error('SerpentinelLayout.leftSpot must be a Spot');
this._leftSpot = val;
this.invalidateLayout();
}
}
/**
* Gets or sets the Spot to use on the right side of a Node.
*
* The default value is {@link go.Spot.Right}.
*/
get rightSpot() {
return this._rightSpot;
}
set rightSpot(val) {
if (!this._rightSpot.equals(val)) {
if (!(val instanceof go.Spot))
throw new Error('SerpentinelLayout.rightSpot must be a Spot');
this._rightSpot = val;
this.invalidateLayout();
}
}
/**
* Copies properties to a cloned Layout.
*/
cloneProtected(copy) {
super.cloneProtected(copy);
copy._spacing = this._spacing;
copy._wrap = this._wrap;
// don't copy _root
copy._leftSpot = this._leftSpot;
copy._rightSpot = this._rightSpot;
}
/**
* This method actually positions all of the Nodes, assuming that the ordering of the nodes
* is given by a single link from one node to the next.
* This respects the {@link spacing} and {@link wrap} properties to affect the layout.
* @param collection - A collection of {@link go.Part}s.
*/
doLayout(collection) {
const diagram = this.diagram;
const coll = this.collectParts(collection);
let root = this.root;
if (root === null) {
// find a root node -- one without any incoming links
const it = coll.iterator;
while (it.next()) {
const n = it.value;
if (!(n instanceof go.Node))
continue;
if (root === null)
root = n;
if (n.findLinksInto().count === 0) {
root = n;
break;
}
}
}
// couldn't find a root node
if (root === null)
return;
const spacing = this.spacing;
// calculate the width at which we should start a new row
let wrap = this.wrap;
if (diagram !== null && isNaN(wrap)) {
if (this.group === null) {
// for a top-level layout, use the Diagram.viewportBounds
const pad = diagram.padding;
wrap = Math.max(spacing.width * 2, diagram.viewportBounds.width - 24 - pad.left - pad.right);
}
else {
wrap = 1000; // provide a better default value?
}
}
// implementations of doLayout that do not make use of a LayoutNetwork
// need to perform their own transactions
if (diagram !== null)
diagram.startTransaction('Serpentine Layout');
// start on the left, at Layout.arrangementOrigin
this.arrangementOrigin = this.initialOrigin(this.arrangementOrigin);
let x = this.arrangementOrigin.x;
let rowh = 0;
let y = this.arrangementOrigin.y;
let increasing = true;
let node = root;
while (node !== null) {
const orignode = node;
if (node.containingGroup !== null)
node = node.containingGroup;
const b = this.getLayoutBounds(node);
// get the next node, if any
let nextlink = null;
for (const it = orignode.findLinksOutOf().iterator; it.next();) {
if (coll.has(it.value)) {
nextlink = it.value;
break;
}
}
let nextnode = nextlink !== null ? nextlink.toNode : null;
const orignextnode = nextnode;
if (nextnode !== null && nextnode.containingGroup !== null)
nextnode = nextnode.containingGroup;
const nb = nextnode !== null ? this.getLayoutBounds(nextnode) : new go.Rect();
if (increasing) {
node.move(new go.Point(x, y));
x += b.width;
rowh = Math.max(rowh, b.height);
if (x + spacing.width + nb.width > wrap) {
y += rowh + spacing.height;
x = wrap - spacing.width;
rowh = 0;
increasing = false;
if (nextlink !== null) {
nextlink.fromSpot = this.rightSpot;
nextlink.toSpot = this.rightSpot;
}
}
else {
x += spacing.width;
if (nextlink !== null) {
nextlink.fromSpot = this.rightSpot;
nextlink.toSpot = this.leftSpot;
}
}
}
else {
x -= b.width;
node.move(new go.Point(x, y));
rowh = Math.max(rowh, b.height);
if (x - spacing.width - nb.width < 0) {
y += rowh + spacing.height;
x = 0;
rowh = 0;
increasing = true;
if (nextlink !== null) {
nextlink.fromSpot = this.leftSpot;
nextlink.toSpot = this.leftSpot;
}
}
else {
x -= spacing.width;
if (nextlink !== null) {
nextlink.fromSpot = this.leftSpot;
nextlink.toSpot = this.rightSpot;
}
}
}
node = orignextnode;
}
if (diagram !== null)
diagram.commitTransaction('Serpentine Layout');
}
}