Skip to content

Instantly share code, notes, and snippets.

@puzzler10
Last active July 16, 2023 11:07
Show Gist options
  • Save puzzler10/2531c035e8d514f125c4d15433f79d74 to your computer and use it in GitHub Desktop.
Save puzzler10/2531c035e8d514f125c4d15433f79d74 to your computer and use it in GitHub Desktop.
Bounding box force directed graph I
<!DOCTYPE html>
<meta charset="utf-8">
<style>
.links line {
stroke: #999;
stroke-opacity: 0.6;
}
.nodes circle {
stroke: black ;
stroke-width: 0px;
}
svg {
border:1px solid black;
}
</style>
<svg width="400" height="400"></svg>
<script src="https://d3js.org/d3.v4.min.js"></script>
<script>
//create somewhere to put the force directed graph
var svg = d3.select("svg"),
width = +svg.attr("width"),
height = +svg.attr("height");
var radius = 15;
var nodes_data = [
{"name": "Lillian", "sex": "F"},
{"name": "Gordon", "sex": "M"},
{"name": "Sylvester", "sex": "M"},
{"name": "Mary", "sex": "F"},
{"name": "Helen", "sex": "F"},
{"name": "Jamie", "sex": "M"},
{"name": "Jessie", "sex": "F"},
{"name": "Ashton", "sex": "M"},
{"name": "Duncan", "sex": "M"},
{"name": "Evette", "sex": "F"},
{"name": "Mauer", "sex": "M"},
{"name": "Fray", "sex": "F"},
{"name": "Duke", "sex": "M"},
{"name": "Baron", "sex": "M"},
{"name": "Infante", "sex": "M"},
{"name": "Percy", "sex": "M"},
{"name": "Cynthia", "sex": "F"},
{"name": "Feyton", "sex": "M"},
{"name": "Lesley", "sex": "F"},
{"name": "Yvette", "sex": "F"},
{"name": "Maria", "sex": "F"},
{"name": "Lexy", "sex": "F"},
{"name": "Peter", "sex": "M"},
{"name": "Ashley", "sex": "F"},
{"name": "Finkler", "sex": "M"},
{"name": "Damo", "sex": "M"},
{"name": "Imogen", "sex": "F"}
]
//Sample links data
//type: A for Ally, E for Enemy
var links_data = [
{"source": "Sylvester", "target": "Gordon", "type":"A" },
{"source": "Sylvester", "target": "Lillian", "type":"A" },
{"source": "Sylvester", "target": "Mary", "type":"A"},
{"source": "Sylvester", "target": "Jamie", "type":"A"},
{"source": "Sylvester", "target": "Jessie", "type":"A"},
{"source": "Sylvester", "target": "Helen", "type":"A"},
{"source": "Helen", "target": "Gordon", "type":"A"},
{"source": "Mary", "target": "Lillian", "type":"A"},
{"source": "Ashton", "target": "Mary", "type":"A"},
{"source": "Duncan", "target": "Jamie", "type":"A"},
{"source": "Gordon", "target": "Jessie", "type":"A"},
{"source": "Sylvester", "target": "Fray", "type":"E"},
{"source": "Fray", "target": "Mauer", "type":"A"},
{"source": "Fray", "target": "Cynthia", "type":"A"},
{"source": "Fray", "target": "Percy", "type":"A"},
{"source": "Percy", "target": "Cynthia", "type":"A"},
{"source": "Infante", "target": "Duke", "type":"A"},
{"source": "Duke", "target": "Gordon", "type":"A"},
{"source": "Duke", "target": "Sylvester", "type":"A"},
{"source": "Baron", "target": "Duke", "type":"A"},
{"source": "Baron", "target": "Sylvester", "type":"E"},
{"source": "Evette", "target": "Sylvester", "type":"E"},
{"source": "Cynthia", "target": "Sylvester", "type":"E"},
{"source": "Cynthia", "target": "Jamie", "type":"E"},
{"source": "Mauer", "target": "Jessie", "type":"E"},
{"source": "Duke", "target": "Lexy", "type":"A"},
{"source": "Feyton", "target": "Lexy", "type":"A"},
{"source": "Maria", "target": "Feyton", "type":"A"},
{"source": "Baron", "target": "Yvette", "type":"E"},
{"source": "Evette", "target": "Maria", "type":"E"},
{"source": "Cynthia", "target": "Yvette", "type":"E"},
{"source": "Maria", "target": "Jamie", "type":"E"},
{"source": "Maria", "target": "Lesley", "type":"E"},
{"source": "Ashley", "target": "Damo", "type":"A"},
{"source": "Damo", "target": "Lexy", "type":"A"},
{"source": "Maria", "target": "Feyton", "type":"A"},
{"source": "Finkler", "target": "Ashley", "type":"E"},
{"source": "Sylvester", "target": "Maria", "type":"E"},
{"source": "Peter", "target": "Finkler", "type":"E"},
{"source": "Ashley", "target": "Gordon", "type":"E"},
{"source": "Maria", "target": "Imogen", "type":"E"}
]
//set up the simulation
var simulation = d3.forceSimulation()
//add nodes
.nodes(nodes_data);
var link_force = d3.forceLink(links_data)
.id(function(d) { return d.name; });
var charge_force = d3.forceManyBody()
.strength(-100);
var center_force = d3.forceCenter(width / 2, height / 2);
simulation
.force("charge_force", charge_force)
.force("center_force", center_force)
.force("links",link_force);
//add tick instructions:
simulation.on("tick", tickActions );
//draw lines for the links
var link = svg.append("g")
.attr("class", "links")
.selectAll("line")
.data(links_data)
.enter().append("line")
.attr("stroke-width", 2)
.style("stroke", linkColour);
//draw circles for the nodes
var node = svg.append("g")
.attr("class", "nodes")
.selectAll("circle")
.data(nodes_data)
.enter()
.append("circle")
.attr("r", radius)
.attr("fill", circleColour);
var drag_handler = d3.drag()
.on("start", drag_start)
.on("drag", drag_drag)
.on("end", drag_end);
drag_handler(node)
/** Functions **/
//Function to choose what color circle we have
//Let's return blue for males and red for females
function circleColour(d){
if(d.sex =="M"){
return "blue";
} else {
return "pink";
}
}
//Function to choose the line colour and thickness
//If the link type is "A" return green
//If the link type is "E" return red
function linkColour(d){
if(d.type == "A"){
return "green";
} else {
return "red";
}
}
//drag handler
//d is the node
function drag_start(d) {
if (!d3.event.active) simulation.alphaTarget(0.3).restart();
d.fx = d.x;
d.fy = d.y;
}
function drag_drag(d) {
d.fx = d3.event.x;
d.fy = d3.event.y;
}
function drag_end(d) {
if (!d3.event.active) simulation.alphaTarget(0);
d.fx = null;
d.fy = null;
}
function tickActions() {
//constrains the nodes to be within a box
node
.attr("cx", function(d) { return d.x = Math.max(radius, Math.min(width - radius, d.x)); })
.attr("cy", function(d) { return d.y = Math.max(radius, Math.min(height - radius, d.y)); });
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; });
}
</script>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment