#include <ejit/ejit.h>
#include <assert.h>
#include "do_jit.h"

int main()
{
	struct ejit_operand operands[2] = {
		EJIT_OPERAND_GPR(0, EJIT_TYPE(long)),
		EJIT_OPERAND_GPR(1, EJIT_TYPE(long))
	};

	struct ejit_func *f = ejit_create_func(EJIT_TYPE(long), 2, operands);

	ejit_lshr(f, EJIT_GPR(0), EJIT_GPR(0), EJIT_GPR(1));
	ejit_retr(f, EJIT_GPR(0));

	ejit_select_compile_func(f, 2, 0, EJIT_USE64(long), do_jit);

	assert(erf2(f,
	            EJIT_ARG(0x7f, long),
	            EJIT_ARG(1, long)) == 0xfe);

	assert(erf2(f,
	            EJIT_ARG(0x7fff, long),
	            EJIT_ARG(2, long)) == 0x1fffc);

	assert(erf2(f,
	            EJIT_ARG(0x81, long),
	            EJIT_ARG(16, long)) == 0x810000);

	assert(erf2(f,
	            EJIT_ARG(0xff, long),
	            EJIT_ARG(15, long)) == 0x7f8000);

	assert(erf2(f,
	            EJIT_ARG(0x7fffffff, long),
	            EJIT_ARG(0, long)) == 0x7fffffff);

	assert(erf2(f,
	            EJIT_ARG(0xffffffff, long),
	            EJIT_ARG(8, long)) == 0xffffffff00);

	assert(erf2(f,
	            EJIT_ARG(0x7fffffff, long),
	            EJIT_ARG(3, long)) == 0x3fffffff8);

	assert(erf2(f,
	            EJIT_ARG(-0x7f, long),
	            EJIT_ARG(31, long)) == (int64_t)0xffffffc080000000);

	assert(erf2(f,
	            EJIT_ARG(-0x7fff, long),
	            EJIT_ARG(30, long)) == (int64_t)0xffffe00040000000);

	assert(erf2(f,
	            EJIT_ARG(-0x7fffffff, long),
	            EJIT_ARG(29, long)) == (int64_t)0xf000000020000000);

	assert(erf2(f,
	            EJIT_ARG(0x80000001, long),
	            EJIT_ARG(28, long)) == (int64_t)0x800000010000000);

	assert(erf2(f,
	            EJIT_ARG(0x8001, long),
	            EJIT_ARG(17, long)) == 0x100020000);

	assert(erf2(f,
	            EJIT_ARG(0x80000001, long),
	            EJIT_ARG(18, long)) == (int64_t)0x2000000040000);

	assert(erf2(f,
	            EJIT_ARG(-0xffff, long),
	            EJIT_ARG(24, long)) == (int64_t)0xffffff0001000000);

	assert(erf2(f,
	            EJIT_ARG(0x7f, long),
	            EJIT_ARG(33, long)) == 0xfe00000000);

	assert(erf2(f,
	            EJIT_ARG(0x7ffff, long),
	            EJIT_ARG(34, long)) == 0x1ffffc00000000);

	assert(erf2(f,
	            EJIT_ARG(0x7fffffff, long),
	            EJIT_ARG(35, long)) == (int64_t)0xfffffff800000000);

	assert(erf2(f,
	            EJIT_ARG(-0x7f, long),
	            EJIT_ARG(63, long)) == (int64_t)0x8000000000000000);

	assert(erf2(f,
	            EJIT_ARG(-0x7fff, long),
	            EJIT_ARG(62, long)) == 0x4000000000000000);

	assert(erf2(f,
	            EJIT_ARG(-0x7fffffff, long),
	            EJIT_ARG(61, long)) == 0x2000000000000000);

	assert(erf2(f,
	            EJIT_ARG(0x80000001, long),
	            EJIT_ARG(60, long)) == 0x1000000000000000);

	assert(erf2(f,
	            EJIT_ARG(0x81, long),
	            EJIT_ARG(48, long)) == 0x81000000000000);

	assert(erf2(f,
	            EJIT_ARG(0x8001, long),
	            EJIT_ARG(49, long)) == 0x2000000000000);

	assert(erf2(f,
	            EJIT_ARG(0x80000001, long),
	            EJIT_ARG(40, long)) == 0x10000000000);

	assert(erf2(f,
	            EJIT_ARG(0xff, long),
	            EJIT_ARG(47, long)) == 0x7f800000000000);

	assert(erf2(f,
	            EJIT_ARG(0xffff0001, long),
	            EJIT_ARG(56, long)) == 0x100000000000000);

	assert(erf2(f,
	            EJIT_ARG(0xffffffff, long),
	            EJIT_ARG(40, long)) == (int64_t)0xffffff0000000000);

	assert(erf2(f,
	            EJIT_ARG(0x7fffffffff, long),
	            EJIT_ARG(33, long)) == (int64_t)0xfffffffe00000000);

	assert(erf2(f,
	            EJIT_ARG(-0x7fffffffff, long),
	            EJIT_ARG(63, long)) == (int64_t)0x8000000000000000);

	assert(erf2(f,
	            EJIT_ARG(0x8000000001,  long),
	            EJIT_ARG(48, long)) == 0x1000000000000);

	assert(erf2(f,
	            EJIT_ARG(0xffffffffff, long),
	            EJIT_ARG(47, long)) == (int64_t)0xffff800000000000);

	ejit_destroy_func(f);
}