src/nituml.nit

   1 # This file is part of NIT ( http://www.nitlanguage.org ).
   2 #
   3 # Licensed under the Apache License, Version 2.0 (the "License");
   4 # you may not use this file except in compliance with the License.
   5 # You may obtain a copy of the License at
   6 #
   7 # http://www.apache.org/licenses/LICENSE-2.0
   8 #
   9 # Unless required by applicable law or agreed to in writing, software
  10 # distributed under the License is distributed on an "AS IS" BASIS,
  11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12 # See the License for the specific language governing permissions and
  13 # limitations under the License.
  14
  15 # UML generator in dot format.
  16 module nituml
  17
  18 import modelbuilder
  19 import frontend
  20 import uml
  21
  22 redef class ToolContext
  23         var umlphase: Phase = new UMLPhase(self, null)
  24
  25         var opt_gen = new OptionEnum(["class", "package"], "Choose which type of uml diagram to generate", 0, "--diagram")
  26
  27         redef init do
  28                 option_context.add_option opt_gen
  29                 super
  30         end
  31 end
  32
  33 private class UMLPhase
  34         super Phase
  35         redef fun process_mainmodule(mainmodule, mmodules)
  36         do
  37                 var d = new UMLModel(mainmodule.model, mainmodule, toolcontext)
  38                 if toolcontext.opt_gen.value == 0 then
  39                         print d.generate_class_uml.write_to_string
  40                 else if toolcontext.opt_gen.value == 1 then
  41                         print d.generate_package_uml.write_to_string
  42                 end
  43         end
  44 end
  45
  46 # process options
  47 var toolcontext = new ToolContext
  48 toolcontext.process_options(args)
  49 var arguments = toolcontext.option_context.rest
  50
  51 # build model
  52 var model = new Model
  53 var mbuilder = new ModelBuilder(model, toolcontext)
  54 var mmodules = mbuilder.parse_full(arguments)
  55
  56 if mmodules.is_empty then return
  57 mbuilder.run_phases
  58 toolcontext.run_global_phases(mmodules)