Extending GoJS

GoJS can be extended in a variety of ways. The most common way to change the standard behavior is to set properties on the GraphObject, Diagram, CommandHandler, Tool, or Layout. But when the desired property does not exist, you might need to override methods of CommandHandler, Tool, Layout, Link, or Node. Methods that you can override are documented in the API reference. This page describes how to override methods, either by replacing a method on an instance (a feature of JavaScript) or by defining a subclass. You should not modify the prototypes of any of the GoJS classes.

Do not modify the prototypes of the GoJS classes.
Only use the properties and methods documented in the API.

In addition to our samples, GoJS provides an extensions gallery, showcasing the creation of custom tools and layouts. Those classes and samples are written in TypeScript, available at ../extensionsJSM/, as ECMAScript/JavaScript modules -- these use the ../release/go-module.js library. We recommend that you copy the files that you need into your project, so that you can adjust how they refer to the GoJS library that you choose and so that you can include them into your own building and packaging procedures.

Note that the API for extension classes 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.

When using modern JavaScript, use arrow functions (e => {}) for event handlers and conversion functions. Use function only when overriding a method and you want access to this object whose method you have overridden.

Command Handler

Overriding the CommandHandler allows you to alter default functionality and create your own key bindings. See the intro page on Commands for more. However, the techniques shown below for overriding methods on Tools and Layouts also applies to the CommandHandler.

Tools and Layouts

GoJS operates on nodes and links using many tools and layouts, all of which are subclasses of the Tool and Layout classes. See the intro page on Tools for more about Tools, and the intro page on Layouts for more about Layouts.

Tools can be modified, or they can be replaced in or added to the Diagram.toolManager. All tools must inherit from the Tool class or from a class that inherits from Tool.

Some of our samples, such as the Pipes sample, contain examples of custom tools. More custom tool examples are available in the extensions gallery. TypeScript versions of those classes are available in ../extensionsJSM/.

Layouts can be modified, or they can be used by setting Diagram.layout or Group.layout. All Layouts must inherit from the Layout class or a class that inherits from Layout.

Some of our samples, such as the Parse Tree sample, contain examples of custom layouts. More custom layout examples are available in the extensions gallery. TypeScript versions of those classes are available in ../extensionsJSM/.

Overriding a Method Without Defining a Subclass

Due to a "feature" of JavaScript, often we can avoid subclassing a Tool in its entirety and merely override a single method. This is common when we want to make a small change to the behavior of a method of a Tool. Here we show how to override the ClickSelectingTool.standardMouseSelect method by modifying the tool instance of a particular Diagram.

One can override Layout methods in this manner also, but that is rarely done -- layouts are almost always subclassed. It cannot be done for layouts that are the value of Group.layout because those layouts may be copied and cannot be shared.

Since we are not creating a new (sub)class, we set the method directly on the Diagram's ClickSelectingTool, which is referenced through its ToolManager. Typical scaffolding for overriding a method in such a manner is as follows:


  var tool = diagram.toolManager.clickSelectingTool;
  tool.standardMouseSelect = function() { // must not be an arrow function =>
    // Maybe do something else before
    // ...

    // In cases where you want normal behavior, call the base functionality.
    // Note the reference to the prototype and the call to 'call' passing it 'this'.
    go.ClickSelectingTool.prototype.standardMouseSelect.call(this);

    // Maybe do something else after
    // ...
  }

As a concrete example, we will override Tool.standardMouseSelect to select only Nodes and Links that have a width and height of less than 50 diagram units. This means that we must find the to-be-selected object using diagram.findPartAt, check its bounds, and quit if the bounds are too large. Otherwise, we call the base functionality to select as we normally might.


  diagram.nodeTemplate =
    $(go.Node, "Auto",
      $(go.Shape, "Rectangle",
        { fill: "white" },
        new go.Binding("fill", "color")),
      $(go.TextBlock,
        { margin: 5 },
        new go.Binding("text", "key"))
    );

  var tool = diagram.toolManager.clickSelectingTool;
  tool.standardMouseSelect = function() { // must not be an arrow function =>
    var diagram = this.diagram;
    var e = diagram.lastInput;
    // to select containing Group if Part.canSelect() is false
    var curobj = diagram.findPartAt(e.documentPoint, false);
    if (curobj !== null) {
      var bounds = curobj.actualBounds;
      // End the selection process if the bounds are greater than 50 width or height
      if (bounds.width > 50 || bounds.height > 50) {
        // If this was a left click with no modifier, we want to act the same as if
        // we are clicking on the Diagram background, so we clear the selection
        if (e.left && !e.control && !e.shift) {
          diagram.clearSelection();
        }
        // Then return, so that we do not call the base functionality
        return;
      }
    }
    // otherwise, call the base functionality
    // (would be a super method when in a subclass)
    go.ClickSelectingTool.prototype.standardMouseSelect.call(this);
  }

  diagram.model = new go.Model([
    { key: "Alpha", color: "lightblue" },
    { key: "Epsilon", color: "thistle" },
    { key: "Psi", color: "lightcoral" },
    { key: "Gamma", color: "lightgreen" }
  ]);

Running this code, we see that the "Epsilon" and "Gamma" nodes are not selectable, because they are both wider than 50. Note that this custom tool does not change the behavior of other tools that might select, such as the DraggingTool or the DragSelectingTool.

Overriding Methods by Subclassing a Layout

Layouts and Tools can be subclassed to create custom classes that inherit the properties and methods of the predefined class.

In modern JavaScript (ECMAScript) or in TypeScript, you can use newer syntax for defining classes:


  export class CascadeLayout extends go.Layout {
    // new data properties (fields) get declared and initialized here

    constructor() {
      super();
      // other initializations can be done here
    }

    // optionally, define property getters and setters here

    // override or define methods
    public doLayout(coll) {
      // Layout logic goes here.
    }
  }

You can define property getters and setters:


  this._offset = new go.Size(12, 12);

  get offset() { return this._offset; }
  set offset(val) {
    if (!(val instanceof go.Size)) {
      throw new Error("new value for CascadeLayout.offset must be a Size, not: " + val);
    }
    if (!this._offset.equals(val)) {
      this._offset = val;
      this.invalidateLayout();
    }
  }

If the layout might be used as the value of Group.layout, you will need to make sure the instance that you set in the Group template can be copied correctly.


  cloneProtected(copy) {
    super.cloneProtected(copy);
    copy._offset = this._offset;
  }

Lastly we'll define a Layout.doLayout, being sure to look at the documentation and accommodate all possible input, as doLayout has one argument that can either be a Diagram, or a Group, or an Iterable collection.

All together, we can see the cascade layout in action:


/**
* This layout arranges nodes in a cascade specified by the offset property
*/
class CascadeLayout extends go.Layout {
  constructor() {
    super();
    this._offset = new go.Size(12, 12);
  }

  cloneProtected(copy) {
    super.cloneProtected(copy);
    copy._offset = this._offset;
  }

  get offset() { return this._offset; }
  set offset(val) {
    if (!(val instanceof go.Size)) {
      throw new Error("new value for CascadeLayout.offset must be a Size, not: " + val);
    }
    if (!this._offset.equals(val)) {
      this._offset = val;
      this.invalidateLayout();
    }
  }

  /**
  * This method positions all Nodes and ignores all Links.
  * @param {Diagram|Group|Iterable} coll the collection of Parts to layout.
  */
  doLayout(coll) {
    // get the Nodes and Links to be laid out
    const parts = this.collectParts(coll);

    // Start the layout at the arrangement origin, a property inherited from Layout
    let x = this.arrangementOrigin.x;
    let y = this.arrangementOrigin.y;
    const offset = this.offset;

    for (let node of parts) {
      if (!(node instanceof go.Node)) continue;  // ignore Links
      node.move(new go.Point(x, y));
      x += offset.width;
      y += offset.height;
    }
  }
}  // end of CascadeLayout

  // Regular diagram setup:

  diagram.layout = new CascadeLayout();

  diagram.nodeTemplate =
    $(go.Node, "Auto",
      $(go.Shape, "Rectangle",
        { fill: "white" },
        new go.Binding("fill", "color")),
      $(go.TextBlock,
        { margin: 5 },
        new go.Binding("text", "key"))
    );

  diagram.model = new go.Model([
    { key: "Alpha", color: "lightblue" },
    { key: "Beta", color: "thistle" },
    { key: "Delta", color: "lightcoral" },
    { key: "Gamma", color: "lightgreen" }
  ]);