optimus_prime optimus_prime - 22 days ago 6
Javascript Question

Filtering nodes and links of a forced directed graph based on a button click

I have a fully working force directed graph . Next stage is to add buttons in the main html data to filter further. I have had no luck so far in making it work. Would appreciate any inputs. Clicking of "High_Txn" should render only those nodes whose links have a value

d.Total_Amount > 100
.

HTML

<style>
.links line {
stroke: #999;
stroke-opacity: 0.6;
}

.nodes circle {
stroke: #fff;
stroke-width: 1.5px;
}

.nodes circle {
stroke: #fff;
stroke-width: 1.5px;
}

div.tooltip {
position: absolute;
background-color: white;
max-width: 200px;
height: auto;
padding: 1px;
border-style: solid;
border-radius: 4px;
border-width: 1px;
box-shadow: 3px 3px 10px rgba(0, 0, 0, .5);
pointer-events: none;
}
</style>
<button type="button" value="Complete">Complete_Data</button>
<button type="button" value="High_Txn">High_Txn</button>




JSON

var IDData = JSON.stringify([
["node/105173", "node/38180995", "Agent", "Customer", "1379644.0", 1, 264, "1374903"],
["node/1061", "node/21373542", "Agent", "Customer", "530848.0", 1, 3000, "529502"],
["node/10750", "node/59648369", "Agent", "Customer", "1454228.0", 1, 120, "1454118"],
["node/10750", "node/78569210", "Agent", "Customer", "1425251.0", 1, 234, "1421416"],
["node/10750", "node/96726118", "Agent", "Customer", "1376239.0", 1, 434, "1376152"],
["node/10946829", "node/11190", "Customer", "Agent", "1409620.0", 20, 3380, "1406665"],
["node/10946829", "node/57774036", "Customer", "Customer", "1460029.0", 3, 960, "1459731"],
["node/109947", "node/97911872", "Agent", "Customer", "1323025.0", 1, 600, "1315582"],..])


I parse through this dynamic JSON data and make it in the format necessary for d3.js as follows:

$(document).ready(function() {

console.log(IDData);
var galData = JSON.parse(IDData);
var startnodes = [];
var endnodes = [];
var startnodetype = [];
var endnodetype = [];
var PayTime = [];
var TXN_COUNT = [];
var Total_Amt = [];
var SendTime = [];
galData.map(function(e, i) {
startnodes.push(e[0]);
endnodes.push(e[1]);
startnodetype.push(e[2]);
endnodetype.push(e[3]);
PayTime.push(e[4]);
TXN_COUNT.push(e[5]);
Total_Amt.push(e[6]);
SendTime.push(e[7]);
});
var final_data = createNodes(startnodes, endnodes, startnodetype, endnodetype, depth, PayTime, TXN_COUNT, Total_Amt, SendTime);
makeGraph("#Network_graph", final_data);

});


Using this I have made a force directed graph inspired from d3.js library

See Jsfiddle for complete code and a working graph.

Now I am trying to add a filter function that will filter the data on the button click:

function filterNetwork() { //not sure if this is the right way to filter , so open to other ideas
force.stop()
originalNodes = force.nodes();
originalLinks = force.links();
influentialNodes = originalNodes.filter(function(d) {
return d.Total_Amount > 100
});
influentialLinks = originalLinks.filter(function(d) {
return influentialNodes.indexOf(d.source) > -1 && influentialNodes.indexOf(d.target) > -1
});

d3.selectAll("g.node")
.data(influentialNodes, function(d) {
return d.id
})
.exit()
.transition()
.duration(12000)
.style("opacity", 0)
.remove();

d3.selectAll("line.link")
.data(influentialLinks, function(d) {
return d.source.id + "-" + d.target.id
})
.exit()
.transition()
.duration(9000)
.style("opacity", 0)
.remove();

force
.nodes(influentialNodes)
.links(influentialLinks)

force.start()
}


Also , how to connect the above filter function to a button click in html? I don't have much expertise in javascript to pull something like this off. So , looking forward for some help

Answer

OK here's the fiddle: https://jsfiddle.net/t6ycuoyo/50/

I've made a button and added an eventListener to it so that when it is clicked it will filter the links and nodes and re-render your graph:

function isUnique(id, nodes){
    for(var i = 0; i < nodes.length; i++){
        if(nodes[i].id == id){
            return false;
        }
    }
    return true;
}

myBtn.addEventListener("click",function(){
  var nodes = [];
  var links = [];
  d3.selectAll("line").filter(function(d,i){
     if(d.total_amt>100){
       if(isUnique(d.source.id, nodes)){
            nodes.push(d.source);
       }

       if(isUnique(d.target.id, nodes)){
            nodes.push(d.target);
       }
       links.push(d);
     }
  });
  final_data.links = links;
  final_data.nodes = nodes;
  d3.select('#Network_graph').selectAll("*").remove();
  makeGraph("#Network_graph", final_data);
});