compile2.py

#!/usr/bin/env python3
import os
import subprocess
import time
from pathlib import Path
import sys
from config import BASE_DIR, CATEGORIES, BUILD_DIR ,BENCHMARK_DIR
from compilers.compiler_setup import get_compilers
#TODO support version in the compiled file name


def compile_files(programdir, builddir, prefix, compilers):
    """Recursively compile files with a prefix in the given directory using multiple compilers."""
    assert isinstance (programdir, Path)
    assert isinstance (builddir, Path)
    builddir.mkdir(parents=True,exist_ok=True)
    files = sorted(programdir.glob('**/*.' + prefix))
    for compiler in compilers:
        print ("Compiling for " + compiler.name)
        length = len (files)
        for i, file in enumerate (files,start = 1):
            target_name = file.stem
            directory = file.parent.name
            assert target_name ==  directory 
            input_file = file.as_posix()
            delimiter = "_"
            if compiler.suffix is None:
                output_filename = delimiter.join([target_name, compiler.name.lower()])
            else:
                output_filename = delimiter.join([target_name, compiler.name.lower(), compiler.suffix])

            output_file = os.path.join(builddir,output_filename)
            print("Compiling: " + file.name + " to " + output_filename + " " + str(i) + "/" + str (length))
            compiler.compile(input_file,output_file)


if __name__ == "__main__":
    benchmark_base_dir = "." 
   
    # Just in case directory to output isn't in repo directory"
    output_base_dir = './out'
    # List of SML compilers as instantiated objects
    compilers_by_lang = get_compilers(base_dir=BASE_DIR)
    
    sml_compilers = compilers_by_lang.get("sml", [])
    cakeml_compilers = compilers_by_lang.get("cakeml", [])
    for category in CATEGORIES:
        directory_to_search = BENCHMARK_DIR.joinpath(category)
        directory_to_output = BUILD_DIR.joinpath(category)
        compile_files(directory_to_search,directory_to_output,"cml", cakeml_compilers)
        compile_files(directory_to_search,directory_to_output,"sml", sml_compilers)