f1vis/js/diagrams.js
2017-12-30 01:18:26 +01:00

370 lines
14 KiB
JavaScript

"use strict";
// https://bl.ocks.org/mbostock/3884955
function createLineGraph(containerId, raceData){
// Rough input validation
if(raceData === undefined || raceData.raceInfo === undefined) {
console.error(["Sorry, that raceData is empty. :-(", raceData]);
return; // early return to avoid errors
} else {
console.log(["raceData", raceData]);
}
var enhancedLapData = processor.getEnhancedLapDataPerDriver(raceData);
attachRaceStatistics(enhancedLapData, raceData);
// Configuration
var height = 720;
var width = 1080;
var linePointSize = 5;
var rectSize = 10;
var amountClickedLines = 0;
//ElemTypes of this graph
var elemTypes = {
line: "line",
linepoint: "linepoint",
pitstoppoint: "pitstoppoint",
endpoint: "endpoint"
};
// set the dimensions and margins of the graph
var margin = {top: 50, right: 100, bottom: 50, left: 100},
width = width - margin.left - margin.right,
height = height - margin.top - margin.bottom;
// set the ranges
var x = d3.scaleLinear().range([0, width]);
var y = d3.scaleLinear().range([height, 0]);
// defines how the passed in Data, at "svg.append" shall be interpreted
var lineDataDefinition = d3.line()
.x(function(d) { return x(d.lap); })
.y(function(d) { return y(d.position); });
// append the svg obgect to the body of the page
// appends a 'group' element to 'svg'
// moves the 'group' element to the top left margin
var svg = d3.select(containerId).append("svg")
.attr("width", width + margin.left + margin.right)
.attr("height", height + margin.top + margin.bottom)
.append("g")
.attr("transform",
"translate(" + margin.left + "," + margin.top + ")");
// Scale the range of the data
x.domain([0, raceData.lapTimes.size]);
y.domain([raceData.drivers.length, 1]);
var enhancedLapData = processor.getEnhancedLapDataPerDriver(raceData);
//console.log(["enhancedLapData", enhancedLapData]);
// Adds all lines
enhancedLapData.forEach((driverLapData, driverIndex) => {
//console.log(driverLapData);
svg.append("path")
.data([driverLapData.laps])
.attr("class", "line")
.attr("data-line", driverLapData.driver.driverId)// custom data to Specify the line
.attr("data-opacitychange", 1)
.attr("data-highlighted", 0)
.attr("data-elemtype", elemTypes.line)
.attr("stroke", getColorValue(driverIndex, enhancedLapData.length) )
.attr("d", lineDataDefinition);
driverLapData.laps.forEach((singleLap, singleLapIndex)=> {
//console.log(["driverLaps.forEach", singleLap, singleLapIndex]);
if(singleLap.pitStop){
//Appends a circle for each datapoint
svg.selectAll(".pitstoppoint")
.data([singleLap])
.enter().append("circle") // Uses the enter().append() method
.attr("class", "dot pitstopdot") // Assign a class for styling
.attr("data-line", driverLapData.driver.driverId)
.attr("data-opacitychange", 1)
.attr("data-highlighted", 0)
.attr("data-elemtype", elemTypes.pitstoppoint)
.attr("fill", getColorValue(driverIndex, enhancedLapData.length))
.attr("cx", function(d, i) {return x(d.lap) })
.attr("cy", function(d, i) { return y(d.position) })
.attr("r", linePointSize * 1.2)
.on("click", handleClickOnPoint)
.on("mouseover", handleMouseOverLinePoint)
.on("mouseout", handleMouseOutLinePoint);
// Remove data from driverLapData, since we don't need a generic datapoint for this
driverLapData.laps[singleLapIndex] = {};
}
});
//Appends a circle for each datapoint
svg.selectAll(".linepoint")
.data(driverLapData.laps)
.enter().append("circle") // Uses the enter().append() method
.attr("class", "dot linedot") // Assign a class for styling
.attr("id", function(d, i) { return "circle-linepoint-" + d.lap + "-" + d.driverId; })
.attr("data-line", driverLapData.driver.driverId)
.attr("data-opacitychange", 0)
.attr("data-highlighted", 0)
.attr("data-elemtype", elemTypes.linepoint)
.attr("fill", getColorValue(driverIndex, enhancedLapData.length))
.attr("cx", function(d, i) {return x(d.lap) })
.attr("cy", function(d, i) { return y(d.position) })
.attr("r", linePointSize)
.style("opacity", 0);
//Appends a circle for each datapoint
svg.selectAll(".invisiblelinepoint")
.data(driverLapData.laps)
.enter().append("circle") // Uses the enter().append() method
.attr("class", "dot linedot") // Assign a class for styling
.attr("circle-id", function(d, i) { return "circle-linepoint-" + d.lap + "-" + d.driverId; })
.attr("data-line", driverLapData.driver.driverId)
.attr("data-opacitychange", 0)
.attr("data-highlighted", 0)
.attr("data-elemtype", elemTypes.linepoint)
.attr("fill", getColorValue(driverIndex, enhancedLapData.length))
.attr("cx", function(d, i) {return x(d.lap) })
.attr("cy", function(d, i) { return y(d.position) })
.attr("r", linePointSize*2.4)
.on("click", handleClickOnPoint)
.on("mouseover", handleMouseOverLinePoint)
.on("mouseout", handleMouseOutLinePoint)
.style("opacity", 0);
// in case the driver ended the race too early, get the status why he quit
/*TODO: Mouseover for Rectangle*/
var resultOfDriver = raceData.results.filter((result) => { return result.driverId == driverLapData.driver.driverId; });
if(resultOfDriver.length > 0 && getValidEndingStatusIds().indexOf(resultOfDriver[0].statusId) < 0){
var triangle = d3.symbol()
.type(d3.symbolTriangle)
.size(25);
//get Data for last round
svg.selectAll(".endpoint")
.data([driverLapData.laps[driverLapData.laps.length - 1]])
.enter().append("rect") // Uses the enter().append() method
//.attr("class", "endpoint") // Assign a class for styling
.attr("data-line", driverLapData.driver.driverId)
.attr("data-opacitychange", 1)
.attr("data-highlighted", 0)
.attr("data-elemtype", elemTypes.endpoint)
.attr("fill", getColorValue(driverIndex, enhancedLapData.length))
.attr("x", function(d, i) {return x(d.lap) - rectSize * 1/2 })
.attr("y", function(d, i) { return y(d.position) - rectSize * 1/2 })
.attr("height", rectSize)
.attr("width", rectSize)
.on("click", handleClickOnPoint)
.on("mouseover", handleMouseOverLinePoint)
.on("mouseout", handleMouseOutLinePoint);
/* tried with Cross, didn't work, don't know why
svg.selectAll(".endpoint")
.data([driverLapData.laps[driverLapData.laps.length - 1]])
.enter().append("symbolCircle") // Uses the enter().append() method
.attr("size", 300)
.attr("class", "endpoint") // Assign a class for styling
.attr("fill", getColorValue(driverIndex, enhancedLapData.length))
.attr("transform", function(d) { return "translate(" + x(d.lap) + "," + y(d.position) + ")"; });
*/
}
});
// Add the X Axis
svg.append("g")
.attr("transform", "translate(0," + height + ")")
.call(d3.axisBottom(x));
// Add the Y Axis on both sides
svg.append("g")
.call(
d3.axisLeft(y)
.ticks(raceData.drivers.length)
.tickFormat(function(d) {
return getDriverCodeFromPosAndLap(raceData, 0, d) + " " + d;
})
);
svg.append("g")
.call(
d3.axisRight(y)
.ticks(raceData.drivers.length)
.tickFormat(function(d) {
var driverCode = "";
if(getDriverCodeFromPosAndLap(raceData, raceData.lapTimes.size, d)){
driverCode = getDriverCodeFromPosAndLap(raceData, raceData.lapTimes.size, d);
}else{
driverCode = getDriverCodeFromPosAndLap(raceData, raceData.lapTimes.size - 1, d);
}
return d + " " + driverCode ;
})
)
.attr("transform", "translate( " + (width) + ", 0 )");
function handleClickOnPoint(d,i){
//select elements that are highlightable but are not highlighted
d3.selectAll("[data-opacitychange='" + 1 +"'][data-highlighted='" + 0 +"']")
.style("opacity", 0.10);
// if clicked on already highlighted line, remove highlight
if(this.getAttribute("data-highlighted") == 1){
d3.selectAll("[data-line='" + d.driverId +"'][data-opacitychange='" + 1 +"']")
.style("opacity", 0.10);
d3.selectAll("[data-line='" + d.driverId +"']")
.attr("data-highlighted", 0);
}else{
//select elements that belong to line and are highlightable
d3.selectAll("[data-line='" + d.driverId +"'][data-opacitychange='" + 1 +"']")
.style("opacity", 1);
d3.selectAll("[data-line='" + d.driverId +"']")
.attr("data-highlighted", 1);
}
// if no element highlighted, then everything normal again
var highlightedElements = d3.selectAll("[data-opacitychange='" + 1 +"'][data-highlighted='" + 1 +"']");
if(highlightedElements.size() == 0){
//select elements that are highlightable but are not highlighted
d3.selectAll("[data-opacitychange='" + 1 +"'][data-highlighted='" + 0 +"']")
.style("opacity", 1);
}
}
function handleMouseOverLinePoint(d, i) {
var elemType = d3.select(this).attr("data-elemtype");
//depending on Pitstop and lap different Texts
var textArr = [];
var circleId = "circle-linepoint-" + d.lap + "-" + d.driverId;
var circle = d3.select("#" + circleId);
// Add interactivity
// Use D3 to select element, change color and size
if(elemType === elemTypes.linepoint){
circle
.style("opacity", 1);
textArr = getLapTextArray(raceData,d);
}else if(elemType === elemTypes.pitstoppoint){
circle
.attr("r", linePointSize * 2);
textArr = getPitStopTextArray(raceData,d);
}else if(elemType === elemTypes.endpoint){
d3.select(this)
.attr("height", rectSize * 1.3)
.attr("width", rectSize * 1.3);
textArr = getEndPointTextArray(raceData,d);
}
var tooltipGroup = svg.append("g")
.attr("id", "t" + d.lap + "-" + d.position + "-" + i); // Unique id so it can be removed later
tooltipGroup.append("rect")
.attr("style", "fill:rgb(225,225,225);stroke:black;stroke-width:2;")
.attr("width", "150")
.attr("height", (textArr.length + 1) + "em")
.attr("x", function() { return x(d.lap) + 10; })
.attr("y", function() { return y(d.position) + 10; });
//Necessary to add Text for each Line
textArr.forEach((text, textIndex) =>{
// Specify where to put label of text
tooltipGroup.append("text")
.attr("dy", (textIndex + 1) + "em")
.attr("x", function() { return x(d.lap) + 15; })
.attr("y", function() { return y(d.position) + 15; })
.text(function() {
return text; // Value of the text
});
});
}
function handleMouseOutLinePoint(d, i) {
var dataType = d3.select(this).attr("data-elemtype");
//depending on Pitstop and lap different Texts
var textArr = [];
var circleId = "circle-linepoint-" + d.lap + "-" + d.driverId;
var circle = d3.select("#" + circleId);
// Use D3 to select element, change color back to normal
if(dataType === elemTypes.linepoint){
circle
.attr("r", linePointSize)
.style("opacity", 0);
textArr = getLapTextArray(raceData,d);
}else if(dataType === elemTypes.pitstoppoint){
circle
.attr("r", linePointSize);
textArr = getPitStopTextArray(raceData,d);
}else if(dataType === elemTypes.endpoint){
d3.select(this)
.attr("height", rectSize)
.attr("width", rectSize);
textArr = getEndPointTextArray(raceData,d);
}
textArr.forEach((text, textIndex)=> {
// Select text by id and then remove
d3.select("#t" + d.lap + "-" + d.position + "-" + i).remove(); // Remove text location
});
}
function getLapTextArray(raceData, d){
var driverText = getDriverCodeById(raceData,d.driverId);
var lapText = "Lap: " + d.lap;
var posText = "Pos: " + d.position;
var returnArr = [driverText, lapText, posText];
if(d.time){
var timeText = "Time: " + d.time;
returnArr.push(timeText);
}
return returnArr;
}
function getPitStopTextArray(raceData, d){
var lapTextArr = getLapTextArray(raceData,d);
lapTextArr.push("Stop Nr: " + d.pitStop.stop);
lapTextArr.push("Duration: " + d.pitStop.duration);
return lapTextArr;
}
function getEndPointTextArray(raceData, d){
var status = "";
var endTextArr = getLapTextArray(raceData,d);
var allStatus = queries.getStatus();
for(var key in allStatus){
if(key === undefined) continue;
if(allStatus[key].statusId === d.finished.statusId){
status = allStatus[key].status;
}
}
endTextArr.push("Reason: " + status);
return endTextArr;
}
function getDriverCodeFromPosAndLap(raceData, lapNr, pos){
var driverCode = "";
if(lapNr == 0){
driverCode = getDriverCodeById(raceData, raceData.qualifying[pos - 1].driverId);
}else if(lapNr == raceData.lapTimes.size){
var resultPos = raceData.results[pos -1].position;
var resultLaps = raceData.results[pos -1].laps;
//making sure the Driver finished and drove all laps
if(resultPos && resultLaps == raceData.lapTimes.size){
driverCode = getDriverCodeById(raceData,raceData.results[pos -1].driverId);
}
}else{
if(raceData.lapTimes.get(lapNr)[pos-1]){
driverCode = getDriverCodeById(raceData, raceData.lapTimes.get(lapNr)[pos-1].driverId);
}
}
return driverCode;
}
}