package com.seaofnodes.simple;

import com.seaofnodes.simple.codegen.CodeGen;
import org.junit.Test;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.fail;

public class Chapter13Test {
    @Test
    public void testJig() {
        CodeGen code = new CodeGen(
"""
return 3;
""");
        code.parse().opto();
        assertEquals("return 3;", code.print());
        assertEquals("3", Eval2.eval(code,  0));
    }

    @Test
    public void testLinkedList0() {
        CodeGen code = new CodeGen(
"""
struct LLI { LLI? next; int i; };
LLI? !head = null;
while( arg ) {
    LLI !x = new LLI;
    x.next = head;
    x.i = arg;
    head = x;
    arg = arg-1;
}
return head.next.i;
""");
        try { code.parse().opto().typeCheck(); fail(); }
        catch( Exception e ) { assertEquals("Might be null accessing 'next'",e.getMessage()); }
    }

    @Test
    public void testLinkedList1() {
        CodeGen code = new CodeGen(
"""
struct _LLI { _LLI? next; int q; };
_LLI? !head = null;
while( arg ) {
    _LLI !x = new _LLI;
    x.next = head;
    x.q = arg;
    head = x;
    arg = arg-1;
}
if( !head ) return 0;
_LLI? next = head.next;
if( next==null ) return 1;
return next.q;
""");
        code.parse().opto().typeCheck();
        assertEquals("return Phi(Region,1,.q,0);", code.print());
        assertEquals("2", Eval2.eval(code,  3));
    }

    @Test
    public void testCoRecur() {
        CodeGen code = new CodeGen(
"""
struct _int0 { int i; Test._flt0? f; };
struct _flt0 { flt f;      _int0? i; };
_int0 !i0 = new _int0;
i0.i = 17;
_flt0 !f0 = new _flt0;
f0.f = 3.14;
i0.f = f0;
f0.i = i0;
return f0.i.f.i.i;
""");
        code.parse().opto().typeCheck();
        assertEquals("return 17;", code.print());
    }

    @Test
    public void testNullRef0() {
        CodeGen code = new CodeGen(
"""
struct _N { _N? next; int i; };
_N n = new _N;
return n.next;
""");
        code.parse().opto();
        assertEquals("return null;", code.print());
    }

    @Test
    public void testNullRef1() {
        CodeGen code = new CodeGen(
"""
struct M { int m; };
struct N { M next; int i; new N = { M m -> next = m; }; };
N n = new N(new M);
return n.next;
""");
        code.parse().opto();
        assertEquals("Stop[ return (const)Test.M; return MEM[ 2:.m=0;]; return Test.M; return MEM[ 2:___ 3:___ 4:___ 5:___ 6:.next=Parm_m(Test.N.N,*Test.M {i64 !m; }); 7:.i=0;]; return MEM[ 2:___ 3:___ 4:___ 5:___ 6:___ 7:.i=0;]; return Test.N; ]", code.print());
    }

    @Test
    public void testNullRef2() {
        CodeGen code = new CodeGen(
"""
struct M { int m; };
struct N { M next; int i; new N = { M m -> next = m; }; };
N n = new N(null);
return n.next;
""");
        try { code.parse().opto().typeCheck(); fail(); }
        catch( Exception e ) { assertEquals("Argument #0 isa null, but must be a *Test.M {i64 !m; }",e.getMessage()); }
    }

    @Test
    public void testNullRef3() {
        CodeGen code = new CodeGen(
"""
struct N { N? next; int i; };
N !n = new N;
n.i = 3.14;
return n.i;
""");
        try { code.parse().opto().typeCheck(); fail(); }
        catch( Exception e ) { assertEquals("Type 3.14 is not of declared type i64",e.getMessage()); }
    }

    @Test
    public void testNullRef4() {
        CodeGen code = new CodeGen("return -null-5/null-5;");
        try { code.parse().opto().typeCheck(); fail(); }
        catch( Exception e ) { assertEquals("Cannot 5 // null",e.getMessage()); }
    }

    @Test public void testNullRef5() {
        CodeGen code = new CodeGen("return null+42;");
        try { code.parse().opto().typeCheck(); fail(); }
        catch( Exception e ) { assertEquals("Cannot null + 42",e.getMessage()); }
    }

    @Test
    public void testEmpty() {
        CodeGen code = new CodeGen(
"""
struct _S{};
return 0;
""");
        code.parse().opto();
        assertEquals("return 0;", code.print());
        assertEquals("0", Eval2.eval(code,  0));
    }

    @Test
    public void testForwardRef0() {
        CodeGen code = new CodeGen(
"""
struct S1 { S2? s; };
return new S2;
""");
        try { code.parse().opto().typeCheck(); fail(); }
        catch( Exception e ) { assertEquals("Unknown struct type 'S2'",e.getMessage()); }
    }

    @Test
    public void testForwardRef1() {
        CodeGen code = new CodeGen(
"""
struct _S1 { _S2? s; };
struct _S2 { int x; };
return (new _S1).s=new _S2;
""");
        code.parse().opto();
        assertEquals("return Test._S2;", code.print());
    }

    @Test
    public void testcheckNull() {
        CodeGen code = new CodeGen(
"""
struct I { int i; };
struct P { I? pi; };
P !p1 = new P;
P !p2 = new P;
p2.pi = new I;
p2.pi.i = 2;
if (arg) p1 = new P;
return p1.pi.i + 1;
""");
        try { code.parse().opto().typeCheck();  fail(); }
        catch( Exception e ) {  assertEquals("Might be null accessing 'i'",e.getMessage());  }
    }

    @Test
    public void testCoRecur2() {
        CodeGen code = new CodeGen(
"""
struct _A { Test._B? f0; Test._C? f1; };  _A !a = new _A;
struct _B { Test._C? f0; Test._A? f1; };  _B !b = new _B;
struct _C { Test._A? f0; Test._B? f1; };  _C !c = new _C;

a.f0=b;  a.f1=c;
b.f0=c;  b.f1=a;
c.f0=a;  c.f1=b;

return a.f0.f1.f0.f1.f0;

""");
        code.parse().opto().typeCheck().GCM().localSched();
        assertEquals("return Test._B;", code.print());
        assertEquals("Test._B{f0=Test._C{f0=Test._A{f0=$cyclic,f1=$cyclic},f1=$cyclic},f1=$cyclic}", Eval2.eval(code,  0));
    }

    @Test
    public void testCoRecur3() {
        CodeGen code = new CodeGen(
"""
struct _A { Test._L? a; Test._T? b; Test._F? c; };  _A !a = new _A;
struct _B { Test._M? a; Test._U? b; Test._G? c; };  _B !b = new _B;
struct _C { Test._N? a; Test._V? b; Test._H? c; };  _C !c = new _C;
struct _D { Test._O? a; Test._W? b; Test._I? c; };  _D !d = new _D;
struct _E { Test._P? a; Test._X? b; Test._J? c; };  _E !e = new _E;
struct _F { Test._Q? a; Test._Y? b; Test._K? c; };  _F !f = new _F;
struct _G { Test._R? a; Test._Z? b; Test._L? c; };  _G !g = new _G;
struct _H { Test._S? a; Test._A? b; Test._M? c; };  _H !h = new _H;
struct _I { Test._T? a; Test._B? b; Test._N? c; };  _I !i = new _I;
struct _J { Test._U? a; Test._C? b; Test._O? c; };  _J !j = new _J;
struct _K { Test._V? a; Test._D? b; Test._P? c; };  _K !k = new _K;
struct _L { Test._W? a; Test._E? b; Test._Q? c; };  _L !l = new _L;
struct _M { Test._X? a; Test._F? b; Test._R? c; };  _M !m = new _M;
struct _N { Test._Y? a; Test._G? b; Test._S? c; };  _N !n = new _N;
struct _O { Test._Z? a; Test._H? b; Test._T? c; };  _O !o = new _O;
struct _P { Test._A? a; Test._I? b; Test._U? c; };  _P !p = new _P;
struct _Q { Test._B? a; Test._J? b; Test._V? c; };  _Q !q = new _Q;
struct _R { Test._C? a; Test._K? b; Test._W? c; };  _R !r = new _R;
struct _S { Test._D? a; Test._L? b; Test._X? c; };  _S !s = new _S;
struct _T { Test._E? a; Test._M? b; Test._Y? c; };  _T !t = new _T;
struct _U { Test._F? a; Test._N? b; Test._Z? c; };  _U !u = new _U;
struct _V { Test._G? a; Test._O? b; Test._A? c; };  _V !v = new _V;
struct _W { Test._H? a; Test._P? b; Test._B? c; };  _W !w = new _W;
struct _X { Test._I? a; Test._Q? b; Test._C? c; };  _X !x = new _X;
struct _Y { Test._J? a; Test._R? b; Test._D? c; };  _Y !y = new _Y;
struct _Z { Test._K? a; Test._S? b; Test._E? c; };  _Z !z = new _Z;

a.a=l;  a.b=t; a.c=f;
b.a=m;  b.b=u; b.c=g;
c.a=n;  c.b=v; c.c=h;
d.a=o;  d.b=w; d.c=i;
e.a=p;  e.b=x; e.c=j;
f.a=q;  f.b=y; f.c=k;
g.a=r;  g.b=z; g.c=l;
h.a=s;  h.b=a; h.c=m;
i.a=t;  i.b=b; i.c=n;
j.a=u;  j.b=c; j.c=o;
k.a=v;  k.b=d; k.c=p;
l.a=w;  l.b=e; l.c=q;
m.a=x;  m.b=f; m.c=r;
n.a=y;  n.b=g; n.c=s;
o.a=z;  o.b=h; o.c=t;
p.a=a;  p.b=i; p.c=u;
q.a=b;  q.b=j; q.c=v;
r.a=c;  r.b=k; r.c=w;
s.a=d;  s.b=l; s.c=x;
t.a=e;  t.b=m; t.c=y;
u.a=f;  u.b=n; u.c=z;
v.a=g;  v.b=o; v.c=a;
w.a=h;  w.b=p; w.c=b;
x.a=i;  x.b=q; x.c=c;
y.a=j;  y.b=r; y.c=d;
z.a=k;  z.b=s; z.c=e;

return a.b.c.a.b.c.a.b.c.a.b.c.a.b.c.a.b.c;

""");
        code.parse().opto().typeCheck().GCM().localSched();
        assertEquals("return Test._R;", code.print());
        assertEquals("Test._R{a=Test._C{a=Test._N{a=Test._Y{a=Test._J{a=Test._U{a=Test._F{a=Test._Q{a=Test._B{a=Test._M{a=Test._X{a=Test._I{a=Test._T{a=Test._E{a=Test._P{a=Test._A{a=Test._L{a=Test._W{a=Test._H{a=Test._S{a=Test._D{a=Test._O{a=Test._Z{a=Test._K{a=Test._V{a=Test._G{a=$cyclic,b=$cyclic,c=$cyclic},b=$cyclic,c=$cyclic},b=$cyclic,c=$cyclic},b=$cyclic,c=$cyclic},b=$cyclic,c=$cyclic},b=$cyclic,c=$cyclic},b=$cyclic,c=$cyclic},b=$cyclic,c=$cyclic},b=$cyclic,c=$cyclic},b=$cyclic,c=$cyclic},b=$cyclic,c=$cyclic},b=$cyclic,c=$cyclic},b=$cyclic,c=$cyclic},b=$cyclic,c=$cyclic},b=$cyclic,c=$cyclic},b=$cyclic,c=$cyclic},b=$cyclic,c=$cyclic},b=$cyclic,c=$cyclic},b=$cyclic,c=$cyclic},b=$cyclic,c=$cyclic},b=$cyclic,c=$cyclic},b=$cyclic,c=$cyclic},b=$cyclic,c=$cyclic},b=$cyclic,c=$cyclic},b=$cyclic,c=$cyclic},b=$cyclic,c=$cyclic}", Eval2.eval(code,  0));
    }
}