cc4a5e8089
Change-Id: Ibd5cbbdfb22a235442cddaebc9eae9a3c4e35ec9 Signed-off-by: Damjan Marion <damarion@cisco.com>
99 lines
3.2 KiB
Python
Executable File
99 lines
3.2 KiB
Python
Executable File
#!/usr/bin/env python2
|
|
#
|
|
# Copyright (c) 2018 Cisco and/or its affiliates.
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at:
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
from string import Template
|
|
|
|
from jvpp_common_gen import generate_hash_code, generate_equals, generate_to_string, generate_fields
|
|
from jvpp_model import Union
|
|
|
|
|
|
def generate_unions(work_dir, model, logger):
|
|
logger.debug("Generating unions for %s " % model.json_api_files)
|
|
|
|
for t in model.types:
|
|
if not isinstance(t, Union):
|
|
continue
|
|
logger.debug("Generating DTO for union %s", t)
|
|
java_union_name = t.java_name
|
|
fields = t.fields
|
|
type_class = _UNION_TEMPLATE.substitute(
|
|
plugin_package=model.plugin_package,
|
|
c_type_name=t.name,
|
|
json_filename=model.json_api_files,
|
|
json_definition=t.doc,
|
|
java_union_name=java_union_name,
|
|
fields=generate_fields(fields, access_modifier="private"),
|
|
constructors=_generate_constructors(java_union_name, fields),
|
|
getters=_generate_getters(fields),
|
|
hash_code=generate_hash_code(fields),
|
|
equals=generate_equals(java_union_name, fields),
|
|
to_string=generate_to_string(java_union_name, fields)
|
|
)
|
|
with open("%s/%s.java" % (work_dir, java_union_name), "w") as f:
|
|
f.write(type_class)
|
|
|
|
_UNION_TEMPLATE = Template("""
|
|
package ${plugin_package}.types;
|
|
|
|
/**
|
|
* <p>This class represents ${c_type_name} union definition.
|
|
* <br>It was generated by unions_gen.py based on ${json_filename}:
|
|
* <pre>
|
|
${json_definition}
|
|
* </pre>
|
|
*/
|
|
public class ${java_union_name} {
|
|
private final int _activeMember;
|
|
${fields}
|
|
private ${java_union_name}() {
|
|
// Constructor for JNI usage. All members can be read.
|
|
_activeMember = -1;
|
|
}
|
|
${constructors}
|
|
${getters}
|
|
${hash_code}
|
|
${equals}
|
|
${to_string}
|
|
}
|
|
""")
|
|
|
|
|
|
def _generate_constructors(union_name, fields):
|
|
return "".join(
|
|
_CONSTRUCTOR_TEMPLATE
|
|
.substitute(union_name=union_name,
|
|
field_type=f.type.java_name_fqn,
|
|
field_name=f.java_name,
|
|
field_index=i) for i, f in enumerate(fields))
|
|
|
|
_CONSTRUCTOR_TEMPLATE = Template("""
|
|
public ${union_name}(${field_type} ${field_name}) {
|
|
this.${field_name} = java.util.Objects.requireNonNull(${field_name}, "${field_name} should not be null");
|
|
_activeMember = $field_index;
|
|
}""")
|
|
|
|
|
|
def _generate_getters(fields):
|
|
return "".join(_GETTER_TEMPLATE.substitute(
|
|
type=f.type.java_name_fqn,
|
|
getter_name=f.java_name_upper,
|
|
field_name=f.java_name
|
|
) for f in fields)
|
|
|
|
_GETTER_TEMPLATE = Template("""
|
|
public ${type} get${getter_name}() {
|
|
return ${field_name};
|
|
}""")
|