dogviz 0.0.17 → 0.0.18

data/lib/dogviz/system.rb

@@ -0,0 +1,95 @@
+require_relative 'parent'
+require_relative 'nominator'
+require_relative 'registry'
+
+module Dogviz
+  class System
+    include Parent
+    include Nominator
+
+    attr_reader :render_hints, :title, :children, :graph
+
+    alias :name :title
+    alias :render_options :render_hints
+
+    def initialize(name, hints = {splines: 'line'})
+      @children = []
+      @by_name = Registry.new name
+      @non_render_hints = remove_dogviz_hints!(hints)
+      @render_hints = hints
+      @title = create_title(name)
+      @rendered = false
+    end
+
+    def output(*args)
+      render
+      out = graph.output *args
+      puts "Created output: #{args.join ' '}" if run_from_command_line?
+      out
+    end
+
+    def flow(name)
+      Flow.new self, name
+    end
+
+    def render(type=:graphviz)
+      return @graph if @rendered
+      renderer = create_renderer(type)
+
+      children.each { |c|
+        c.render renderer
+      }
+      children.each { |c|
+        c.render_edges renderer
+      }
+      @rendered = true
+      @graph = renderer.graph
+    end
+
+    def create_renderer(type)
+      if type == :graphviz
+        GraphvizRenderer.new @title, render_hints
+      elsif type == :sigma
+        SigmaRenderer.new @title
+      else
+        raise "dunno bout that '#{type}', only know :graphviz or :sigma"
+      end
+
+    end
+
+    def rollup?
+      false
+    end
+
+    def skip?
+      false
+    end
+
+    def register(name, thing)
+      @by_name.register name, thing
+    end
+
+    def colorize_edges?
+      @non_render_hints[:colorize_edges]
+    end
+
+    private
+
+    def remove_dogviz_hints!(hints)
+      dogviz_only_hints = {}
+      %i(colorize_edges).each { |k|
+        dogviz_only_hints[k] = hints.delete k
+      }
+      dogviz_only_hints
+    end
+
+    def create_title(name)
+      now = DateTime.now
+      "#{now.strftime '%H:%M'} #{name} #{now.strftime '%F'}"
+    end
+
+    def run_from_command_line?
+      $stdout.isatty
+    end
+  end
+end

data/lib/dogviz/thing.rb

@@ -0,0 +1,135 @@
+require_relative 'common'
+require_relative 'nominator'
+require_relative 'flowable'
+require_relative 'colorizer'
+
+module Dogviz
+  class Thing
+    include Common
+    include Nominator
+    include Flowable
+    attr_reader :parent
+    attr_reader :name, :id, :pointers, :edge_heads
+
+    @@colorizer = Colorizer.new
+
+    def initialize(parent, name, options = {})
+      @parent = parent
+      @name = name
+      @id = create_id(name, parent)
+      @pointers = []
+      @rollup = false
+      @skip = false
+      @info = {}
+      @edge_heads = []
+
+      rollup! if options[:rollup]
+      options.delete(:rollup)
+
+      @render_options = options
+      setup_render_attributes({label: name}.merge inherited_render_options)
+
+      parent.register name, self
+    end
+
+    def points_to_all(*others)
+      others.each { |other|
+        points_to other
+      }
+    end
+
+    def points_to(other, options = {})
+      setup_render_edge(other, options)
+      other
+    end
+
+    def render(renderer)
+      do_render_node(renderer) unless in_rollup? || in_skip?
+    end
+
+    def render_edges(renderer)
+      pointers.each { |p|
+        render_pointer p, renderer
+      }
+    end
+
+    private
+
+    def do_render_node(renderer)
+      render_options = @render_options
+      attributes = @attributes
+      renderer.render_node(parent, id, render_options, attributes)
+    end
+
+    def setup_render_edge(other, options)
+      pointers << {
+          other: other,
+          options: {
+              xlabel: options[:name],
+              style: options[:style]
+          }.merge(inherited_render_options)
+      }
+
+      if options[:colorize] || root.colorize_edges?
+        edge_color = next_colorizer_color
+        pointers.last[:options].merge!({
+                                           color: edge_color,
+                                           fontcolor: edge_color
+                                       })
+      end
+
+    end
+
+    def render_pointer(pointer, renderer)
+      other = pointer[:other]
+      while (other.in_rollup? && !other.on_top_rollup?) do
+        other = other.parent
+      end
+      return if other.under_rollup?
+
+      from = self
+      while (from.in_rollup? && !from.on_top_rollup?) do
+        from = from.parent
+      end
+
+      return if from.in_skip?
+
+      return if from == self && from.in_rollup?
+      return if from == other
+      return if already_added_connection?(other)
+
+      if other.in_skip?
+        others = resolve_skipped_others other
+      else
+        others = [other]
+      end
+
+      others.each do |other|
+        edge_heads << other
+        render_options = pointer[:options]
+        renderer.render_edge(from, other, render_options)
+      end
+    end
+
+    def already_added_connection?(other)
+      edge_heads.include? other
+    end
+
+    def resolve_skipped_others(skipped)
+      resolved = []
+      skipped.pointers.each { |pointer|
+        next_in_line = pointer[:other]
+        if next_in_line.in_skip?
+          resolved += resolve_skipped_others next_in_line
+        else
+          resolved << next_in_line
+        end
+      }
+      resolved
+    end
+
+    def next_colorizer_color
+      @@colorizer.next
+    end
+  end
+end

data/lib/dogviz/version.rb

@@ -1,3 +1,3 @@
 module Dogviz
-  VERSION = '0.0.17'
+  VERSION = '0.0.18'
 end