steffi 0.0.1

data/Gemfile

@@ -0,0 +1,10 @@
+source :rubygems
+gemspec
+
+group :development do
+  gem 'pry'
+end
+
+group :test do
+  gem 'rspec'
+end

data/lib/steffi/communities.rb

@@ -0,0 +1,14 @@
+module Steffi
+
+  module Igraph
+    bind :community_multilevel, [:pointer, :pointer, :pointer, :pointer, :pointer], :int
+  end
+
+  class Graph
+    def communities
+      membership = Vector.new
+      Igraph.community_multilevel ptr, nil, membership.ptr, nil, nil
+      membership
+    end
+  end
+end

data/lib/steffi/create.rb

@@ -0,0 +1,78 @@
+module Steffi
+
+  module Igraph
+    bind :read_graph_edgelist, [:pointer, :pointer, :int, :bool], :int
+    bind :atlas, [:pointer, :int], :int
+    bind :cited_type_game, [:pointer, :int, :pointer, :pointer, :int, :bool], :int
+    bind :erdos_renyi_game, [:pointer, :int, :int, :double, :bool, :bool], :int
+    bind :forest_fire_game, [:pointer, :int, :double, :double, :int, :bool], :int
+    bind :famous, [:pointer, :string], :int
+    bind :full, [:pointer, :int, :bool, :bool], :int
+    bind :ring, [:pointer, :int, :bool, :bool, :bool], :int
+    bind :star, [:pointer, :int, :int, :int], :int
+    bind :tree, [:pointer, :int, :int, :int], :int
+  end
+
+  class Graph
+
+    class << self
+      
+      def load path
+        stream = C.fopen path, 'r'
+        g = from :read_graph_edgelist, stream, 0, false
+        C.fclose stream
+        g
+      end
+
+      def atlas i
+        from :atlas, i
+      end
+
+      def citations n, p
+        types = (0...n).to_a
+        types = Vector.from_a(types).ptr
+        attractivity = 1.0 / p
+        prefs = [attractivity] * (n+1)
+        prefs = Vector.from_a(prefs).ptr
+        from :cited_type_game, n, types, prefs, 1, false
+      end
+
+      def erdos_renyi n, k
+        from :erdos_renyi_game, 1, n, k, false, false
+      end
+
+      def famous name
+        from :famous, name.to_s
+      end
+
+      def forest_fire n, k
+        from :forest_fire_game, n, 0.01/k, 1, 2, false
+      end
+
+      def full n
+        from :full, n, false, false
+      end
+
+      def ring n
+        from :ring, n, false, false, true
+      end
+
+      def star n
+        from :star, n, 2, 0
+      end
+
+      def tree n, k
+        from :tree, n, k, 2
+      end
+
+      private #-------------------------------------------------------
+
+      def from name, *args
+        Graph.new do |g|
+          Igraph.send name, g.ptr, *args
+        end
+      end
+
+    end
+  end
+end

data/lib/steffi/edge.rb

@@ -0,0 +1,15 @@
+module Steffi
+  class Edge
+    attr_accessor :graph, :from, :to
+
+    def initialize graph,from,to
+      @graph = graph
+      @from = from
+      @to = to
+    end
+
+    def to_s
+      "(#{from},#{to})"
+    end
+  end
+end

data/lib/steffi/error.rb

@@ -0,0 +1,16 @@
+module Steffi
+  module Igraph
+    callback :error_handler, [:string, :string, :int, :int], :void
+
+    bind :set_error_handler, [:error_handler], :error_handler
+    bind :strerror, [:int], :string
+
+    class Error < StandardError; end
+
+    ErrorHandler = Proc.new do |reason, file, line, code|
+      raise Error.new "#{strerror code}: #{reason} (#{file}:#{line})"
+    end
+
+    set_error_handler ErrorHandler
+  end
+end

data/lib/steffi/ffi.rb

@@ -0,0 +1,26 @@
+require 'ffi'
+
+module Steffi
+
+  module C
+    extend FFI::Library
+    ffi_lib :c
+
+    attach_function :fopen, [:pointer, :pointer], :pointer
+    attach_function :fclose, [:pointer], :int
+  end
+
+  module Igraph
+    extend FFI::Library
+    ffi_lib File.expand_path('../../../ext/libigraph.0.6.dylib', __FILE__)
+
+    def self.bind name, in_t, out_t
+      attach_function :"igraph_#{name}", in_t, out_t
+    end
+
+    def self.method_missing name, *args, &block
+      iname = :"igraph_#{name}"
+      respond_to?(iname) ? send(iname, *args, &block) : super
+    end
+  end
+end

data/lib/steffi/graph.rb

@@ -0,0 +1,47 @@
+module Steffi
+
+  module Igraph
+    bind :write_graph_edgelist, [:pointer, :pointer], :int
+    bind :destroy, [:pointer], :int
+  end
+
+  class Graph
+
+    class Struct < FFI::ManagedStruct
+      layout :n,        :int,
+             :directed, :bool,
+             :from,     Vector::Struct,
+             :to,       Vector::Struct,
+             :oi,       Vector::Struct,
+             :ii,       Vector::Struct,
+             :os,       Vector::Struct,
+             :is,       Vector::Struct,
+             :attr,     :pointer
+
+      def self.release ptr
+        Igraph.destroy ptr
+      end
+    end
+
+    attr_reader :ptr
+
+    def initialize 
+      @ptr = FFI::MemoryPointer.new Struct
+      yield self if block_given?
+    end
+
+    def dump path
+      stream = C.fopen path, 'w'
+      Igraph.write_graph_edgelist ptr, stream
+      C.fclose stream
+    end
+
+    def d3
+      {
+        nodes: communities.each_with_index.map { |c,i| { name: i, group: c.to_i } },
+        links: edges.map { |e| { source: e.from, target: e.to, weight: 1 } }
+      }
+    end
+
+  end
+end

data/lib/steffi/metrics.rb

@@ -0,0 +1,41 @@
+module Steffi
+  module Igraph
+    bind :diameter, [:pointer, :pointer, :pointer, :pointer, :pointer, :int, :int], :int
+    bind :ecount,   [:pointer], :int
+    bind :vcount,   [:pointer], :int
+
+    bind :edge,     [:pointer, :int, :pointer, :pointer], :int
+  end
+
+  class Graph
+    def diameter
+      d = FFI::MemoryPointer.new :int
+      Igraph.diameter ptr, d, nil, nil, nil, 0, 1
+      d.get_int 0
+    end
+
+    def ecount
+      Igraph.ecount ptr
+    end
+
+    def vcount
+      Igraph.vcount ptr
+    end
+
+    def stats
+      Hash[ %w{ diameter ecount vcount }.map { |n| [ n, send(n) ] } ]
+    end
+
+    def edge i
+      from = FFI::MemoryPointer.new :int
+      to = FFI::MemoryPointer.new :int
+      Igraph.edge ptr, i, from, to
+      Edge.new self, from.get_int(0), to.get_int(0)
+    end
+
+    def edges
+      0.upto(ecount - 1).map { |i| edge i }
+    end
+
+  end
+end

data/lib/steffi/vector.rb

@@ -0,0 +1,65 @@
+module Steffi
+
+  module Igraph
+    bind :vector_init, [:pointer, :long], :int
+    bind :vector_e, [:pointer, :long], :double
+    bind :vector_set, [:pointer, :long, :double], :void
+    bind :vector_size, [:pointer], :long
+    bind :vector_push_back, [:pointer, :double], :int
+    bind :vector_destroy, [:pointer], :void
+  end
+
+  class Vector
+
+    class Struct < FFI::ManagedStruct
+      layout :stor_begin, :pointer,
+             :stor_end,   :pointer,
+             :end,        :pointer
+
+      def self.release ptr
+        Igraph.vector_destroy ptr
+      end
+    end
+
+    include Enumerable
+
+    attr_reader :ptr
+
+    def initialize
+      @ptr = FFI::MemoryPointer.new Vector::Struct
+      Igraph.vector_init ptr, 0
+    end
+
+    def << obj
+      Igraph.vector_push_back ptr, obj.to_f
+    end
+
+    def [] i
+      Igraph.vector_e ptr, i
+    end
+
+    def []= i, obj
+      Igraph.vector_set ptr, i, obj.to_f
+    end
+
+    def size
+      Igraph.vector_size ptr
+    end
+
+    def each
+      0.upto(size-1) { |i| yield self[i] }
+    end
+
+    def to_s
+      to_a.to_s
+    end
+
+    def self.from_a ary
+      v = new
+      ary.each { |i| v << i }
+      v
+    end
+
+  end
+
+end

data/lib/steffi/version.rb

@@ -0,0 +1,12 @@
+module Steffi
+  class Version
+    MAJOR = 0
+    MINOR = 0
+    PATCH = 1
+    PRE = nil
+
+    def self.to_s
+      [MAJOR, MINOR, PATCH, PRE].compact.join '.'
+    end
+  end
+end

data/lib/steffi/vertex.rb

@@ -0,0 +1,7 @@
+module Steffi
+  class Vertex
+    def initialize graph
+      @graph = graph
+    end
+  end
+end

data/lib/steffi.rb

@@ -0,0 +1,15 @@
+require 'steffi/ffi'
+require 'steffi/error'
+
+require 'steffi/vector'
+
+require 'steffi/graph'
+require 'steffi/vertex'
+require 'steffi/edge'
+
+require 'steffi/create'
+require 'steffi/metrics'
+require 'steffi/communities'
+
+module Steffi
+end

data/steffi.gemspec

@@ -0,0 +1,18 @@
+lib = File.expand_path '../lib', __FILE__
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'steffi/version'
+
+Gem::Specification.new do |s|
+  s.name        = 'steffi'
+  s.version     = Steffi::Version
+  s.date        = '2013-01-27'
+  s.summary     = 'FFI bindings to the igraph library'
+  s.description = s.summary
+  s.authors     = ['James Dabbs']
+  s.email       = 'jamesdabbs@gmail.com'
+  s.files       = %w{ Gemfile steffi.gemspec }
+  s.files      += Dir['lib/**/*.rb'] + Dir['ext/**/*']
+  s.homepage    = 'http://github.com/jamesdabbs/steffi'
+
+  s.add_dependency 'ffi'
+end

metadata

@@ -0,0 +1,74 @@
+--- !ruby/object:Gem::Specification
+name: steffi
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- James Dabbs
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-01-27 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: ffi
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: FFI bindings to the igraph library
+email: jamesdabbs@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- Gemfile
+- steffi.gemspec
+- lib/steffi/communities.rb
+- lib/steffi/create.rb
+- lib/steffi/edge.rb
+- lib/steffi/error.rb
+- lib/steffi/ffi.rb
+- lib/steffi/graph.rb
+- lib/steffi/metrics.rb
+- lib/steffi/vector.rb
+- lib/steffi/version.rb
+- lib/steffi/vertex.rb
+- lib/steffi.rb
+- ext/libigraph.0.6.dylib
+homepage: http://github.com/jamesdabbs/steffi
+licenses: []
+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.24
+signing_key: 
+specification_version: 3
+summary: FFI bindings to the igraph library
+test_files: []