package com.seaofnodes.simple;

import com.seaofnodes.isa.eval.EvalRisc5;

import com.seaofnodes.isa.eval.EvalArm64;

import com.seaofnodes.simple.codegen.RegAllocTestSupport.CheckedCodeGen;
import com.seaofnodes.simple.node.*;
import com.seaofnodes.simple.type.*;


import com.seaofnodes.simple.codegen.CodeGen;
import com.seaofnodes.simple.node.cpus.riscv.riscv;
import java.io.IOException;
import org.junit.Ignore;
import org.junit.Test;
import static org.junit.Assert.*;

public class Chapter24Test {
    @Test public void testRecursiveWidening() {
        String[] srcs = {
            // Parm(x) must widen even though the recursive result is constant.
            "val f = { int x, int limit -> x >= limit ? 0 : f(x+1,limit); }; return f(0,arg);",
            // The growing result has its own cycle, which never crosses Parm(x).
            "val f = { int x -> x <= 0 ? 0 : f(x-1)+1; }; return f(3);"
        };
        for( int i=0; i<srcs.length; i++ )
            for( long seed : new long[]{0,42,126} ) {
                var code = new CodeGen(srcs[i],TypeInteger.BOT,seed,true).driver(CodeGen.Phase.LocalSched);
                assertEquals(i==0 ? "0" : "3",Eval2.eval(code,3));
            }
    }

    @Test public void testColdLoopSelfConflict() { com.seaofnodes.simple.codegen.RegAllocTestSupport.coldLoopSelfConflict(); }

    @Test public void testPrintingDeadFunction() throws Exception {
        var code = new CodeGen("return 0;").parse();
        var tfp = TypeFunPtr.TEST;
        var fun = new FunNode(null,tfp,"deadPrinterTarget");
        org.junit.Assert.assertTrue(fun.isDead());
        code.link(fun);
        var con = new ConstantNode(tfp);
        var call = new CallNode(null,con,con,con);
        con.toString();
        call.name();

        org.junit.Assert.assertSame(fun,code._link(tfp));
        // Optimizer lookup still performs its usual lazy cleanup.
        org.junit.Assert.assertNull(code.link(tfp));
        org.junit.Assert.assertNull(code._link(tfp));
    }


    @Ignore
    @Test
    public void testJig() throws IOException {
        String src = "int v0=0;  v0 = arg && 0== 0 !=0;";
        testCPU(src,"x86_64_v2", "Win64"  ,-1,null);
        testCPU(src,"riscv"    , "SystemV",-1,null);
        testCPU(src,"arm"      , "SystemV",-1,null);
    }

    static CodeGen testCPU( String src, String cpu, String os, int spills, String stop ) {
        CodeGen code = new CheckedCodeGen(src).driver(CodeGen.Phase.Encoding,cpu,os);
        SpillStats.record(code,"Chapter24",cpu,os);
        SpillStats.checkSpills(spills,code._regAlloc._spillScaled);
        if( stop != null )
            assertEquals(stop, code._stop.toString());
        return code;
    }

    @Test
    public void testOr() throws IOException {
        String src =
"""
int a = 1;
int b = 0;

if(a++ || b++ ) {
    if(b == 0 && a == 2) {
        sys.io.p("Or");
    }
} else{
    sys.io.p("And");
}
return 0;
""";

        TestC.runSF("or1", src, null, "Or", 0);

        // Evaluate on RISC5 emulator
        EvalRisc5 R5 = TestRisc5.build("or1", src, 0, 2, false);
        int trap = R5.step(100);
        assertEquals(0,trap);
        assertEquals(0,R5.regs[riscv.A0]);
        assertEquals("Or",R5._stdout.toString());

        // Evaluate on ARM emulator
        EvalArm64 arm = TestArm64.build("or1", src,0, 2, false);
        trap = arm.step(100);
        assertEquals(0,trap);
        assertEquals(0,arm.regs[0]);
        assertEquals("Or",arm._stdout.toString());
    }

    @Test
    public void testAnd() throws IOException {
        String src =
"""
int a = 1;
int b = 1;

if(a && b) {
    sys.io.p("And");
} else {
    sys.io.p("Or");
}
return 0;
""";
        TestC.runSF("and1", src, null, "And", 0);

        // Evaluate on RISC5 emulator
        EvalRisc5 R5 = TestRisc5.build("and1", src, 0, 2, false);
        int trap = R5.step(100);
        assertEquals(0,trap);
        assertEquals(0,R5.regs[riscv.A0]);
        assertEquals("And",R5._stdout.toString());

        // Evaluate on ARM emulator
        EvalArm64 arm = TestArm64.build("and1", src, 0, 2, false);
        trap = arm.step(100);
        assertEquals(0,trap);
        assertEquals(0,arm.regs[0]);
        assertEquals("And",arm._stdout.toString());
    }

    @Test
    public void testAndPtr() throws IOException {
        // Todo: have one src here
        String src =
"""
struct S { !S? !fld; };
val ptr = arg == 1 ? null : new S{fld = arg==1 ? null : new S{fld = null;};};
return ptr && ptr.fld ? "true" : "false";
""";
        String src2 =
"""
struct S { !S? !fld; };

val ptr = arg == 1 ? null : new S{fld = arg==1 ? null : new S{fld = null;};};
if( ptr && ptr.fld ) {
  sys.io.p("true");
} else {
  sys.io.p("false");
}

return 0;
""";
        CodeGen code = new CodeGen(src).driver(CodeGen.Phase.TypeCheck);
        assertEquals("[97-117][ 116,114,117,101]", Eval2.eval(code, 0));

        // Evaluate on RISC5 emulator
        EvalRisc5 R5 = TestRisc5.build("ptrand",src2, 1, 8, false);
        int trap = R5.step(100);
        assertEquals(0,trap);
        assertEquals(0,R5.regs[riscv.A0]);
        assertEquals("false",R5._stdout.toString());

        // Evaluate on ARM emulator
        EvalArm64 arm = TestArm64.build("ptrand", src2, 1, 8, false);
        trap = arm.step(100);
        assertEquals(0,trap);
        assertEquals(0,arm.regs[0]);
        assertEquals("false",arm._stdout.toString());
    }

    // conditional side effect
    @Test
    public void testCondSideEffAnd() throws IOException {
        String src =
"""
int a = 1;
int b = 1;

int x=1;
int y=1;
int z=0;

int g = x++ && y++ && z++;

if(x == 2 && y == 2 && z == 1 && g == 0) {
    sys.io.p("Effected");
} else {
    sys.io.p("Not effected");
}
return 0;
""";

        TestC.runSF("and2", src, null, "Effected", 0);

        // Evaluate on RISC5 emulator
        EvalRisc5 R5 = TestRisc5.build("and2", src, 0, 2, false);
        int trap = R5.step(100);
        assertEquals(0,trap);
        assertEquals(0,R5.regs[riscv.A0]);
        assertEquals("Effected",R5._stdout.toString());

        // Evaluate on ARM emulator
        EvalArm64 arm = TestArm64.build("and2", src, 0, 2, false);
        trap = arm.step(100);
        assertEquals(0,trap);
        assertEquals(0,arm.regs[0]);
        assertEquals("Effected",arm._stdout.toString());
    }


    @Test
    public void testCondSideEffOr() throws IOException {
        String src =
"""
int a = 1;
int b = 1;

int x = -1;
int y=1;
int z= -1;

int g = x++ || y++ || z++;

int switch = 0;
if(x == 4 || y == 4 || z == 4) {
    switch = -1;
} else {
    switch = 1;
}
int cd = 1;
if((x == 0 && y == 1 && z==-1) && switch && g) {
    sys.io.p("Effected");
} else {
    sys.io.p("Not effected");
}
return 0;
""";
        TestC.runSF("or2", src, null, "Effected", 0);

        // Evaluate on RISC5 emulator
        EvalRisc5 R5 = TestRisc5.build("or2", src, 0, 2, false);
        int trap = R5.step(100);
        assertEquals(0,trap);
        assertEquals(0,R5.regs[riscv.A0]);
        assertEquals("Effected",R5._stdout.toString());

        // Evaluate on ARM emulator
        EvalArm64 arm = TestArm64.build("or2", src, 0, 2, false);
        trap = arm.step(100);
        assertEquals(0,trap);
        assertEquals(0,arm.regs[0]);
        assertEquals("Effected",arm._stdout.toString());
    }

    // test it with function calls
    @Test
    public void testFuncCall() throws IOException {
        String src =
"""
// -*- mode: java;  -*-
int a = 1;
int b = 1;

var sq = { int x ->
    x*x;
};

if(a && sq(0)) {
    sys.io.p("And");
} else {
    sys.io.p("Or");
}
return 0;
""";
        TestC.runSF("and3", src, null, "Or", 0);

        // Evaluate on RISC5 emulator
        EvalRisc5 R5 = TestRisc5.build("and3", src, 0, 2, false);
        int trap = R5.step(100);
        assertEquals(0,trap);
        assertEquals(0,R5.regs[riscv.A0]);
        assertEquals("Or",R5._stdout.toString());

        // Evaluate on ARM emulator
        EvalArm64 arm = TestArm64.build("and3", src, 0, 2, false);
        trap = arm.step(100);
        assertEquals(0,trap);
        assertEquals(0,arm.regs[0]);
        assertEquals("Or",arm._stdout.toString());
    }

    // test it with function calls
    @Test
    public void testStack1() throws IOException {
        String src =
"""
int something_specific = 11;

if (10 < something_specific < 20) {
    sys.io.p("In Range");
} else {
   sys.io.p("Out of Range");
}
return 0;
""";

        TestC.runSF("stacked_r_1", src, null, "In Range", 0);

        // Evaluate on RISC5 emulator
        EvalRisc5 R5 = TestRisc5.build("stacked_r_1", src, 0, 2, false);
        int trap = R5.step(100);
        assertEquals(0,trap);
        assertEquals(0,R5.regs[riscv.A0]);
        assertEquals("In Range",R5._stdout.toString());

        // Evaluate on ARM emulator
        EvalArm64 arm = TestArm64.build("stacked_r_1", src, 0, 2, false);
        trap = arm.step(100);
        assertEquals(0,trap);
        assertEquals(0,arm.regs[0]);
        assertEquals("In Range",arm._stdout.toString());
    }

    @Test
    public void testStack2() throws IOException {
        String src =
"""
int score = 75;

if (60 <= score < 90) {
    sys.io.p("In Range");
} else {
sys.io.p("Out of range");
}
return 0;
""";

        TestC.runSF("stacked_r_2", src, null, "In Range", 0);

        // Evaluate on RISC5 emulator
        EvalRisc5 R5 = TestRisc5.build("stacked_r_2", src, 0, 2, false);
        int trap = R5.step(100);
        assertEquals(0,trap);
        assertEquals(0,R5.regs[riscv.A0]);
        assertEquals("In Range",R5._stdout.toString());

        // Evaluate on ARM emulator
        EvalArm64 arm = TestArm64.build("stacked_r_2", src, 0, 2, false);
        trap = arm.step(100);
        assertEquals(0,trap);
        assertEquals(0,arm.regs[0]);
        assertEquals("In Range",arm._stdout.toString());
    }

    @Test
    public void testStack3() throws IOException {
        String src =
"""
var sq = { int x ->
    x*x;
};

int score = 5;

val str = (sq(2) <= score < sq(3)) ? "In Range" : "Out of range";
sys.io.p(str);
return 0;
""";

        TestC.runSF("stacked_r_3", src, null, "In Range", 5);

        // Evaluate on RISC5 emulator
        EvalRisc5 R5 = TestRisc5.build("stacked_r_3", src, 0, 5, false);
        int trap = R5.step(100);
        assertEquals(0,trap);
        assertEquals(0,R5.regs[riscv.A0]);
        assertEquals("In Range",R5._stdout.toString());

        // Evaluate on ARM emulator
        EvalArm64 arm = TestArm64.build("stacked_r_3", src, 0, 5, false);
        int trap_1 = arm.step(100);
        assertEquals(0,trap_1);
        assertEquals(0,arm.regs[0]);
        assertEquals("In Range",arm._stdout.toString());
    }

    @Test
    public void testStack4() throws IOException {
        String src =
"""
int something_specific = 9;

if (10 < something_specific < 20) {
    sys.io.p("In Range");
} else {
   sys.io.p("Out of Range");
}
return 0;
""";

        TestC.runSF("stacked_r_4", src, null, "Out of Range", 0);

        // Evaluate on RISC5 emulator
        EvalRisc5 R5 = TestRisc5.build("stacked_r_4", src,0, 2, false);
        int trap = R5.step(100);
        assertEquals(0,trap);
        assertEquals(0,R5.regs[riscv.A0]);
        assertEquals("Out of Range",R5._stdout.toString());

        // Evaluate on ARM emulator
        EvalArm64 arm = TestArm64.build("stacked_r_4", src, 0, 2, false);
        trap = arm.step(100);
        assertEquals(0,trap);
        assertEquals(0,arm.regs[0]);
        assertEquals("Out of Range",arm._stdout.toString());
    }

    @Test
    public void testStack5() throws IOException {
        String src =
"""
var sq = { int x ->
    x*x;
};

int score = 5;

if (sq(2) <= score < sq(3)) {
sys.io.p("In Range");
} else {
sys.io.p("Out of range");
}

return 0;
""";
        TestC.runSF("stacked_r_5", src, null, "In Range", 5);

        // Evaluate on RISC5 emulator
        EvalRisc5 R5 = TestRisc5.build("stacked_r_5", src, 0, 7, false);
        int trap = R5.step(100);
        assertEquals(0,trap);
        assertEquals(0,R5.regs[riscv.A0]);
        assertEquals("In Range",R5._stdout.toString());

        // Evaluate on ARM emulator
        EvalArm64 arm = TestArm64.build("stacked_r_5", src, 0, 7, false);
        trap = arm.step(100);
        assertEquals(0,trap);
        assertEquals(0,arm.regs[0]);
        assertEquals("In Range",arm._stdout.toString());
    }


    @Test
    public void testStack6() {
        try { new CodeGen(
                    """
                        return 0 < arg > 1;
                        """
        ).parse(); fail(); }
        catch( RuntimeException e ) { assertEquals("Mixing relational directions in a chained relational test",e.getMessage()); }
    }

    @Test
    public void testStack9() throws IOException {
        String stack9 = "1";
        String src =
"""
var stack9 = { int x ->
    (x < 1) == 1;
};
""";
        String src2 =
"""
return (0 < 1) == 1;
""";
        TestC.run(src, "stacked_r_9x", null, stack9,0);

        // Evaluate on RISC5 emulator
        EvalRisc5 R5 = TestRisc5.build("stacked_r_9", src2, 0, 0, false);
        int trap = R5.step(100);
        assertEquals(0,trap);
        assertEquals(1,R5.regs[riscv.A0]);

        // Evaluate on ARM emulator
        EvalArm64 arm = TestArm64.build("stacked_r_9", src2, 0, 0, false);
        trap = arm.step(100);
        assertEquals(0,trap);
        assertEquals(1,arm.regs[0]);
    }

    @Test
    public void testStack10() throws IOException {
        String stack10 = "1";
        String src =
"""
var stack10 = { int x ->
    (x < 1) == (1>x);
};
""";
    String src2 =
"""
return (0 < 1) == (1>0);
""";
        TestC.run(src, "stacked_r_10x", null, stack10,0);

        // Evaluate on RISC5 emulator
        EvalRisc5 R5 = TestRisc5.build("stacked_r_10", src2, 0, 0, false);
        int trap = R5.step(100);
        assertEquals(0,trap);
        assertEquals(1,R5.regs[riscv.A0]);

        // Evaluate on ARM emulator
        EvalArm64 arm = TestArm64.build("stacked_r_10", src2,0, 0, false);
        trap = arm.step(100);
        assertEquals(0,trap);
        assertEquals(1,arm.regs[0]);
    }

    @Test
    public void testStack11() throws IOException {
        String stack11 = "1";
        String src =
"""
var stack11 = { int x ->
    0 < x < x+1 < 4;
};
""";
        String src2 =
"""
return 0 < arg < arg+1 < 4;
""";
        TestC.run(src, "stacked_r_11x", null, stack11,2);

        // Evaluate on RISC5 emulator
        EvalRisc5 R5 = TestRisc5.build("stacked_r_11", src2, 1, 2, false);
        int trap = R5.step(100);
        assertEquals(0,trap);
        assertEquals(1,R5.regs[riscv.A0]);

        // Evaluate on ARM emulator
        EvalArm64 arm = TestArm64.build("stacked_r_11", src2, 1, 0, false);
        trap = arm.step(100);
        assertEquals(0,trap);
        assertEquals(1,arm.regs[0]);
    }

    // shows that it is parsed as
    // ((0 != x) != 2)
    @Test
    public void testStack12() throws IOException {
        String stack11 = "1";
        String src =
                """
                var stack12 = { int x ->
                    0 != x != 2;
                };
                """;
        String src2 =
                """
                return 0 != arg != 2;
                """;
        TestC.run(src, "stacked_r_12x", null, stack11,0);

        // Evaluate on RISC5 emulator
        EvalRisc5 R5 = TestRisc5.build("stacked_r_12", src2, 1, 0, false);
        int trap = R5.step(100);
        assertEquals(0,trap);
        assertEquals(1,R5.regs[riscv.A0]);

        // Evaluate on ARM emulator
        EvalArm64 arm = TestArm64.build("stacked_r_12", src2, 1, 0, false);
        trap = arm.step(100);
        assertEquals(0,trap);
        assertEquals(1,arm.regs[0]);
    }


    // shows that it is parsed as
    // ((0 == x) == 1)
    @Test
    public void testStack13() throws IOException {
        String stack12 = "1";
        String src =
                """
                var stack13 = { int x ->
                    0 == x == 1;
                };
                """;
        String src2 =
                """
                return 0 == arg == 1;
                """;
        TestC.run(src, "stacked_r_13x", null, stack12,0);

        // Compile once per target, then check false, true and negative inputs.
        EvalRisc5 R5 = TestRisc5.build("stacked_r_13", src2, 1, 0, false);
        for( int arg : new int[]{-1,0,1} ) {
            R5._pc = 0;
            R5.regs[riscv.A0] = arg;
            assertEquals(0,R5.step(100));
            assertEquals(arg==0 ? 1 : 0,R5.regs[riscv.A0]);
        }
        EvalArm64 arm = TestArm64.build("stacked_r_13", src2, 1, 0, false);
        for( int arg : new int[]{-1,0,1} ) {
            arm._pc = 0;
            arm.regs[0] = arg;
            assertEquals(0,arm.step(100));
            assertEquals(arg==0 ? 1 : 0,arm.regs[0]);
        }
    }

    @Test
    public void testFRefFields() {
        String src =
"""
// A hypothetical scanner class
struct Scan {
    int !x;
    u8[~] !buf;
    // Skip whitespace
    val skip = { !Scan s ->
        while( s.buf[s.x] <= ' ' )
            s.x++;
    };
    // Peek a character; if matched consume it, else false.
    val peek = { !Scan !s, u8 c ->
        skip(s);
        if( s.buf[s.x] != c ) return false;
        s.x++;
        return true;
    };
};
!Scan !s = new Scan{ buf = "  q"; };
return Scan.peek(s,'q');
""";

        CodeGen code = new CodeGen(src).driver(CodeGen.Phase.TypeCheck);
        assertEquals("return Phi(Region,0,1);", code._stop.toString());
        assertEquals("1", Eval2.eval(code, 0));
        testCPU(src,"x86_64_v2", "Win64"  ,23,null);
    };

    @Test
    public void testFRefFields2() {
        String src =
"""
// A hypothetical scanner class
struct Scan {
    int !x;
    u8[~] !buf;
    // Skip whitespace
    val skip = { ->
        while( buf[x] <= ' ' )
            x++;
        return self;
    };
};
~Scan s = new Scan{ buf = "  q"; };
return s.skip().x;
""";

        try { new CodeGen(src).driver(CodeGen.Phase.TypeCheck); fail(); }
        catch( Exception e ) { org.junit.Assert.assertTrue(e.getMessage().startsWith("Argument #0 isa")); }
    };


    @Test
    public void testMethod() {
        String src =
"""
// A hypothetical scanner class
struct Scan {
    int !x;
    u8[~] !buf;
    // Skip whitespace
    val skip = { ->
        while( buf[x] <= ' ' )
            x++;
    };
    val require = { u8 ch ->
        skip();
        buf[x++]==ch;
    };
};
!Scan !s = new Scan{ buf = "  [1,2]"; };
return s.require('[');
""";

        CodeGen code = new CodeGen(src).driver(CodeGen.Phase.TypeCheck);
        assertEquals("return (.[]==91);", code._stop.toString());
        assertEquals("1", Eval2.eval(code, 0));
        testCPU(src,"x86_64_v2", "Win64"  ,4,null);
    };

    @Test
    public void testSCCP0() {
        String src =
"""
int x = 1;
int cnt = 5;
while (cnt) {
  cnt -= 1;
  if (x == 0) {
    x += 1;
  }
}
return x;
""";
        CodeGen code = new CodeGen(src).driver(CodeGen.Phase.TypeCheck);
        assertEquals("return 1;", code._stop.toString());
    }

    @Test
    public void testSCCP1() {
        String src =
"""
int x = 1;
int cnt = 5;
while( cnt-- )
  if( x == 1 )
    x = 2-x;
return x;
""";
        CodeGen code = new CodeGen(src).driver(CodeGen.Phase.TypeCheck);
        assertEquals("return 1;", code._stop.toString());
    }

}