Last week we saw Sigma.js, and as promised here is a graph visualization with Three.js and Neo4j. Three.js is a lightweight 3D library, written by Mr. Doob and a small army of contributors.

The things you can do with Three.js are amazing, and my little demo here doesn’t give it justice, but nonetheless I’ll show you how to build it.



We need to pass the nodes and relationships to Three.js, one of the ways we can do that easily is with the Gon gem, since we’re using sinatra, we’ll use the special gon-sinatra gem.

class App < Sinatra::Base register Gon::Sinatra def nodes neo = Neography::Rest.new cypher_query = " START node = node:nodes_index(type='User')" cypher_query << " RETURN ID(node), node" neo.execute_query(cypher_query)["data"].collect{|n| {"id" => n[0]}.merge(n[1]["data"])} end def edges neo = Neography::Rest.new cypher_query = " START source = node:nodes_index(type='User')" cypher_query << " MATCH source -[rel]-> target" cypher_query << " RETURN ID(rel), ID(source), ID(target)" neo.execute_query(cypher_query)["data"].collect{|n| {"id" => n[0], "source" => n[1], "target" => n[2]} } end get '/' do neo = Neography::Rest.new gon.nodes = nodes gon.edges = edges erb :index end end

In our view we will add”include_gon” which ties our nodes and edges to our html page.

<!doctype html> <html lang="en"> <head> <title>Three.js and Neo4j</title> <%= include_gon %> <meta charset="utf-8"> <script type="text/javascript" src="Three.js"></script> <link type="text/css" rel="stylesheet" href="neo_three.css"/> </head> <body> <script type="text/javascript" src="neo_three.js"></script> </body> </html>

If you view the source of index.html, you’ll see our nodes and edges.

window.gon = {}; gon.nodes=[{"id":1, "rotation_x":5.061454830783556, "name":"zfbushqe", "position_y":256, "position_x":658, "position_z":577, "rotation_y":3.543018381548489}, {"id":2, "rotation_x":4.572762640225143, "name":"afntayhh", "position_y":-22, "position_x":510, "position_z":404, "rotation_y":2.2689280275926285} ... gon.edges=[{"id":3,"source":1,"target":198}, {"id":2,"source":1,"target":39}, {"id":1,"source":1,"target":21} ... ]

To use Three.js, we’ll need to create a camera, a scene and pick a renderer to use.

camera = new THREE.PerspectiveCamera( 75, window.innerWidth / window.innerHeight, 1, 10000 ); camera.position.z = 100; scene = new THREE.Scene(); scene.add( camera ); renderer = new THREE.CanvasRenderer(); renderer.setSize( window.innerWidth, window.innerHeight ); container.appendChild( renderer.domElement );

To create our nodes, we’ll use Spheres and use a normal material. We’ll grab the nodes from gon and use their properties to position and set the orientation of our spheres.

var geometry = new THREE.SphereGeometry( 50, 8, 7, false ); var material = new THREE.MeshNormalMaterial(); group = new THREE.Object3D(); for (n in gon.nodes) { var mesh = new THREE.Mesh( geometry, material ); mesh.position.x = gon.nodes[n].position_x; mesh.position.y = gon.nodes[n].position_y; mesh.position.z = gon.nodes[n].position_z; mesh.rotation.x = gon.nodes[n].rotation_x; mesh.rotation.y = gon.nodes[n].rotation_y; mesh.matrixAutoUpdate = false; mesh.updateMatrix(); group.add( mesh ); } scene.add( group );

For our edges, we’ll create simple lines of random colors between the source and target nodes.

for (n in gon.edges) { var line_segment = new THREE.Geometry(); line_segment.vertices.push( new THREE.Vertex( group.children[gon.edges[n].source - 1].position ) ); line_segment.vertices.push( new THREE.Vertex( group.children[gon.edges[n].target - 1].position ) ); var line = new THREE.Line( line_segment, new THREE.LineBasicMaterial( { color: Math.random() * 0xffffff, opacity: 0.5 } ) ); scene.add(line) }

We’ll animate our visualization by rendering the scene and controlling the camera with the mouse.

document.addEventListener( 'mousemove', onDocumentMouseMove, false ); function onDocumentMouseMove(event) { mouseX = event.clientX - windowHalfX; mouseY = event.clientY - windowHalfY; } function animate() { requestAnimationFrame( animate ); render(); } function render() { camera.position.x += ( mouseX - camera.position.x ) * .05; camera.position.y += ( - mouseY + 200 - camera.position.y ) * .05; camera.lookAt( scene.position ); renderer.render( scene, camera ); }

I’m going to skip the graph creation but all the code is available on github as usual, and you’ll definitively want to see this live on Heroku.

If you want to learn more about Three.js, check out these tutorial and these videos.