/**************************************************************************
 *
 * Copyright 2007-2008 VMware, Inc.
 * 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 VMWARE 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.
 *
 **************************************************************************/


#ifndef CSO_CONTEXT_H
#define CSO_CONTEXT_H

#include "pipe/p_context.h"
#include "pipe/p_state.h"
#include "pipe/p_defines.h"
#include "cso_cache.h"


#ifdef __cplusplus
extern "C" {
#endif

struct cso_context {
   struct pipe_context *pipe;

   /* This is equal to either pipe_context::draw_vbo or u_vbuf_draw_vbo. */
   pipe_draw_func draw_vbo;
};

#define CSO_NO_USER_VERTEX_BUFFERS (1 << 0)
#define CSO_NO_64B_VERTEX_BUFFERS  (1 << 1)
#define CSO_NO_VBUF  (1 << 2)

struct cso_context *
cso_create_context(struct pipe_context *pipe, unsigned flags);

void
cso_unbind_context(struct cso_context *ctx);

void
cso_destroy_context(struct cso_context *cso);

enum pipe_error
cso_set_blend(struct cso_context *cso, const struct pipe_blend_state *blend);

enum pipe_error
cso_set_depth_stencil_alpha(struct cso_context *cso,
                            const struct pipe_depth_stencil_alpha_state *dsa);

enum pipe_error
cso_set_rasterizer(struct cso_context *cso,
                   const struct pipe_rasterizer_state *rasterizer);

void
cso_set_samplers(struct cso_context *cso,
                 mesa_shader_stage shader_stage,
                 unsigned count,
                 const struct pipe_sampler_state **states);


/* Alternate interface to support gallium frontends that like to modify
 * samplers one at a time:
 */
void
cso_single_sampler(struct cso_context *cso, mesa_shader_stage shader_stage,
                   unsigned idx, const struct pipe_sampler_state *states);

void
cso_single_sampler_done(struct cso_context *cso,
                        mesa_shader_stage shader_stage);

void *
cso_get_vertex_elements_for_bind(struct cso_context *cso,
                                 const struct cso_velems_state *velems);

enum pipe_error
cso_set_vertex_elements(struct cso_context *ctx,
                        const struct cso_velems_state *velems);

void cso_set_vertex_buffers(struct cso_context *ctx,
                            unsigned count,
                            const struct pipe_vertex_buffer *buffers);

void cso_set_stream_outputs(struct cso_context *ctx,
                            unsigned num_targets,
                            struct pipe_stream_output_target **targets,
                            const unsigned *offsets,
                            enum mesa_prim output_prim);


enum cso_unbind_flags {
   CSO_UNBIND_FS_SAMPLERVIEWS = (1 << 0),
   CSO_UNBIND_FS_SAMPLERVIEW0 = (1 << 1),
   CSO_UNBIND_FS_IMAGE0 = (1 << 2),
   CSO_UNBIND_VS_CONSTANTS = (1 << 3),
   CSO_UNBIND_FS_CONSTANTS = (1 << 4),
};

/*
 * We don't provide shader caching in CSO.  Most of the time the api provides
 * object semantics for shaders anyway, and the cases where it doesn't
 * (eg mesa's internally-generated texenv programs), it will be up to
 * gallium frontends to implement their own specialized caching.
 */

void cso_set_fragment_shader_handle(struct cso_context *ctx, void *handle);
void cso_set_vertex_shader_handle(struct cso_context *ctx, void *handle);
void cso_set_geometry_shader_handle(struct cso_context *ctx, void *handle);
void cso_set_tessctrl_shader_handle(struct cso_context *ctx, void *handle);
void cso_set_tesseval_shader_handle(struct cso_context *ctx, void *handle);
void cso_set_compute_shader_handle(struct cso_context *ctx, void *handle);
void cso_set_task_shader_handle(struct cso_context *ctx, void *handle);
void cso_set_mesh_shader_handle(struct cso_context *ctx, void *handle);


void
cso_set_framebuffer(struct cso_context *cso,
                    const struct pipe_framebuffer_state *fb);

void
cso_set_viewport(struct cso_context *cso,
                 const struct pipe_viewport_state *vp);

void
cso_set_viewport_dims(struct cso_context *ctx,
                      float width, float height, bool invert);

void
cso_set_sample_mask(struct cso_context *cso, unsigned stencil_mask);

void
cso_set_min_samples(struct cso_context *cso, unsigned min_samples);

void
cso_set_stencil_ref(struct cso_context *cso,
                    const struct pipe_stencil_ref sr);

void
cso_set_render_condition(struct cso_context *cso,
                         struct pipe_query *query,
                         bool condition,
                         enum pipe_render_cond_flag mode);

/* gap */
#define CSO_BIT_BLEND                     0x2
#define CSO_BIT_DEPTH_STENCIL_ALPHA       0x4
#define CSO_BIT_FRAGMENT_SAMPLERS         0x8
/* gap */
#define CSO_BIT_FRAGMENT_SHADER          0x20
#define CSO_BIT_FRAMEBUFFER              0x40
#define CSO_BIT_GEOMETRY_SHADER          0x80
#define CSO_BIT_MIN_SAMPLES             0x100
#define CSO_BIT_RASTERIZER              0x200
#define CSO_BIT_RENDER_CONDITION        0x400
#define CSO_BIT_SAMPLE_MASK             0x800
#define CSO_BIT_STENCIL_REF            0x1000
#define CSO_BIT_STREAM_OUTPUTS         0x2000
#define CSO_BIT_TESSCTRL_SHADER        0x4000
#define CSO_BIT_TESSEVAL_SHADER        0x8000
#define CSO_BIT_VERTEX_ELEMENTS       0x10000
#define CSO_BIT_VERTEX_SHADER         0x20000
#define CSO_BIT_VIEWPORT              0x40000
#define CSO_BIT_PAUSE_QUERIES         0x80000
#define CSO_BIT_TASK_SHADER          0x100000
#define CSO_BIT_MESH_SHADER          0x200000

#define CSO_BITS_VERTEX_PIPE_SHADERS \
   (CSO_BIT_VERTEX_SHADER | \
    CSO_BIT_FRAGMENT_SHADER | \
    CSO_BIT_GEOMETRY_SHADER | \
    CSO_BIT_TESSCTRL_SHADER | \
    CSO_BIT_TESSEVAL_SHADER)

#define CSO_BITS_MESH_PIPE_SHADERS \
   (CSO_BIT_TASK_SHADER | \
    CSO_BIT_MESH_SHADER | \
    CSO_BIT_FRAGMENT_SHADER)

#define CSO_BIT_COMPUTE_SHADER   (1<<0)
#define CSO_BIT_COMPUTE_SAMPLERS (1<<1)

void
cso_save_state(struct cso_context *cso, unsigned state_mask);

void
cso_restore_state(struct cso_context *cso, unsigned unbind);

void
cso_save_compute_state(struct cso_context *cso, unsigned state_mask);

void
cso_restore_compute_state(struct cso_context *cso);


/* Optimized version. */
void
cso_set_vertex_buffers_and_elements(struct cso_context *ctx,
                                    const struct cso_velems_state *velems,
                                    unsigned vb_count,
                                    bool uses_user_vertex_buffers,
                                    const struct pipe_vertex_buffer *vbuffers);

void
cso_draw_arrays_instanced(struct cso_context *cso, unsigned mode,
                          unsigned start, unsigned count,
                          unsigned start_instance, unsigned instance_count);

void
cso_draw_arrays(struct cso_context *cso, unsigned mode, unsigned start, unsigned count);

/* Inline functions. */

static ALWAYS_INLINE void
cso_draw_vbo(struct cso_context *cso,
             const struct pipe_draw_info *info,
             unsigned drawid_offset,
             const struct pipe_draw_indirect_info *indirect,
             const struct pipe_draw_start_count_bias *draws,
             unsigned num_draws)
{
   /* We can't have both indirect drawing and SO-vertex-count drawing */
   assert(!indirect ||
          indirect->buffer == NULL ||
          indirect->count_from_stream_output == NULL);

   /* We can't have SO-vertex-count drawing with an index buffer */
   assert(info->index_size == 0 ||
          !indirect ||
          indirect->count_from_stream_output == NULL);

   /* Indirect only uses indirect->draw_count, not num_draws. */
   assert(!indirect || num_draws == 1);

   cso->draw_vbo(cso->pipe, info, drawid_offset, indirect, draws, num_draws);
}

#ifdef __cplusplus
}
#endif

#endif
