blob: 1e68acdaa652dd93ad4dc799510affad42f54dd5 [file] [log] [blame]
Andreas Wundsam27303462013-07-16 12:52:35 -07001# Copyright 2013, Big Switch Networks, Inc.
2#
3# LoxiGen is licensed under the Eclipse Public License, version 1.0 (EPL), with
4# the following special exception:
5#
6# LOXI Exception
7#
8# As a special exception to the terms of the EPL, you may distribute libraries
9# generated by LoxiGen (LoxiGen Libraries) under the terms of your choice, provided
10# that copyright and licensing notices generated by LoxiGen are not altered or removed
11# from the LoxiGen Libraries and the notice provided below is (i) included in
12# the LoxiGen Libraries, if distributed in source code form and (ii) included in any
13# documentation for the LoxiGen Libraries, if distributed in binary form.
14#
15# Notice: "Copyright 2013, Big Switch Networks, Inc. This library was generated by the LoxiGen Compiler."
16#
17# You may not use this file except in compliance with the EPL or LOXI Exception. You may obtain
18# a copy of the EPL at:
19#
20# http://www.eclipse.org/legal/epl-v10.html
21#
22# Unless required by applicable law or agreed to in writing, software
23# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
24# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
25# EPL for the specific language governing permissions and limitations
26# under the EPL.
27
28"""
29@brief Main Java Generation module
30"""
31
32import pdb
33import os
34import shutil
35
36import of_g
37from loxi_ir import *
38import lang_java
Andreas Wundsame916d6f2013-07-30 11:33:58 -070039import test_data
Yotam Harchol466b3212013-08-15 12:14:46 -070040from import_cleaner import ImportCleaner
Andreas Wundsam27303462013-07-16 12:52:35 -070041
42import loxi_utils.loxi_utils as loxi_utils
43
44import java_gen.java_model as java_model
45
46def gen_all_java(out, name):
Andreas Wundsamd83883e2013-07-25 14:50:51 -070047 basedir= '%s/openflowj' % of_g.options.install_dir
Andreas Wundsam27303462013-07-16 12:52:35 -070048 print "Outputting to %s" % basedir
49 if os.path.exists(basedir):
50 shutil.rmtree(basedir)
51 os.makedirs(basedir)
52 copy_prewrite_tree(basedir)
53
Andreas Wundsame916d6f2013-07-30 11:33:58 -070054 gen = JavaGenerator(basedir)
Andreas Wundsam27303462013-07-16 12:52:35 -070055 gen.create_of_interfaces()
56 gen.create_of_classes()
57 gen.create_of_const_enums()
Andreas Wundsam5204de22013-07-30 11:34:45 -070058 gen.create_of_factories()
Andreas Wundsam27303462013-07-16 12:52:35 -070059
Andreas Wundsamd83883e2013-07-25 14:50:51 -070060 with open('%s/README.java-lang' % os.path.dirname(__file__)) as readme_src:
Andreas Wundsam27303462013-07-16 12:52:35 -070061 out.writelines(readme_src.readlines())
62 out.close()
63
64
65class JavaGenerator(object):
66 templates_dir = os.path.join(os.path.dirname(os.path.realpath(__file__)), 'templates')
67
68 def __init__(self, basedir):
69 self.basedir = basedir
70 self.java_model = java_model.model
71
Andreas Wundsame916d6f2013-07-30 11:33:58 -070072 def render_class(self, clazz, template, src_dir=None, **context):
73 if not src_dir:
74 src_dir = "src/main/java/"
75
Andreas Wundsam27303462013-07-16 12:52:35 -070076 context['class_name'] = clazz.name
77 context['package'] = clazz.package
Andreas Wundsamd10e1c52013-08-01 22:02:49 -070078 context['template_dir'] = self.templates_dir
Andreas Wundsam27303462013-07-16 12:52:35 -070079
Andreas Wundsame916d6f2013-07-30 11:33:58 -070080 filename = os.path.join(self.basedir, src_dir, "%s/%s.java" % (clazz.package.replace(".", "/"), clazz.name))
Andreas Wundsam27303462013-07-16 12:52:35 -070081 dirname = os.path.dirname(filename)
82 if not os.path.exists(dirname):
83 os.makedirs(dirname)
84 prefix = '//::(?=[ \t]|$)'
85 print "filename: %s" % filename
86 with open(filename, "w") as f:
87 loxi_utils.render_template(f, template, [self.templates_dir], context, prefix=prefix)
Yotam Harchol466b3212013-08-15 12:14:46 -070088
89 try:
90 cleaner = ImportCleaner(filename)
91 cleaner.find_used_imports()
92 cleaner.rewrite_file(filename)
93 except:
94 print 'Cannot clean imports from file %s' % filename
95
Andreas Wundsam27303462013-07-16 12:52:35 -070096
97 def create_of_const_enums(self):
98 for enum in self.java_model.enums:
99 if enum.name in ["OFPort"]:
100 continue
101 self.render_class(clazz=enum,
102 template='const.java', enum=enum, all_versions=self.java_model.versions)
103
Andreas Wundsambf1dbbd2013-07-30 11:07:59 -0700104 for version in enum.versions:
105 clazz = java_model.OFGenericClass(package="org.openflow.protocol.ver{}".format(version.of_version), name="{}SerializerVer{}".format(enum.name, version.of_version))
106 self.render_class(clazz=clazz, template="const_serializer.java", enum=enum, version=version)
107
Andreas Wundsam27303462013-07-16 12:52:35 -0700108 def create_of_interfaces(self):
109 """ Create the base interfaces for of classes"""
110 for interface in self.java_model.interfaces:
111 #if not utils.class_is_message(interface.c_name):
112 # continue
113 self.render_class(clazz=interface,
114 template="of_interface.java", msg=interface)
115
116 def create_of_classes(self):
117 """ Create the OF classes with implementations for each of the interfaces and versions """
118 for interface in self.java_model.interfaces:
Andreas Wundsam27303462013-07-16 12:52:35 -0700119 for java_class in interface.versioned_classes:
Andreas Wundsam001b1822013-08-02 22:25:55 -0700120 if self.java_model.generate_class(java_class):
121 if not java_class.is_virtual:
122 self.render_class(clazz=java_class,
123 template='of_class.java', version=java_class.version, msg=java_class,
124 impl_class=java_class.name)
Andreas Wundsam27303462013-07-16 12:52:35 -0700125
Andreas Wundsam001b1822013-08-02 22:25:55 -0700126 self.create_unit_test(java_class.unit_test)
127 else:
128 disc = java_class.discriminator
129 if disc:
130 self.render_class(clazz=java_class,
131 template='of_virtual_class.java', version=java_class.version, msg=java_class,
132 impl_class=java_class.name, model=self.java_model)
133 else:
134 print "Class %s virtual but no discriminator" % java_class.name
135 else:
136 print "Class %s ignored by generate_class" % java_class.name
Andreas Wundsame916d6f2013-07-30 11:33:58 -0700137
Yotam Harchol466b3212013-08-15 12:14:46 -0700138 def create_unit_test(self, unit_tests):
139 if unit_tests.has_test_data:
140 for i in range(unit_tests.length):
141 unit_test = unit_tests.get_test_unit(i)
142 if unit_test.has_test_data:
143 self.render_class(clazz=unit_test,
144 template='unit_test.java', src_dir="src/test/java",
145 version=unit_test.java_class.version,
146 test=unit_test, msg=unit_test.java_class,
147 test_data=unit_test.test_data)
Andreas Wundsame916d6f2013-07-30 11:33:58 -0700148
149 def create_of_factories(self):
Andreas Wundsame0d52be2013-08-22 07:52:13 -0700150 for factory in self.java_model.of_factories:
151 self.render_class(clazz=factory, template="of_factory_interface.java", factory=factory)
152 for factory_class in factory.factory_classes:
153 self.render_class(clazz=factory_class, template="of_factory_class.java", factory=factory_class, model=self.java_model)
154 self.render_class(clazz=java_model.OFGenericClass(package="org.openflow.protocol", name="OFFactories"), template="of_factories.java", versions=self.java_model.versions)
Andreas Wundsame916d6f2013-07-30 11:33:58 -0700155
Andreas Wundsam27303462013-07-16 12:52:35 -0700156def copy_prewrite_tree(basedir):
157 """ Recursively copy the directory structure from ./java_gen/pre-write
158 into $basedir"""
159 print "Copying pre-written files into %s" % basedir
Andreas Wundsamac285ba2013-07-24 20:29:42 -0700160 #subprocess.call("cd java_gen/pre-written && tar cpf - pom.xml | ( cd ../../%s && tar xvpf - )" % basedir,
161 # shell=True)
Andreas Wundsamd83883e2013-07-25 14:50:51 -0700162 os.symlink(os.path.abspath("%s/pre-written/pom.xml" % os.path.dirname(__file__)), "%s/pom.xml" % basedir)