nit: Added new binary nituml, for generation of UML diagrams from Nit source code.
[nit.git] / 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 end
25
26 private class UMLPhase
27 super Phase
28 redef fun process_mainmodule(mainmodule, mmodules)
29 do
30 var d = new UMLModel(mainmodule.model, mainmodule, toolcontext)
31 print d.generate_class_uml.write_to_string
32 end
33 end
34
35 # process options
36 var toolcontext = new ToolContext
37 toolcontext.process_options(args)
38 var arguments = toolcontext.option_context.rest
39
40 # build model
41 var model = new Model
42 var mbuilder = new ModelBuilder(model, toolcontext)
43 var mmodules = mbuilder.parse(arguments)
44
45 if mmodules.is_empty then return
46 mbuilder.run_phases
47 toolcontext.run_global_phases(mmodules)