/**
 * $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");
    test_data_block_t sortconf = get_test_data(test_case, "sortconf");
    test_data_block_t sorttab = get_test_data(test_case, "sorttab");
    
    if( input == NULL || output == NULL ) {
	fprintf( stderr, "Skipping test case '%s': data incomplete\n", test_case->test_name );
	return -1;
    }

    if( sortconf != NULL && sorttab != NULL ) {
        if( sortconf->length != 8 ) {
            fprintf( stderr, "Invalid sort config length: %d - abort test %s\n", 
                     sortconf->length, test_case->test_name );
            return -1;
        }
        uint32_t *sc = (uint32_t *)sortconf->data;
        if( sort_dma_write( sorttab->data, sorttab->length, input->data, input->length, *sc, *(sc+1) ) == -1 ){
            return -1;
        }
    } else {
        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;
    }
}