bud 0.0.3 → 0.0.4

data/lib/bud/graphs.rb

@@ -1,32 +1,21 @@
-# -*- coding: utf-8 -*-
 require 'rubygems'
+require 'digest/md5'
 require 'graphviz'
 
 class GraphGen #:nodoc: all
+  attr_reader :nodes
 
-  def initialize(mapping, tableinfo, cycle, name, budtime, vizlevel, pics_dir, collapse=false, depanalysis=nil, cardinalities={})
-    #@graph = GraphViz.new(:G, :type => :digraph, :label => "", :ratio => 0.85 )
+  def initialize(tableinfo, cycle, name, budtime, collapse=false, cardinalities={})
     @graph = GraphViz.new(:G, :type => :digraph, :label => "")
     @graph.node[:fontname] = "Times-Roman"
     @graph.node[:fontsize] = 18
     @graph.edge[:fontname] = "Times-Roman"
     @graph.edge[:fontsize] = 18
-    @tiers = []
     @cards = cardinalities
     @name = name
     @collapse = collapse
-    @depanalysis = depanalysis
     @budtime = budtime
-    @vizlevel = vizlevel
-    @pics_dir = pics_dir
-    #@internals = {'count' => 1, 'localtick' => 1, 'stdio' => 1, 't_rules' => 1, 't_depends' => 1, 't_depends_tc' => 1, 't_provides' => 1, 't_cycle' => 1}
-    @internals = {'count' => 1, 'localtick' => 1, 'stdio' => 1} #, 't_rules' => 1, 't_depends' => 1, 't_depends_tc' => 1, 't_provides' => 1, 't_cycle' => 1}
-
-    # map: table -> stratum
-    @t2s = {}
-    mapping.each do |m|
-      @t2s[m[0]] = m[1].to_i
-    end
+    @internals = {'localtick' => 1, 'stdio' => 1}
 
     # map: table -> type
     @tabinf = {}
@@ -36,7 +25,7 @@ class GraphGen #:nodoc: all
 
     @redcycle = {}
     cycle.each do |c|
-      puts "CYCLE: #{c.inspect}"
+      # assumption: !(c[2] and !c[3]), or stratification would have bombed out
       if c[2] and c[3]
         if !@redcycle[c[0]]
           @redcycle[c[0]] = []
@@ -50,21 +39,6 @@ class GraphGen #:nodoc: all
     @labels = {}
   end
   
-  def safe_t2s(tab)
-    if @t2s[tab]
-      @t2s[tab]
-    else  
-      words = tab.split(",")
-      maxs = 0
-      words.each do |w|
-        if @t2s[w] and @t2s[w] > maxs
-          maxs = @t2s[w]
-        end
-      end
-      return maxs
-    end
-  end
-
   def name_bag(predicate, bag)
     if bag[predicate]
       return bag
@@ -84,12 +58,9 @@ class GraphGen #:nodoc: all
 
   def name_of(predicate)
     # consider doing this in bud
-    # PAA
     if @redcycle[predicate] and @collapse
-      puts "collapse #{predicate}, redcycle #{@redcycle[predicate].inspect}"
       via = @redcycle[predicate]
       bag = name_bag(predicate, {})
-      #str = bag.key_cols.sort.join(", ")
       str = bag.keys.sort.join(", ")
       return str
     else
@@ -98,14 +69,11 @@ class GraphGen #:nodoc: all
   end
 
   def process(depends)
-
     # collapsing NEG/+ cycles.
     # we want to create a function from any predicate to (cycle_name or bottom)
     # bottom if the predicate is not in a NEG/+ cycle.  otherwise,
     # its name is "CYC" + concat(sort(predicate names))
-
     depends.each do |d|
-      #puts "DEP: #{d.inspect}"
       head = d[1]
       body = d[3]
 
@@ -116,32 +84,30 @@ class GraphGen #:nodoc: all
 
       head = name_of(head)
       body = name_of(body)
-      addonce(head, (head != d[1]))
+      addonce(head, (head != d[1]), true)
       addonce(body, (body != d[3]))
       addedge(body, head, d[2], d[4], (head != d[1]), d[0])
     end
   end
 
-  def addonce(node, negcluster)
-    #puts "ADD NODE #{node}"
+  def addonce(node, negcluster, inhead=false)
     if !@nodes[node]
       @nodes[node] = @graph.add_node(node)
       if @cards and @cards[node]
         @nodes[node].label = node +"\n (#{@cards[node].to_s})"
-        puts "IMAGE IS #{@cards[node]}"
-        #@nodes[node].image = @cards[node]
-      end
- 
-      if @vizlevel >= 3 
-        @nodes[node].URL = "javascript:openWin(\"#{node}\", #{@budtime})"
-      else
-        @nodes[node].URL = "#{node}.html"
       end
+    end 
+
+    if @budtime == -1
+      @nodes[node].URL = "#{node}.html" if inhead
+    else
+      @nodes[node].URL = "javascript:openWin(\"#{node}\", #{@budtime})"
     end
 
     if negcluster
       # cleaning 
       res = node
+      # pretty-printing issues
       node.split(", ").each_with_index do |p, i|
         if i == 0
           res = p
@@ -151,12 +117,11 @@ class GraphGen #:nodoc: all
           res = res + ", " + p
         end
       end
-      #@nodes[node].label = "<b>" + res + "</b>"
       @nodes[node].label = res
       @nodes[node].color = "red"
       @nodes[node].shape = "octagon"
       @nodes[node].penwidth = 3
-      @nodes[node].URL = "#{File.basename(@name)}_expanded.svg"
+      @nodes[node].URL = "#{File.basename(@name).gsub(".staging", "").gsub("collapsed", "expanded")}.svg"
     elsif @tabinf[node] and (@tabinf[node] == "Bud::BudTable")
       @nodes[node].shape = "rect"
     end
@@ -180,11 +145,8 @@ class GraphGen #:nodoc: all
         @edges[ekey].minlen = 1.5
       end
       @labels[ekey] = {}
-
     end
 
-    #@edges[ekey].minlen = 5 if negcluster and body == head
-
     if op == '<+'
       @labels[ekey][' +/-'] = true
     elsif op == "<~"
@@ -200,9 +162,8 @@ class GraphGen #:nodoc: all
     end
   end
 
-  def finish
+  def finish(depanalysis=nil, output=nil)
     @labels.each_key do |k|
-      #@edges[k].label = @labels[k].key_cols.join(" ")
       @edges[k].label = @labels[k].keys.join(" ")
     end
 
@@ -221,28 +182,26 @@ class GraphGen #:nodoc: all
     @nodes["T"].penwidth = 3
 
     @tabinf.each_pair do |k, v|
-
       unless @nodes[name_of(k.to_s)] or k.to_s =~ /_tbl/ or @internals[k.to_s] or (k.to_s =~ /^t_/ and @budtime != 0)
         addonce(k.to_s, false)
       end
       if v == "Bud::BudPeriodic"
-        puts "adding edge S -> #{@nodes[k.to_s]}"
         addedge("S", k.to_s, false, false, false)
       end
     end
 
-    unless @depanalysis.nil? 
-      @depanalysis.source.each {|s| addedge("S", s.pred, false, false, false) }
-      @depanalysis.sink.each {|s| addedge(s.pred, "T", false, false, false) }
+    unless depanalysis.nil? 
+      depanalysis.source.each {|s| addedge("S", s.pred, false, false, false)}
+      depanalysis.sink.each {|s| addedge(s.pred, "T", false, false, false)}
 
-      unless @depanalysis.underspecified.empty?
+      unless depanalysis.underspecified.empty?
         addonce("??", false)
         @nodes["??"].color = "red"
         @nodes["??"].shape = "diamond"
         @nodes["??"].penwidth = 2
       end
 
-      @depanalysis.underspecified.each do |u|
+      depanalysis.underspecified.each do |u|
         if u.input
           addedge(u.pred, "??", false, false, false)
         else
@@ -251,155 +210,100 @@ class GraphGen #:nodoc: all
       end
     end
 
-    suffix = @collapse ? "collapsed" : "expanded"
-    fn = "#{@name}_#{suffix}.svg"
-    puts "fn is #{fn}"
-    staging = "#{fn}_staging"
-    @graph.output(:svg => staging)
-    @graph.output(:dot => "#{fn}.dot")
-    @graph.output(:png => "#{fn}.png")
-    fin = File.open(staging, "r")
-    fout = File.open(fn, "w")
-    while line = fin.gets
-      fout.puts line.gsub("<title>G</title>", svg_javascript())
+    if output.nil?
+      @graph.output(:svg => @name)
+    else
+      @graph.output(output => @name)
     end
-    fin.close
-    fout.close
-    File.delete(staging) 
   end
+end
 
-  def output_base
-    if @vizlevel >= 3
-      @pics_dir
-    else
-      "bud_doc"
-    end 
+class SpaceTime    
+  def initialize(input, links = false)
+    @input = input 
+    @links = links
+    processes = input.map {|i| i[1]}
+    input.map{|i| processes << i[2]}
+    processes.uniq!
+
+    @queues = {} 
+    
+    @g = GraphViz.new(:G, :type => :digraph, :rankdir => "LR", :outputorder => "nodesfirst", :splines => "line")#, :clusterrank => "none")
+    @hdr = @g.subgraph("cluster_0")
+    
+    @subs = {}
+    @head = {}
+    last = nil
+    processes.each_with_index do |p, i|
+      #@head[p] = @hdr.add_node("process #{p}(#{i})")#, :color => "white", :label => "")
+      @subs[p] = @g.subgraph("buster_#{i+1}")
+      @head[p] = @hdr.add_node("process #{p}(#{i})", :group => p)#, :color => "white", :label => "")
+      
+    end
   end
 
-  def dump(shredded_rules)
-    return if shredded_rules.nil?
-
-    fout = File.new("#{output_base}/style.css", "w")
-    fout.puts css
-    fout.close
-
-    code = {}
-    rules = {}
-    convertor = Syntax::Convertors::HTML.for_syntax "ruby"
-    shredded_rules.each do |s|
-      #fout = File.new("#{output_base}/#{s[0]}.html", "w+")
-      fout = File.new("#{output_base}/#{s[0]}.html", "w+")
-      fout.puts header
-      fout.puts "<h1>Rule #{s[0]}</h1><br>"
-
-      c = convertor.convert(s[3])
-      c.sub!(/^<pre>/, "<pre class=\"code\" style='font-size:20px'>\n")
-      fout.puts c
-      rules[s[0]] = [s[1], s[3]]
-      fout.close
+  def msg_edge(f, t, l)
+    lbl = "#{f}#{t}#{l}"
+    if @edges[lbl]
+      prev = @edges[lbl]
+      @edges[lbl] = [prev[0], prev[1], prev[2], prev[3] + 1]
+    else
+      @edges[lbl] = [f, t, l, 1]
+    end
+  end
+  
+  def process
+    @edges = {}
+    queues = {}
+    @input.each do |i|
+      queues[i[1]] = [] unless queues[i[1]]
+      queues[i[2]] = [] unless queues[i[2]]
+      queues[i[1]] << i[3]
+      queues[i[2]] << i[4]
     end
 
-    rules.each_pair do |k, v|
-      if !code[v[0]]
-        code[v[0]] = ""
-      end
-      #code[v[0]] = "<br># RULE #{k}<br> " + code[v[0]] + "<br>" + v[1]
-      code[v[0]] = "\n# RULE #{k}\n " + code[v[0]] + "\n" + v[1]
+    squeues = {}
+    queues.each_pair do |k, v|
+      squeues[k] = v.sort{|a, b| a <=> b}
     end
-    @nodes.each_pair do |k, v|
-      fout = File.new("#{output_base}/#{k}.html", "w+")
-      fout.puts header
-      k.split(", ").each do |i|
-        unless code[i].nil?
-          c = convertor.convert(code[i])
-          c.sub!(/^<pre>/, "<pre class=\"code\">\n")
-          fout.puts c
+
+    # create the nodes and the timeline edges first.
+    squeues.each do |k, v|
+      v.each_with_index do |item, i|
+        label = "#{k}-#{item}"
+        if @links
+          url = "DBM_#{k}_/tm_#{item}.svg"
+          #puts "URL is #{url}"
+        end
+        snd = @subs[k].add_node(label, {:label => item.to_s, :width => 0.1, :height => 0.1, :fontsize => 6, :pos => [1, i], :group => k, :URL => url})  
+
+        unless @head[k].id == snd.id
+          @subs[k].add_edge(@head[k], snd, :weight => 2)
+          @head[k] = snd
         end
       end
-      fout.puts("</body></html>")
-      fout.close
     end
-  end
-
 
-  def header
-      return "<html><meta content='text/html; charset=UTF-8' http-equiv='Content-Type'/>\n<head><link rel=\"stylesheet\" type=\"text/css\" href=\"style.css\" /></head><body>"
-  end
-
-  def css
-    return "pre.code {
-  padding: 1ex 1ex 1ex 1ex;
-  border: 4px groove #CC0000;
-  overflow-x: auto;
-}
-
-pre.code span.attribute { color: #009900; }
-pre.code span.char { color: #F00; }
-pre.code span.class { color: #A020F0; font-weight: bold; }
-pre.code span.comment { color: #0000FF; }
-pre.code span.constant { color: #008B8B; }
-pre.code span.escape { color: #6A5ACD; }
-pre.code span.expr { color: #2222CC; }
-pre.code span.global { color: #11AA44; }
-pre.code span.ident { color: #000000; }
-pre.code span.keyword { color: #A52A2A; font-weight: bold; }
-pre.code span.method { color: #008B8B; }
-pre.code span.module { color: #A020F0; font-weight: bold; }
-pre.code span.number { color: #DD00DD; }
-pre.code span.punct { color: #6A5ACD; }
-pre.code span.regex { color: #DD00DD; }
-pre.code span.string { color: #DD00DD; }
-pre.code span.symbol { color: #008B8B; }
-"
+    #@input.sort{|a, b| a[3] <=> b[3]}.each do |i|
+    @input.each do |i|
+      snd_loc = i[1]
+      rcv_loc = i[2]
+      snd_label = "#{snd_loc}-#{i[3]}"
+      rcv_label = "#{rcv_loc}-#{i[4]}"
+      #@g.add_edge(snd_label, rcv_label, :color => "red", :weight => 1, :label => i[5])
+      msg_edge(snd_label, rcv_label, i[5])
+    end
   end
-
   
-end
-
-  def svg_javascript
-    return "
-<script type='text/javascript'>
-  <![CDATA[
-
-var windows = new Array()
-var info = new Array()
-
-function openWin(target, time) {
-  win = window.open(target + \"_\" + time + \".html\", target, \"location=no,width=400,height=180,left=0,status=no\");
-  // hm, an associative array, how strange.
-  info[target] = 1
-}
-
-function advanceTo(time) {
-  arr = gup(\"wins\").split(\",\");
-  for (i=0; i < arr.length; i++) {
-    if (arr[i] != \"\") {
-      openWin(arr[i], time);
-    }
-  }
-  str = '';
-  // getting 'key_cols'
-  for (var i in info) {
-    str = str + ',' + i;
-  }
-  self.window.location.href = 'tm_' + time + '_expanded.svg?wins=' + str;
-}
-
-// off the netz
-function gup( name )
-{
-  name = name.replace(/[\[]/,\"\\\[\").replace(/[\]]/,\"\\\]\");
-  var regexS = \"[\\?&]\"+name+\"=([^&#]*)\";
-  var regex = new RegExp( regexS );
-  var results = regex.exec( window.location.href );
-  if( results == null )
-    return \"\";
-  else
-    return results[1];
-}
-
-  ]]>
-</script>
-"
+  def finish(file, fmt=nil)
+    @edges.each_pair do |k, v|
+      lbl =  v[3] > 1 ? "#{v[2]}(#{v[3]})" : v[2]
+      @g.add_edge(v[0], v[1], :label => lbl, :color => "red", :weight => 1)
+    end
+    if fmt.nil?
+      @g.output(:svg => "#{file}.svg")
+    else
+      eval("@g.output(:#{fmt} => \"\#{file}.#{fmt}\")")
+    end
   end
-
+end