jsontonetworkgraph 0.0.1

data/lib/jsontonetworkgraph.rb

@@ -0,0 +1,71 @@
+require 'json'
+load 'node.rb'
+load 'link.rb'
+
+class JSONToNetworkGraph
+
+  def initialize(input, field1, field2)
+    @input = JSON.parse(input)
+    @field1 = field1
+    @field2 = field2
+    @nodehash = Hash.new
+    @linkhash = Hash.new
+    @nodeindex = 0
+  end
+
+  # Generate all the nodes
+  def genNodes
+    @input.each do |i|
+      addupdateNode(@field1, i)
+      addupdateNode(@field2, i)
+    end
+  end
+
+  # Create or update the appropriate node
+  def addupdateNode(fieldname, i)
+    if !(@nodehash.include? i[fieldname])
+      @nodehash[i[fieldname]] = Node.new(@nodeindex, i[fieldname], fieldname, i)
+      @nodeindex += 1
+    else
+      @nodehash[i[fieldname]].update(i)
+    end
+  end
+
+  # Generate all the links
+  def genLinks
+    @input.each do |i|
+      identifier = i[@field1]+ "-" + i[@field2]
+      if !@linkhash.include? identifier
+        source = @nodehash[i[@field1]].getID
+        target = @nodehash[i[@field2]].getID
+
+        @nodehash[i[@field1]].addLink
+        @nodehash[i[@field2]].addLink
+
+        @linkhash[identifier] = Link.new(source, target, i[@field1], i[@field2], i)
+      else
+        @linkhash[identifier].update(i)
+      end
+    end
+  end
+
+  # Generate JSON with nodes and links
+  def genJSON
+    genNodes
+    genLinks
+    
+    nodearray = Array.new
+    @nodehash.each_value do |n|
+      nodearray.push(n.nodeData)
+    end
+
+    linkarray = Array.new
+    @linkhash.each_value do |l|
+      linkarray.push(l.linkData)
+    end
+
+    jsonhash = {:nodes => nodearray, :links => linkarray}
+    JSON.pretty_generate(jsonhash)
+  end
+
+end

data/lib/link.rb

@@ -0,0 +1,18 @@
+class Link
+  def initialize(source, target, field1, field2, value)
+    @source = source
+    @target = target
+    @field1 = field1
+    @field2 = field2
+    @value = Array.new
+    @value.push(value)
+  end
+
+  def update(toadd)
+    @value.push(toadd)
+  end
+
+  def linkData
+    link_hash = {:source => @source, :target => @target, :field1 => @field1, :field2 => @field2, :data => @value}
+  end
+end

data/lib/node.rb

@@ -0,0 +1,26 @@
+class Node
+  def initialize(num, name, type, value)
+    @num = num
+    @name = name
+    @type = type
+    @linkcount = 0
+    @value = Array.new
+    @value.push(value)
+  end
+
+  def update(toadd)
+    @value.push(toadd)
+  end
+  
+  def getID
+    return @num
+  end
+
+  def addLink
+    @linkcount += 1
+  end
+
+  def nodeData
+    json_hash = {:id => @name, :type => @type, :linkcount => @linkcount, :data => @value}
+  end
+end

metadata

@@ -0,0 +1,49 @@
+--- !ruby/object:Gem::Specification
+name: jsontonetworkgraph
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- M. C. McGrath
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-02-20 00:00:00.000000000 Z
+dependencies: []
+description: Generates node and link data from any JSON.
+email: shidash@shidash.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/jsontonetworkgraph.rb
+- lib/node.rb
+- lib/link.rb
+homepage: https://github.com/Shidash/JSONToNetworkGraph
+licenses:
+- GPL
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.23
+signing_key: 
+specification_version: 3
+summary: Turn JSONs into network graphs
+test_files: []
+has_rdoc: