/**

 * $Id$

 * 

 * Tile Accelerator test cases 

 *

 * Copyright (c) 2006 Nathan Keynes.

 *

 * This program is free software; you can redistribute it and/or modify

 * it under the terms of the GNU General Public License as published by

 * the Free Software Foundation; either version 2 of the License, or

 * (at your option) any later version.

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 */

#include <stdio.h>

#include "testdata.h"

#include "pvr.h"

#include "lib.h"

#include "asic.h"

#define DMA_ALIGN(x)   ((void *)((((unsigned int)(x))+0x1F)&0xFFFFFFE0))

#define OBJ_START 0x00010000

#define OBJ_LENGTH 0x00010000

#define TILE_START 0x00060000

#define TILE_LENGTH 0x00010000

#define MEM_FILL 0xFE

int ta_tile_sizes[4] = { 0, 32, 64, 128 };

#define TILE_SIZE(cfg, tile) ta_tile_sizes[((((cfg->ta_cfg) >> (4*tile))&0x03))]

struct ta_config default_ta_config = { 0x00111111, GRID_SIZE(640,480), OBJ_START,

				       OBJ_START+OBJ_LENGTH, TILE_START+TILE_LENGTH,

				       TILE_START, TILE_START+TILE_LENGTH };

int tile_sizes[5];

int tile_events[5] = { EVENT_PVR_OPAQUE_DONE, EVENT_PVR_OPAQUEMOD_DONE,

		       EVENT_PVR_TRANS_DONE, EVENT_PVR_TRANSMOD_DONE,

		       EVENT_PVR_PUNCHOUT_DONE };

char *tile_names[5] = { "Opaque", "Opaque Mod", "Trans", "Trans Mod", "Punch Out" };

#define FLOAT(p) *((float *)(p))

void make_expected_buffer( test_data_block_t expected_block, char *expect, int length )

{

    memset( expect, MEM_FILL,length );

    if( expected_block != NULL ) {

	if( expected_block->length > length ) {

	    fprintf( stderr, "Test data error: expected tile length is %d, but tile size is only %d\n", expected_block->length, length );

	    return;

	}

	memcpy( expect, expected_block->data, expected_block->length );

	if( expected_block->length <= length-4 ) {

	    *((unsigned int *)&expect[expected_block->length]) = 0xF0000000;

	}

    }

}

int tilematrix_block_compare( test_data_block_t expected_block, char *tile_ptrs[], int tile_type, int offset )

{

    int tile_size = tile_sizes[tile_type];

    char expect[tile_size];

    make_expected_buffer(expected_block, expect, tile_size);

    return memcmp( expect, tile_ptrs[tile_type]+(offset*tile_sizes[tile_type]), tile_size );

}

/**

 * Copy from vram, wrapping appropriately 

 */

int memcpy_from_vram( char *dest, char *src, int len ) 

{

    while( len > 0 ) {

	*dest++ = *src++;

	src = (char *)( ((unsigned int)src) & 0xFF7FFFFF );

	len--;

    }

}

int test_ta( test_data_t test_case )

{

    char buf[1024];

    unsigned int *p = DMA_ALIGN(buf);

    unsigned int *data = p;

    int haveFailure = 0;

    int checkedTile[5] = {0,0,0,0,0};

    int i;

    int hsegs,vsegs;

    char *tile_ptrs[5];

    asic_clear();

    memset( PVR_VRAM_BASE, MEM_FILL,  0x00090000 );

    test_data_block_t config_data = get_test_data( test_case, "config" );

    struct ta_config *config = &default_ta_config;

    if( config_data != NULL ) {

	if( config_data->length != sizeof(struct ta_config) ) {

	    fprintf( stderr, "Invalid config data length %d - aborting test %s\n",

		     config_data->length, test_case->test_name );

	    return -1;

	}

	config = (struct ta_config *)config_data->data;

    }

    char *result = (char *)(PVR_VRAM_BASE+config->obj_start);

    char *tilematrix = (char *)(PVR_VRAM_BASE+config->tile_start);

    ta_init(config);

    for( i=0; i<5; i++ ) {

	tile_sizes[i] = TILE_SIZE(config,i);

    }

    hsegs = (config->grid_size & 0xFFFF)+1;

    vsegs = (config->grid_size >> 16) + 1;

    tile_ptrs[0] = tilematrix;

    tile_ptrs[1] = tile_ptrs[0] + (hsegs*vsegs*tile_sizes[0]);

    tile_ptrs[2] = tile_ptrs[1] + (hsegs*vsegs*tile_sizes[1]);

    tile_ptrs[3] = tile_ptrs[2] + (hsegs*vsegs*tile_sizes[2]);

    tile_ptrs[4] = tile_ptrs[3] + (hsegs*vsegs*tile_sizes[3]);

    test_data_block_t input = get_test_data(test_case, "input");

    test_data_block_t input2 = get_test_data(test_case, "input2");

    test_data_block_t output = get_test_data(test_case, "output");

    test_data_block_t error = get_test_data(test_case, "error");

    if( input == NULL || output == NULL ) {

	fprintf( stderr, "Skipping test case '%s': data incomplete\n", test_case->test_name );

	return -1;

    }

    if( pvr_dma_write( 0x10000000, input->data, input->length, 0 ) == -1 ) {

	return -1;

    }

    if( input2 != NULL ) {

	ta_reinit();

	pvr_dma_write( 0x10000000, input2->data, input2->length, 0 );

    }

    if( error != NULL ) {

	for( i=0; i<error->length; i++ ) {

	    if( asic_wait( error->data[i] ) == -1 ) {

		fprintf( stderr, "Test %s: failed (Timeout waiting for error event %d)\n",

			 test_case->test_name, error->data[i] );

		asic_dump( stderr );

		return -1;

	    }

	}

    }

    for( i=0; i<MAX_DATA_BLOCKS; i++ ) {

	test_data_block_t data = &test_case->item[i];

	int tile, x, y, offset;

	if( data->name != NULL ) {

	    int result = sscanf( data->name, "tile %d %dx%d", &tile, &x, &y );

	    if( result == 1 ) {

		x = y = 0;

	    } else if( result != 3 ) {

		continue;

	    }

	    tile--;

	    offset = x + (y * hsegs);

	    if( checkedTile[tile] == 0 ) {

		if( asic_wait( tile_events[tile] ) == -1 ) {

		    fprintf( stderr, "Test %s: failed (Timeout waiting for %s done event)\n", 

			     test_case->test_name, tile_names[tile] );

		    ta_dump_regs();

		    asic_dump( stderr );

		    haveFailure = 1;

		}

	    }

	    if( tilematrix_block_compare( data, tile_ptrs, tile, offset ) != 0 ) {

		fprintf( stderr, "Test %s: Failed (%s matrix %dx%d). ", 

			 test_case->test_name, tile_names[tile], x, y );

		fwrite_diff32( stderr, data->data, data->length, 

			       tile_ptrs[tile] + (tile_sizes[tile]*offset), tile_sizes[tile] );

		haveFailure = 1;

	    }

	    checkedTile[tile] = 1;

	}

    }

    /* Overflow */

    test_data_block_t plist = get_test_data(test_case, "plist" );

    if( plist != NULL ) {

	unsigned int plist_posn, plist_end;

	if( config->ta_cfg & 0x00100000 ) { /* Descending */

	    plist_posn = pvr_get_plist_posn(); //+ tile_sizes[0];

	    plist_end = config->plist_start;

	} else {

	    plist_posn = config->plist_start;

	    plist_end = pvr_get_plist_posn() + tile_sizes[0];

	}

	char *plist_data = (char *)(PVR_VRAM_BASE + plist_posn);

	if( test_block_compare( plist, plist_data, plist_end-plist_posn ) != 0 ) {

	    fprintf( stderr, "Test %s: Failed (Plist buffer)", test_case->test_name );

	    fwrite_diff32( stderr, plist->data, plist->length, (char *)plist_data, 

			   plist_end - plist_posn );

	    haveFailure = 1;

	}

	char block[tile_sizes[0]];

	memset( block, MEM_FILL, tile_sizes[0] );

	if( memcmp( block, plist_data - tile_sizes[0], tile_sizes[0] ) != 0 ) {

	    fprintf( stderr, "Test %s: Failed (Plist buffer)", test_case->test_name );

	    fwrite_diff32( stderr, block, tile_sizes[0], plist_data - tile_sizes[0],

			   tile_sizes[0]);

	    haveFailure = 1;

	}

    }

    /* Vertex buffer */

    int result_length = pvr_get_objbuf_size();

    char tmp[result_length];

    memcpy_from_vram( tmp, result, result_length );

    if( test_block_compare( output, tmp, result_length ) != 0 ) {

	fprintf( stderr, "Test %s: Failed (Vertex buffer). ", test_case->test_name );

	fwrite_diff32( stderr, output->data, output->length, tmp, result_length );

	haveFailure = 1;

    }

    for( i=0; i<5; i++ ) {

	if( checkedTile[i] == 0 ) {

	    if( tilematrix_block_compare( NULL, tile_ptrs, i, 0 ) != 0 ) {

		fprintf( stderr, "Test %s: Failed (%s matrix). ", test_case->test_name, tile_names[i] );

                fprintf( stderr, "Expected empty buffer at %08X, but was =>\n", 

			 (unsigned int)(tile_ptrs[i]) );

		fwrite_dump( stderr, tile_ptrs[i], tile_sizes[i] );

		//		fwrite_dump( stderr, tile_ptrs[i] - 128, 256 );

	    }

	}

    }

    if( error == NULL ) {

	if( asic_check(EVENT_TA_ERROR) || asic_check(EVENT_PVR_PRIM_ALLOC_FAIL) ||

	    asic_check(EVENT_PVR_MATRIX_ALLOC_FAIL) || asic_check(EVENT_PVR_BAD_INPUT) ) {

	    fprintf( stderr, "Test %s: Failed (unexpected error events)\n", test_case->test_name );

	    asic_dump( stderr );

	    haveFailure = 1;

	}

    }

    if( haveFailure )

	return -1;

    fprintf( stdout, "Test %s: OK\n", test_case->test_name );

    return 0;

}

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

{

    int test_cases = 0;

    int test_failures = 0;

    test_data_t test_data = load_test_dataset(stdin);

    test_data_t test_case = test_data;

    asic_mask_all();

    pvr_init();

    while( test_case != NULL ) {

	test_cases++;

	int result = test_ta(test_case);

	if( result != 0 ) {

	    test_failures++;

	}

	test_case = test_case->next;

    }

    free_test_dataset(test_data);

    if( test_failures != 0 ) {

	fprintf( stderr, "%d/%d test failures!\n", test_failures, test_cases );

	return 1;

    } else {

	fprintf( stderr, "%d tests OK\n", test_cases );

	return 0;

    }

}