Function main

Test/astc_size_binary.py:145–196

The main function. Return: The process return code.

()

def main() -> int:
    '''
    The main function.

    Return:
        The process return code.
    '''
    args = parse_command_line()

    # Preflight - check that size exists
    path = shutil.which('size')
    if not path:
        print('ERROR: The size utility is not installed')
        return 1

    # Pick an appropriate implementation based on host OS
    run_size = run_size_linux
    if platform.system() == 'Darwin':
        run_size = run_size_macos

    # Collect the data
    try:
        ref_dat = None
        if args.ref_bin:
            ref_dat = run_size(args.ref_bin.name)

        new_dat = run_size(args.bin.name)

    except sp.CalledProcessError as ex:
        print('ERROR: The size utility failed')
        print(ex.stderr.strip())
        return 1

    # Common header
    print('              Code        RO        ZI')

    # One binary, no reference comparison
    if not ref_dat:
        print(f'     New{new_dat[0]:>10}{new_dat[1]:>10}{new_dat[2]:>10}')

    # Two binaries with reference comparison
    else:
        print(f'     Ref{ref_dat[0]:>10}{ref_dat[1]:>10}{ref_dat[2]:>10}')
        print(f'     New{new_dat[0]:>10}{new_dat[1]:>10}{new_dat[2]:>10}')

        df_a = [y - x for (x, y) in zip(ref_dat, new_dat)]
        df_r = [((y - x) / x) * 100.0 for (x, y) in zip(ref_dat, new_dat)]

        print(f'Diff abs{df_a[0]:>+10}{df_a[1]:>+10}{df_a[2]:>+10}')
        print(f'Diff rel{df_r[0]:>+10.2f}%{df_r[1]:>+9.2f}%{df_r[2]:>+9.2f}%')

    return 0


if __name__ == '__main__':