/**************************************************************************
 *
 * 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:
  *   Zack Rusin <zack@tungstengraphics.com>
  */

#include "st_cache.h"

#include "st_context.h"

#include "pipe/p_state.h"

#include "cso_cache/cso_cache.h"
#include "cso_cache/cso_hash.h"


/* Those function will either find the state of the given template
 * in the cache or they will create a new state from the given
 * template, insert it in the cache and return it.
 */

/*
 * If the driver returns 0 from the create method then they will assign
 * the data member of the cso to be the template itself.
 */

const struct cso_blend * st_cached_blend_state(struct st_context *st,
                                               const struct pipe_blend_state *templ)
{
   unsigned hash_key = cso_construct_key((void*)templ, sizeof(struct pipe_blend_state));
   struct cso_hash_iter iter = cso_find_state_template(st->cache,
                                                       hash_key, CSO_BLEND,
                                                       (void*)templ);
   if (cso_hash_iter_is_null(iter)) {
      struct cso_blend *cso = malloc(sizeof(struct cso_blend));
      memcpy(&cso->state, templ, sizeof(struct pipe_blend_state));
      cso->data = st->pipe->create_blend_state(st->pipe, &cso->state);
      if (!cso->data)
         cso->data = &cso->state;
      cso->delete_state = (cso_state_callback)st->pipe->delete_blend_state;
      cso->context = st->pipe;
      iter = cso_insert_state(st->cache, hash_key, CSO_BLEND, cso);
   }
   return ((struct cso_blend *)cso_hash_iter_data(iter));
}

const struct cso_sampler *
st_cached_sampler_state(struct st_context *st,
                        const struct pipe_sampler_state *templ)
{
   unsigned hash_key = cso_construct_key((void*)templ, sizeof(struct pipe_sampler_state));
   struct cso_hash_iter iter = cso_find_state_template(st->cache,
                                                       hash_key, CSO_SAMPLER,
                                                       (void*)templ);
   if (cso_hash_iter_is_null(iter)) {
      struct cso_sampler *cso = malloc(sizeof(struct cso_sampler));
      memcpy(&cso->state, templ, sizeof(struct pipe_sampler_state));
      cso->data = st->pipe->create_sampler_state(st->pipe, &cso->state);
      if (!cso->data)
         cso->data = &cso->state;
      cso->delete_state = (cso_state_callback)st->pipe->delete_sampler_state;
      cso->context = st->pipe;
      iter = cso_insert_state(st->cache, hash_key, CSO_SAMPLER, cso);
   }
   return (struct cso_sampler*)(cso_hash_iter_data(iter));
}

const struct cso_depth_stencil_alpha *
st_cached_depth_stencil_alpha_state(struct st_context *st,
                              const struct pipe_depth_stencil_alpha_state *templ)
{
   unsigned hash_key = cso_construct_key((void*)templ,
                                         sizeof(struct pipe_depth_stencil_alpha_state));
   struct cso_hash_iter iter = cso_find_state_template(st->cache,
                                                       hash_key, 
						       CSO_DEPTH_STENCIL_ALPHA,
                                                       (void*)templ);
   if (cso_hash_iter_is_null(iter)) {
      struct cso_depth_stencil_alpha *cso = malloc(sizeof(struct cso_depth_stencil_alpha));
      memcpy(&cso->state, templ, sizeof(struct pipe_depth_stencil_alpha_state));
      cso->data = st->pipe->create_depth_stencil_alpha_state(st->pipe, &cso->state);
      if (!cso->data)
         cso->data = &cso->state;
      cso->delete_state = (cso_state_callback)st->pipe->delete_depth_stencil_alpha_state;
      cso->context = st->pipe;
      iter = cso_insert_state(st->cache, hash_key, CSO_DEPTH_STENCIL_ALPHA, cso);
   }
   return (struct cso_depth_stencil_alpha*)(cso_hash_iter_data(iter));
}

const struct cso_rasterizer* st_cached_rasterizer_state(
   struct st_context *st,
   const struct pipe_rasterizer_state *templ)
{
   unsigned hash_key = cso_construct_key((void*)templ,
                                         sizeof(struct pipe_rasterizer_state));
   struct cso_hash_iter iter = cso_find_state_template(st->cache,
                                                       hash_key, CSO_RASTERIZER,
                                                       (void*)templ);
   if (cso_hash_iter_is_null(iter)) {
      struct cso_rasterizer *cso = malloc(sizeof(struct cso_rasterizer));
      memcpy(&cso->state, templ, sizeof(struct pipe_rasterizer_state));
      cso->data = st->pipe->create_rasterizer_state(st->pipe, &cso->state);
      if (!cso->data)
         cso->data = &cso->state;
      cso->delete_state = (cso_state_callback)st->pipe->delete_rasterizer_state;
      cso->context = st->pipe;
      iter = cso_insert_state(st->cache, hash_key, CSO_RASTERIZER, cso);
   }
   return (struct cso_rasterizer*)(cso_hash_iter_data(iter));
}

const struct cso_fragment_shader *
st_cached_fs_state(struct st_context *st,
                   const struct pipe_shader_state *templ)
{
   unsigned hash_key = cso_construct_key((void*)templ,
                                         sizeof(struct pipe_shader_state));
   struct cso_hash_iter iter = cso_find_state_template(st->cache,
                                                       hash_key, CSO_FRAGMENT_SHADER,
                                                       (void*)templ);
   if (cso_hash_iter_is_null(iter)) {
      struct cso_fragment_shader *cso = malloc(sizeof(struct cso_fragment_shader));
      memcpy(&cso->state, templ, sizeof(struct pipe_shader_state));
      cso->data = st->pipe->create_fs_state(st->pipe, &cso->state);
      if (!cso->data)
         cso->data = &cso->state;
      cso->delete_state = (cso_state_callback)st->pipe->delete_fs_state;
      cso->context = st->pipe;
      iter = cso_insert_state(st->cache, hash_key, CSO_FRAGMENT_SHADER, cso);
   }
   return (struct cso_fragment_shader*)(cso_hash_iter_data(iter));
}

const struct cso_vertex_shader *
st_cached_vs_state(struct st_context *st,
                   const struct pipe_shader_state *templ)
{
   unsigned hash_key = cso_construct_key((void*)templ,
                                         sizeof(struct pipe_shader_state));
   struct cso_hash_iter iter = cso_find_state_template(st->cache,
                                                       hash_key, CSO_VERTEX_SHADER,
                                                       (void*)templ);
   if (cso_hash_iter_is_null(iter)) {
      struct cso_vertex_shader *cso = malloc(sizeof(struct cso_vertex_shader));
      memcpy(&cso->state, templ, sizeof(struct pipe_shader_state));
      cso->data = st->pipe->create_vs_state(st->pipe, &cso->state);
      if (!cso->data)
         cso->data = &cso->state;
      cso->delete_state = (cso_state_callback)st->pipe->delete_vs_state;
      cso->context = st->pipe;
      iter = cso_insert_state(st->cache, hash_key, CSO_VERTEX_SHADER, cso);
   }
   return (struct cso_vertex_shader*)(cso_hash_iter_data(iter));
}