lib/railroad/app_diagram.rb in railroad-0.4.0 vs lib/railroad/app_diagram.rb in railroad-0.4.1

@@ -30,13 +30,17 @@
       rescue
         STDERR.print "Error: Cannot write diagram to #{@options.output}\n\n"
         exit 2
       end
     end
- 
-    STDERR.print "Generating DOT graph\n" if @options.verbose
-
-    STDOUT.print @graph.to_dot 
+    
+    if @options.xmi 
+        STDERR.print "Generating XMI diagram\n" if @options.verbose
+    	STDOUT.print @graph.to_xmi
+    else
+        STDERR.print "Generating DOT graph\n" if @options.verbose
+        STDOUT.print @graph.to_dot 
+    end
 
     if @options.output
       STDOUT.reopen(old_stdout)
     end
   end # print