package com.seaofnodes.simple;

import com.seaofnodes.isa.eval.EvalRisc5;

import com.seaofnodes.simple.codegen.RegAllocTestSupport.CheckedCodeGen;
import com.seaofnodes.simple.codegen.CodeGen;
import com.seaofnodes.simple.node.FunNode;
import com.seaofnodes.simple.node.Node;
import com.seaofnodes.simple.node.cpus.riscv.riscv;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;
import static org.junit.Assert.*;

// Runs 32-bit R5 code in an emulator
public abstract class TestRisc5 {

    public static EvalRisc5 build( String file, int arg, int spills, boolean print ) throws IOException {
        String dir = "src/test/java/com/seaofnodes/simple/progs";
        String src = Files.readString(Path.of(dir + "/"+file+".smp"));
        return build(dir,src, file, null, arg, spills, print);
    }

    public static EvalRisc5 build( String name, String src, int arg, int spills, boolean print ) throws IOException {
        String dir = "src/test/java/com/seaofnodes/simple/progs";
        return build(dir, src ,name, null, arg, spills, print);
    }

    // Compile and run a simple program
    public static EvalRisc5 build( String dir, String src, String file, String main, int arg, int spills, boolean print ) throws IOException {
        // Compile and export Simple
        CodeGen code = new CheckedCodeGen(src).driver("riscv", "SystemV",null);
        SpillStats.recordNative(code,"riscv","SystemV");
        if( print ) { code.print_as_hex(); System.out.print(code.asm()); }

        // Allocation quality not degraded
        SpillStats.checkSpills(spills,code._regAlloc._spillScaled);

        // Image
        byte[] image = new byte[1<<20]; // A megabyte (1024*1024 bytes)
        EvalRisc5 R5 = new EvalRisc5(image, 1<<16);
        // Code at offset 0
        System.arraycopy(code._encoding.bits(), 0, image, 0, code._encoding.bits().length);
        // Initial incoming int arg
        R5.regs[riscv.A0] = arg;
        // malloc memory starts at 64K and goes up

        // Look up starting PC or zero if not given
        if( main!=null )
            for( Node use : code._start._outputs )
                if( use instanceof FunNode fun && main.equals(fun._name) )
                    R5._pc = code._encoding._opStart[fun._nid];

        return R5;
    }

}