/* test_shift.c

 * vi:ts=4 sw=4

 *

 * (c) Copyright 2012 Hewlett-Packard Development Company, L.P.

 * Licensed under the Apache License, Version 2.0 (the "License"); you may

 * not use this file except in compliance with the License. You may obtain

 * a copy of the License at http://www.apache.org/licenses/LICENSE-2.0.

 * Unless required by applicable law or agreed to in writing, software

 * distributed under the License is distributed on an "AS IS" BASIS,

 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express

 * or implied. See the License for the specific language governing

 * permissions and limitations under the License.

 */

#ifdef HAVE_CONFIG_H

#include "config.h"

#endif

#include <winpr/sysinfo.h>

#include "prim_test.h"

#define FUNC_TEST_SIZE 65536

static BOOL test_lShift_16s_func(void)

{

	pstatus_t status;

	INT16 ALIGN(src[FUNC_TEST_SIZE+3]);

	INT16 ALIGN(d1[FUNC_TEST_SIZE+3]);

	UINT32 val;

	winpr_RAND((BYTE*)&val, sizeof(val));

	winpr_RAND((BYTE*)src, sizeof(src));

	val = (val % (FUNC_TEST_SIZE - 1)) + 1;

	/* Aligned */

	status = generic->lShiftC_16s(src+1, val, d1+1, FUNC_TEST_SIZE);

	if (status != PRIMITIVES_SUCCESS)

		return FALSE;

	status = optimized->lShiftC_16s(src+1, val, d1+1, FUNC_TEST_SIZE);

	if (status != PRIMITIVES_SUCCESS)

		return FALSE;

	/* Unaligned */

	status = generic->lShiftC_16s(src+1, val, d1+2, FUNC_TEST_SIZE);

	if (status != PRIMITIVES_SUCCESS)

		return FALSE;

	status = optimized->lShiftC_16s(src+1, val, d1+2, FUNC_TEST_SIZE);

	if (status != PRIMITIVES_SUCCESS)

		return FALSE;

	return TRUE;

}

static BOOL test_lShift_16u_func(void)

{

	pstatus_t status;

	UINT16 ALIGN(src[FUNC_TEST_SIZE+3]);

	UINT16 ALIGN(d1[FUNC_TEST_SIZE+3]);

	UINT32 val;

	winpr_RAND((BYTE*)&val, sizeof(val));

	winpr_RAND((BYTE*)src, sizeof(src));

	val = (val % (FUNC_TEST_SIZE - 1)) + 1;

	/* Aligned */

	status = generic->lShiftC_16u(src+1, val, d1+1, FUNC_TEST_SIZE);

	if (status != PRIMITIVES_SUCCESS)

		return FALSE;

	status = optimized->lShiftC_16u(src+1, val, d1+1, FUNC_TEST_SIZE);

	if (status != PRIMITIVES_SUCCESS)

		return FALSE;

	/* Unaligned */

	status = generic->lShiftC_16u(src+1, val, d1+2, FUNC_TEST_SIZE);

	if (status != PRIMITIVES_SUCCESS)

		return FALSE;

	status = optimized->lShiftC_16u(src+1, val, d1+2, FUNC_TEST_SIZE);

	if (status != PRIMITIVES_SUCCESS)

		return FALSE;

	return TRUE;

}

static BOOL test_rShift_16s_func(void)

{

	pstatus_t status;

	INT16 ALIGN(src[FUNC_TEST_SIZE+3]);

	INT16 ALIGN(d1[FUNC_TEST_SIZE+3]);

	UINT32 val;

	winpr_RAND((BYTE*)&val, sizeof(val));

	winpr_RAND((BYTE*)src, sizeof(src));

	val = (val % (FUNC_TEST_SIZE - 1)) + 1;

	/* Aligned */

	status = generic->rShiftC_16s(src+1, val, d1+1, FUNC_TEST_SIZE);

	if (status != PRIMITIVES_SUCCESS)

		return FALSE;

	status = optimized->rShiftC_16s(src+1, val, d1+1, FUNC_TEST_SIZE);

	if (status != PRIMITIVES_SUCCESS)

		return FALSE;

	/* Unaligned */

	status = generic->rShiftC_16s(src+1, val, d1+2, FUNC_TEST_SIZE);

	if (status != PRIMITIVES_SUCCESS)

		return FALSE;

	status = optimized->rShiftC_16s(src+1, val, d1+2, FUNC_TEST_SIZE);

	if (status != PRIMITIVES_SUCCESS)

		return FALSE;

	return TRUE;

}

static BOOL test_rShift_16u_func(void)

{

	pstatus_t status;

	UINT16 ALIGN(src[FUNC_TEST_SIZE+3]);

	UINT16 ALIGN(d1[FUNC_TEST_SIZE+3]);

	UINT32 val;

	winpr_RAND((BYTE*)&val, sizeof(val));

	winpr_RAND((BYTE*)src, sizeof(src));

	val = (val % (FUNC_TEST_SIZE - 1)) + 1;

	/* Aligned */

	status = generic->rShiftC_16u(src+1, val, d1+1, FUNC_TEST_SIZE);

	if (status != PRIMITIVES_SUCCESS)

		return FALSE;

	status = optimized->rShiftC_16u(src+1, val, d1+1, FUNC_TEST_SIZE);

	if (status != PRIMITIVES_SUCCESS)

		return FALSE;

	/* Unaligned */

	status = generic->rShiftC_16u(src+1, val, d1+2, FUNC_TEST_SIZE);

	if (status != PRIMITIVES_SUCCESS)

		return FALSE;

	status = optimized->rShiftC_16u(src+1, val, d1+2, FUNC_TEST_SIZE);

	if (status != PRIMITIVES_SUCCESS)

		return FALSE;

	return TRUE;

}

static BOOL test_ShiftWrapper_16s_func(void)

{

	pstatus_t status;

	INT16 ALIGN(src[FUNC_TEST_SIZE+3]);

	INT16 ALIGN(d1[FUNC_TEST_SIZE+3]);

	UINT32 tmp;

	INT32 val;

	winpr_RAND((BYTE*)&tmp, sizeof(tmp));

	winpr_RAND((BYTE*)src, sizeof(src));

	val = (tmp % (FUNC_TEST_SIZE - 1)) + 1;

	/* Aligned */

	status = generic->shiftC_16s(src+1, val, d1+1, FUNC_TEST_SIZE);

	if (status != PRIMITIVES_SUCCESS)

		return FALSE;

	status = optimized->shiftC_16s(src+1, val, d1+1, FUNC_TEST_SIZE);

	if (status != PRIMITIVES_SUCCESS)

		return FALSE;

	status = generic->shiftC_16s(src+1, -val, d1+1, FUNC_TEST_SIZE);

	if (status != PRIMITIVES_SUCCESS)

		return FALSE;

	status = optimized->shiftC_16s(src+1, -val, d1+1, FUNC_TEST_SIZE);

	if (status != PRIMITIVES_SUCCESS)

		return FALSE;

	/* Unaligned */

	status = generic->shiftC_16s(src+1, val, d1+2, FUNC_TEST_SIZE);

	if (status != PRIMITIVES_SUCCESS)

		return FALSE;

	status = optimized->shiftC_16s(src+1, val, d1+2, FUNC_TEST_SIZE);

	if (status != PRIMITIVES_SUCCESS)

		return FALSE;

	status = generic->shiftC_16s(src+1, -val, d1+2, FUNC_TEST_SIZE);

	if (status != PRIMITIVES_SUCCESS)

		return FALSE;

	status = optimized->shiftC_16s(src+1, -val, d1+2, FUNC_TEST_SIZE);

	if (status != PRIMITIVES_SUCCESS)

		return FALSE;

	return TRUE;

}

static BOOL test_ShiftWrapper_16u_func(void)

{

	pstatus_t status;

	UINT16 ALIGN(src[FUNC_TEST_SIZE+3]);

	UINT16 ALIGN(d1[FUNC_TEST_SIZE+3]);

	UINT32 tmp;

	INT32 val;

	winpr_RAND((BYTE*)&tmp, sizeof(tmp));

	winpr_RAND((BYTE*)src, sizeof(src));

	val = (tmp % (FUNC_TEST_SIZE - 1)) + 1;

	/* Aligned */

	status = generic->shiftC_16u(src+1, val, d1+1, FUNC_TEST_SIZE);

	if (status != PRIMITIVES_SUCCESS)

		return FALSE;

	status = optimized->shiftC_16u(src+1, val, d1+1, FUNC_TEST_SIZE);

	if (status != PRIMITIVES_SUCCESS)

		return FALSE;

	status = generic->shiftC_16u(src+1, -val, d1+1, FUNC_TEST_SIZE);

	if (status != PRIMITIVES_SUCCESS)

		return FALSE;

	status = optimized->shiftC_16u(src+1, -val, d1+1, FUNC_TEST_SIZE);

	if (status != PRIMITIVES_SUCCESS)

		return FALSE;

	/* Unaligned */

	status = generic->shiftC_16u(src+1, val, d1+2, FUNC_TEST_SIZE);

	if (status != PRIMITIVES_SUCCESS)

		return FALSE;

	status = optimized->shiftC_16u(src+1, val, d1+2, FUNC_TEST_SIZE);

	if (status != PRIMITIVES_SUCCESS)

		return FALSE;

	status = generic->shiftC_16u(src+1, -val, d1+2, FUNC_TEST_SIZE);

	if (status != PRIMITIVES_SUCCESS)

		return FALSE;

	status = optimized->shiftC_16u(src+1, -val, d1+2, FUNC_TEST_SIZE);

	if (status != PRIMITIVES_SUCCESS)

		return FALSE;

	return TRUE;

}

/* ------------------------------------------------------------------------- */

static BOOL test_lShift_16s_speed(void)

{

	UINT32 val;

	INT16 ALIGN(src[MAX_TEST_SIZE+1]), ALIGN(dst[MAX_TEST_SIZE+1]);

	winpr_RAND((BYTE*)src, sizeof(src));

	winpr_RAND((BYTE*)&val, sizeof(val));

	if (!speed_test("lShift_16s", "aligned", g_Iterations,

			(speed_test_fkt)generic->lShiftC_16s,

			(speed_test_fkt)optimized->lShiftC_16s, src, val,

			dst, MAX_TEST_SIZE))

		return FALSE;

	if (!speed_test("lShift_16s", "unaligned", g_Iterations,

			(speed_test_fkt)generic->lShiftC_16s,

			(speed_test_fkt)optimized->lShiftC_16s, src + 1, val,

			dst, MAX_TEST_SIZE))

		return FALSE;

	return TRUE;

}

/* ------------------------------------------------------------------------- */

static BOOL test_lShift_16u_speed(void)

{

	UINT32 val;

	UINT16 ALIGN(src[MAX_TEST_SIZE + 1]), ALIGN(dst[MAX_TEST_SIZE + 1]);

	winpr_RAND((BYTE*)&val, sizeof(val));

	winpr_RAND((BYTE*)src, sizeof(src));

	if (!speed_test("lShift_16u", "aligned", g_Iterations,

			(speed_test_fkt)generic->lShiftC_16u,

			(speed_test_fkt)optimized->lShiftC_16u, src, val,

			dst, MAX_TEST_SIZE))

		return FALSE;

	if (!speed_test("lShift_16u", "unaligned", g_Iterations,

			(speed_test_fkt)generic->lShiftC_16u,

			(speed_test_fkt)optimized->lShiftC_16u, src + 1, val,

			dst, MAX_TEST_SIZE))

		return FALSE;

	return TRUE;

}

/* ------------------------------------------------------------------------- */

static BOOL test_rShift_16s_speed(void)

{

	UINT32 val;

	INT16 ALIGN(src[MAX_TEST_SIZE+1]), ALIGN(dst[MAX_TEST_SIZE+1]);

	winpr_RAND((BYTE*)src, sizeof(src));

	winpr_RAND((BYTE*)&val, sizeof(val));

	if (!speed_test("rShift_16s", "aligned", g_Iterations,

			(speed_test_fkt)generic->rShiftC_16s,

			(speed_test_fkt)optimized->rShiftC_16s, src, val,

			dst, MAX_TEST_SIZE))

		return FALSE;

	if (!speed_test("rShift_16s", "unaligned", g_Iterations,

			(speed_test_fkt)generic->rShiftC_16s,

			(speed_test_fkt)optimized->rShiftC_16s, src + 1, val,

			dst, MAX_TEST_SIZE))

		return FALSE;

	return TRUE;

}

/* ------------------------------------------------------------------------- */

static BOOL test_rShift_16u_speed(void)

{

	UINT32 val;

	UINT16 ALIGN(src[MAX_TEST_SIZE + 1]), ALIGN(dst[MAX_TEST_SIZE + 1]);

	winpr_RAND((BYTE*)&val, sizeof(val));

	winpr_RAND((BYTE*)src, sizeof(src));

	if (!speed_test("rShift_16u", "aligned", g_Iterations,

			(speed_test_fkt)generic->rShiftC_16u,

			(speed_test_fkt)optimized->rShiftC_16u, src, val,

			dst, MAX_TEST_SIZE))

		return FALSE;

	if (!speed_test("rShift_16u", "unaligned", g_Iterations,

			(speed_test_fkt)generic->rShiftC_16u,

			(speed_test_fkt)optimized->rShiftC_16u, src + 1, val,

			dst, MAX_TEST_SIZE))

		return FALSE;

	return TRUE;

}

int TestPrimitivesShift(int argc, char* argv[])

{

	prim_test_setup(FALSE);

	if (!test_lShift_16s_func())

		return 1;

	if (g_TestPrimitivesPerformance)

	{

		if (!test_lShift_16s_speed())

			return 1;

	}

	if (!test_lShift_16u_func())

		return 1;

	if (g_TestPrimitivesPerformance)

	{

		if (!test_lShift_16u_speed())

			return 1;

	}

	if (!test_rShift_16s_func())

		return 1;

	if (g_TestPrimitivesPerformance)

	{

		if (!test_rShift_16s_speed())

			return 1;

	}

	if (!test_rShift_16u_func())

		return 1;

	if (g_TestPrimitivesPerformance)

	{

		if (!test_rShift_16u_speed())

			return 1;

	}

	if (!test_ShiftWrapper_16s_func())

		return 1;

	if (!test_ShiftWrapper_16u_func())

		return 1;

	return 0;

}