/**

  * $Id$

  *

  * Manage the internal vertex/polygon buffers and scene data structure. 

  * Where possible this uses VBOs for the vertex + index data.

  *

  * Copyright (c) 2005 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 <assert.h>

 #include <string.h>

 #include "lxdream.h"

 #include "display.h"

 #include "pvr2/pvr2.h"

 #include "pvr2/glutil.h"

 #include "pvr2/scene.h"

 #define VBO_EXT_STRING "GL_ARB_vertex_buffer_object"

 #define PBO_EXT_STRING "GL_ARB_pixel_buffer_object"

 static inline uint32_t bgra_to_rgba(uint32_t bgra)

 {

     return (bgra&0xFF00FF00) | ((bgra&0x00FF0000)>>16) | ((bgra&0x000000FF)<<16);

 }

 struct pvr2_scene_struct pvr2_scene;

 static gboolean vbo_init = FALSE;

 static gboolean vbo_supported = FALSE;

 /**

  * Test for VBO support, and allocate all the system memory needed for the

  * temporary structures. GL context must have been initialized before this

  * point.

  */

 void pvr2_scene_init()

 {

     if( !vbo_init ) {

 	if( isGLExtensionSupported(VBO_EXT_STRING) ) {

 	    vbo_supported = TRUE;

 	    pvr2_scene.vbo_id = 1;

 	}

 	pvr2_scene.vertex_array = NULL;

 	pvr2_scene.vertex_array_size = 0;

 	pvr2_scene.poly_array = g_malloc( MAX_POLY_BUFFER_SIZE );

 	pvr2_scene.buf_to_poly_map = g_malloc0( BUF_POLY_MAP_SIZE );

 	vbo_init = TRUE;

     }

 }

 /**

  * Clear the scene data structures in preparation for fresh data

  */

 void pvr2_scene_reset()

 {

     pvr2_scene.poly_count = 0;

     pvr2_scene.vertex_count = 0;

     memset( pvr2_scene.buf_to_poly_map, 0, BUF_POLY_MAP_SIZE );

 }

 void pvr2_scene_shutdown()

 {

     if( vbo_supported ) {

 	glBindBufferARB( GL_ARRAY_BUFFER_ARB, 0 );

 	glDeleteBuffersARB( 1, &pvr2_scene.vbo_id );

 	pvr2_scene.vbo_id = 0;

     } else {

 	g_free( pvr2_scene.vertex_array );

 	pvr2_scene.vertex_array = NULL;

     }

     g_free( pvr2_scene.poly_array );

     g_free( pvr2_scene.buf_to_poly_map );

     vbo_init = FALSE;

 }

 void *vertex_buffer_map()

 {

     glGetError();

     uint32_t size = pvr2_scene.vertex_count * sizeof(struct vertex_struct);

     if( vbo_supported ) {

 	glBindBufferARB( GL_ARRAY_BUFFER_ARB, pvr2_scene.vbo_id );

 	assert( glGetError() == 0 );

 	if( size > pvr2_scene.vertex_array_size ) {

 	    glBufferDataARB( GL_ARRAY_BUFFER_ARB, size, NULL, GL_DYNAMIC_DRAW_ARB );

 	    int status = glGetError();

 	    if( status != 0 ) {

 		fprintf( stderr, "Error %08X allocating vertex buffer\n", status );

 		abort();

 	    }

 	    pvr2_scene.vertex_array_size = size;

 	}

 	pvr2_scene.vertex_array = glMapBufferARB( GL_ARRAY_BUFFER_ARB, GL_WRITE_ONLY_ARB );

 	assert(pvr2_scene.vertex_array != NULL );

     } else {

 	if( size > pvr2_scene.vertex_array_size ) {

 	    pvr2_scene.vertex_array = g_realloc( pvr2_scene.vertex_array, size );

 	}

     }

     return pvr2_scene.vertex_array;

 }

 gboolean vertex_buffer_unmap()

 {

     if( vbo_supported ) {

 	pvr2_scene.vertex_array = NULL;

 	return glUnmapBufferARB( GL_ARRAY_BUFFER_ARB );

     } else {

 	return TRUE;

     }

 }

 static struct polygon_struct *scene_add_polygon( pvraddr_t poly_idx, int vertex_count,

 							 gboolean is_modified ) 

 {

     int vert_mul = is_modified ? 2 : 1;

     if( pvr2_scene.buf_to_poly_map[poly_idx] != NULL ) {

 	if( vertex_count > pvr2_scene.buf_to_poly_map[poly_idx]->vertex_count ) {

 	    pvr2_scene.vertex_count += (vertex_count - pvr2_scene.buf_to_poly_map[poly_idx]->vertex_count) * vert_mul;

 	    pvr2_scene.buf_to_poly_map[poly_idx]->vertex_count = vertex_count;

 	}

 	return pvr2_scene.buf_to_poly_map[poly_idx];

     } else {

 	struct polygon_struct *poly = &pvr2_scene.poly_array[pvr2_scene.poly_count++];

 	poly->context = (uint32_t *)(video_base + MMIO_READ(PVR2,RENDER_POLYBASE) + (poly_idx<<2));

 	poly->vertex_count = vertex_count;

 	poly->vertex_index = -1;

 	poly->next = NULL;

 	pvr2_scene.buf_to_poly_map[poly_idx] = poly;

 	pvr2_scene.vertex_count += (vertex_count * vert_mul);

 	return poly;

     }

 }

 /**

  * Decode a single PVR2 renderable vertex (opaque/trans/punch-out, but not shadow

  * volume)

  * @param vert Pointer to output vertex structure

  * @param poly1 First word of polygon context (needed to understand vertex)

  * @param poly2 Second word of polygon context

  * @param pvr2_data Pointer to raw pvr2 vertex data (in VRAM)

  * @param modify_offset Offset in 32-bit words to the tex/color data. 0 for

  *        the normal vertex, half the vertex length for the modified vertex.

  */

 static void pvr2_decode_render_vertex( struct vertex_struct *vert, uint32_t poly1, 

 				       uint32_t poly2, uint32_t *pvr2_data, 

 				       int modify_offset )

 {

     gboolean force_alpha = !POLY2_ALPHA_ENABLE(poly2);

     union pvr2_data_type {

 	uint32_t *ival;

 	float *fval;

     } data;

     data.ival = pvr2_data;

     vert->x = *data.fval++;

     vert->y = *data.fval++;

     float z = *data.fval++;

     if( z > pvr2_scene.bounds[5] ) {

 	pvr2_scene.bounds[5] = z;

     } else if( z < pvr2_scene.bounds[4] && z != 0 ) {

 	pvr2_scene.bounds[4] = z;

     }

     vert->z = z;

     data.ival += modify_offset;

     if( POLY1_TEXTURED(poly1) ) {

 	if( POLY1_UV16(poly1) ) {

 	    vert->u = halftofloat( *data.ival>>16 );

 	    vert->v = halftofloat( *data.ival );

 	    data.ival++;

 	} else {

 	    vert->u = *data.fval++;

 	    vert->v = *data.fval++;

 	}

 	if( POLY2_TEX_BLEND(poly2) == 1 ) {

 	    force_alpha = TRUE;

 	}

     }

     if( force_alpha ) {

 	vert->rgba = bgra_to_rgba((*data.ival++) | 0xFF000000);

 	if( POLY1_SPECULAR(poly1) ) {

 	    vert->offset_rgba = bgra_to_rgba((*data.ival++) | 0xFF000000);

 	}

     } else {

 	vert->rgba = bgra_to_rgba(*data.ival++);

 	if( POLY1_SPECULAR(poly1) ) {

 	    vert->offset_rgba = bgra_to_rgba(*data.ival++);

 	}

     }

 }

 /**

  * Compute texture, colour, and z values for a result point by interpolating from

  * a set of 3 input points. The result point must define its x,y.

  */

 static void scene_compute_vertex( struct vertex_struct *result, 

 					  struct vertex_struct *input,

 					  gboolean is_solid_shaded )

 {

     int i;

     float sx = input[2].x - input[1].x;

     float sy = input[2].y - input[1].y;

     float tx = input[0].x - input[1].x;

     float ty = input[0].y - input[1].y;

     float detxy = ((sy) * (tx)) - ((ty) * (sx));

     if( detxy == 0 ) {

 	result->z = input[2].z;

 	result->u = input[2].u;

 	result->v = input[2].v;

 	result->rgba = input[2].rgba;

 	result->offset_rgba = input[2].offset_rgba;

 	return;

     }

     float t = ((result->x - input[1].x) * sy -

 	       (result->y - input[1].y) * sx) / detxy;

     float s = ((result->y - input[1].y) * tx -

 	       (result->x - input[1].x) * ty) / detxy;

     float sz = input[2].z - input[1].z;

     float tz = input[0].z - input[1].z;

     float su = input[2].u - input[1].u;

     float tu = input[0].u - input[1].u;

     float sv = input[2].v - input[1].v;

     float tv = input[0].v - input[1].v;

     float rz = input[1].z + (t*tz) + (s*sz);

     if( rz > pvr2_scene.bounds[5] ) {

 	pvr2_scene.bounds[5] = rz;

     } else if( rz < pvr2_scene.bounds[4] ) {

 	pvr2_scene.bounds[4] = rz; 

     }

     result->z = rz;

     result->u = input[1].u + (t*tu) + (s*su);

     result->v = input[1].v + (t*tv) + (s*sv);

     if( is_solid_shaded ) {

 	result->rgba = input[2].rgba;

 	result->offset_rgba = input[2].offset_rgba;

     } else {

 	uint8_t *rgba0 = (uint8_t *)&input[0].rgba;

 	uint8_t *rgba1 = (uint8_t *)&input[1].rgba;

 	uint8_t *rgba2 = (uint8_t *)&input[2].rgba;

 	uint8_t *rgba3 = (uint8_t *)&result->rgba;

 	for( i=0; i<8; i++ ) { // note: depends on rgba & offset_rgba being adjacent

 	    float tc = *rgba0++ - *rgba1;

 	    float sc = *rgba2++ - *rgba1;

 	    float rc = *rgba1++ + (t*tc) + (s*sc);

 	    if( rc < 0 ) {

 		rc = 0;

 	    } else if( rc > 255 ) {

 		rc = 255;

 	    }

 	    *rgba3++ = rc;

 	}

     }    

 }

 static void scene_add_vertexes( pvraddr_t poly_idx, int vertex_length,

 					gboolean is_modified )

 {

     struct polygon_struct *poly = pvr2_scene.buf_to_poly_map[poly_idx];

     uint32_t *ptr = &pvr2_scene.pvr2_pbuf[poly_idx];

     uint32_t *context = ptr;

     unsigned int i;

     assert( poly != NULL );

     if( poly->vertex_index == -1 ) {

 	ptr += (is_modified ? 5 : 3 );

 	poly->vertex_index = pvr2_scene.vertex_index;

 	assert( pvr2_scene.vertex_index + poly->vertex_count <= pvr2_scene.vertex_count );

 	for( i=0; i<poly->vertex_count; i++ ) {

 	    pvr2_decode_render_vertex( &pvr2_scene.vertex_array[pvr2_scene.vertex_index++], context[0], context[1], ptr, 0 );

 	    ptr += vertex_length;

 	}

 	if( is_modified ) {

 	    int mod_offset = (vertex_length - 3)>>1;

 	    ptr = &pvr2_scene.pvr2_pbuf[poly_idx] + 5;

 	    poly->mod_vertex_index = pvr2_scene.vertex_index;

 	    for( i=0; i<poly->vertex_count; i++ ) {

 		pvr2_decode_render_vertex( &pvr2_scene.vertex_array[pvr2_scene.vertex_index++], context[0], context[3], ptr, mod_offset );

 		ptr += vertex_length;

 	    }

 	}

     }

 }

 static void scene_add_quad_vertexes( pvraddr_t poly_idx, int vertex_length, 

 					     gboolean is_modified )

 {

     struct polygon_struct *poly = pvr2_scene.buf_to_poly_map[poly_idx];

     uint32_t *ptr = &pvr2_scene.pvr2_pbuf[poly_idx];

     uint32_t *context = ptr;

     unsigned int i;

     if( poly->vertex_index == -1 ) {

 	// Construct it locally and copy to the vertex buffer, as the VBO is 

 	// allowed to be horribly slow for reads (ie it could be direct-mapped

 	// vram).

 	struct vertex_struct quad[4];

 	assert( poly != NULL );

 	ptr += (is_modified ? 5 : 3 );

 	poly->vertex_index = pvr2_scene.vertex_index;

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

 	    pvr2_decode_render_vertex( &quad[i], context[0], context[1], ptr, 0 );

 	    ptr += vertex_length;

 	}

 	scene_compute_vertex( &quad[3], &quad[0], !POLY1_GOURAUD_SHADED(context[0]) );

 	// Swap last two vertexes (quad arrangement => tri strip arrangement)

 	memcpy( &pvr2_scene.vertex_array[pvr2_scene.vertex_index], quad, sizeof(struct vertex_struct)*2 );

 	memcpy( &pvr2_scene.vertex_array[pvr2_scene.vertex_index+2], &quad[3], sizeof(struct vertex_struct) );

 	memcpy( &pvr2_scene.vertex_array[pvr2_scene.vertex_index+3], &quad[2], sizeof(struct vertex_struct) );

 	pvr2_scene.vertex_index += 4;

 	if( is_modified ) {

 	    int mod_offset = (vertex_length - 3)>>1;

 	    ptr = &pvr2_scene.pvr2_pbuf[poly_idx] + 5;

 	    poly->mod_vertex_index = pvr2_scene.vertex_index;

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

 		pvr2_decode_render_vertex( &quad[4], context[0], context[3], ptr, mod_offset );

 		ptr += vertex_length;

 	    }

 	    scene_compute_vertex( &quad[3], &quad[0], !POLY1_GOURAUD_SHADED(context[0]) );

 	    memcpy( &pvr2_scene.vertex_array[pvr2_scene.vertex_index], quad, sizeof(struct vertex_struct)*2 );

 	    memcpy( &pvr2_scene.vertex_array[pvr2_scene.vertex_index+2], &quad[3], sizeof(struct vertex_struct) );

 	    memcpy( &pvr2_scene.vertex_array[pvr2_scene.vertex_index+3], &quad[2], sizeof(struct vertex_struct) );

 	    pvr2_scene.vertex_index += 4;

 	}

     }

 }

 static void scene_extract_polygons( pvraddr_t tile_entry )

 {

     uint32_t *tile_list = (uint32_t *)(video_base+tile_entry);

     do {

 	uint32_t entry = *tile_list++;

 	if( entry >> 28 == 0x0F ) {

 	    break;

 	} else if( entry >> 28 == 0x0E ) {

 	    tile_list = (uint32_t *)(video_base + (entry&0x007FFFFF));

 	} else {

 	    pvraddr_t polyaddr = entry&0x000FFFFF;

 	    int is_modified = (entry & 0x01000000) && pvr2_scene.full_shadow;

 	    int vertex_length = (entry >> 21) & 0x07;

 	    int context_length = 3;

 	    if( is_modified ) {

 		context_length = 5;

 		vertex_length <<= 1 ;

 	    }

 	    vertex_length += 3;

 	    if( (entry & 0xE0000000) == 0x80000000 ) {

 		/* Triangle(s) */

 		int strip_count = ((entry >> 25) & 0x0F)+1;

 		int polygon_length = 3 * vertex_length + context_length;

 		int i;

 		struct polygon_struct *last_poly = NULL;

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

 		    struct polygon_struct *poly = scene_add_polygon( polyaddr, 3, is_modified );

 		    polyaddr += polygon_length;

 		    if( last_poly != NULL && last_poly->next == NULL ) {

 			last_poly->next = poly;

 		    }

 		    last_poly = poly;

 		}

 	    } else if( (entry & 0xE0000000) == 0xA0000000 ) {

 		/* Sprite(s) */

 		int strip_count = ((entry >> 25) & 0x0F)+1;

 		int polygon_length = 4 * vertex_length + context_length;

 		int i;

 		struct polygon_struct *last_poly = NULL;

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

 		    struct polygon_struct *poly = scene_add_polygon( polyaddr, 4, is_modified );

 		    polyaddr += polygon_length;

 		    if( last_poly != NULL && last_poly->next == NULL ) {

 			last_poly->next = poly;

 		    }

 		    last_poly = poly;

 		}

 	    } else {

 		/* Polygon */

 		int i, last = -1;

 		for( i=5; i>=0; i-- ) {

 		    if( entry & (0x40000000>>i) ) {

 			last = i;

 			break;

 		    }

 		}

 		if( last != -1 ) {

 		    scene_add_polygon( polyaddr, last+3, is_modified );

 		}

 	    }

 	}

     } while( 1 );

 }

 static void scene_extract_vertexes( pvraddr_t tile_entry )

 {

     uint32_t *tile_list = (uint32_t *)(video_base+tile_entry);

     do {

 	uint32_t entry = *tile_list++;

 	if( entry >> 28 == 0x0F ) {

 	    break;

 	} else if( entry >> 28 == 0x0E ) {

 	    tile_list = (uint32_t *)(video_base + (entry&0x007FFFFF));

 	} else {

 	    pvraddr_t polyaddr = entry&0x000FFFFF;

 	    int is_modified = (entry & 0x01000000) && pvr2_scene.full_shadow;

 	    int vertex_length = (entry >> 21) & 0x07;

 	    int context_length = 3;

 	    if( is_modified ) {

 		context_length = 5;

 		vertex_length <<=1 ;

 	    }

 	    vertex_length += 3;

 	    if( (entry & 0xE0000000) == 0x80000000 ) {

 		/* Triangle(s) */

 		int strip_count = ((entry >> 25) & 0x0F)+1;

 		int polygon_length = 3 * vertex_length + context_length;

 		int i;

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

 		    scene_add_vertexes( polyaddr, vertex_length, is_modified );

 		    polyaddr += polygon_length;

 		}

 	    } else if( (entry & 0xE0000000) == 0xA0000000 ) {

 		/* Sprite(s) */

 		int strip_count = ((entry >> 25) & 0x0F)+1;

 		int polygon_length = 4 * vertex_length + context_length;

 		int i;

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

 		    scene_add_quad_vertexes( polyaddr, vertex_length, is_modified );

 		    polyaddr += polygon_length;

 		}

 	    } else {

 		/* Polygon */

 		int i, last = -1;

 		for( i=5; i>=0; i-- ) {

 		    if( entry & (0x40000000>>i) ) {

 			last = i;

 			break;

 		    }

 		}

 		if( last != -1 ) {

 		    scene_add_vertexes( polyaddr, vertex_length, is_modified );

 		}

 	    }

 	}

     } while( 1 );    

 }

 uint32_t pvr2_scene_buffer_width()

 {

     return pvr2_scene.buffer_width;

 }

 uint32_t pvr2_scene_buffer_height()

 {

     return pvr2_scene.buffer_height;

 }

 /**

  * Extract the current scene into the rendering structures. We run two passes

  * - first pass extracts the polygons into pvr2_scene.poly_array (finding vertex counts), 

  * second pass extracts the vertex data into the VBO/vertex array.

  *

  * Difficult to do in single pass as we don't generally know the size of a 

  * polygon for certain until we've seen all tiles containing it. It also means we

  * can count the vertexes and allocate the appropriate size VBO.

  *

  * FIXME: accesses into VRAM need to be bounds-checked properly

  */

 void pvr2_scene_read( void )

 {

     pvr2_scene_init();

     pvr2_scene_reset();

     pvr2_scene.bounds[0] = MMIO_READ( PVR2, RENDER_HCLIP ) & 0x03FF;

     pvr2_scene.bounds[1] = ((MMIO_READ( PVR2, RENDER_HCLIP ) >> 16) & 0x03FF) + 1;

     pvr2_scene.bounds[2] = MMIO_READ( PVR2, RENDER_VCLIP ) & 0x03FF;

     pvr2_scene.bounds[3] = ((MMIO_READ( PVR2, RENDER_VCLIP ) >> 16) & 0x03FF) + 1;

     pvr2_scene.bounds[4] = pvr2_scene.bounds[5] = MMIO_READF( PVR2, RENDER_FARCLIP );

     uint32_t *tilebuffer = (uint32_t *)(video_base + MMIO_READ( PVR2, RENDER_TILEBASE ));

     uint32_t *segment = tilebuffer;

     pvr2_scene.segment_list = (struct tile_segment *)tilebuffer;

     pvr2_scene.pvr2_pbuf = (uint32_t *)(video_base + MMIO_READ(PVR2,RENDER_POLYBASE));

     pvr2_scene.full_shadow = MMIO_READ( PVR2, RENDER_SHADOW ) & 0x100 ? FALSE : TRUE;

     int max_tile_x = 0;

     int max_tile_y = 0;

     int obj_config = MMIO_READ( PVR2, RENDER_OBJCFG );

     int isp_config = MMIO_READ( PVR2, RENDER_ISPCFG );

     if( (obj_config & 0x00200000) == 0 ) {

 	if( isp_config & 1 ) {

 	    pvr2_scene.sort_mode = SORT_NEVER;

 	} else {

 	    pvr2_scene.sort_mode = SORT_ALWAYS;

 	}

     } else {

 	pvr2_scene.sort_mode = SORT_BYFLAG;

     }

     // Pass 1: Extract polygon list 

     uint32_t control;

     int i;

     do {

 	control = *segment++;

 	int tile_x = SEGMENT_X(control);

 	int tile_y = SEGMENT_Y(control);

 	if( tile_x > max_tile_x ) {

 	    max_tile_x = tile_x;

 	} 

 	if( tile_y > max_tile_y ) {

 	    max_tile_y = tile_y;

 	}

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

 	    if( (*segment & NO_POINTER) == 0 ) {

 		scene_extract_polygons( *segment );

 	    }

 	    segment++;

 	}

     } while( (control & SEGMENT_END) == 0 );

     pvr2_scene.buffer_width = (max_tile_x+1)<<5;

     pvr2_scene.buffer_height = (max_tile_y+1)<<5;

     if( pvr2_scene.vertex_count > 0 ) {

 	// Pass 2: Extract vertex data

 	vertex_buffer_map();

 	pvr2_scene.vertex_index = 0;

 	segment = tilebuffer;

 	do {

 	    control = *segment++;

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

 		if( (*segment & NO_POINTER) == 0 ) {

 		    scene_extract_vertexes( *segment );

 		}

 		segment++;

 	    }

 	} while( (control & SEGMENT_END) == 0 );

 	vertex_buffer_unmap();

     }

 }