courgette 0.3.0 → 0.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: f615e4ef69593e21d1de8c8e0f5a1f4137777bd6
-  data.tar.gz: 9972586e2328530973db1e9234d6fd6978a214e8
+  metadata.gz: 9543a488037562ed482f67d8854b97598ebdc75a
+  data.tar.gz: 60a7b5203affdcdc38a50aef87197659cde5d2ff
 SHA512:
-  metadata.gz: 31414ebd38fb6d5a3328e0c4e46be2d6bdc7fd2b97b66beabcc7f081140c2fe2ea31ca489168cf538fee32e025f88d9a7268b08f46b3f1720c56b1b5314ca3c5
-  data.tar.gz: 86c6a07660d04acffb78c4909b1a90fd67548f0952b7e0876b0a9de511608b3b144134ada8869de57b8f5992f287d68272709ac05b83ef21219cc205630b5890
+  metadata.gz: b7349b7c7037c3005ac85bb70f9be4e485466865e92e38911363ccb7b31d41308b31b09448fcc27dad01c6fb813a6f6481e75cb6a0c805b17d04820f45f15e70
+  data.tar.gz: 498be2f6041a00fbd12a9f066285817af1c3cdb09ff304b52e1db5651104a559df04aa69b5bee8e37d1e8bbf8c61ad8f3ff3fc46b85897d01bcb2b5f49ced924

data/bin/courgette

@@ -31,7 +31,8 @@ command :graph do |c|
   c.example 'description', 'command example'
   c.option '--glob STRING', String, 'Glob to run metrics over (default: **/*.rb)'
   c.option '--name STRING', String, 'Glob to run metrics over (default: graph)'
-  c.option '--format STRING', String, 'Format (default: png)'
+  c.option '--format STRING', String, 'Format (default: svg)'
+  c.option '--filter STRING', String, 'Comma-separated class names to filter'
   c.action do |args, options|
     options.default :glob => '**/*.rb', :format => 'svg', :name => 'graph'
 

data/lib/courgette/commands/graph.rb

@@ -9,7 +9,7 @@ module Courgette
       end
 
       def run
-        factory.load graph
+        factory.load filtered_graph
 
         formats.each do |fmt|
           factory.save name, fmt
@@ -36,6 +36,18 @@ module Courgette
         end
       end
 
+      def filtered_graph
+        @filtered_graph ||= graph.filter filter_nodes
+      end
+
+      def filter_nodes
+        return nil if @options.filter.nil?
+
+        @options.filter.split(',').map do |root|
+          root.split("::").map(&:to_sym)
+        end
+      end
+
       def graph
         @graph ||= Courgette::DirectoryAnalyser.new.tap do |da|
           da.analyse glob

data/lib/courgette/graph.rb

@@ -1,3 +1,5 @@
+require 'set'
+
 module Courgette
   class Graph
     attr_reader :nodes, :edges
@@ -5,6 +7,8 @@ module Courgette
     def initialize nodes, edges
       @nodes = nodes
       @edges = edges
+
+      setup_graph
     end
 
     def dependency_count node
@@ -15,6 +19,39 @@ module Courgette
       edges.select { |r| r.reference == node }.count
     end
 
+    def filter roots
+      return self if roots.nil?
+
+      visited = Set.new
+
+      roots.each do |r|
+        dfs(r, visited)
+      end
+
+      filtered_edges = edges.select { |r| visited.include? r.referrer }
+
+      Graph.new visited.to_a, filtered_edges
+    end
+
+    attr_accessor :adjacency_list
+
     private
+    def dfs v, visited
+      return if visited.include? v
+
+      visited << v
+
+      adjacency_list[v].each do |w|
+        dfs(w, visited) unless visited.include? w
+      end
+    end
+
+    def setup_graph
+      self.adjacency_list = Hash.new { |h,k| h[k] = Set.new }
+
+      edges.each do |e|
+        self.adjacency_list[e.referrer] << e.reference
+      end
+    end
   end
 end

data/lib/courgette/version.rb

@@ -1,3 +1,3 @@
 module Courgette
-  VERSION = "0.3.0"
+  VERSION = "0.4.0"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: courgette
 version: !ruby/object:Gem::Version
-  version: 0.3.0
+  version: 0.4.0
 platform: ruby
 authors:
 - Hugo Peixoto
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-02-06 00:00:00.000000000 Z
+date: 2015-07-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: parser
@@ -171,7 +171,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.2.2
+rubygems_version: 2.4.5
 signing_key: 
 specification_version: 4
 summary: Courgette analyses a set of files, and calculates a dependency graph