basicprimitives/src/graphics/structs/PaletteManager.js

Basic Primitives Diagrams for JavaScript - data visualization components library that implements organizational chart and multi-parent dependency diagrams, contains implementations of JavaScript Controls and PDF rendering plugins.

import PaletteItem from './PaletteItem';
import { ConnectorStyleType } from '../../enums';

export default function PaletteManager(options, linesPalette) {
  this.palette = [];
  this.cursor = 0;

  var index, len;

  /* pallete based connectors */
  if (linesPalette.length === 0) {
    /* draw all extra as regular */
    this.palette = [new PaletteItem({
      lineColor: options.linesColor,
      lineWidth: options.linesWidth,
      lineType: options.linesType
    })];
    this.paletteLength = this.palette.length;

    this.regularIndex = 0;
  } else {
    for (index = 0, len = linesPalette.length; index < len; index += 1) {
      this.palette.push(new PaletteItem(linesPalette[index]));
    }
    this.paletteLength = this.palette.length;

    /* regular */
    this.palette.push(new PaletteItem({
      lineColor: options.linesColor,
      lineWidth: options.linesWidth,
      lineType: options.linesType
    }));
    this.regularIndex = this.palette.length - 1;
  }

  /* highlight */
  this.palette.push(new PaletteItem({
    lineColor: options.highlightLinesColor,
    lineWidth: options.highlightLinesWidth,
    lineType: options.highlightLinesType
  }));
  this.highlightIndex = this.palette.length - 1;
};

PaletteManager.prototype.selectPalette = function (index) {
  this.cursor = index % this.paletteLength;
};

PaletteManager.prototype.getPalette = function (connectorStyleType) {
  var index = null;
  switch (connectorStyleType) {
    case ConnectorStyleType.Regular:
      index = this.regularIndex;
      break;
    case ConnectorStyleType.Highlight:
      index = this.highlightIndex;
      break;
    case ConnectorStyleType.Extra:
      index = this.cursor;
      break;
  }
  return this.palette[index];
};