@@ -1,14 +1,58 @@ class MethodTrails module DotOutput + class OutputGenerator + def initialize(graph) + @graph = graph + @o = [] + end + attr_accessor :graph, :o + + def output_classes_and_i_methods + if classes_and_i_methods = graph[:classes_and_i_methods] + o << "/* Classes and their Instance Methods */" + classes_and_i_methods.each_with_index do |hash, i| + pairs = hash["class"].zip(hash["def"]) + hash["class"].uniq.each do |klass| + o.concat cluster_start(klass) + pairs.find_all { |p| p[0] == klass }.each do |p| + o << dot_indent(1) + %{"#{p[1]}" [color=green];} + end + o.concat cluster_stop + end + end + o << nil + end + end + + def output_i_methods_and_ids + if i_methods_and_ids = graph[:i_methods_and_ids] + i_methods_and_ids.each do |hash| + pairs = hash["def"].zip(hash["ident"]) + pairs.each do |pair| + o << %{"#{pair[0]}" -> "#{pair[1]}";} + end + end + end + end + + def run + output_classes_and_i_methods + output_i_methods_and_ids + end + + def to_graph + @o + end + + def self.run(graph) + new(graph).run.to_graph + end + + end + def dot_file_contents(graph) - o = [] - # output_classes(graph, o) - # output_i_methods(graph, o) - # output_c_methods(graph, o) - output_classes_and_i_methods(graph, o) - output_i_methods_and_ids(graph, o) - o + OutputGenerator.run(graph) end def dot_file_header lib/method_trails/dot_output.rb 51d58e3788f9f08b95262a30146be6fd8b15c9e6 David James djwonk@collectiveinsight.net http://github.com/djwonk/method_trails/commit/c23ac940c2a9e1f67839a6441bea65ca6d7448e5 c23ac940c2a9e1f67839a6441bea65ca6d7448e5 2008-06-18T13:04:42-07:00 2008-06-18T13:04:42-07:00 Refactoring help from Rein Henrichs 27ef09fdac3dffc0c332cce34c6072620816bc68 David James djwonk@collectiveinsight.net