bpmn-js/test/integration/custom-elements/CustomElementFactory.js

78 lines
1.8 KiB
JavaScript
Raw Normal View History

'use strict';
var assign = require('lodash/object/assign'),
inherits = require('inherits');
var BpmnElementFactory = require('../../../lib/features/modeling/ElementFactory'),
LabelUtil = require('../../../lib/util/LabelUtil');
function CustomElementFactory(bpmnFactory, moddle) {
BpmnElementFactory.call(this, bpmnFactory, moddle);
var self = this;
this.create = function(elementType, attrs) {
var type = attrs.type,
businessObject,
size;
if (elementType === 'label') {
return self.baseCreate(elementType, assign({ type: 'label' }, LabelUtil.DEFAULT_LABEL_SIZE, attrs));
}
if (/^custom\:/.test(type)) {
type = attrs.type.replace(/^custom\:/, '');
businessObject = {};
size = self._getCustomElementSize(type);
return self.baseCreate(elementType,
assign({ type: elementType, businessObject: businessObject }, attrs, size));
}
return self.createBpmnElement(elementType, attrs);
};
}
inherits(CustomElementFactory, BpmnElementFactory);
module.exports = CustomElementFactory;
CustomElementFactory.$inject = [ 'bpmnFactory', 'moddle' ];
/**
* Sets the *width* and *height* for custom shapes.
*
* The following example shows an interface on how
* to setup the custom element's dimensions.
*
* @example
*
2015-12-07 16:46:53 +00:00
* var shapes = {
* triangle: { width: 40, height: 40 },
* rectangle: { width: 100, height: 20 }
* };
*
2015-12-07 16:46:53 +00:00
* return shapes[type];
*
*
* @param {String} type
*
* @return {Bounds} { width, height}
*/
CustomElementFactory.prototype._getCustomElementSize = function (type) {
if (!type) {
return { width: 100, height: 80 };
}
var shapes = {
triangle: { width: 40, height: 40 },
circle: { width: 140, height: 140 }
};
return shapes[type];
};