#include <stdlib.h>
#include <string.h>
#include "DrawSpan.h"
#include "polygon_pack.h"
#include "SpanPack.h"
#include "texture.h"
#include "viewer_types.h"

#define SPAN_PACK_LOAD 0
#define TEX_LOAD 1
#define TILE_INFO_LOAD 2

SchedDefineTask(DrawSpan);

static const int hashsize = 263;

static TilePtr hash_table[hashsize] = {0};

unsigned short PRIME[8] = {
    0x002, 0x065, 0x0c7, 0x133, 0x191, 0x1f3, 0x259, 0x2bd,
};

static TileListPtr tileList;

/**
 * �ƥ�������ϡ�TEXTURE_SPLIT_PIXEL^2 �Υ֥��å���ʬ�䤹��
 *
 * +---+---+---+---+---+---+
 * | 0 | 1 | 2 | 3 | 4 | 5 |
 * +---+---+---+---+---+---+
 * |   |   |   |   |   |11 |
 * +---+---+---+---+---+---+
 * |   |   |   |   |   |17 |
 * +---+---+---+---+---+---+
 * |   |   |   |   |   |23 |
 * +---+---+---+---+---+---+
 * |   |   |   |   |   |29 |
 * +---+---+---+---+---+---+
 * |   |   |   |   |   |35 |
 * +---+---+---+---+---+---+
 *
 * ���դ� TEXTURE_SPLIT �Ȥ���
 * �ƥ֥��å��ο������֥��å�ID�Ȥʤ롣
 */

/**
 * �ƥ�������κ�ɸ���顢
 * �ƥ�������ΤɤΥ֥��å��������
 *
 * @param[in] tx X coordinates of texture
 * @param[in] tx Y coordinates of texture
 * @param[in] twidth  Width of texture
 * @return block ID
 */
static inline int
get_tex_block(int tx, int ty, int twidth)
{
     int blockX, blockY;

     blockX = tx / TEXTURE_SPLIT_PIXEL;
     blockY = ty / TEXTURE_SPLIT_PIXEL;

     return blockX + (twidth/TEXTURE_SPLIT_PIXEL)*blockY;
}

/**
 * block ID �ȡ��ƥ�������� TOP address ����
 * (tx,ty) �ǻȤ���ƥ�������� Tile addres �����
 *
 * @param[in] tx X coordinates of texture
 * @param[in] tx Y coordinates of texture
 * @param[in] tw Width of texture
 * @param[in] tex_addr_top (tx,ty) �ǻȤ��ƥ����������Ƭaddress
 * @return block ID
 */
static inline uint32*
getTile(int tx, int ty, int tw, uint32 *tex_addr_top)
{
    int block = get_tex_block(tx, ty, tw);
    return tex_addr_top + block*TEXTURE_BLOCK_SIZE;
}


static int
hash(uint32 data)
{
    int value = 0;
    int n = 0;
    int key;

    for (int i = 0; i < 8; i ++) {
	key = data & 0xf;
	value += key * PRIME[n++];
	data >>= 4;
    }

    return value % hashsize;
}

static int
put(void *key, TilePtr data)
{
    int hashval = hash((uint32)key);

    for (int i = 0; i < hashsize/2; i++) {
	int index = (hashval + i*i)%hashsize;

	if (hash_table[index] == 0) {
	    hash_table[index] = data;
	    return index;
	}
    }

    return -1;
}

static TilePtr
get(void *key)
{
    int hashval = hash((uint32)key);

    for (int i = 0; i < hashsize/2; i++) {
	int index = (hashval + i*i)%hashsize;
	
	if (hash_table[index] != NULL &&
	    hash_table[index]->texture_addr == key) {
	    return hash_table[index];
	}
    }

    return NULL;
}

void
DrawSpan::linebuf_init(int *buf, int x, int rgb)
{
    for (int i = 0; i < x; i++) {
	buf[i] = rgb;
    }
}

float*
DrawSpan::zRow_init(int w, int h)
{
    float *zRow = NULL;
    float z = 65535.0f;
    int length = w*h;

    zRow = (float*)smanager->allocate(sizeof(float)*length);

    for (int i = 0; i < length; i++) {
	zRow[i] = z;
    }

    return zRow;
}


char*
DrawSpan::get_pixel(int tx, int ty, void *texture_image)
{
    return (char*)texture_image+(4*((TEXTURE_SPLIT_PIXEL)*ty+tx));
}

Uint32
DrawSpan::get_rgb(int tx, int ty, void *addr)
{
    Uint8 red, green, blue, alpha;
    TilePtr tile;

    /**
     * get,put �ϥ��֥�������(Hash�Ȥ���)���äƥ����������뤫��
     */
    tile = get(addr);
    if (tile == NULL) {
	if (tileList->size >= MAX_TILE) {
	    tileList->init();
	    bzero(hash_table, sizeof(TilePtr)*hashsize);
	}

	tile = &tileList->tile[tileList->size];
	tile->texture_addr = addr;

	smanager->dma_load(tile->pixel, (uint32)addr,
			   sizeof(uint32)*TEXTURE_BLOCK_SIZE, TEX_LOAD);

	int index = put(tile->texture_addr, tile);

	/**
	 * TODO:
	 *   ����ʤ��ä���Ĥ�
	 *   ���ޤǤΤ�Ĥ����褷�Ƥ���³���򤹤�Ȥ�
	 */
	if (index < 0) {
	    printf("[%p] Can't entry\n", tile);
	    return 0xff0000;
	}

	tileList->size++;

	smanager->dma_wait(TEX_LOAD);
    }

    char *p = get_pixel(tx, ty, tile->pixel);
    
    alpha = 255;
    red   = (Uint8) p[0];
    green = (Uint8) p[1];
    blue  = (Uint8) p[2];

    return (red & 0xff) * 0x10000 + (green & 0xff) * 0x100
	+ (blue & 0xff) + (alpha << 24);
}



int
DrawSpan::run(void *rbuf, void *wbuf)
{
    SpanPack *sp = (SpanPack*)smanager->get_input(0);
    SpanPack *next_sp =
	(SpanPack*)smanager->allocate(sizeof(SpanPack));
    SpanPack *free_sp = next_sp; // next_sp �� free() ��
    SpanPack *tmp_sp = NULL;
    Span *span;

    tileList = (TileListPtr)smanager->allocate(sizeof(TileList));
    tileList->init();

    int rangex_start  = get_param(0); // ���Υ�������ô������ x ���ϰϤλ���
    int rangex_end    = get_param(1); // ���� (start <= x <= end)
    int rangey        = get_param(2); // y ���ϰ� (render_y + rangey - 1)
    int rangex        = rangex_end - rangex_start + 1;

    float *zRow = zRow_init(rangex, rangey);

    int **linebuf = (int**)smanager->allocate(sizeof(int*)*rangey);

    for (int i = 0; i < rangey; i++) {
	linebuf[i] = (int*)smanager->get_output(i);
	linebuf_init(linebuf[i], rangex, 0xffffff);
    }

    bzero(hash_table, sizeof(TilePtr)*hashsize);

    do {
	/**
	 * SpanPack->next ��¸�ߤ����硢
	 * ���ߤ� SpanPack ��������Ƥ�֤�
	 * ���� SpanPack �� DMA ž����Ԥ�
	 */
	if (sp->next != NULL) {
	    smanager->dma_load(next_sp, (uint32)sp->next,
			       sizeof(SpanPack), SPAN_PACK_LOAD);
	} else {
	    next_sp = NULL;
	}

	for (int t = 0; t < sp->info.size; t++) {	  
	    span = &sp->span[t];

	    Uint32 rgb = 0x00ff00;
	    float tex1 = span->tex_x1;
	    float tex2 = span->tex_x2;
	    float tey1 = span->tex_y1;
	    float tey2 = span->tex_y2;

	    int tex_xpos;
	    int tex_ypos;
	    int tex_zpos;

	    int tex_localx;
	    int tex_localy;
	    uint32 *tex_addr;

	    int x = span->x;
	    int y = span->y;
	    int x_len = span->length_x;
	    float z = span->start_z;
	    float zpos = span->end_z;

	    // ��ɸ�� [0 .. split_screen_w-1] ������褦�� x,y �� -1
	    int localx = getLocalX(x-1);
	    int localy = getLocalY(y-1);

	    if (x_len == 1) {
		if (x < rangex_start || rangex_end < x) {
		    continue;
		}

		tex_xpos = (int)((span->tex_width-1) * tex1);
		tex_ypos = (int)((span->tex_height-1) * tey1);
		tex_zpos = (int)z;

		if (zpos < zRow[localx + (rangex * localy)]) {
		    tex_localx = tex_xpos % TEXTURE_SPLIT_PIXEL;
		    tex_localy = tex_ypos % TEXTURE_SPLIT_PIXEL;
		    tex_addr = getTile(tex_xpos, tex_ypos,
				       span->tex_width, span->tex_addr);

		    rgb = get_rgb(tex_localx, tex_localy, tex_addr);

		    zRow[localx + (rangex * localy)] = zpos;
		    linebuf[localy][localx] = rgb;
		}
	    } else {
		int js = (x < rangex_start) ? rangex_start - x : 0;
		int je = (x + x_len > rangex_end) ? rangex_end - x : x_len;
		float tex_x, tex_y, tex_z;

		for (int j = js; j <= je; j++) {
		    localx = getLocalX(x-1+j);

		    tex_z = z*(x_len-1-j)/(x_len-1) + zpos*j/(x_len-1);

		    tex_x = tex1*(x_len-1-j)/(x_len-1) + tex2*j/(x_len-1);
		    tex_y = tey1*(x_len-1-j)/(x_len-1) + tey2*j/(x_len-1);
		    if (tex_x > 1) tex_x = 1;
		    if (tex_y > 1) tex_y = 1;
		    tex_xpos = (int)((span->tex_width-1) * tex_x);
		    tex_ypos = (int)((span->tex_height-1) * tex_y);
		    
		    if (tex_z < zRow[localx + (rangex*localy)]) {
			tex_localx = tex_xpos % TEXTURE_SPLIT_PIXEL;
			tex_localy = tex_ypos % TEXTURE_SPLIT_PIXEL;
			tex_addr = getTile(tex_xpos, tex_ypos,
					   span->tex_width, span->tex_addr);
			
			rgb = get_rgb(tex_localx, tex_localy, tex_addr);

			zRow[localx + (rangex*localy)] = tex_z;
			linebuf[localy][localx] = rgb;
		    }
		}
	    }
	}

	smanager->dma_wait(SPAN_PACK_LOAD);

	tmp_sp = sp;
	sp = next_sp;
	next_sp = tmp_sp;
    } while (sp);

    free(free_sp);
    free(linebuf);
    free(zRow);
    free(tileList);

    return 0;
}