lib/activefacts/generator/doc/graphviz.rb in activefacts-compositions-1.9.8 vs lib/activefacts/generator/doc/graphviz.rb in activefacts-compositions-1.9.9

@@ -2,11 +2,10 @@
 #       ActiveFacts GraphViz generator
 #
 # Copyright (c) 2009-2016 Clifford Heath. Read the LICENSE file.
 #
 require 'activefacts/metamodel'
-require 'activefacts/registry'
 require 'activefacts/compositions'
 require 'activefacts/generator'
 
 module ActiveFacts
   module Generators
@@ -18,11 +17,12 @@
         def self.options
           {
           }
         end
 
-        def initialize composition, options = {}
-          @composition = composition
+        def initialize compositions, options = {}
+          raise "--graphviz only processes a single composition" if compositions.size > 1
+          @composition = compositions[0]
           @options = options
         end
 
         def generate
           composites = @composition.all_composite.sort_by{|c| c.mapping.name }