r600g: do fine-grained depth texture flushing
[mesa.git] / src / gallium / drivers / r600 / r600_state_common.c
index 9b8a2296ee8931fc527b4bb742fc434581a61453..e62cd2e9280abb289e68c31aa0df687dc9ce78dd 100644 (file)
  * Authors: Dave Airlie <airlied@redhat.com>
  *          Jerome Glisse <jglisse@redhat.com>
  */
-#include "util/u_blitter.h"
-#include "util/u_memory.h"
-#include "util/u_format.h"
-#include "pipebuffer/pb_buffer.h"
-#include "pipe/p_shader_tokens.h"
-#include "tgsi/tgsi_parse.h"
 #include "r600_formats.h"
-#include "r600_pipe.h"
 #include "r600d.h"
-#include "r600_hw_context_priv.h"
+
+#include "util/u_blitter.h"
+#include "util/u_upload_mgr.h"
+#include "tgsi/tgsi_parse.h"
+#include <byteswap.h>
 
 static void r600_emit_command_buffer(struct r600_context *rctx, struct r600_atom *atom)
 {
@@ -62,7 +59,7 @@ void r600_release_command_buffer(struct r600_command_buffer *cb)
 static void r600_emit_surface_sync(struct r600_context *rctx, struct r600_atom *atom)
 {
        struct radeon_winsys_cs *cs = rctx->cs;
-       struct r600_atom_surface_sync *a = (struct r600_atom_surface_sync*)atom;
+       struct r600_surface_sync_cmd *a = (struct r600_surface_sync_cmd*)atom;
 
        cs->buf[cs->cdw++] = PKT3(PKT3_SURFACE_SYNC, 3, 0);
        cs->buf[cs->cdw++] = a->flush_flags;  /* CP_COHER_CNTL */
@@ -91,8 +88,8 @@ void r600_init_atom(struct r600_atom *atom,
 
 void r600_init_common_atoms(struct r600_context *rctx)
 {
-       r600_init_atom(&rctx->atom_surface_sync.atom,   r600_emit_surface_sync,         5, EMIT_EARLY);
-       r600_init_atom(&rctx->atom_r6xx_flush_and_inv,  r600_emit_r6xx_flush_and_inv,   2, EMIT_EARLY);
+       r600_init_atom(&rctx->surface_sync_cmd.atom,    r600_emit_surface_sync,         5, EMIT_EARLY);
+       r600_init_atom(&rctx->r6xx_flush_and_inv_cmd,   r600_emit_r6xx_flush_and_inv,   2, EMIT_EARLY);
 }
 
 unsigned r600_get_cb_flush_flags(struct r600_context *rctx)
@@ -118,8 +115,8 @@ void r600_texture_barrier(struct pipe_context *ctx)
 {
        struct r600_context *rctx = (struct r600_context *)ctx;
 
-       rctx->atom_surface_sync.flush_flags |= S_0085F0_TC_ACTION_ENA(1) | r600_get_cb_flush_flags(rctx);
-       r600_atom_dirty(rctx, &rctx->atom_surface_sync.atom);
+       rctx->surface_sync_cmd.flush_flags |= S_0085F0_TC_ACTION_ENA(1) | r600_get_cb_flush_flags(rctx);
+       r600_atom_dirty(rctx, &rctx->surface_sync_cmd.atom);
 }
 
 static bool r600_conv_pipe_prim(unsigned pprim, unsigned *prim)
@@ -155,18 +152,31 @@ void r600_bind_blend_state(struct pipe_context *ctx, void *state)
        struct r600_context *rctx = (struct r600_context *)ctx;
        struct r600_pipe_blend *blend = (struct r600_pipe_blend *)state;
        struct r600_pipe_state *rstate;
+       bool update_cb = false;
 
        if (state == NULL)
                return;
        rstate = &blend->rstate;
        rctx->states[rstate->id] = rstate;
-       rctx->cb_target_mask = blend->cb_target_mask;
-
-       /* Replace every bit except MULTIWRITE_ENABLE. */
-       rctx->cb_color_control &= ~C_028808_MULTIWRITE_ENABLE;
-       rctx->cb_color_control |= blend->cb_color_control & C_028808_MULTIWRITE_ENABLE;
-
+       rctx->dual_src_blend = blend->dual_src_blend;
        r600_context_pipe_state_set(rctx, rstate);
+
+       if (rctx->cb_misc_state.blend_colormask != blend->cb_target_mask) {
+               rctx->cb_misc_state.blend_colormask = blend->cb_target_mask;
+               update_cb = true;
+       }
+       if (rctx->chip_class <= R700 &&
+           rctx->cb_misc_state.cb_color_control != blend->cb_color_control) {
+               rctx->cb_misc_state.cb_color_control = blend->cb_color_control;
+               update_cb = true;
+       }
+       if (rctx->cb_misc_state.dual_src_blend != blend->dual_src_blend) {
+               rctx->cb_misc_state.dual_src_blend = blend->dual_src_blend;
+               update_cb = true;
+       }
+       if (update_cb) {
+               r600_atom_dirty(rctx, &rctx->cb_misc_state.atom);
+       }
 }
 
 void r600_set_blend_color(struct pipe_context *ctx,
@@ -179,10 +189,10 @@ void r600_set_blend_color(struct pipe_context *ctx,
                return;
 
        rstate->id = R600_PIPE_STATE_BLEND_COLOR;
-       r600_pipe_state_add_reg(rstate, R_028414_CB_BLEND_RED, fui(state->color[0]), NULL, 0);
-       r600_pipe_state_add_reg(rstate, R_028418_CB_BLEND_GREEN, fui(state->color[1]), NULL, 0);
-       r600_pipe_state_add_reg(rstate, R_02841C_CB_BLEND_BLUE, fui(state->color[2]), NULL, 0);
-       r600_pipe_state_add_reg(rstate, R_028420_CB_BLEND_ALPHA, fui(state->color[3]), NULL, 0);
+       r600_pipe_state_add_reg(rstate, R_028414_CB_BLEND_RED, fui(state->color[0]));
+       r600_pipe_state_add_reg(rstate, R_028418_CB_BLEND_GREEN, fui(state->color[1]));
+       r600_pipe_state_add_reg(rstate, R_02841C_CB_BLEND_BLUE, fui(state->color[2]));
+       r600_pipe_state_add_reg(rstate, R_028420_CB_BLEND_ALPHA, fui(state->color[3]));
 
        free(rctx->states[R600_PIPE_STATE_BLEND_COLOR]);
        rctx->states[R600_PIPE_STATE_BLEND_COLOR] = rstate;
@@ -203,14 +213,12 @@ static void r600_set_stencil_ref(struct pipe_context *ctx,
                                R_028430_DB_STENCILREFMASK,
                                S_028430_STENCILREF(state->ref_value[0]) |
                                S_028430_STENCILMASK(state->valuemask[0]) |
-                               S_028430_STENCILWRITEMASK(state->writemask[0]),
-                               NULL, 0);
+                               S_028430_STENCILWRITEMASK(state->writemask[0]));
        r600_pipe_state_add_reg(rstate,
                                R_028434_DB_STENCILREFMASK_BF,
                                S_028434_STENCILREF_BF(state->ref_value[1]) |
                                S_028434_STENCILMASK_BF(state->valuemask[1]) |
-                               S_028434_STENCILWRITEMASK_BF(state->writemask[1]),
-                               NULL, 0);
+                               S_028434_STENCILWRITEMASK_BF(state->writemask[1]));
 
        free(rctx->states[R600_PIPE_STATE_STENCIL_REF]);
        rctx->states[R600_PIPE_STATE_STENCIL_REF] = rstate;
@@ -250,6 +258,8 @@ void r600_bind_dsa_state(struct pipe_context *ctx, void *state)
                return;
        rstate = &dsa->rstate;
        rctx->states[rstate->id] = rstate;
+       rctx->sx_alpha_test_control &= ~0xff;
+       rctx->sx_alpha_test_control |= dsa->sx_alpha_test_control;
        rctx->alpha_ref = dsa->alpha_ref;
        rctx->alpha_ref_dirty = true;
        r600_context_pipe_state_set(rctx, rstate);
@@ -262,11 +272,6 @@ void r600_bind_dsa_state(struct pipe_context *ctx, void *state)
        ref.writemask[1] = dsa->writemask[1];
 
        r600_set_stencil_ref(ctx, &ref);
-
-       if (rctx->atom_db_misc_state.flush_depthstencil_enabled != dsa->is_flush) {
-               rctx->atom_db_misc_state.flush_depthstencil_enabled = dsa->is_flush;
-               r600_atom_dirty(rctx, &rctx->atom_db_misc_state.atom);
-       }
 }
 
 void r600_set_max_scissor(struct r600_context *rctx)
@@ -301,7 +306,6 @@ void r600_bind_rs_state(struct pipe_context *ctx, void *state)
 
        if (rctx->chip_class >= EVERGREEN) {
                evergreen_polygon_offset_update(rctx);
-               evergreen_set_rasterizer_discard(ctx, rs->rasterizer_discard);
        } else {
                r600_polygon_offset_update(rctx);
        }
@@ -365,8 +369,6 @@ void r600_bind_vertex_elements(struct pipe_context *ctx, void *state)
        rctx->vertex_elements = v;
        if (v) {
                r600_inval_shader_cache(rctx);
-               u_vbuf_bind_vertex_elements(rctx->vbuf_mgr, state,
-                                               v->vmgr_elements);
 
                rctx->states[v->rstate.id] = &v->rstate;
                r600_context_pipe_state_set(rctx, &v->rstate);
@@ -385,36 +387,33 @@ void r600_delete_vertex_element(struct pipe_context *ctx, void *state)
                rctx->vertex_elements = NULL;
 
        pipe_resource_reference((struct pipe_resource**)&v->fetch_shader, NULL);
-       u_vbuf_destroy_vertex_elements(rctx->vbuf_mgr, v->vmgr_elements);
        FREE(state);
 }
 
-
 void r600_set_index_buffer(struct pipe_context *ctx,
                           const struct pipe_index_buffer *ib)
 {
        struct r600_context *rctx = (struct r600_context *)ctx;
 
-       u_vbuf_set_index_buffer(rctx->vbuf_mgr, ib);
+       if (ib) {
+               pipe_resource_reference(&rctx->index_buffer.buffer, ib->buffer);
+               memcpy(&rctx->index_buffer, ib, sizeof(*ib));
+       } else {
+               pipe_resource_reference(&rctx->index_buffer.buffer, NULL);
+       }
 }
 
 void r600_set_vertex_buffers(struct pipe_context *ctx, unsigned count,
                             const struct pipe_vertex_buffer *buffers)
 {
        struct r600_context *rctx = (struct r600_context *)ctx;
-       int i;
 
-       /* Zero states. */
-       for (i = 0; i < count; i++) {
-               if (!buffers[i].buffer) {
-                       r600_context_pipe_state_set_fs_resource(rctx, NULL, i);
-               }
-       }
-       for (; i < rctx->vbuf_mgr->nr_real_vertex_buffers; i++) {
-               r600_context_pipe_state_set_fs_resource(rctx, NULL, i);
-       }
+       util_copy_vertex_buffers(rctx->vertex_buffer, &rctx->nr_vertex_buffers, buffers, count);
 
-       u_vbuf_set_vertex_buffers(rctx->vbuf_mgr, count, buffers);
+       r600_inval_vertex_cache(rctx);
+       rctx->vertex_buffer_state.num_dw = (rctx->chip_class >= EVERGREEN ? 12 : 10) *
+                                          rctx->nr_vertex_buffers;
+       r600_atom_dirty(rctx, &rctx->vertex_buffer_state);
 }
 
 void *r600_create_vertex_elements(struct pipe_context *ctx,
@@ -429,9 +428,7 @@ void *r600_create_vertex_elements(struct pipe_context *ctx,
                return NULL;
 
        v->count = count;
-       v->vmgr_elements =
-               u_vbuf_create_vertex_elements(rctx->vbuf_mgr, count,
-                                                 elements, v->elements);
+       memcpy(v->elements, elements, sizeof(struct pipe_vertex_element) * count);
 
        if (r600_vertex_elements_build_fetch_shader(rctx, v)) {
                FREE(v);
@@ -441,40 +438,156 @@ void *r600_create_vertex_elements(struct pipe_context *ctx,
        return v;
 }
 
-void *r600_create_shader_state(struct pipe_context *ctx,
-                              const struct pipe_shader_state *state)
+/* Compute the key for the hw shader variant */
+static INLINE unsigned r600_shader_selector_key(struct pipe_context * ctx,
+               struct r600_pipe_shader_selector * sel)
 {
-       struct r600_pipe_shader *shader = CALLOC_STRUCT(r600_pipe_shader);
+       struct r600_context *rctx = (struct r600_context *)ctx;
+       unsigned key;
+
+       if (sel->type == PIPE_SHADER_FRAGMENT) {
+               key = rctx->two_side |
+                               MIN2(sel->nr_ps_max_color_exports, rctx->nr_cbufs + rctx->dual_src_blend) << 1;
+       } else
+               key = 0;
+
+       return key;
+}
+
+/* Select the hw shader variant depending on the current state.
+ * (*dirty) is set to 1 if current variant was changed */
+static int r600_shader_select(struct pipe_context *ctx,
+        struct r600_pipe_shader_selector* sel,
+        unsigned *dirty)
+{
+       unsigned key;
+       struct r600_context *rctx = (struct r600_context *)ctx;
+       struct r600_pipe_shader * shader = NULL;
        int r;
 
-       shader->tokens = tgsi_dup_tokens(state->tokens);
-       shader->so = state->stream_output;
+       key = r600_shader_selector_key(ctx, sel);
 
-       r =  r600_pipe_shader_create(ctx, shader);
-       if (r) {
-               return NULL;
+       /* Check if we don't need to change anything.
+        * This path is also used for most shaders that don't need multiple
+        * variants, it will cost just a computation of the key and this
+        * test. */
+       if (likely(sel->current && sel->current->key == key)) {
+               return 0;
+       }
+
+       /* lookup if we have other variants in the list */
+       if (sel->num_shaders > 1) {
+               struct r600_pipe_shader *p = sel->current, *c = p->next_variant;
+
+               while (c && c->key != key) {
+                       p = c;
+                       c = c->next_variant;
+               }
+
+               if (c) {
+                       p->next_variant = c->next_variant;
+                       shader = c;
+               }
+       }
+
+       if (unlikely(!shader)) {
+               shader = CALLOC(1, sizeof(struct r600_pipe_shader));
+               shader->selector = sel;
+
+               r = r600_pipe_shader_create(ctx, shader);
+               if (unlikely(r)) {
+                       R600_ERR("Failed to build shader variant (type=%u, key=%u) %d\n",
+                                       sel->type, key, r);
+                       sel->current = NULL;
+                       return r;
+               }
+
+               /* We don't know the value of nr_ps_max_color_exports until we built
+                * at least one variant, so we may need to recompute the key after
+                * building first variant. */
+               if (sel->type == PIPE_SHADER_FRAGMENT &&
+                               sel->num_shaders == 0) {
+                       sel->nr_ps_max_color_exports = shader->shader.nr_ps_max_color_exports;
+                       key = r600_shader_selector_key(ctx, sel);
+               }
+
+               shader->key = key;
+               sel->num_shaders++;
        }
-       return shader;
+
+       if (dirty)
+               *dirty = 1;
+
+       shader->next_variant = sel->current;
+       sel->current = shader;
+
+       if (rctx->chip_class < EVERGREEN && rctx->ps_shader && rctx->vs_shader) {
+               r600_adjust_gprs(rctx);
+       }
+
+       if (rctx->ps_shader &&
+           rctx->cb_misc_state.nr_ps_color_outputs != rctx->ps_shader->current->nr_ps_color_outputs) {
+               rctx->cb_misc_state.nr_ps_color_outputs = rctx->ps_shader->current->nr_ps_color_outputs;
+               r600_atom_dirty(rctx, &rctx->cb_misc_state.atom);
+       }
+       return 0;
+}
+
+static void *r600_create_shader_state(struct pipe_context *ctx,
+                              const struct pipe_shader_state *state,
+                              unsigned pipe_shader_type)
+{
+       struct r600_pipe_shader_selector *sel = CALLOC_STRUCT(r600_pipe_shader_selector);
+       int r;
+
+       sel->type = pipe_shader_type;
+       sel->tokens = tgsi_dup_tokens(state->tokens);
+       sel->so = state->stream_output;
+
+       r = r600_shader_select(ctx, sel, NULL);
+       if (r)
+           return NULL;
+
+       return sel;
+}
+
+void *r600_create_shader_state_ps(struct pipe_context *ctx,
+               const struct pipe_shader_state *state)
+{
+       return r600_create_shader_state(ctx, state, PIPE_SHADER_FRAGMENT);
+}
+
+void *r600_create_shader_state_vs(struct pipe_context *ctx,
+               const struct pipe_shader_state *state)
+{
+       return r600_create_shader_state(ctx, state, PIPE_SHADER_VERTEX);
 }
 
 void r600_bind_ps_shader(struct pipe_context *ctx, void *state)
 {
        struct r600_context *rctx = (struct r600_context *)ctx;
 
-       if (!state) {
+       if (!state)
                state = rctx->dummy_pixel_shader;
-       }
 
-       rctx->ps_shader = (struct r600_pipe_shader *)state;
+       rctx->ps_shader = (struct r600_pipe_shader_selector *)state;
+       r600_context_pipe_state_set(rctx, &rctx->ps_shader->current->rstate);
 
-       r600_inval_shader_cache(rctx);
-       r600_context_pipe_state_set(rctx, &rctx->ps_shader->rstate);
+       if (rctx->chip_class <= R700) {
+               bool multiwrite = rctx->ps_shader->current->shader.fs_write_all;
+
+               if (rctx->cb_misc_state.multiwrite != multiwrite) {
+                       rctx->cb_misc_state.multiwrite = multiwrite;
+                       r600_atom_dirty(rctx, &rctx->cb_misc_state.atom);
+               }
 
-       rctx->cb_color_control &= C_028808_MULTIWRITE_ENABLE;
-       rctx->cb_color_control |= S_028808_MULTIWRITE_ENABLE(!!rctx->ps_shader->shader.fs_write_all);
+               if (rctx->vs_shader)
+                       r600_adjust_gprs(rctx);
+       }
 
-       if (rctx->ps_shader && rctx->vs_shader) {
-               r600_adjust_gprs(rctx);
+       if (rctx->cb_misc_state.nr_ps_color_outputs != rctx->ps_shader->current->nr_ps_color_outputs) {
+               rctx->cb_misc_state.nr_ps_color_outputs = rctx->ps_shader->current->nr_ps_color_outputs;
+               r600_atom_dirty(rctx, &rctx->cb_misc_state.atom);
        }
 }
 
@@ -482,42 +595,53 @@ void r600_bind_vs_shader(struct pipe_context *ctx, void *state)
 {
        struct r600_context *rctx = (struct r600_context *)ctx;
 
-       rctx->vs_shader = (struct r600_pipe_shader *)state;
+       rctx->vs_shader = (struct r600_pipe_shader_selector *)state;
        if (state) {
-               r600_inval_shader_cache(rctx);
-               r600_context_pipe_state_set(rctx, &rctx->vs_shader->rstate);
+               r600_context_pipe_state_set(rctx, &rctx->vs_shader->current->rstate);
+
+               if (rctx->chip_class < EVERGREEN && rctx->ps_shader)
+                       r600_adjust_gprs(rctx);
        }
-       if (rctx->ps_shader && rctx->vs_shader) {
-               r600_adjust_gprs(rctx);
+}
+
+static void r600_delete_shader_selector(struct pipe_context *ctx,
+               struct r600_pipe_shader_selector *sel)
+{
+       struct r600_pipe_shader *p = sel->current, *c;
+       while (p) {
+               c = p->next_variant;
+               r600_pipe_shader_destroy(ctx, p);
+               free(p);
+               p = c;
        }
+
+       free(sel->tokens);
+       free(sel);
 }
 
+
 void r600_delete_ps_shader(struct pipe_context *ctx, void *state)
 {
        struct r600_context *rctx = (struct r600_context *)ctx;
-       struct r600_pipe_shader *shader = (struct r600_pipe_shader *)state;
+       struct r600_pipe_shader_selector *sel = (struct r600_pipe_shader_selector *)state;
 
-       if (rctx->ps_shader == shader) {
+       if (rctx->ps_shader == sel) {
                rctx->ps_shader = NULL;
        }
 
-       free(shader->tokens);
-       r600_pipe_shader_destroy(ctx, shader);
-       free(shader);
+       r600_delete_shader_selector(ctx, sel);
 }
 
 void r600_delete_vs_shader(struct pipe_context *ctx, void *state)
 {
        struct r600_context *rctx = (struct r600_context *)ctx;
-       struct r600_pipe_shader *shader = (struct r600_pipe_shader *)state;
+       struct r600_pipe_shader_selector *sel = (struct r600_pipe_shader_selector *)state;
 
-       if (rctx->vs_shader == shader) {
+       if (rctx->vs_shader == sel) {
                rctx->vs_shader = NULL;
        }
 
-       free(shader->tokens);
-       r600_pipe_shader_destroy(ctx, shader);
-       free(shader);
+       r600_delete_shader_selector(ctx, sel);
 }
 
 static void r600_update_alpha_ref(struct r600_context *rctx)
@@ -527,98 +651,86 @@ static void r600_update_alpha_ref(struct r600_context *rctx)
 
        alpha_ref = rctx->alpha_ref;
        rstate.nregs = 0;
-       if (rctx->export_16bpc)
+       if (rctx->export_16bpc && rctx->chip_class >= EVERGREEN) {
                alpha_ref &= ~0x1FFF;
-       r600_pipe_state_add_reg(&rstate, R_028438_SX_ALPHA_REF, alpha_ref, NULL, 0);
+       }
+       r600_pipe_state_add_reg(&rstate, R_028438_SX_ALPHA_REF, alpha_ref);
 
        r600_context_pipe_state_set(rctx, &rstate);
        rctx->alpha_ref_dirty = false;
 }
 
+void r600_constant_buffers_dirty(struct r600_context *rctx, struct r600_constbuf_state *state)
+{
+       r600_inval_shader_cache(rctx);
+       state->atom.num_dw = rctx->chip_class >= EVERGREEN ? util_bitcount(state->dirty_mask)*20
+                                                          : util_bitcount(state->dirty_mask)*19;
+       r600_atom_dirty(rctx, &state->atom);
+}
+
 void r600_set_constant_buffer(struct pipe_context *ctx, uint shader, uint index,
-                             struct pipe_resource *buffer)
+                             struct pipe_constant_buffer *input)
 {
        struct r600_context *rctx = (struct r600_context *)ctx;
-       struct r600_resource *rbuffer = r600_resource(buffer);
-       struct r600_pipe_resource_state *rstate;
-       uint64_t va_offset;
-       uint32_t offset;
+       struct r600_constbuf_state *state;
+       struct pipe_constant_buffer *cb;
+       const uint8_t *ptr;
+
+       switch (shader) {
+       case PIPE_SHADER_VERTEX:
+               state = &rctx->vs_constbuf_state;
+               break;
+       case PIPE_SHADER_FRAGMENT:
+               state = &rctx->ps_constbuf_state;
+               break;
+       default:
+               return;
+       }
 
        /* Note that the state tracker can unbind constant buffers by
         * passing NULL here.
         */
-       if (buffer == NULL) {
+       if (unlikely(!input)) {
+               state->enabled_mask &= ~(1 << index);
+               state->dirty_mask &= ~(1 << index);
+               pipe_resource_reference(&state->cb[index].buffer, NULL);
                return;
        }
 
-       r600_inval_shader_cache(rctx);
+       cb = &state->cb[index];
+       cb->buffer_size = input->buffer_size;
 
-       r600_upload_const_buffer(rctx, &rbuffer, &offset);
-       va_offset = r600_resource_va(ctx->screen, (void*)rbuffer);
-       va_offset += offset;
-       va_offset >>= 8;
+       ptr = input->user_buffer;
 
-       switch (shader) {
-       case PIPE_SHADER_VERTEX:
-               rctx->vs_const_buffer.nregs = 0;
-               r600_pipe_state_add_reg(&rctx->vs_const_buffer,
-                                       R_028180_ALU_CONST_BUFFER_SIZE_VS_0 + index * 4,
-                                       ALIGN_DIVUP(buffer->width0 >> 4, 16),
-                                       NULL, 0);
-               r600_pipe_state_add_reg(&rctx->vs_const_buffer,
-                                       R_028980_ALU_CONST_CACHE_VS_0 + index * 4,
-                                       va_offset, rbuffer, RADEON_USAGE_READ);
-               r600_context_pipe_state_set(rctx, &rctx->vs_const_buffer);
-
-               rstate = &rctx->vs_const_buffer_resource[index];
-               if (!rstate->id) {
-                       if (rctx->chip_class >= EVERGREEN) {
-                               evergreen_pipe_init_buffer_resource(rctx, rstate);
-                       } else {
-                               r600_pipe_init_buffer_resource(rctx, rstate);
+       if (ptr) {
+               /* Upload the user buffer. */
+               if (R600_BIG_ENDIAN) {
+                       uint32_t *tmpPtr;
+                       unsigned i, size = input->buffer_size;
+
+                       if (!(tmpPtr = malloc(size))) {
+                               R600_ERR("Failed to allocate BE swap buffer.\n");
+                               return;
                        }
-               }
 
-               if (rctx->chip_class >= EVERGREEN) {
-                       evergreen_pipe_mod_buffer_resource(ctx, rstate, rbuffer, offset, 16, RADEON_USAGE_READ);
-               } else {
-                       r600_pipe_mod_buffer_resource(rstate, rbuffer, offset, 16, RADEON_USAGE_READ);
-               }
-               r600_context_pipe_state_set_vs_resource(rctx, rstate, index);
-               break;
-       case PIPE_SHADER_FRAGMENT:
-               rctx->ps_const_buffer.nregs = 0;
-               r600_pipe_state_add_reg(&rctx->ps_const_buffer,
-                                       R_028140_ALU_CONST_BUFFER_SIZE_PS_0,
-                                       ALIGN_DIVUP(buffer->width0 >> 4, 16),
-                                       NULL, 0);
-               r600_pipe_state_add_reg(&rctx->ps_const_buffer,
-                                       R_028940_ALU_CONST_CACHE_PS_0,
-                                       va_offset, rbuffer, RADEON_USAGE_READ);
-               r600_context_pipe_state_set(rctx, &rctx->ps_const_buffer);
-
-               rstate = &rctx->ps_const_buffer_resource[index];
-               if (!rstate->id) {
-                       if (rctx->chip_class >= EVERGREEN) {
-                               evergreen_pipe_init_buffer_resource(rctx, rstate);
-                       } else {
-                               r600_pipe_init_buffer_resource(rctx, rstate);
+                       for (i = 0; i < size / 4; ++i) {
+                               tmpPtr[i] = bswap_32(((uint32_t *)ptr)[i]);
                        }
-               }
-               if (rctx->chip_class >= EVERGREEN) {
-                       evergreen_pipe_mod_buffer_resource(ctx, rstate, rbuffer, offset, 16, RADEON_USAGE_READ);
+
+                       u_upload_data(rctx->uploader, 0, size, tmpPtr, &cb->buffer_offset, &cb->buffer);
+                       free(tmpPtr);
                } else {
-                       r600_pipe_mod_buffer_resource(rstate, rbuffer, offset, 16, RADEON_USAGE_READ);
+                       u_upload_data(rctx->uploader, 0, input->buffer_size, ptr, &cb->buffer_offset, &cb->buffer);
                }
-               r600_context_pipe_state_set_ps_resource(rctx, rstate, index);
-               break;
-       default:
-               R600_ERR("unsupported %d\n", shader);
-               return;
+       } else {
+               /* Setup the hw buffer. */
+               cb->buffer_offset = input->buffer_offset;
+               pipe_resource_reference(&cb->buffer, input->buffer);
        }
 
-       if (buffer != &rbuffer->b.b.b)
-               pipe_resource_reference((struct pipe_resource**)&rbuffer, NULL);
+       state->enabled_mask |= 1 << index;
+       state->dirty_mask |= 1 << index;
+       r600_constant_buffers_dirty(rctx, state);
 }
 
 struct pipe_stream_output_target *
@@ -644,9 +756,9 @@ r600_create_so_target(struct pipe_context *ctx,
 
        t->filled_size = (struct r600_resource*)
                pipe_buffer_create(ctx->screen, PIPE_BIND_CUSTOM, PIPE_USAGE_STATIC, 4);
-       ptr = rctx->ws->buffer_map(t->filled_size->buf, rctx->cs, PIPE_TRANSFER_WRITE);
+       ptr = rctx->ws->buffer_map(t->filled_size->cs_buf, rctx->cs, PIPE_TRANSFER_WRITE);
        memset(ptr, 0, t->filled_size->buf->size);
-       rctx->ws->buffer_unmap(t->filled_size->buf);
+       rctx->ws->buffer_unmap(t->filled_size->cs_buf);
 
        return &t->b;
 }
@@ -669,7 +781,7 @@ void r600_set_so_targets(struct pipe_context *ctx,
        unsigned i;
 
        /* Stop streamout. */
-       if (rctx->num_so_targets) {
+       if (rctx->num_so_targets && !rctx->streamout_start) {
                r600_context_streamout_end(rctx);
        }
 
@@ -686,84 +798,10 @@ void r600_set_so_targets(struct pipe_context *ctx,
        rctx->streamout_append_bitmask = append_bitmask;
 }
 
-static void r600_vertex_buffer_update(struct r600_context *rctx)
-{
-       struct r600_pipe_resource_state *rstate;
-       struct r600_resource *rbuffer;
-       struct pipe_vertex_buffer *vertex_buffer;
-       unsigned i, count, offset;
-
-       r600_inval_vertex_cache(rctx);
-
-       if (rctx->vertex_elements->vbuffer_need_offset) {
-               /* one resource per vertex elements */
-               count = rctx->vertex_elements->count;
-       } else {
-               /* bind vertex buffer once */
-               count = rctx->vbuf_mgr->nr_real_vertex_buffers;
-       }
-
-       for (i = 0 ; i < count; i++) {
-               rstate = &rctx->fs_resource[i];
-
-               if (rctx->vertex_elements->vbuffer_need_offset) {
-                       /* one resource per vertex elements */
-                       unsigned vbuffer_index;
-                       vbuffer_index = rctx->vertex_elements->elements[i].vertex_buffer_index;
-                       vertex_buffer = &rctx->vbuf_mgr->real_vertex_buffer[vbuffer_index];
-                       rbuffer = (struct r600_resource*)vertex_buffer->buffer;
-                       offset = rctx->vertex_elements->vbuffer_offset[i];
-               } else {
-                       /* bind vertex buffer once */
-                       vertex_buffer = &rctx->vbuf_mgr->real_vertex_buffer[i];
-                       rbuffer = (struct r600_resource*)vertex_buffer->buffer;
-                       offset = 0;
-               }
-               if (vertex_buffer == NULL || rbuffer == NULL)
-                       continue;
-               offset += vertex_buffer->buffer_offset;
-
-               if (!rstate->id) {
-                       if (rctx->chip_class >= EVERGREEN) {
-                               evergreen_pipe_init_buffer_resource(rctx, rstate);
-                       } else {
-                               r600_pipe_init_buffer_resource(rctx, rstate);
-                       }
-               }
-
-               if (rctx->chip_class >= EVERGREEN) {
-                       evergreen_pipe_mod_buffer_resource(&rctx->context, rstate, rbuffer, offset, vertex_buffer->stride, RADEON_USAGE_READ);
-               } else {
-                       r600_pipe_mod_buffer_resource(rstate, rbuffer, offset, vertex_buffer->stride, RADEON_USAGE_READ);
-               }
-               r600_context_pipe_state_set_fs_resource(rctx, rstate, i);
-       }
-}
-
-static int r600_shader_rebuild(struct pipe_context * ctx, struct r600_pipe_shader * shader)
-{
-       struct r600_context *rctx = (struct r600_context *)ctx;
-       int r;
-
-       r600_pipe_shader_destroy(ctx, shader);
-       r = r600_pipe_shader_create(ctx, shader);
-       if (r) {
-               return r;
-       }
-       r600_context_pipe_state_set(rctx, &shader->rstate);
-
-       return 0;
-}
-
 static void r600_update_derived_state(struct r600_context *rctx)
 {
        struct pipe_context * ctx = (struct pipe_context*)rctx;
-       struct r600_pipe_state rstate;
-
-       rstate.nregs = 0;
-
-       if (rstate.nregs)
-               r600_context_pipe_state_set(rctx, &rstate);
+       unsigned ps_dirty = 0;
 
        if (!rctx->blitter->running) {
                if (rctx->have_depth_fb || rctx->have_depth_texture)
@@ -774,28 +812,55 @@ static void r600_update_derived_state(struct r600_context *rctx)
                r600_update_sampler_states(rctx);
        }
 
-       if ((rctx->ps_shader->shader.two_side != rctx->two_side) ||
-           ((rctx->chip_class >= EVERGREEN) && rctx->ps_shader->shader.fs_write_all &&
-            (rctx->ps_shader->shader.nr_cbufs != rctx->nr_cbufs))) {
-               r600_shader_rebuild(&rctx->context, rctx->ps_shader);
-       }
+       r600_shader_select(ctx, rctx->ps_shader, &ps_dirty);
 
        if (rctx->alpha_ref_dirty) {
                r600_update_alpha_ref(rctx);
        }
 
        if (rctx->ps_shader && ((rctx->sprite_coord_enable &&
-               (rctx->ps_shader->sprite_coord_enable != rctx->sprite_coord_enable)) ||
-               (rctx->rasterizer && rctx->rasterizer->flatshade != rctx->ps_shader->flatshade))) {
+               (rctx->ps_shader->current->sprite_coord_enable != rctx->sprite_coord_enable)) ||
+               (rctx->rasterizer && rctx->rasterizer->flatshade != rctx->ps_shader->current->flatshade))) {
 
                if (rctx->chip_class >= EVERGREEN)
-                       evergreen_pipe_shader_ps(ctx, rctx->ps_shader);
+                       evergreen_pipe_shader_ps(ctx, rctx->ps_shader->current);
                else
-                       r600_pipe_shader_ps(ctx, rctx->ps_shader);
+                       r600_pipe_shader_ps(ctx, rctx->ps_shader->current);
 
-               r600_context_pipe_state_set(rctx, &rctx->ps_shader->rstate);
+               ps_dirty = 1;
        }
 
+       if (ps_dirty)
+               r600_context_pipe_state_set(rctx, &rctx->ps_shader->current->rstate);
+               
+       if (rctx->chip_class >= EVERGREEN) {
+               evergreen_update_dual_export_state(rctx);
+       } else {
+               r600_update_dual_export_state(rctx);
+       }
+}
+
+static unsigned r600_conv_prim_to_gs_out(unsigned mode)
+{
+       static const int prim_conv[] = {
+               V_028A6C_OUTPRIM_TYPE_POINTLIST,
+               V_028A6C_OUTPRIM_TYPE_LINESTRIP,
+               V_028A6C_OUTPRIM_TYPE_LINESTRIP,
+               V_028A6C_OUTPRIM_TYPE_LINESTRIP,
+               V_028A6C_OUTPRIM_TYPE_TRISTRIP,
+               V_028A6C_OUTPRIM_TYPE_TRISTRIP,
+               V_028A6C_OUTPRIM_TYPE_TRISTRIP,
+               V_028A6C_OUTPRIM_TYPE_TRISTRIP,
+               V_028A6C_OUTPRIM_TYPE_TRISTRIP,
+               V_028A6C_OUTPRIM_TYPE_TRISTRIP,
+               V_028A6C_OUTPRIM_TYPE_LINESTRIP,
+               V_028A6C_OUTPRIM_TYPE_LINESTRIP,
+               V_028A6C_OUTPRIM_TYPE_TRISTRIP,
+               V_028A6C_OUTPRIM_TYPE_TRISTRIP
+       };
+       assert(mode < Elements(prim_conv));
+
+       return prim_conv[mode];
 }
 
 void r600_draw_vbo(struct pipe_context *ctx, const struct pipe_draw_info *dinfo)
@@ -803,14 +868,14 @@ void r600_draw_vbo(struct pipe_context *ctx, const struct pipe_draw_info *dinfo)
        struct r600_context *rctx = (struct r600_context *)ctx;
        struct pipe_draw_info info = *dinfo;
        struct pipe_index_buffer ib = {};
-       unsigned prim, mask, ls_mask = 0;
+       unsigned prim, ls_mask = 0;
        struct r600_block *dirty_block = NULL, *next_block = NULL;
        struct r600_atom *state = NULL, *next_state = NULL;
        struct radeon_winsys_cs *cs = rctx->cs;
        uint64_t va;
+       uint8_t *ptr;
 
        if ((!info.count && (info.indexed || !info.count_from_stream_output)) ||
-           (info.indexed && !rctx->vbuf_mgr->index_buffer.buffer) ||
            !r600_conv_pipe_prim(info.mode, &prim)) {
                assert(0);
                return;
@@ -823,20 +888,20 @@ void r600_draw_vbo(struct pipe_context *ctx, const struct pipe_draw_info *dinfo)
 
        r600_update_derived_state(rctx);
 
-       u_vbuf_draw_begin(rctx->vbuf_mgr, &info);
-       r600_vertex_buffer_update(rctx);
-
        if (info.indexed) {
                /* Initialize the index buffer struct. */
-               pipe_resource_reference(&ib.buffer, rctx->vbuf_mgr->index_buffer.buffer);
-               ib.index_size = rctx->vbuf_mgr->index_buffer.index_size;
-               ib.offset = rctx->vbuf_mgr->index_buffer.offset + info.start * ib.index_size;
+               pipe_resource_reference(&ib.buffer, rctx->index_buffer.buffer);
+               ib.user_buffer = rctx->index_buffer.user_buffer;
+               ib.index_size = rctx->index_buffer.index_size;
+               ib.offset = rctx->index_buffer.offset + info.start * ib.index_size;
 
                /* Translate or upload, if needed. */
                r600_translate_index_buffer(rctx, &ib, info.count);
 
-               if (u_vbuf_resource(ib.buffer)->user_ptr) {
-                       r600_upload_index_buffer(rctx, &ib, info.count);
+               ptr = (uint8_t*)ib.user_buffer;
+               if (!ib.buffer && ptr) {
+                       u_upload_data(rctx->uploader, 0, info.count * ib.index_size,
+                                     ptr, &ib.offset, &ib.buffer);
                }
        } else {
                info.index_bias = info.start;
@@ -845,29 +910,27 @@ void r600_draw_vbo(struct pipe_context *ctx, const struct pipe_draw_info *dinfo)
                }
        }
 
-       mask = (1ULL << ((unsigned)rctx->framebuffer.nr_cbufs * 4)) - 1;
-
        if (rctx->vgt.id != R600_PIPE_STATE_VGT) {
                rctx->vgt.id = R600_PIPE_STATE_VGT;
                rctx->vgt.nregs = 0;
-               r600_pipe_state_add_reg(&rctx->vgt, R_008958_VGT_PRIMITIVE_TYPE, prim, NULL, 0);
-               r600_pipe_state_add_reg(&rctx->vgt, R_028238_CB_TARGET_MASK, rctx->cb_target_mask & mask, NULL, 0);
-               r600_pipe_state_add_reg(&rctx->vgt, R_028408_VGT_INDX_OFFSET, info.index_bias, NULL, 0);
-               r600_pipe_state_add_reg(&rctx->vgt, R_02840C_VGT_MULTI_PRIM_IB_RESET_INDX, info.restart_index, NULL, 0);
-               r600_pipe_state_add_reg(&rctx->vgt, R_028A94_VGT_MULTI_PRIM_IB_RESET_EN, info.primitive_restart, NULL, 0);
-               r600_pipe_state_add_reg(&rctx->vgt, R_03CFF4_SQ_VTX_START_INST_LOC, info.start_instance, NULL, 0);
-               r600_pipe_state_add_reg(&rctx->vgt, R_028A0C_PA_SC_LINE_STIPPLE, 0, NULL, 0);
-               if (rctx->chip_class <= R700)
-                       r600_pipe_state_add_reg(&rctx->vgt, R_028808_CB_COLOR_CONTROL, rctx->cb_color_control, NULL, 0);
-               r600_pipe_state_add_reg(&rctx->vgt, R_02881C_PA_CL_VS_OUT_CNTL, 0, NULL, 0);
-               r600_pipe_state_add_reg(&rctx->vgt, R_028810_PA_CL_CLIP_CNTL, 0, NULL, 0);
+               r600_pipe_state_add_reg(&rctx->vgt, R_008958_VGT_PRIMITIVE_TYPE, prim);
+               r600_pipe_state_add_reg(&rctx->vgt, R_028A6C_VGT_GS_OUT_PRIM_TYPE, 0);
+               r600_pipe_state_add_reg(&rctx->vgt, R_028408_VGT_INDX_OFFSET, info.index_bias);
+               r600_pipe_state_add_reg(&rctx->vgt, R_02840C_VGT_MULTI_PRIM_IB_RESET_INDX, info.restart_index);
+               r600_pipe_state_add_reg(&rctx->vgt, R_028410_SX_ALPHA_TEST_CONTROL, 0);
+               r600_pipe_state_add_reg(&rctx->vgt, R_028A94_VGT_MULTI_PRIM_IB_RESET_EN, info.primitive_restart);
+               r600_pipe_state_add_reg(&rctx->vgt, R_03CFF4_SQ_VTX_START_INST_LOC, info.start_instance);
+               r600_pipe_state_add_reg(&rctx->vgt, R_028A0C_PA_SC_LINE_STIPPLE, 0);
+               r600_pipe_state_add_reg(&rctx->vgt, R_02881C_PA_CL_VS_OUT_CNTL, 0);
+               r600_pipe_state_add_reg(&rctx->vgt, R_028810_PA_CL_CLIP_CNTL, 0);
        }
 
        rctx->vgt.nregs = 0;
        r600_pipe_state_mod_reg(&rctx->vgt, prim);
-       r600_pipe_state_mod_reg(&rctx->vgt, rctx->cb_target_mask & mask);
+       r600_pipe_state_mod_reg(&rctx->vgt, r600_conv_prim_to_gs_out(info.mode));
        r600_pipe_state_mod_reg(&rctx->vgt, info.index_bias);
        r600_pipe_state_mod_reg(&rctx->vgt, info.restart_index);
+       r600_pipe_state_mod_reg(&rctx->vgt, rctx->sx_alpha_test_control);
        r600_pipe_state_mod_reg(&rctx->vgt, info.primitive_restart);
        r600_pipe_state_mod_reg(&rctx->vgt, info.start_instance);
 
@@ -876,15 +939,13 @@ void r600_draw_vbo(struct pipe_context *ctx, const struct pipe_draw_info *dinfo)
        else if (prim == V_008958_DI_PT_LINESTRIP) 
                ls_mask = 2;
        r600_pipe_state_mod_reg(&rctx->vgt, S_028A0C_AUTO_RESET_CNTL(ls_mask) | rctx->pa_sc_line_stipple);
-       if (rctx->chip_class <= R700)
-               r600_pipe_state_mod_reg(&rctx->vgt, rctx->cb_color_control);
        r600_pipe_state_mod_reg(&rctx->vgt,
-                               rctx->vs_shader->pa_cl_vs_out_cntl |
-                               (rctx->rasterizer->clip_plane_enable & rctx->vs_shader->shader.clip_dist_write));
+                               rctx->vs_shader->current->pa_cl_vs_out_cntl |
+                               (rctx->rasterizer->clip_plane_enable & rctx->vs_shader->current->shader.clip_dist_write));
        r600_pipe_state_mod_reg(&rctx->vgt,
                                rctx->pa_cl_clip_cntl |
-                               (rctx->vs_shader->shader.clip_dist_write ||
-                                rctx->vs_shader->shader.vs_prohibit_ucps ?
+                               (rctx->vs_shader->current->shader.clip_dist_write ||
+                                rctx->vs_shader->current->shader.vs_prohibit_ucps ?
                                 0 : rctx->rasterizer->clip_plane_enable & 0x3F));
 
        r600_context_pipe_state_set(rctx, &rctx->vgt);
@@ -896,7 +957,7 @@ void r600_draw_vbo(struct pipe_context *ctx, const struct pipe_draw_info *dinfo)
                r600_emit_atom(rctx, state);
        }
        LIST_FOR_EACH_ENTRY_SAFE(dirty_block, next_block, &rctx->dirty,list) {
-               r600_context_block_emit_dirty(rctx, dirty_block);
+               r600_context_block_emit_dirty(rctx, dirty_block, 0 /* pkt_flags */);
        }
        LIST_FOR_EACH_ENTRY_SAFE(dirty_block, next_block, &rctx->resource_dirty,list) {
                r600_context_block_resource_emit_dirty(rctx, dirty_block);
@@ -935,22 +996,24 @@ void r600_draw_vbo(struct pipe_context *ctx, const struct pipe_draw_info *dinfo)
 
        rctx->flags |= R600_CONTEXT_DST_CACHES_DIRTY | R600_CONTEXT_DRAW_PENDING;
 
-       if (rctx->framebuffer.zsbuf)
-       {
-               struct pipe_resource *tex = rctx->framebuffer.zsbuf->texture;
-               ((struct r600_resource_texture *)tex)->dirty_db = TRUE;
+       /* Set the depth buffer as dirty. */
+       if (rctx->framebuffer.zsbuf) {
+               struct pipe_surface *surf = rctx->framebuffer.zsbuf;
+               struct r600_resource_texture *rtex = (struct r600_resource_texture *)surf->texture;
+
+               rtex->dirty_db_mask |= 1 << surf->u.tex.level;
        }
 
        pipe_resource_reference(&ib.buffer, NULL);
-       u_vbuf_draw_end(rctx->vbuf_mgr);
 }
 
-void _r600_pipe_state_add_reg(struct r600_context *ctx,
-                             struct r600_pipe_state *state,
-                             uint32_t offset, uint32_t value,
-                             uint32_t range_id, uint32_t block_id,
-                             struct r600_resource *bo,
-                             enum radeon_bo_usage usage)
+void _r600_pipe_state_add_reg_bo(struct r600_context *ctx,
+                                struct r600_pipe_state *state,
+                                uint32_t offset, uint32_t value,
+                                uint32_t range_id, uint32_t block_id,
+                                struct r600_resource *bo,
+                                enum radeon_bo_usage usage)
+                             
 {
        struct r600_range *range;
        struct r600_block *block;
@@ -970,6 +1033,15 @@ void _r600_pipe_state_add_reg(struct r600_context *ctx,
        assert(state->nregs < R600_BLOCK_MAX_REG);
 }
 
+void _r600_pipe_state_add_reg(struct r600_context *ctx,
+                             struct r600_pipe_state *state,
+                             uint32_t offset, uint32_t value,
+                             uint32_t range_id, uint32_t block_id)
+{
+       _r600_pipe_state_add_reg_bo(ctx, state, offset, value,
+                                   range_id, block_id, NULL, 0);
+}
+
 void r600_pipe_state_add_reg_noblock(struct r600_pipe_state *state,
                                     uint32_t offset, uint32_t value,
                                     struct r600_resource *bo,