blob: 20cf48645ec6162b95430f5352d1a95a7471f92c [file] [log] [blame] [edit]
# Copyright (c) 2019, Alliance for Open Media. All rights reserved
#
# This source code is subject to the terms of the BSD 2 Clause License and
# the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License
# was not distributed with this source code in the LICENSE file, you can
# obtain it at www.aomedia.org/license/software. If the Alliance for Open
# Media Patent License 1.0 was not distributed with this source code in the
# PATENTS file, you can obtain it at www.aomedia.org/license/patent.
#
r"""Utility file containing StructNode class."""
import collections
class StructNode(collections.OrderedDict):
"""Encapsulates the human-readable data of a C language struct."""
def __init__(self, *args, **kwargs):
self._name = kwargs.get("name")
super(StructNode, self).__init__(*args, **kwargs)
def __getitem__(self, key):
return super(StructNode, self).__getitem__(key)
def __setitem__(self, key, value):
super(StructNode, self).__setitem__(key, value)
def write_fields_to_output(self, output_file, include_names=False):
"""Writes the contents of this struct to output_file.
Args:
output_file: The file object to write output.
include_names: Boolean used to indicate if the include .field notation.
"""
if self._name:
output_file.write(" // %s" % self._name)
for key, value in super(StructNode, self).items():
if key == "name":
continue
output_file.write("\n ")
if include_names:
output_file.write(".{0} = ".format(key))
if isinstance(value, StructNode):
output_file.write("{")
if value.items():
value.write_fields_to_output(output_file, include_names)
output_file.write("\n },")
else:
output_file.write(" 0 },")
elif isinstance(value, list):
# If list, then it it is the cnn_config layer_configs.
output_file.write("{")
for idx, element in enumerate(value):
output_file.write("\n { // layer_%d" % idx)
element.write_fields_to_output(output_file, include_names)
output_file.write("\n },")
output_file.write("\n }")
else:
if value is not None:
output_file.write("{0},".format(value))
if not include_names:
output_file.write("\t// {0}".format(key))
else:
output_file.write("NULL,")