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

int main(int argc, char *argv[])
{
	(void)argv;
	bool do_jit = argc > 1;
	struct ejit_operand operands[2] = {
		EJIT_OPERAND_GPR(0, EJIT_TYPE(int64_t)),
		EJIT_OPERAND_GPR(1, EJIT_TYPE(int64_t))
	};
	struct ejit_func *f = ejit_create_func(EJIT_TYPE(int64_t), 2, operands);

	ejit_rshr(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(int64_t), do_jit, true);

	assert(erfl2(f, EJIT_ARG(0xfe, int64_t), EJIT_ARG(1, int64_t)) == 0x7f);
	assert(erfl2(f, EJIT_ARG(0x1fffc, int64_t),
	            EJIT_ARG(2, int64_t)) == 0x7fff);

	assert(erfl2(f, EJIT_ARG(0x40000000, int64_t),
	            EJIT_ARG(30, int64_t)) == 1);

	assert(erfl2(f, EJIT_ARG(0x20000000, int64_t),
	            EJIT_ARG(29, int64_t)) == 1);

	assert(erfl2(f, EJIT_ARG(0x10000000, int64_t),
	            EJIT_ARG(28, int64_t)) == 1);

	assert(erfl2(f, EJIT_ARG(0x810000, int64_t),
	            EJIT_ARG(16, int64_t)) == 0x81);

	assert(erfl2(f, EJIT_ARG(0x20000, int64_t), EJIT_ARG(17, int64_t)) == 1);
	assert(erfl2(f, EJIT_ARG(0x40000, int64_t), EJIT_ARG(18, int64_t)) == 1);
	assert(erfl2(f, EJIT_ARG(0x7f8000, int64_t),
	            EJIT_ARG(15, int64_t)) == 0xff);

	assert(erfl2(f, EJIT_ARG(0x1000000, int64_t),
	            EJIT_ARG(24, int64_t)) == 1);

	assert(erfl2(f, EJIT_ARG(0x7fffffff, int64_t), EJIT_ARG(0, int64_t))
			== 0x7fffffff);

	assert(erfl2(f, EJIT_ARG(0x3fffffff8, int64_t), EJIT_ARG(3, int64_t))
			== 0x7fffffff);

	assert(erfl2(f, EJIT_ARG(0xffffff00, int64_t), EJIT_ARG(8, int64_t))
			== 0xffffff);

	assert(erfl2(f, EJIT_ARG(0xfe00000000, int64_t), EJIT_ARG(33, int64_t))
			== 0x7f);

	assert(erfl2(f,
				EJIT_ARG(0x1ffffc00000000, int64_t),
				EJIT_ARG(34, int64_t)) == 0x7ffff);

	assert(erfl2(f,
				EJIT_ARG(0xfffffff800000000, int64_t),
				EJIT_ARG(29, int64_t))
			== (int64_t)0xffffffffffffffc0);

	assert(erfl2(f,
				EJIT_ARG(0x8000000000000000, int64_t),
				EJIT_ARG(63, int64_t))
			== (int64_t)0xffffffffffffffff);

	assert(erfl2(f,
				EJIT_ARG(0x4000000000000000, int64_t),
				EJIT_ARG(62, int64_t))
			== 1);

	assert(erfl2(f,
				EJIT_ARG(0x2000000000000000, int64_t),
				EJIT_ARG(61, int64_t))
			== 1);

	assert(erfl2(f,
				EJIT_ARG(0x1000000000000000, int64_t),
				EJIT_ARG(60, int64_t))
			== 1);

	assert(erfl2(f,
				EJIT_ARG(0xffffffc080000000, int64_t),
				EJIT_ARG(31, int64_t))
			== (int64_t)0xffffffffffffff81);

	assert(erfl2(f,
				EJIT_ARG(0x81000000000000, int64_t),
				EJIT_ARG(48, int64_t))
			== 0x81);

	assert(erfl2(f,
				EJIT_ARG(0x2000000000000, int64_t),
				EJIT_ARG(49, int64_t))
			== 1);

	assert(erfl2(f,
				EJIT_ARG(0x10000000000, int64_t),
				EJIT_ARG(40, int64_t))
			== 1);

	assert(erfl2(f,
				EJIT_ARG(0x7f800000000000, int64_t),
				EJIT_ARG(47, int64_t))
			== 0xff);

	assert(erfl2(f,
				EJIT_ARG(0x100000000000000, int64_t),
				EJIT_ARG(56, int64_t))
			== 1);

	assert(erfl2(f,
				EJIT_ARG(0xffffff0000000000, int64_t),
				EJIT_ARG(40, int64_t))
			== (int64_t)0xffffffffffffffff);

	assert(erfl2(f,
				EJIT_ARG(0xfffffffe00000000, int64_t),
				EJIT_ARG(33, int64_t))
			== (int64_t)0xffffffffffffffff);

	assert(erfl2(f,
				EJIT_ARG(0x8000000000000001, int64_t),
				EJIT_ARG(63, int64_t))
			== (int64_t)0xffffffffffffffff);

	assert(erfl2(f,
				EJIT_ARG(0x1000000000000, int64_t),
				EJIT_ARG(48, int64_t))
			== 1);

	assert(erfl2(f,
				EJIT_ARG(0xffff800000000000, int64_t),
				EJIT_ARG(47, int64_t))
			== (int64_t)0xffffffffffffffff);

	ejit_destroy_func(f);
}