1 # This file is part of NIT ( http://www.nitlanguage.org ).
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
7 # http://www.apache.org/licenses/LICENSE-2.0
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.
15 # UML generator in dot format.
22 redef class ToolContext
23 var umlphase
: Phase = new UMLPhase(self, null)
25 var opt_gen
= new OptionEnum(["class", "package"], "Choose which type of uml diagram to generate", 0, "--diagram")
28 option_context
.add_option opt_gen
33 private class UMLPhase
35 redef fun process_mainmodule
(mainmodule
, mmodules
)
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
47 var toolcontext
= new ToolContext
48 toolcontext
.process_options
(args
)
49 var arguments
= toolcontext
.option_context
.rest
53 var mbuilder
= new ModelBuilder(model
, toolcontext
)
54 var mmodules
= mbuilder
.parse
(arguments
)
56 if mmodules
.is_empty
then return
58 toolcontext
.run_global_phases
(mmodules
)