Files
fa-diagrams/src/rendering.js
T
2019-07-16 10:19:04 +02:00

349 lines
10 KiB
JavaScript

const convert = require('xml-js');
const utils = require('./utils');
let resources = {
name: 'error',
height: 512,
index: [],
links: {
'arrow-head': {},
'arrow-head-reverse': {},
'line-start': {},
'line-end': {},
'dashed-step': {}
},
icons: {}
};
try {
resources = require('../resources.json');
} catch (err) {
console.error('fa-diagrams: SVG resources could not be loaded: ' + err);
}
/**
* @typedef Node2
* @property {string} name
* @property {number} x
* @property {number} y
* @property {string} icon
*/
/**
* @typedef Link2
* @property {string} from
* @property {string} to
* @property {string|undefined} type
*/
const SUB_DEF = {
'_': 'string',
'text': 'string',
'icon': 'string',
'color': 'string',
'font': 'string',
'font-size': 'number',
'font-style': 'string',
'scale': 'number',
};
const NODE_DEF = {
'name': '!string',
'icon': '!string',
'x': '!number',
'y': '!number',
'color': 'string',
'scale': 'number',
'top': SUB_DEF,
'bottom': SUB_DEF,
'left': SUB_DEF,
'right': SUB_DEF
};
const LINK_DEF = {
'from': '!string',
'to': '!string',
'type': 'string',
'color': 'string',
'scale': 'number',
'size': 'number',
'top': SUB_DEF,
'bottom': SUB_DEF,
};
const DEFAULT_OPTIONS = {
'beautify': false,
'scale': 128,
'h-spacing': 1.3,
'color': 'black',
'icons': {
'scale': 1,
'color': ''
},
'links': {
'scale': 1,
'color': '',
'size': 0
},
'texts': {
'font': 'sans-serif',
'font-size': '20',
'font-style': 'normal',
'color': ''
}
};
const SUBSTITUTES = {
'font-awesome': {
'fas': 'solid',
'far': 'regular',
'fab': 'brands'
}
};
const DEFAULT_SCALE = 0.4;
const LINK_MARGIN = (1 - DEFAULT_SCALE) / 2;
module.exports = (options) => {
options = utils.merge(DEFAULT_OPTIONS, options);
const self = {
/**
* Find icon data from given name
* @param {string} name
* @returns {null|{path: string, width: number}}
*/
getIcon: (name) => {
if (!name || !name.trim())
return null;
let search = utils.ezClone(resources.index);
const spl = name.trim().split(' ').map(t => t.indexOf('fa-') === 0 ? t.substr(3) : t);
for (let i = 0; i < spl.length; i++) {
//replace fas by regular for example
if (Object.keys(SUBSTITUTES[resources.name] || {}).includes(spl[i])) {
spl[i] = SUBSTITUTES[resources.name][spl[i]];
}
if (resources.index.includes(spl[i])) {
search = [spl.splice(i, 1)];
}
}
name = spl[0];
for (let i = 0; i < search.length; i++) {
if (resources.icons[search[i]] && resources.icons[search[i]][name]) {
return resources.icons[search[i]][name];
}
}
return null;
},
/**
* Create the correct path from the type and width
* @param {string} type
* @param {number} width
* @return {string|null}
*/
getLinkPath: (type, width) => {
const arrowHead = 'v46.059c0 21.382 25.851 32.09 40.971 16.971 l86.059 -86.059c9.373-9.373 9.373-24.569 0-33.941l-86.059-86.059c-15.119-15.119-40.971-4.411-40.971 16.971v46.059'; //134.059
const arrowHeadR = 'v-46.059c0-21.382-25.851-32.09-40.971-16.971l-86.059 86.059c-9.373 9.373-9.373 24.568 0 33.941l86.059 86.059c15.119 15.119 40.971 4.411 40.971-16.971v-46.059'; //134.059
const lineEnd = 'c6.627 0 12 -5.373 12 -12v-56c0 -6.627 -5.373 -12 -12 -12'; // 12
const lineStart = 'c-6.627 0-12 5.373-12 12v56c0 6.627 5.373 12 12 12'; // 12
const dashedStep = `${lineStart}h56${lineEnd}`; // 80
switch (type) {
case 'none':
return null;
default:
return `M12 216${resources.links['arrow-head'].path}h${width * 512 - 146.059}${arrowHead}z`;
case 'line':
return `M12 216${lineStart}h${width * 512 - 24}${lineEnd}z`;
case 'double':
return `M134.059 216${arrowHeadR}h${width * 512 - 268.06}${arrowHead}z`;
case 'split-double':
return `M12 126${lineStart}h${width * 512 - 146.059}${arrowHead}M134.059 306${arrowHeadR}h${width * 512 - 146.059}${lineEnd}z`;
case 'dashed':
const n1 = Math.floor((width * 512 - 134.059) / (80 * 2.1));
const space1 = ((width * 512 - 134.059) - n1 * 56 - 24) / (n1);
return `M12 216${new Array(n1).fill(`${dashedStep}h${space1}`).join('')}v80${arrowHead}z`;
case 'dashed-line':
const n2 = Math.floor(width * 512 / (80 * 2.1));
const space2 = (width * 512 - n2 * 56 - 24) / (n2 - 1);
return `M12 216${new Array(n2).fill(dashedStep).join(`h${space2}`)}z`;
case 'dashed-double':
const n3 = Math.floor((width * 512 - 134.059 * 2) / (80 * 2.1));
const space3 = ((width * 512 - 134.059 * 2) - n3 * 56 - 24) / (n3 + 1);
return `M134.059 216${arrowHeadR}m${space3}-80${new Array(n3).fill(`${dashedStep}h${space3}`).join('')}v80${arrowHead}z`;
case 'dashed-split-double':
const n4 = Math.floor((width * 512 - 134.059) / (80 * 2.1));
const space4 = ((width * 512 - 134.059) - n4 * 56 - 24) / (n4);
return `M12 126${new Array(n4).fill(`${dashedStep}h${space4}`).join('')}v80${arrowHead}M134.059 306${arrowHeadR}m${space4} -80${new Array(n4).fill(`${dashedStep}`).join(`h${space4}`)}z`;
}
},
/**
* Get the width and height of the graph of nodes
* @param {Object<string,Node2>} nodes
* @returns {{w: number, h: number}}
*/
getBounds: (nodes) => {
const list = Object.values(nodes);
if (list.length === 0)
return {w: 0, h: 0}; //empty
let maxX = 0;
let maxY = 0;
list.forEach(n => {
maxX = Math.max(n.x, maxX);
maxY = Math.max(n.y, maxY);
});
return {w: maxX + 1, h: maxY + 1};
},
/**
* @param {{g:Object[]}} data
* @param {Object<string,Node2>} nodes
*/
renderNodes: (data, nodes) => {
Object.values(nodes).forEach(node => {
const icon = self.getIcon(node.icon);
if (icon) {
const scale = (node['scale'] || options['icons']['scale']) * DEFAULT_SCALE;
const group = {
'_attributes': {
'transform': `translate(${(node.x + 0.5) * options['h-spacing']} ${node.y + 0.5})`,
},
'g': {
'_attributes': {
'transform': `scale(${scale / 512} ${scale / 512}) translate(${-icon.width / 2} ${-256})`,
'stroke': (node['color'] || options['icons']['color'] || undefined),
'fill': (node['color'] || options['icons']['color'] || undefined)
},
'path': {
'_attributes': {
'd': icon.path,
}
}
}
};
data['g'].push(group);
}
});
},
/**
* @param {{g:Object[]}} data
* @param {Object<string,Node2>} nodes
* @param {Link2[]} links
*/
renderLinks: (data, nodes, links) => {
links.forEach(link => {
const src = nodes[link.from];
const dst = nodes[link.to];
const posX = ((src.x + dst.x) / 2 + 0.5) * options['h-spacing'];
const posY = (src.y + dst.y) / 2 + 0.5;
const angle = Math.atan2(dst.y - src.y, (dst.x - src.x) * options['h-spacing']) * 180 / Math.PI;
const scale = (link['scale'] || options['links']['scale']) * DEFAULT_SCALE;
let size = link['size'] || options['links']['size'];
if (!size) {
let dx = Math.abs(dst.x - src.x) * options['h-spacing'];
if (dx > 0)
dx -= LINK_MARGIN * 2;
let dy = Math.abs(dst.y - src.y);
if (dy > 0)
dy -= LINK_MARGIN * 2;
size = Math.sqrt(Math.pow(dx, 2) + Math.pow(dy, 2)) / scale;
}
const path = self.getLinkPath(link.type, size);
if (!path)
return;
const group = {
'_attributes': {
'transform': `translate(${posX} ${posY}) rotate(${angle})`
},
'g': {
'_attributes': {
'transform': `scale(${scale / 512} ${scale / 512}) translate(${(-256 * size)} ${-256})`,
'stroke': (link['color'] || options['links']['color'] || undefined),
'fill': (link['color'] || options['links']['color'] || undefined)
},
'path': {
'_attributes': {
'd': path
}
}
}
};
data['g'].push(group);
});
},
/**
* Convert xml-js data into correct svg xml string
* @param {Object} data
* @param {{w:number, h:number}} bounds
* @returns {string}
*/
toXML: (data, bounds) => {
const xml = {
'svg': {
'_attributes': {
'xmlns': 'http://www.w3.org/2000/svg',
'viewBox': `0 0 ${bounds.w * options['h-spacing']} ${bounds.h}`,
'width': bounds.w * options['h-spacing'] * options['scale'] / DEFAULT_SCALE,
'height': bounds.h * options['scale'] / DEFAULT_SCALE,
'stroke': options['color'],
'fill': options['color']
}
}
};
Object.keys(data).forEach(key => {
xml['svg'][key] = data[key];
});
return convert.js2xml(xml, {
compact: true,
spaces: options['beautify'] ? '\t' : 0
});
},
/**
* @param {Object<string,Node2>} nodes
* @param {Link2[]} links
*/
compute: (nodes, links) => {
Object.keys(nodes).forEach(key => {
const res = utils.isValid(nodes[key], NODE_DEF);
if (res)
throw `node '${key}' is invalid at key ${res}`;
});
links.forEach((link, i) => {
const res = utils.isValid(link, LINK_DEF);
if (res)
throw `link ${i} (${link.from}->${link.to}) is invalid at key ${res}`;
});
const bounds = self.getBounds(nodes);
const data = {'g': []};
self.renderNodes(data, nodes);
self.renderLinks(data, nodes, links);
return self.toXML(data, bounds);
}
};
return self;
};