hw/ip/prim/util/generate_prim_mubi_pkg.py

#!/usr/bin/env python3
# Copyright lowRISC contributors.
# Licensed under the Apache License, Version 2.0, see LICENSE for details.
# SPDX-License-Identifier: Apache-2.0
r"""Convert mako template to Hjson register description
"""
import argparse
import sys
from io import StringIO

from mako.template import Template


def main():
    parser = argparse.ArgumentParser(prog="generate_prim_mubi_pkg")
    parser.add_argument('input',
                        nargs='?',
                        metavar='file',
                        type=argparse.FileType('r'),
                        default=sys.stdin,
                        help='input template file')
    parser.add_argument('--n_max_nibbles',
                        type=int,
                        help='The script will create all multibit types with 1 to n_max_nibbles.',
                        default=4)
    
    args = parser.parse_args()

    # Determine output: if stdin then stdout if not then ??
    out = StringIO()

    reg_tpl = Template(args.input.read())
    out.write(reg_tpl.render(n_max_nibbles=args.n_max_nibbles))

    print(out.getvalue())

    out.close()


if __name__ == "__main__":
    main()