make.py

#!/usr/bin/env python3

import argparse
import os

from litex.build.tools import write_to_file
from litex.soc.integration.soc_sdram import *
from litex.soc.integration.builder import *


def make_args(parser, platform='opsis', target='hdmi2usb'):
    parser.add_argument("--platform", action="store", default=os.environ.get('PLATFORM', platform))
    parser.add_argument("--target", action="store", default=os.environ.get('TARGET', target))
    parser.add_argument("--iprange", default="192.168.100")
    parser.add_argument("--cpu-type", default=os.environ.get('CPU', 'lm32'))

    parser.add_argument("-Op", "--platform-option", default=[], nargs=2, action="append", help="set platform-specific option")
    parser.add_argument("-Ot", "--target-option", default=[], nargs=2, action="append", help="set target-specific option")
    parser.add_argument("-Ob", "--build-option", default=[], nargs=2, action="append", help="set build option")
    parser.add_argument("--no-compile-firmware", action="store_true", help="do not compile the firmware")


def make_builddir(args):
    return "build/{}_{}_{}/".format(args.platform, args.target.lower(), args.cpu_type)


def make_testdir(args):
    builddir = make_builddir(args)
    testdir = "{}/test".format(builddir)
    return testdir


def make_platform(args):
    exec("from platforms.{} import Platform".format(args.platform), globals())
    return Platform(**dict(args.platform_option))


def main():
    parser = argparse.ArgumentParser(description="Opsis LiteX SoC", conflict_handler='resolve')
    make_args(parser)
    builder_args(parser)
    soc_sdram_args(parser)

    args = parser.parse_args()
    assert args.platform is not None
    assert args.target is not None

    platform = make_platform(args)

    exec("from targets.{}.{} import SoC".format(args.platform, args.target.lower(), args.target), globals())
    soc = SoC(platform, **soc_sdram_argdict(args), **dict(args.target_option))
    if hasattr(soc, 'configure_iprange'):
        soc.configure_iprange(args.iprange)

    builddir = make_builddir(args)
    testdir = make_testdir(args)

    buildargs = builder_argdict(args)
    if not buildargs.get('output_dir', None):
        buildargs['output_dir'] = builddir
    if not buildargs.get('csr_csv', None):
        buildargs['csr_csv'] = os.path.join(testdir, "csr.csv")

    builder = Builder(soc, **buildargs)
    if not args.no_compile_firmware:
        builder.add_software_package("uip", "{}/firmware/uip".format(os.getcwd()))
        builder.add_software_package("firmware", "{}/firmware".format(os.getcwd()))
    vns = builder.build(**dict(args.build_option))

    if hasattr(soc, 'pcie_phy'):
        from targets.common import cpu_interface
        csr_header = cpu_interface.get_csr_header(soc.get_csr_regions(), soc.get_constants())
        kerneldir = os.path.join(builddir, "software", "pcie", "kernel")
        os.makedirs(kerneldir, exist_ok=True)
        write_to_file(os.path.join(kerneldir, "csr.h"), csr_header)

    if hasattr(soc, 'do_exit'):
        soc.do_exit(vns, filename="{}/analyzer.csv".format(testdir))


if __name__ == "__main__":
    main()