/**************************************************************************
 * 
 * Copyright 2007 Tungsten Graphics, Inc., Cedar Park, Texas.
 * All Rights Reserved.
 * 
 * Permission is hereby granted, free of charge, to any person obtaining a
 * copy of this software and associated documentation files (the
 * "Software"), to deal in the Software without restriction, including
 * without limitation the rights to use, copy, modify, merge, publish,
 * distribute, sub license, and/or sell copies of the Software, and to
 * permit persons to whom the Software is furnished to do so, subject to
 * the following conditions:
 * 
 * The above copyright notice and this permission notice (including the
 * next paragraph) shall be included in all copies or substantial portions
 * of the Software.
 * 
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
 * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.
 * IN NO EVENT SHALL TUNGSTEN GRAPHICS AND/OR ITS SUPPLIERS BE LIABLE FOR
 * ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
 * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
 * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 * 
 **************************************************************************/

 /*
  * Authors:
  *   Keith Whitwell <keith@tungstengraphics.com>
  */


#include "pipe/p_util.h"
#include "draw_context.h"
#include "draw_private.h"



struct draw_context *draw_create( void )
{
   struct draw_context *draw = CALLOC_STRUCT( draw_context );

#if defined(__i386__) || defined(__386__)
   draw->use_sse = GETENV( "GALLIUM_NOSSE" ) == NULL;
#else
   draw->use_sse = FALSE;
#endif

   /* create pipeline stages */
   draw->pipeline.wide      = draw_wide_stage( draw );
   draw->pipeline.unfilled  = draw_unfilled_stage( draw );
   draw->pipeline.twoside   = draw_twoside_stage( draw );
   draw->pipeline.offset    = draw_offset_stage( draw );
   draw->pipeline.clip      = draw_clip_stage( draw );
   draw->pipeline.flatshade = draw_flatshade_stage( draw );
   draw->pipeline.cull      = draw_cull_stage( draw );
   draw->pipeline.feedback  = draw_feedback_stage( draw );
   draw->pipeline.validate  = draw_validate_stage( draw );
   draw->pipeline.first     = draw->pipeline.validate;

   ASSIGN_4V( draw->plane[0], -1,  0,  0, 1 );
   ASSIGN_4V( draw->plane[1],  1,  0,  0, 1 );
   ASSIGN_4V( draw->plane[2],  0, -1,  0, 1 );
   ASSIGN_4V( draw->plane[3],  0,  1,  0, 1 );
   ASSIGN_4V( draw->plane[4],  0,  0,  1, 1 ); /* yes these are correct */
   ASSIGN_4V( draw->plane[5],  0,  0, -1, 1 ); /* mesa's a bit wonky */
   draw->nr_planes = 6;

   /* Statically allocate maximum sized vertices for the cache - could be cleverer...
    */
   {
      int i;
      char *tmp = MALLOC( Elements(draw->vcache.vertex) * MAX_VERTEX_SIZE );

      for (i = 0; i < Elements(draw->vcache.vertex); i++)
	 draw->vcache.vertex[i] = (struct vertex_header *)(tmp + i * MAX_VERTEX_SIZE);
   }

   draw->attrib_front0 = -1;
   draw->attrib_back0 = -1;
   draw->attrib_front1 = -1;
   draw->attrib_back1 = -1;

   draw->prim = ~0; /* != any of PIPE_PRIM_x */

   draw_vertex_cache_invalidate( draw );
   draw_set_mapped_element_buffer( draw, 0, NULL );

   return draw;
}


void draw_destroy( struct draw_context *draw )
{
   draw->pipeline.wide->destroy( draw->pipeline.wide );
   draw->pipeline.unfilled->destroy( draw->pipeline.unfilled );
   draw->pipeline.twoside->destroy( draw->pipeline.twoside );
   draw->pipeline.offset->destroy( draw->pipeline.offset );
   draw->pipeline.clip->destroy( draw->pipeline.clip );
   draw->pipeline.flatshade->destroy( draw->pipeline.flatshade );
   draw->pipeline.cull->destroy( draw->pipeline.cull );
   draw->pipeline.feedback->destroy( draw->pipeline.feedback );
   draw->pipeline.validate->destroy( draw->pipeline.validate );
   if (draw->pipeline.rasterize)
      draw->pipeline.rasterize->destroy( draw->pipeline.rasterize );
   FREE( draw->vcache.vertex[0] ); /* Frees all the vertices. */
   FREE( draw );
}



void draw_flush( struct draw_context *draw )
{
   if (draw->drawing)
      draw_do_flush( draw, DRAW_FLUSH_DRAW );
}


void draw_set_feedback_state( struct draw_context *draw,
                              const struct pipe_feedback_state *feedback )
{
   draw_flush( draw );
   draw->feedback = *feedback;
}


/**
 * Register new primitive rasterization/rendering state.
 * This causes the drawing pipeline to be rebuilt.
 */
void draw_set_rasterizer_state( struct draw_context *draw,
                                const struct pipe_rasterizer_state *raster )
{
   draw_flush( draw );
   draw->rasterizer = raster;
}


/** 
 * Plug in the primitive rendering/rasterization stage (which is the last
 * stage in the drawing pipeline).
 * This is provided by the device driver.
 */
void draw_set_rasterize_stage( struct draw_context *draw,
                               struct draw_stage *stage )
{
   draw_flush( draw );
   draw->pipeline.rasterize = stage;
}


/**
 * Set the draw module's clipping state.
 */
void draw_set_clip_state( struct draw_context *draw,
                          const struct pipe_clip_state *clip )
{
   draw_flush( draw );

   assert(clip->nr <= PIPE_MAX_CLIP_PLANES);
   memcpy(&draw->plane[6], clip->ucp, clip->nr * sizeof(clip->ucp[0]));
   draw->nr_planes = 6 + clip->nr;
}


/**
 * Set the draw module's viewport state.
 */
void draw_set_viewport_state( struct draw_context *draw,
                              const struct pipe_viewport_state *viewport )
{
   draw_flush( draw );
   draw->viewport = *viewport; /* struct copy */
}



void
draw_set_vertex_buffer(struct draw_context *draw,
                       unsigned attr,
                       const struct pipe_vertex_buffer *buffer)
{
   draw_flush( draw );

   assert(attr < PIPE_ATTRIB_MAX);
   draw->vertex_buffer[attr] = *buffer;
}


void
draw_set_vertex_element(struct draw_context *draw,
                        unsigned attr,
                        const struct pipe_vertex_element *element)
{
   draw_flush( draw );

   assert(attr < PIPE_ATTRIB_MAX);
   draw->vertex_element[attr] = *element;
}


/**
 * Tell drawing context where to find mapped vertex buffers.
 */
void
draw_set_mapped_vertex_buffer(struct draw_context *draw,
                              unsigned attr, const void *buffer)
{
   draw_flush( draw );

   draw->user.vbuffer[attr] = buffer;
}


void
draw_set_mapped_constant_buffer(struct draw_context *draw,
                                const void *buffer)
{
   draw_flush( draw );

   draw->user.constants = buffer;
}


void
draw_set_mapped_feedback_buffer(struct draw_context *draw, uint index,
                                void *buffer, uint size)
{
   draw_flush( draw );

   assert(index < PIPE_MAX_FEEDBACK_ATTRIBS);
   draw->user.feedback_buffer[index] = buffer;
   draw->user.feedback_buffer_size[index] = size; /* in bytes */
}





/**
 * Allocate space for temporary post-transform vertices, such as for clipping.
 */
void draw_alloc_tmps( struct draw_stage *stage, unsigned nr )
{
   assert(!stage->tmp);

   stage->nr_tmps = nr;

   if (nr) {
      ubyte *store = (ubyte *) MALLOC( MAX_VERTEX_SIZE * nr );
      unsigned i;

      stage->tmp = (struct vertex_header **) MALLOC( sizeof(struct vertex_header *) * nr );
      
      for (i = 0; i < nr; i++)
	 stage->tmp[i] = (struct vertex_header *)(store + i * MAX_VERTEX_SIZE);
   }
}

/**
 * Reset the vertex ids for the stage's temp verts.
 */
void draw_reset_tmps( struct draw_stage *stage )
{
   unsigned i;
   
   if (stage->tmp)
      for (i = 0; i < stage->nr_tmps; i++)
	 stage->tmp[i]->vertex_id = UNDEFINED_VERTEX_ID;
}

void draw_free_tmps( struct draw_stage *stage )
{
   if (stage->tmp) {
      FREE( stage->tmp[0] );
      FREE( stage->tmp );
      stage->tmp = NULL;
   }
}


boolean draw_use_sse(struct draw_context *draw)
{
   return (boolean) draw->use_sse;
}


void draw_reset_vertex_ids(struct draw_context *draw)
{
   struct draw_stage *stage = draw->pipeline.first;
   
   while (stage) {
      if (stage->reset_tmps)
	 stage->reset_tmps(stage);
      else
	 draw_reset_tmps(stage);
      stage = stage->next;
   }

   draw_vertex_cache_reset_vertex_ids(draw);
}