@@ -11,26 +11,33 @@ class PartialDependencies @views = nil end - def dot(type = "png", fn = "partial_dependencies") - IO.popen("dot -T#{type} -o #{fn}.#{type}", "w") do |pipe| - pipe.puts dot_input + def dot(type = "png", view_set = "used", fn = "partial_dependencies") + prog = view_set == "unused" ? "neato" : "dot" + IO.popen("#{prog} -T#{type} -o #{fn}-#{view_set}.#{type}", "w") do |pipe| + pipe.puts dot_input(view_set) end end private - def dot_input + def dot_input(view_set) + possible_view_sets = ["used", "unused", "all"] + unless possible_view_sets.include?(view_set) + raise "Wrong view_set. Only #{possible_view_sets.inspect} possible. Was #{view_set.inspect}" + end str = StringIO.new parse_files str.puts "digraph partial_dependencies {" name_to_node = {} - @used_views.each_with_index do |view, index| + instance_variable_get("@#{view_set}_views").each_with_index do |view, index| str.puts "Node#{index} [label=\"#{view}\"]" name_to_node[view] = "Node#{index}" end - @edges.each do |view, partials| - partials.each do |partial| - str.puts "#{name_to_node[view]}->#{name_to_node[partial]}" + if (["used", "all"].include?(view_set)) + @edges.each do |view, partials| + partials.each do |partial| + str.puts "#{name_to_node[view]}->#{name_to_node[partial]}" + end end end str.puts "}" @@ -42,7 +49,9 @@ class PartialDependencies def parse_files @edges = Hash.new {|hash,key| hash[key] = []} @used_views = {} + @all_views = [] views.each do |view| + @all_views << pwfe(view) File.open("#{view[:path]}") do |contents| contents.each do |line| if line =~ /=\s*render\s*\(?:partial\s*=>\s*["'](.*?)["']/ @@ -60,6 +69,7 @@ class PartialDependencies end end @used_views = @used_views.keys + @unused_views = @all_views - @used_views end def pwfe(view) @@ -77,13 +87,8 @@ end namespace :partial_dependencies do desc "Generate a graphical (PNG) representation of the partial dependencies" - task :generate_picture do - pd = PartialDependencies.new - pd.dot - end - desc "Generate a SVG representation (same as partial_dependencies:generate_picture apart from output format)" - task :generate_svg do + task :generate_picture, :file_type, :view_set do |t, args| pd = PartialDependencies.new - pd.dot("svg") + pd.dot(args.file_type || "png", args.view_set || "used") end end tasks/partial_dependencies.rake 0436344fded3d18476349f6b5dcfc81c84df93cb 148dfbb46f5be21146105b5bd8666bfb798cc8f0 phinze paul.t.hinze@gmail.com http://github.com/msales/partial_dependencies/commit/2674721ae12148f756cce51402ed237e4d2980ea 2674721ae12148f756cce51402ed237e4d2980ea 2009-03-11T06:19:30-07:00 2009-03-11T06:19:30-07:00 Merge commit 'msales/master' f99a3a803ed254f22ba1e56b15f537c2f555a756 phinze paul.t.hinze@gmail.com