#include <setjmp.h>

jmp_buf g_stopped;
#define occam_stop(pos, nargs, format, args...) longjmp(g_stopped, 1)

#define occam_INT_size SIZEOF_VOIDP
#define occam_extra_param 
#include "support/tock_support.h"

#define report_failure(msg, args...) { printf(msg, ##args); }

#define testf(call) do { \
	if (setjmp(g_stopped) == 0) { \
		call; \
		failures++; \
		report_failure(#call " failed, expected to stop, got: %lld\n", (int64_t)call); \
	} else { \
		passes++; \
	} \
  } while(0)

#define testp(exp, call) do { \
	if (setjmp(g_stopped) == 0) { \
		if (exp == call) { \
			passes++; \
		} else { \
			failures++; \
			report_failure(#call " failed, expected %lld, got %lld\n", (int64_t)exp, (int64_t)call); \
		} \
	} else { \
		failures++; \
		report_failure(#call "failed, unexpectedly stopped\n"); \
	} \
  } while (0)

#define mult(x,y) (x*y)
#define add(x,y) (x+y)

#define test_commutative(t,f,fe) \
	do {int done_x = 0; for (t x = 0; !(x == 0 && done_x == 1);x+=1) { done_x = 1;\
		int done_y = 0; for (t y = 0; !(y == x && done_y == 1);y+=1) { done_y = 1;\
			t r0 = 0, r1 = 0; \
			int stopped_earlier = 0, stopped_later = 0; \
			if (setjmp(g_stopped) == 0) { \
				r0 = f(x,y,""); \
			} else { \
				stopped_earlier = 1; \
			} \
			if (setjmp(g_stopped) == 0) { \
				r1 = f(y,x,""); \
			} else { \
				stopped_later = 1; \
			} \
			if (stopped_later == 1 && stopped_earlier == 1) {} else if \
			  (stopped_earlier == 0 && stopped_later == 0 && r0 == r1) { \
				if ((int)(t)(fe((int)x,(int)y)) != (int)(fe((int)x,(int)y))) { \
				  failures++; report_failure(#f " failed, overflowed but did not stop: %d %d",x,y);goto comm_label_##f; \
				} else if (r0 != (t)(fe((int)x,(int)y))) { failures++; \
				  report_failure(#f " failed, result (%d) not as expected (%d)", r0, (t)fe((int)x,(int)y));goto comm_label_##f;} \
			} else { \
				failures++; \
				report_failure(#f " failed, non-commutative with %d, %d (stop: %d, %d) (res: %d, %d)\n", \
				  (int)x, (int)y, stopped_earlier, stopped_later, r0, r1); \
				goto comm_label_##f; \
			} \
		} \
	}passes++; comm_label_##f:;} while (0)

// All the addition cases I wrote were effectively the same,
// so I just made it a macro.  We define min to be -max-1, which gets around
// the fact that the C standard apparently says that the largest
// constant for a value is -max (not -max-1).

//Since we test commutativity (at least for 8- and 16-bits),
//we only need to test one arrangement of each addition
//and multiplication test
#define check_addition(max, f) \
	do { testp(2,f(1,1,"")); \
		testp(0,f(0,0,"")); \
		testp(-1,f(0,-1,"")); \
		testp(max,f((max>>1)+1,max>>1,"")); \
		testf(f((max>>1)+1,(max>>1)+1,"")); \
		testf(f(max,max,"")); \
		testf(f(-max-1,-max-1,"")); \
		testp(-1,f(-max-1,max,"")); \
		testp(-max-1,f(-max-1,0,"")); \
		testp(max,f(max,0,"")); \
	} while (0)

#define check_multiplication(max, f) \
	do { testp(0, f(0,0,"")); \
		testp(max, f(1,max,"")); \
		testp(-max-1, f(1,-max-1,"")); \
		testp(-max, f(-1,max,"")); \
		testf(f(-1,-max-1,"")); \
		testp(-max-1,f(-2,(max>>1)+1,"")); \
		testp(-max-1,f(2,-((max>>1)+1),"")); \
		testf(f(-2,(max>>1)+2,"")); \
        testf(f(-max-1,-max-1,"")); \
        testf(f(max,max,"")); \
        testf(f(max,-max-1,"")); \
        testf(f(max-(max>>1),max-(max>>1),"")); \
        testf(f(5,(-max-1)/4,"")); \
        testf(f(5,(max>>2)+1,"")); \
	} while (0)

#define check_subtraction(max, f) \
	do { testp(0,f(1,1,"")); \
		testp(0,f(0,0,"")); \
		testp(0,f(max,max,"")); \
		testp(0,f(-max-1,-max-1,"")); \
		testp(1,f(0,-1,"")); \
		testp(-1,f(0,1,"")); \
		testp(-max,f(0,max,"")); \
		testp(-max-1,f(-1,max,"")); \
		testf(f(-2,max,"")); \
		testp(max,f(0,-max,"")); \
		testp(max,f(1,1-max,"")); \
		testf(f(0,-max-1,"")); \
		testf(f(1,-max,"")); \
	} while (0)

#define check_subtraction_u(max, f) \
	do { testp(0,f(1,1,"")); \
		testp(0,f(0,0,"")); \
		testp(0,f(max,max,"")); \
		testf(f(0,1,"")); \
		testf(f(max-1,max,"")); \
		testf(f(0,max,"")); \
	} while (0)

#define check_negation(max, f) \
	do { testp(0,f(0,"")); \
		testp(1,f(-1,"")); \
		testp(-1,f(1,"")); \
		testp(max,f(-max,"")); \
		testp(-max,f(max,"")); \
		testf(f(-max-1,"")); \
	} while (0)

#define check_division(max, f) \
	do { testp(0,f(0,1,"")); \
		testp(0,f(0,-1,"")); \
		testf(f(0,0,"")); \
		testf(f(1,0,"")); \
		testf(f(-1,0,"")); \
		testf(f(max,0,"")); \
		testf(f(-max-1,0,"")); \
		testf(f(-max-1,-1,"")); \
		testp(-max-1,f(-max-1,1,"")); \
		testp(max,f(-max,-1,"")); \
		testp(-max,f(max,-1,"")); \
		testp(max,f(max,1,"")); \
	} while (0)

#define check_division_u(max, f) \
	do { testp(0,f(0,1,"")); \
		testf(f(0,0,"")); \
		testf(f(1,0,"")); \
		testf(f(max,0,"")); \
		testp(max,f(max,1,"")); \
	} while (0)

#define check_rem(max, f) \
	do { testp(0, f(1,1,"")); \
		testp(0, f(0,1,"")); \
		testp(0, f(0,-1,"")); \
		testf(f(0,0,"")); \
		testf(f(1,0,"")); \
		testf(f(-1,0,"")); \
		testf(f(max,0,"")); \
		testf(f(-max-1,0,"")); \
		testp(0,f(max,max,"")); \
		testp(0,f(-max-1,-max-1,"")); \
		testp(-2,f(-max-1,max>>1,"")); \
		testp(-2,f(-max-1,-(max>>1),"")); \
		testp(-1,f(-max,max>>1,"")); \
		testp(-1,f(-max,-(max>>1),"")); \
		testp(-1,f(-max-1,-max,"")); \
		testp(-1,f(-max-1,max,"")); \
		testp(3,f(3,-max-1,"")); \
		testp(-3,f(-3,-max-1,"")); \
		testp(max,f(max,-max-1,"")); \
		testp(-max,f(-max,-max-1,"")); \
	} while (0)

#define check_rem_u(max, f) \
	do { testp(0, f(1,1,"")); \
		testp(0, f(0,1,"")); \
		testf(f(0,0,"")); \
		testf(f(1,0,"")); \
		testf(f(max,0,"")); \
		testp(0,f(max,max,"")); \
		testp(1,f(max,max-1,"")); \
	} while (0)

#define check_shift(max,bits,f) \
	do { testp(0, f(0,0,"")); \
		testp(max, f(max,0,"")); \
		testp(-max, f(-max,0,"")); \
		testp(-max-1, f(-max-1,0,"")); \
		testp(1,f(1,0,"")); \
		testp(-1,f(-1,0,"")); \
		testp(0, f(0,bits,"")); \
		testp(0, f(max,bits,"")); \
		testp(0, f(-max,bits,"")); \
		testp(0, f(-max-1,bits,"")); \
		testp(0, f(1,bits,"")); \
		testp(0, f(-1,bits,"")); \
		testf(f(0,-1,"")); \
		testf(f(1,-1,"")); \
		testf(f(-1,-1,"")); \
		testf(f(max,-1,"")); \
		testf(f(-max-1,-1,"")); \
		testf(f(0,bits+1,"")); \
		testf(f(1,bits+1,"")); \
		testf(f(-1,bits+1,"")); \
		testf(f(max,bits+1,"")); \
		testf(f(-max-1,bits+1,"")); \
	} while (0)

#define check_shift_u(max,bits,f) \
	do { testp(0, f(0,0,"")); \
		testp(max, f(max,0,"")); \
		testp(1,f(1,0,"")); \
		testp(0, f(0,bits,"")); \
		testp(0, f(max,bits,"")); \
		testp(0, f(1,bits,"")); \
		testf(f(0,-1,"")); \
		testf(f(1,-1,"")); \
		testf(f(max,-1,"")); \
		testf(f(0,bits+1,"")); \
		testf(f(1,bits+1,"")); \
		testf(f(max,bits+1,"")); \
	} while (0)

#define check_all_b(max,bits,type,otype) \
	check_addition(max,occam_add_##otype##_##otype); \
	check_subtraction(max,occam_subtr_##otype##_##otype); \
	check_multiplication(max,occam_mul_##otype##_##otype); \
	check_negation(max,occam_subtr_##otype); \
	check_division(max,occam_div_##otype##_##otype); \
	check_rem(max,occam_rem_##otype##_##otype); \
	check_shift(max,bits,occam_lshift_##otype##_INT);\
	check_shift(max,bits,occam_rshift_##otype##_INT);

#define check_all(max,bits) check_all_b(max,bits,int##bits##_t,INT##bits)

#define check_all_b_u(max,bits,type,otype) \
	check_subtraction_u(max,occam_subtr_##otype##_##otype); \
	check_division_u(max,occam_div_##otype##_##otype); \
	check_rem_u(max,occam_rem_##otype##_##otype); \
	check_shift_u(max,bits,occam_lshift_##otype##_INT); \
	check_shift_u(max,bits,occam_rshift_##otype##_INT);

#define check_all_u(max,bits) check_all_b_u(max,bits,uint##bits##_t,UINT##bits)



// The values of various operations (REM, shifts and so on)
// are checked by the cgtest.  All we are concerned with
// testing here is that the various occam support functions
// STOP when they are supposed to, which is something that
// the cgtests do not check.  But it can't hurt that we check
// the various corner cases at the same time.

int main(int argc, char** argv)
{
	int passes = 0;
	int failures = 0;
	

	check_all(127,8);
	check_all(32767,16);
	check_all(2147483647,32);
	check_all(9223372036854775807,64);
	check_all_b_u(255,8,uint8_t,BYTE);
	
	test_commutative(int8_t,occam_add_INT8_INT8,add);
	test_commutative(int8_t,occam_mul_INT8_INT8,mult);
	test_commutative(uint8_t,occam_add_BYTE_BYTE,add);
	test_commutative(uint8_t,occam_mul_BYTE_BYTE,mult);

	test_commutative(int16_t,occam_add_INT16_INT16,add);
	test_commutative(int16_t,occam_mul_INT16_INT16,mult);

	//TODO add tests for the index-checking functions too

	//Arithmetic:
	testp(-2,occam_LONGADD(-1,-1,0,""));
	testp(-1,occam_LONGADD(-1,0,0,""));
	testp(2147483647,occam_LONGADD(2147483647,0,0,""));
	testf(occam_LONGADD(2147483647,0,1,""));
	testp(2,occam_LONGADD(-1,3,0,""));
	testf(occam_LONGADD(2147483647,2147483647,0,""));
	testf(occam_LONGADD(2147483647,2147483647,1,""));
	//TODO LONGSUB	
	
	testf(occam_ASHIFTLEFT(1,33,""));
	testf(occam_ASHIFTLEFT(1,32,""));
	testf(occam_ASHIFTLEFT(1,31,""));
	testp(0,occam_ASHIFTLEFT(0,32,""));
	testf(occam_ASHIFTLEFT(2,31,""));
	testf(occam_ASHIFTLEFT(INT_MIN,1,""));
	testp(INT_MIN,occam_ASHIFTLEFT(INT_MIN,0,""));
	testp(-4,occam_ASHIFTLEFT(-1,2,""));

	//Floating point:
	testf(occam_ABS(NAN,""));
	testf(occam_DABS(NAN,""));
	testf(occam_ABS(INFINITY,""));
	testf(occam_DABS(INFINITY,""));
	
	testf(occam_SQRT(NAN,""));
	testf(occam_DSQRT(NAN,""));
	testf(occam_SQRT(INFINITY,""));
	testf(occam_DSQRT(INFINITY,""));
	testf(occam_SQRT(-1,""));
	testf(occam_DSQRT(-1,""));

	testf(occam_SCALEB(NAN,1,""));
	testf(occam_DSCALEB(NAN,1,""));
	testf(occam_SCALEB(INFINITY,1,""));
	testf(occam_DSCALEB(INFINITY,1,""));

	testf(occam_MULBY2(NAN,""));
	testf(occam_DMULBY2(NAN,""));
	testf(occam_MULBY2(INFINITY,""));
	testf(occam_DMULBY2(INFINITY,""));

	testf(occam_DIVBY2(NAN,""));
	testf(occam_DDIVBY2(NAN,""));
	testf(occam_DIVBY2(INFINITY,""));
	testf(occam_DDIVBY2(INFINITY,""));


	printf("Tests complete, passed: %d, failed: %d\n", passes, failures);
	
	return -failures;
}