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

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

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

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

	assert(erfl2(f,
	            EJIT_ARG(0xff, int64_t),
	            EJIT_ARG(15, int64_t)) == 0x7f8000);

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

	assert(erfl2(f,
	            EJIT_ARG(0xffffffff, int64_t),
	            EJIT_ARG(8, int64_t)) == 0xffffffff00);

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

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

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

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

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

	assert(erfl2(f,
	            EJIT_ARG(0x8001, int64_t),
	            EJIT_ARG(17, int64_t)) == 0x100020000);

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

	ejit_destroy_func(f);
}