chart = { const links = data.links.map(d => Object.create(d)); const nodes = data.nodes.map(d => Object.create(d)); const [width, height] = [1000, 800]; const svg = d3.select(DOM.svg(width, height)); const aiScoreMinDiff = 0.2; // create link reference let linkedByIndex = {}; data.links.forEach(d => { linkedByIndex[`${d.source},${d.target}`] = true; }); // nodes map let nodesById = {}; data.nodes.forEach(d => { nodesById[d.id] = {...d}; }) console.log(nodesById); const isConnectedAsSource = (a, b) => linkedByIndex[`${a},${b}`]; const isConnectedAsTarget = (a, b) => linkedByIndex[`${b},${a}`]; const isConnected = (a, b) => isConnectedAsTarget(a, b) || isConnectedAsSource(a, b) || a === b; const isEqual = (a, b) => a === b; const nodeRadius = d => .00005 * d.support; //const nodeRadius = 15; const baseGroup = svg.append("g"); function zoomed() { baseGroup.attr("transform", d3.event.transform); } const zoom = d3.zoom() .scaleExtent([0.4, 8]) //.translateExtent([[-100, -100], [width + 1000, height + 1000]]) .on("zoom", zoomed); svg.call(zoom); let ifClicked = false; const simulation = d3.forceSimulation() .force("link", d3.forceLink().id(function(d) { return d.id; }).strength(0.3)) .force("charge", d3.forceManyBody()) .force("y", d3.forceY(height/2).strength(0.1)) .force("center", d3.forceCenter(width / 2, height / 2)) .force("x", d3.forceX(d => { // if(d.network === "FX"){return width/7}else if(d.netwrok ==="Nickelodeon" etc if (d.level === 1) { return width/9 - 50; // width/5 - 50 - 400; } return width/9 * d.level; // width/5 * d.group - 400; }).strength(0.95)) .force("collide", d3.forceCollide().radius(d => nodeRadius(d) + 3).iterations(2)); const link = baseGroup.append("g") .selectAll("line") .data(links) .join("line") .classed('link', true) .style('stroke', d => { //console.log(d.target) // link colors change so set to race //const pplRace = nodesById[d.target].race; //const aiScoreDiff = nodesById[d.target].aiScore - nodesById[d.source].aiScore; const aiScoreDiff = nodesById[d.target].id || nodesById[d.source].id; if(aiScoreDiff < 50){ return'#999'; } else{ // make networks this color if(aiScoreDiff >50 && aiScoreDiff < 65){ return"#e85335"; //make people this color } else{ return "#33FFA7" } } }) .style("stroke-opacity", 0.5); // if (pplRace === "White") { // return '#999'; // } else { // if (pplRace ==="Black") { // return "#84c942"; // inactive -> active // } else { // return "#e85335"; // active -> inactive // } // } // }) const node = baseGroup.append("g") .selectAll("circle") .data(nodes) .join("circle") .classed('node', true) .attr("r", d => nodeRadius(d)) .attr("fill",d =>{ // //make all shows this color // if(d.race === "White"){ // return'#999'; // } else{ // // make networks this color // if(d.race == "Black"){ // return"#84c942"; // //make people this color // } else{ // return "#e85335" // } // } if(Math.abs(d.id) < 50){ return'#999'; } else{ // make networks this color if(d.id >50 && d.id < 65){ return"#e85335"; //make people this color } else{ return "#33FFA7" } } }); // edit context menu const rightClickItems = ['Expand sub-network of the pattern', 'Mark the pattern', 'Mark compound population']; const menuItems = baseGroup.selectAll(".menuitems") .data(rightClickItems) .join('g') .classed('menuitems', true) .attr('visibility', "hidden") .attr('transform', `translate(${0}, ${0})`) //.on('click', rightClickActions); menuItems.append('rect') .attr('x', 0) .attr('y', (d,i) => i * 20) .attr('width', 180) .attr('height', 20); menuItems.append('text') .text(d => d) .attr('x', 3) .attr('y', (d,i) => 13 + i * 20) .style('fill', 'black') .style('font-size', '11px'); function ticked() { link .attr("x1", function(d) { return d.source.x; }) .attr("y1", function(d) { return d.source.y; }) .attr("x2", function(d) { return d.target.x; }) .attr("y2", function(d) { return d.target.y; }); node .attr("cx", function(d) { return d.x; }) .attr("cy", function(d) { return d.y; }); } simulation .nodes(nodes) .on("tick", ticked); simulation.force("link") .links(links); const mouseOverFunction = d => { tooltip.style("visibility", "visible") .html(() => { const content = `Name: ${d.name}`+'
' + `id: ${d.id}`+'
' + `Total Votes: ${d.support}`+'
' +`Average Rating: ${d3.format('.2f')(d.aiScore)} `; return content; }); if (ifClicked) return; node .transition(500) .style('opacity', o => { const isConnectedValue = isConnected(o.id, d.id); if (isConnectedValue) { return 1.0; } return 0.1; }); link .transition(500) .style('stroke-opacity', o => { console.log(o.source === d) return (o.source === d || o.target === d ? 1 : 0.1)}) .transition(500) .attr('marker-end', o => (o.source === d || o.target === d ? 'url(#arrowhead)' : 'url()')); }; const mouseOutFunction = d => { tooltip.style("visibility", "hidden"); if (ifClicked) return; node .transition(500) .style('opacity', 1); link .transition(500) .style("stroke-opacity", o => { console.log(o.value) }); }; const mouseClickFunction = d => { // we don't want the click event bubble up to svg d3.event.stopPropagation(); menuItems.attr('visibility', "hidden"); ifClicked = true; node .transition(500) .style('opacity', 1) link .transition(500); node .transition(500) .style('opacity', o => { const isConnectedValue = isConnected(o.id, d.id); if (isConnectedValue) { return 1.0; } return 0.1 }) link .transition(500) .style('stroke-opacity', o => (o.source === d || o.target === d ? 1 : 0.1)) .transition(500) .attr('marker-end', o => (o.source === d || o.target === d ? 'url(#arrowhead)' : 'url()')); }; const rightClickActions = (d, menuItem, i) => { // d is the clicked node, i is the menuItem index console.log(d.id, menuItem, i) // Expand sub-network of the patter if (i === 0) { // sending d.id (pattern) to backend to get sub-network }; menuItems.attr('visibility', "hidden"); }; const rightClickFunction = d => { d3.event.preventDefault(); // add clicked effect: highlight right-clicked item ifClicked = true; node .transition(500) .style('opacity', 1) link .transition(500); node .transition(500) .style('opacity', o => { const isConnectedValue = isConnected(o.id, d.id); if (isConnectedValue) { return 1.0; } return 0.1 }) link .transition(500) .style('stroke-opacity', o => (o.source === d || o.target === d ? 1 : 0.1)) .transition(500) .attr('marker-end', o => (o.source === d || o.target === d ? 'url(#arrowhead)' : 'url()')); tooltip.style("visibility", "hidden"); const position = {x: d.x, y: d.y}; menuItems.attr('visibility', "visible") .attr('transform', `translate(${position.x}, ${position.y})`) .on('click', (menuItem, i) => rightClickActions(d, menuItem, i)); }; node.on('mouseover', mouseOverFunction) .on('mouseout', mouseOutFunction) .on('click', mouseClickFunction) .on('contextmenu', rightClickFunction) .on('mousemove', () => tooltip.style("top", (d3.event.pageY-10)+"px").style("left",(d3.event.pageX+10)+"px")); svg.on('click', () => { ifClicked = false; node .transition(500) .style('opacity', 1); link .transition(500) .style("stroke-opacity", 0.5) menuItems.attr('visibility', "hidden"); }); //invalidation.then(() => layout.stop()); return svg.node(); }