mirror of
https://github.com/mermaid-js/mermaid.git
synced 2025-09-07 17:46:44 +02:00
482 lines
13 KiB
JavaScript
482 lines
13 KiB
JavaScript
import {
|
|
select,
|
|
scaleTime,
|
|
min,
|
|
max,
|
|
scaleLinear,
|
|
interpolateHcl,
|
|
axisBottom,
|
|
timeFormat
|
|
} from 'd3';
|
|
import { parser } from './parser/gantt';
|
|
import common from '../common/common';
|
|
import ganttDb from './ganttDb';
|
|
import { configureSvgSize } from '../../utils';
|
|
|
|
parser.yy = ganttDb;
|
|
|
|
const conf = {
|
|
titleTopMargin: 25,
|
|
barHeight: 20,
|
|
barGap: 4,
|
|
topPadding: 50,
|
|
rightPadding: 75,
|
|
leftPadding: 75,
|
|
gridLineStartPadding: 35,
|
|
fontSize: 11,
|
|
fontFamily: '"Open-Sans", "sans-serif"'
|
|
};
|
|
export const setConf = function(cnf) {
|
|
const keys = Object.keys(cnf);
|
|
|
|
keys.forEach(function(key) {
|
|
conf[key] = cnf[key];
|
|
});
|
|
};
|
|
let w;
|
|
export const draw = function(text, id) {
|
|
parser.yy.clear();
|
|
parser.parse(text);
|
|
|
|
const elem = document.getElementById(id);
|
|
w = elem.parentElement.offsetWidth;
|
|
|
|
if (typeof w === 'undefined') {
|
|
w = 1200;
|
|
}
|
|
|
|
if (typeof conf.useWidth !== 'undefined') {
|
|
w = conf.useWidth;
|
|
}
|
|
|
|
const taskArray = parser.yy.getTasks();
|
|
|
|
// Set height based on number of tasks
|
|
const h = taskArray.length * (conf.barHeight + conf.barGap) + 2 * conf.topPadding;
|
|
|
|
// Set viewBox
|
|
elem.setAttribute('viewBox', '0 0 ' + w + ' ' + h);
|
|
const svg = select(`[id="${id}"]`);
|
|
|
|
// Set timescale
|
|
const timeScale = scaleTime()
|
|
.domain([
|
|
min(taskArray, function(d) {
|
|
return d.startTime;
|
|
}),
|
|
max(taskArray, function(d) {
|
|
return d.endTime;
|
|
})
|
|
])
|
|
.rangeRound([0, w - conf.leftPadding - conf.rightPadding]);
|
|
|
|
let categories = [];
|
|
|
|
for (let i = 0; i < taskArray.length; i++) {
|
|
categories.push(taskArray[i].type);
|
|
}
|
|
|
|
const catsUnfiltered = categories; // for vert labels
|
|
|
|
categories = checkUnique(categories);
|
|
|
|
function taskCompare(a, b) {
|
|
const taskA = a.startTime;
|
|
const taskB = b.startTime;
|
|
let result = 0;
|
|
if (taskA > taskB) {
|
|
result = 1;
|
|
} else if (taskA < taskB) {
|
|
result = -1;
|
|
}
|
|
return result;
|
|
}
|
|
|
|
// Sort the task array using the above taskCompare() so that
|
|
// tasks are created based on their order of startTime
|
|
taskArray.sort(taskCompare);
|
|
|
|
makeGant(taskArray, w, h);
|
|
|
|
configureSvgSize(svg, h, w, conf.useMaxWidth);
|
|
|
|
svg
|
|
.append('text')
|
|
.text(parser.yy.getTitle())
|
|
.attr('x', w / 2)
|
|
.attr('y', conf.titleTopMargin)
|
|
.attr('class', 'titleText');
|
|
|
|
function makeGant(tasks, pageWidth, pageHeight) {
|
|
const barHeight = conf.barHeight;
|
|
const gap = barHeight + conf.barGap;
|
|
const topPadding = conf.topPadding;
|
|
const leftPadding = conf.leftPadding;
|
|
|
|
const colorScale = scaleLinear()
|
|
.domain([0, categories.length])
|
|
.range(['#00B9FA', '#F95002'])
|
|
.interpolate(interpolateHcl);
|
|
|
|
makeGrid(leftPadding, topPadding, pageWidth, pageHeight);
|
|
drawRects(tasks, gap, topPadding, leftPadding, barHeight, colorScale, pageWidth, pageHeight);
|
|
vertLabels(gap, topPadding, leftPadding, barHeight, colorScale);
|
|
drawToday(leftPadding, topPadding, pageWidth, pageHeight);
|
|
}
|
|
|
|
function drawRects(theArray, theGap, theTopPad, theSidePad, theBarHeight, theColorScale, w) {
|
|
// Draw background rects covering the entire width of the graph, these form the section rows.
|
|
svg
|
|
.append('g')
|
|
.selectAll('rect')
|
|
.data(theArray)
|
|
.enter()
|
|
.append('rect')
|
|
.attr('x', 0)
|
|
.attr('y', function(d, i) {
|
|
// Ignore the incoming i value and use our order instead
|
|
i = d.order;
|
|
return i * theGap + theTopPad - 2;
|
|
})
|
|
.attr('width', function() {
|
|
return w - conf.rightPadding / 2;
|
|
})
|
|
.attr('height', theGap)
|
|
.attr('class', function(d) {
|
|
for (let i = 0; i < categories.length; i++) {
|
|
if (d.type === categories[i]) {
|
|
return 'section section' + (i % conf.numberSectionStyles);
|
|
}
|
|
}
|
|
return 'section section0';
|
|
});
|
|
|
|
// Draw the rects representing the tasks
|
|
const rectangles = svg
|
|
.append('g')
|
|
.selectAll('rect')
|
|
.data(theArray)
|
|
.enter();
|
|
|
|
rectangles
|
|
.append('rect')
|
|
.attr('id', function(d) {
|
|
return d.id;
|
|
})
|
|
.attr('rx', 3)
|
|
.attr('ry', 3)
|
|
.attr('x', function(d) {
|
|
if (d.milestone) {
|
|
return (
|
|
timeScale(d.startTime) +
|
|
theSidePad +
|
|
0.5 * (timeScale(d.endTime) - timeScale(d.startTime)) -
|
|
0.5 * theBarHeight
|
|
);
|
|
}
|
|
return timeScale(d.startTime) + theSidePad;
|
|
})
|
|
.attr('y', function(d, i) {
|
|
// Ignore the incoming i value and use our order instead
|
|
i = d.order;
|
|
return i * theGap + theTopPad;
|
|
})
|
|
.attr('width', function(d) {
|
|
if (d.milestone) {
|
|
return theBarHeight;
|
|
}
|
|
return timeScale(d.renderEndTime || d.endTime) - timeScale(d.startTime);
|
|
})
|
|
.attr('height', theBarHeight)
|
|
.attr('transform-origin', function(d, i) {
|
|
return (
|
|
(
|
|
timeScale(d.startTime) +
|
|
theSidePad +
|
|
0.5 * (timeScale(d.endTime) - timeScale(d.startTime))
|
|
).toString() +
|
|
'px ' +
|
|
(i * theGap + theTopPad + 0.5 * theBarHeight).toString() +
|
|
'px'
|
|
);
|
|
})
|
|
.attr('class', function(d) {
|
|
const res = 'task';
|
|
|
|
let classStr = '';
|
|
if (d.classes.length > 0) {
|
|
classStr = d.classes.join(' ');
|
|
}
|
|
|
|
let secNum = 0;
|
|
for (let i = 0; i < categories.length; i++) {
|
|
if (d.type === categories[i]) {
|
|
secNum = i % conf.numberSectionStyles;
|
|
}
|
|
}
|
|
|
|
let taskClass = '';
|
|
if (d.active) {
|
|
if (d.crit) {
|
|
taskClass += ' activeCrit';
|
|
} else {
|
|
taskClass = ' active';
|
|
}
|
|
} else if (d.done) {
|
|
if (d.crit) {
|
|
taskClass = ' doneCrit';
|
|
} else {
|
|
taskClass = ' done';
|
|
}
|
|
} else {
|
|
if (d.crit) {
|
|
taskClass += ' crit';
|
|
}
|
|
}
|
|
|
|
if (taskClass.length === 0) {
|
|
taskClass = ' task';
|
|
}
|
|
|
|
if (d.milestone) {
|
|
taskClass = ' milestone ' + taskClass;
|
|
}
|
|
|
|
taskClass += secNum;
|
|
|
|
taskClass += ' ' + classStr;
|
|
|
|
return res + taskClass;
|
|
});
|
|
|
|
// Append task labels
|
|
rectangles
|
|
.append('text')
|
|
.attr('id', function(d) {
|
|
return d.id + '-text';
|
|
})
|
|
.text(function(d) {
|
|
return d.task;
|
|
})
|
|
.attr('font-size', conf.fontSize)
|
|
.attr('x', function(d) {
|
|
let startX = timeScale(d.startTime);
|
|
let endX = timeScale(d.renderEndTime || d.endTime);
|
|
if (d.milestone) {
|
|
startX += 0.5 * (timeScale(d.endTime) - timeScale(d.startTime)) - 0.5 * theBarHeight;
|
|
}
|
|
if (d.milestone) {
|
|
endX = startX + theBarHeight;
|
|
}
|
|
const textWidth = this.getBBox().width;
|
|
|
|
// Check id text width > width of rectangle
|
|
if (textWidth > endX - startX) {
|
|
if (endX + textWidth + 1.5 * conf.leftPadding > w) {
|
|
return startX + theSidePad - 5;
|
|
} else {
|
|
return endX + theSidePad + 5;
|
|
}
|
|
} else {
|
|
return (endX - startX) / 2 + startX + theSidePad;
|
|
}
|
|
})
|
|
.attr('y', function(d, i) {
|
|
// Ignore the incoming i value and use our order instead
|
|
i = d.order;
|
|
return i * theGap + conf.barHeight / 2 + (conf.fontSize / 2 - 2) + theTopPad;
|
|
})
|
|
.attr('text-height', theBarHeight)
|
|
.attr('class', function(d) {
|
|
const startX = timeScale(d.startTime);
|
|
let endX = timeScale(d.endTime);
|
|
if (d.milestone) {
|
|
endX = startX + theBarHeight;
|
|
}
|
|
const textWidth = this.getBBox().width;
|
|
|
|
let classStr = '';
|
|
if (d.classes.length > 0) {
|
|
classStr = d.classes.join(' ');
|
|
}
|
|
|
|
let secNum = 0;
|
|
console.log(conf);
|
|
for (let i = 0; i < categories.length; i++) {
|
|
if (d.type === categories[i]) {
|
|
secNum = i % conf.numberSectionStyles;
|
|
}
|
|
}
|
|
|
|
let taskType = '';
|
|
if (d.active) {
|
|
if (d.crit) {
|
|
taskType = 'activeCritText' + secNum;
|
|
} else {
|
|
taskType = 'activeText' + secNum;
|
|
}
|
|
}
|
|
|
|
if (d.done) {
|
|
if (d.crit) {
|
|
taskType = taskType + ' doneCritText' + secNum;
|
|
} else {
|
|
taskType = taskType + ' doneText' + secNum;
|
|
}
|
|
} else {
|
|
if (d.crit) {
|
|
taskType = taskType + ' critText' + secNum;
|
|
}
|
|
}
|
|
|
|
if (d.milestone) {
|
|
taskType += ' milestoneText';
|
|
}
|
|
|
|
// Check id text width > width of rectangle
|
|
if (textWidth > endX - startX) {
|
|
if (endX + textWidth + 1.5 * conf.leftPadding > w) {
|
|
return classStr + ' taskTextOutsideLeft taskTextOutside' + secNum + ' ' + taskType;
|
|
} else {
|
|
return (
|
|
classStr +
|
|
' taskTextOutsideRight taskTextOutside' +
|
|
secNum +
|
|
' ' +
|
|
taskType +
|
|
' width-' +
|
|
textWidth
|
|
);
|
|
}
|
|
} else {
|
|
return classStr + ' taskText taskText' + secNum + ' ' + taskType + ' width-' + textWidth;
|
|
}
|
|
});
|
|
}
|
|
|
|
function makeGrid(theSidePad, theTopPad, w, h) {
|
|
let xAxis = axisBottom(timeScale)
|
|
.tickSize(-h + theTopPad + conf.gridLineStartPadding)
|
|
.tickFormat(timeFormat(parser.yy.getAxisFormat() || conf.axisFormat || '%Y-%m-%d'));
|
|
|
|
svg
|
|
.append('g')
|
|
.attr('class', 'grid')
|
|
.attr('transform', 'translate(' + theSidePad + ', ' + (h - 50) + ')')
|
|
.call(xAxis)
|
|
.selectAll('text')
|
|
.style('text-anchor', 'middle')
|
|
.attr('fill', '#000')
|
|
.attr('stroke', 'none')
|
|
.attr('font-size', 10)
|
|
.attr('dy', '1em');
|
|
}
|
|
|
|
function vertLabels(theGap, theTopPad) {
|
|
const numOccurances = [];
|
|
let prevGap = 0;
|
|
|
|
for (let i = 0; i < categories.length; i++) {
|
|
numOccurances[i] = [categories[i], getCount(categories[i], catsUnfiltered)];
|
|
}
|
|
|
|
svg
|
|
.append('g') // without doing this, impossible to put grid lines behind text
|
|
.selectAll('text')
|
|
.data(numOccurances)
|
|
.enter()
|
|
.append(function(d) {
|
|
const rows = d[0].split(common.lineBreakRegex);
|
|
const dy = -(rows.length - 1) / 2;
|
|
|
|
const svgLabel = document.createElementNS('http://www.w3.org/2000/svg', 'text');
|
|
svgLabel.setAttribute('dy', dy + 'em');
|
|
|
|
for (let j = 0; j < rows.length; j++) {
|
|
const tspan = document.createElementNS('http://www.w3.org/2000/svg', 'tspan');
|
|
tspan.setAttribute('alignment-baseline', 'central');
|
|
tspan.setAttribute('x', '10');
|
|
if (j > 0) tspan.setAttribute('dy', '1em');
|
|
tspan.textContent = rows[j];
|
|
svgLabel.appendChild(tspan);
|
|
}
|
|
return svgLabel;
|
|
})
|
|
.attr('x', 10)
|
|
.attr('y', function(d, i) {
|
|
if (i > 0) {
|
|
for (let j = 0; j < i; j++) {
|
|
prevGap += numOccurances[i - 1][1];
|
|
return (d[1] * theGap) / 2 + prevGap * theGap + theTopPad;
|
|
}
|
|
} else {
|
|
return (d[1] * theGap) / 2 + theTopPad;
|
|
}
|
|
})
|
|
.attr('class', function(d) {
|
|
for (let i = 0; i < categories.length; i++) {
|
|
if (d[0] === categories[i]) {
|
|
return 'sectionTitle sectionTitle' + (i % conf.numberSectionStyles);
|
|
}
|
|
}
|
|
return 'sectionTitle';
|
|
});
|
|
}
|
|
|
|
function drawToday(theSidePad, theTopPad, w, h) {
|
|
const todayMarker = ganttDb.getTodayMarker();
|
|
if (todayMarker === 'off') {
|
|
return;
|
|
}
|
|
|
|
const todayG = svg.append('g').attr('class', 'today');
|
|
const today = new Date();
|
|
const todayLine = todayG.append('line');
|
|
|
|
todayLine
|
|
.attr('x1', timeScale(today) + theSidePad)
|
|
.attr('x2', timeScale(today) + theSidePad)
|
|
.attr('y1', conf.titleTopMargin)
|
|
.attr('y2', h - conf.titleTopMargin)
|
|
.attr('class', 'today');
|
|
|
|
if (todayMarker !== '') {
|
|
todayLine.attr('style', todayMarker.replace(/,/g, ';'));
|
|
}
|
|
}
|
|
|
|
// from this stackexchange question: http://stackoverflow.com/questions/1890203/unique-for-arrays-in-javascript
|
|
function checkUnique(arr) {
|
|
const hash = {};
|
|
const result = [];
|
|
for (let i = 0, l = arr.length; i < l; ++i) {
|
|
if (!hash.hasOwnProperty(arr[i])) { // eslint-disable-line
|
|
// it works with objects! in FF, at least
|
|
hash[arr[i]] = true;
|
|
result.push(arr[i]);
|
|
}
|
|
}
|
|
return result;
|
|
}
|
|
|
|
// from this stackexchange question: http://stackoverflow.com/questions/14227981/count-how-many-strings-in-an-array-have-duplicates-in-the-same-array
|
|
function getCounts(arr) {
|
|
let i = arr.length; // const to loop over
|
|
const obj = {}; // obj to store results
|
|
while (i) {
|
|
obj[arr[--i]] = (obj[arr[i]] || 0) + 1; // count occurrences
|
|
}
|
|
return obj;
|
|
}
|
|
|
|
// get specific from everything
|
|
function getCount(word, arr) {
|
|
return getCounts(arr)[word] || 0;
|
|
}
|
|
};
|
|
|
|
export default {
|
|
setConf,
|
|
draw
|
|
};
|