bpmn-js/lib/features/modeling/ElementFactory.js

102 lines
2.3 KiB
JavaScript
Raw Normal View History

'use strict';
var _ = require('lodash');
var BaseElementFactory = require('diagram-js/lib/core/ElementFactory');
var LabelUtil = require('../../util/Label');
/**
* A bpmn-aware factory for diagram-js shapes
*/
function ElementFactory(bpmnFactory) {
BaseElementFactory.call(this);
this._bpmnFactory = bpmnFactory;
}
ElementFactory.prototype = Object.create(BaseElementFactory.prototype);
2014-08-05 15:35:54 +00:00
ElementFactory.$inject = [ 'bpmnFactory' ];
module.exports = ElementFactory;
ElementFactory.prototype.baseCreate = BaseElementFactory.prototype.create;
ElementFactory.prototype.create = function(elementType, attrs) {
// no special magic for labels,
// we assume their businessObjects have already been created
// and wired via attrs
if (elementType === 'label') {
return this.baseCreate(elementType, _.extend({ type: 'label' }, LabelUtil.DEFAULT_LABEL_SIZE, attrs));
}
attrs = attrs || {};
var businessObject = attrs.businessObject,
size;
if (!businessObject) {
if (!attrs.type) {
throw new Error('no shape type specified');
}
businessObject = this._bpmnFactory.create(attrs.type);
}
if (!businessObject.di) {
if (elementType === 'connection') {
businessObject.di = this._bpmnFactory.createDiEdge(businessObject, [], {
id: businessObject.id + '_di'
});
} else {
businessObject.di = this._bpmnFactory.createDiShape(businessObject, {}, {
id: businessObject.id + '_di'
});
}
}
if (!!attrs.isExpanded) {
businessObject.di.isExpanded = attrs.isExpanded;
}
size = this._getDefaultSize(businessObject);
attrs = _.extend({
businessObject: businessObject,
id: businessObject.id
}, size, attrs);
return this.baseCreate(elementType, attrs);
};
ElementFactory.prototype._getDefaultSize = function(semantic) {
if (semantic.$instanceOf('bpmn:SubProcess')) {
var isExpanded = semantic.di.isExpanded === true;
if (isExpanded) {
return { width: 350, height: 200 };
} else {
return { width: 100, height: 80 };
}
}
if (semantic.$instanceOf('bpmn:Task')) {
return { width: 100, height: 80 };
}
if (semantic.$instanceOf('bpmn:Gateway')) {
return { width: 50, height: 50 };
}
if (semantic.$instanceOf('bpmn:Event')) {
return { width: 36, height: 36 };
}
return { width: 100, height: 80 };
};