| /************************************************************************** |
| * |
| * Copyright 2010 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. |
| * |
| **************************************************************************/ |
| |
| #include "draw_llvm.h" |
| |
| #include "draw_context.h" |
| #include "draw_vs.h" |
| |
| #include "gallivm/lp_bld_arit.h" |
| #include "gallivm/lp_bld_logic.h" |
| #include "gallivm/lp_bld_const.h" |
| #include "gallivm/lp_bld_swizzle.h" |
| #include "gallivm/lp_bld_struct.h" |
| #include "gallivm/lp_bld_type.h" |
| #include "gallivm/lp_bld_flow.h" |
| #include "gallivm/lp_bld_debug.h" |
| #include "gallivm/lp_bld_tgsi.h" |
| #include "gallivm/lp_bld_printf.h" |
| #include "gallivm/lp_bld_intr.h" |
| #include "gallivm/lp_bld_init.h" |
| #include "gallivm/lp_bld_type.h" |
| #include "gallivm/lp_bld_pack.h" |
| #include "gallivm/lp_bld_format.h" |
| |
| #include "tgsi/tgsi_exec.h" |
| #include "tgsi/tgsi_dump.h" |
| |
| #include "util/u_math.h" |
| #include "util/u_pointer.h" |
| #include "util/u_string.h" |
| #include "util/u_simple_list.h" |
| |
| |
| #define DEBUG_STORE 0 |
| |
| |
| static void |
| draw_llvm_generate(struct draw_llvm *llvm, struct draw_llvm_variant *var, |
| boolean elts); |
| |
| |
| /** |
| * Create LLVM type for struct draw_jit_texture |
| */ |
| static LLVMTypeRef |
| create_jit_texture_type(struct gallivm_state *gallivm, const char *struct_name) |
| { |
| LLVMTargetDataRef target = gallivm->target; |
| LLVMTypeRef texture_type; |
| LLVMTypeRef elem_types[DRAW_JIT_TEXTURE_NUM_FIELDS]; |
| LLVMTypeRef int32_type = LLVMInt32TypeInContext(gallivm->context); |
| |
| elem_types[DRAW_JIT_TEXTURE_WIDTH] = |
| elem_types[DRAW_JIT_TEXTURE_HEIGHT] = |
| elem_types[DRAW_JIT_TEXTURE_DEPTH] = |
| elem_types[DRAW_JIT_TEXTURE_FIRST_LEVEL] = |
| elem_types[DRAW_JIT_TEXTURE_LAST_LEVEL] = int32_type; |
| elem_types[DRAW_JIT_TEXTURE_ROW_STRIDE] = |
| elem_types[DRAW_JIT_TEXTURE_IMG_STRIDE] = |
| LLVMArrayType(int32_type, PIPE_MAX_TEXTURE_LEVELS); |
| elem_types[DRAW_JIT_TEXTURE_DATA] = |
| LLVMArrayType(LLVMPointerType(LLVMInt8TypeInContext(gallivm->context), 0), |
| PIPE_MAX_TEXTURE_LEVELS); |
| elem_types[DRAW_JIT_TEXTURE_MIN_LOD] = |
| elem_types[DRAW_JIT_TEXTURE_MAX_LOD] = |
| elem_types[DRAW_JIT_TEXTURE_LOD_BIAS] = LLVMFloatTypeInContext(gallivm->context); |
| elem_types[DRAW_JIT_TEXTURE_BORDER_COLOR] = |
| LLVMArrayType(LLVMFloatTypeInContext(gallivm->context), 4); |
| |
| texture_type = LLVMStructTypeInContext(gallivm->context, elem_types, |
| Elements(elem_types), 0); |
| |
| #if HAVE_LLVM < 0x0300 |
| LLVMAddTypeName(gallivm->module, struct_name, texture_type); |
| |
| /* Make sure the target's struct layout cache doesn't return |
| * stale/invalid data. |
| */ |
| LLVMInvalidateStructLayout(gallivm->target, texture_type); |
| #endif |
| |
| LP_CHECK_MEMBER_OFFSET(struct draw_jit_texture, width, |
| target, texture_type, |
| DRAW_JIT_TEXTURE_WIDTH); |
| LP_CHECK_MEMBER_OFFSET(struct draw_jit_texture, height, |
| target, texture_type, |
| DRAW_JIT_TEXTURE_HEIGHT); |
| LP_CHECK_MEMBER_OFFSET(struct draw_jit_texture, depth, |
| target, texture_type, |
| DRAW_JIT_TEXTURE_DEPTH); |
| LP_CHECK_MEMBER_OFFSET(struct draw_jit_texture, first_level, |
| target, texture_type, |
| DRAW_JIT_TEXTURE_FIRST_LEVEL); |
| LP_CHECK_MEMBER_OFFSET(struct draw_jit_texture, last_level, |
| target, texture_type, |
| DRAW_JIT_TEXTURE_LAST_LEVEL); |
| LP_CHECK_MEMBER_OFFSET(struct draw_jit_texture, row_stride, |
| target, texture_type, |
| DRAW_JIT_TEXTURE_ROW_STRIDE); |
| LP_CHECK_MEMBER_OFFSET(struct draw_jit_texture, img_stride, |
| target, texture_type, |
| DRAW_JIT_TEXTURE_IMG_STRIDE); |
| LP_CHECK_MEMBER_OFFSET(struct draw_jit_texture, data, |
| target, texture_type, |
| DRAW_JIT_TEXTURE_DATA); |
| LP_CHECK_MEMBER_OFFSET(struct draw_jit_texture, min_lod, |
| target, texture_type, |
| DRAW_JIT_TEXTURE_MIN_LOD); |
| LP_CHECK_MEMBER_OFFSET(struct draw_jit_texture, max_lod, |
| target, texture_type, |
| DRAW_JIT_TEXTURE_MAX_LOD); |
| LP_CHECK_MEMBER_OFFSET(struct draw_jit_texture, lod_bias, |
| target, texture_type, |
| DRAW_JIT_TEXTURE_LOD_BIAS); |
| LP_CHECK_MEMBER_OFFSET(struct draw_jit_texture, border_color, |
| target, texture_type, |
| DRAW_JIT_TEXTURE_BORDER_COLOR); |
| |
| LP_CHECK_STRUCT_SIZE(struct draw_jit_texture, target, texture_type); |
| |
| return texture_type; |
| } |
| |
| |
| /** |
| * Create LLVM type for struct draw_jit_texture |
| */ |
| static LLVMTypeRef |
| create_jit_context_type(struct gallivm_state *gallivm, |
| LLVMTypeRef texture_type, const char *struct_name) |
| { |
| LLVMTargetDataRef target = gallivm->target; |
| LLVMTypeRef float_type = LLVMFloatTypeInContext(gallivm->context); |
| LLVMTypeRef elem_types[5]; |
| LLVMTypeRef context_type; |
| |
| elem_types[0] = LLVMPointerType(float_type, 0); /* vs_constants */ |
| elem_types[1] = LLVMPointerType(float_type, 0); /* gs_constants */ |
| elem_types[2] = LLVMPointerType(LLVMArrayType(LLVMArrayType(float_type, 4), |
| DRAW_TOTAL_CLIP_PLANES), 0); |
| elem_types[3] = LLVMPointerType(float_type, 0); /* viewport */ |
| elem_types[4] = LLVMArrayType(texture_type, |
| PIPE_MAX_SAMPLERS); /* textures */ |
| context_type = LLVMStructTypeInContext(gallivm->context, elem_types, |
| Elements(elem_types), 0); |
| #if HAVE_LLVM < 0x0300 |
| LLVMAddTypeName(gallivm->module, struct_name, context_type); |
| |
| LLVMInvalidateStructLayout(gallivm->target, context_type); |
| #endif |
| |
| LP_CHECK_MEMBER_OFFSET(struct draw_jit_context, vs_constants, |
| target, context_type, 0); |
| LP_CHECK_MEMBER_OFFSET(struct draw_jit_context, gs_constants, |
| target, context_type, 1); |
| LP_CHECK_MEMBER_OFFSET(struct draw_jit_context, planes, |
| target, context_type, 2); |
| LP_CHECK_MEMBER_OFFSET(struct draw_jit_context, textures, |
| target, context_type, |
| DRAW_JIT_CTX_TEXTURES); |
| LP_CHECK_STRUCT_SIZE(struct draw_jit_context, |
| target, context_type); |
| |
| return context_type; |
| } |
| |
| |
| /** |
| * Create LLVM type for struct pipe_vertex_buffer |
| */ |
| static LLVMTypeRef |
| create_jit_vertex_buffer_type(struct gallivm_state *gallivm, const char *struct_name) |
| { |
| LLVMTargetDataRef target = gallivm->target; |
| LLVMTypeRef elem_types[4]; |
| LLVMTypeRef vb_type; |
| |
| elem_types[0] = |
| elem_types[1] = LLVMInt32TypeInContext(gallivm->context); |
| elem_types[2] = |
| elem_types[3] = LLVMPointerType(LLVMInt8TypeInContext(gallivm->context), 0); /* vs_constants */ |
| |
| vb_type = LLVMStructTypeInContext(gallivm->context, elem_types, |
| Elements(elem_types), 0); |
| #if HAVE_LLVM < 0x0300 |
| LLVMAddTypeName(gallivm->module, struct_name, vb_type); |
| |
| LLVMInvalidateStructLayout(gallivm->target, vb_type); |
| #endif |
| |
| LP_CHECK_MEMBER_OFFSET(struct pipe_vertex_buffer, stride, |
| target, vb_type, 0); |
| LP_CHECK_MEMBER_OFFSET(struct pipe_vertex_buffer, buffer_offset, |
| target, vb_type, 1); |
| |
| LP_CHECK_STRUCT_SIZE(struct pipe_vertex_buffer, target, vb_type); |
| |
| return vb_type; |
| } |
| |
| |
| /** |
| * Create LLVM type for struct vertex_header; |
| */ |
| static LLVMTypeRef |
| create_jit_vertex_header(struct gallivm_state *gallivm, int data_elems) |
| { |
| LLVMTargetDataRef target = gallivm->target; |
| LLVMTypeRef elem_types[4]; |
| LLVMTypeRef vertex_header; |
| char struct_name[24]; |
| |
| util_snprintf(struct_name, 23, "vertex_header%d", data_elems); |
| |
| elem_types[DRAW_JIT_VERTEX_VERTEX_ID] = LLVMIntTypeInContext(gallivm->context, 32); |
| elem_types[DRAW_JIT_VERTEX_CLIP] = LLVMArrayType(LLVMFloatTypeInContext(gallivm->context), 4); |
| elem_types[DRAW_JIT_VERTEX_PRE_CLIP_POS] = LLVMArrayType(LLVMFloatTypeInContext(gallivm->context), 4); |
| elem_types[DRAW_JIT_VERTEX_DATA] = LLVMArrayType(elem_types[1], data_elems); |
| |
| vertex_header = LLVMStructTypeInContext(gallivm->context, elem_types, |
| Elements(elem_types), 0); |
| #if HAVE_LLVM < 0x0300 |
| LLVMAddTypeName(gallivm->module, struct_name, vertex_header); |
| |
| LLVMInvalidateStructLayout(gallivm->target, vertex_header); |
| #endif |
| |
| /* these are bit-fields and we can't take address of them |
| LP_CHECK_MEMBER_OFFSET(struct vertex_header, clipmask, |
| target, vertex_header, |
| DRAW_JIT_VERTEX_CLIPMASK); |
| LP_CHECK_MEMBER_OFFSET(struct vertex_header, edgeflag, |
| target, vertex_header, |
| DRAW_JIT_VERTEX_EDGEFLAG); |
| LP_CHECK_MEMBER_OFFSET(struct vertex_header, pad, |
| target, vertex_header, |
| DRAW_JIT_VERTEX_PAD); |
| LP_CHECK_MEMBER_OFFSET(struct vertex_header, vertex_id, |
| target, vertex_header, |
| DRAW_JIT_VERTEX_VERTEX_ID); |
| */ |
| LP_CHECK_MEMBER_OFFSET(struct vertex_header, clip, |
| target, vertex_header, |
| DRAW_JIT_VERTEX_CLIP); |
| LP_CHECK_MEMBER_OFFSET(struct vertex_header, pre_clip_pos, |
| target, vertex_header, |
| DRAW_JIT_VERTEX_PRE_CLIP_POS); |
| LP_CHECK_MEMBER_OFFSET(struct vertex_header, data, |
| target, vertex_header, |
| DRAW_JIT_VERTEX_DATA); |
| |
| assert(LLVMABISizeOfType(target, vertex_header) == |
| offsetof(struct vertex_header, data[data_elems])); |
| |
| return vertex_header; |
| } |
| |
| |
| /** |
| * Create LLVM types for various structures. |
| */ |
| static void |
| create_jit_types(struct draw_llvm_variant *variant) |
| { |
| struct gallivm_state *gallivm = variant->gallivm; |
| LLVMTypeRef texture_type, context_type, buffer_type, vb_type; |
| |
| texture_type = create_jit_texture_type(gallivm, "texture"); |
| |
| context_type = create_jit_context_type(gallivm, texture_type, "draw_jit_context"); |
| variant->context_ptr_type = LLVMPointerType(context_type, 0); |
| |
| buffer_type = LLVMPointerType(LLVMIntTypeInContext(gallivm->context, 8), 0); |
| variant->buffer_ptr_type = LLVMPointerType(buffer_type, 0); |
| |
| vb_type = create_jit_vertex_buffer_type(gallivm, "pipe_vertex_buffer"); |
| variant->vb_ptr_type = LLVMPointerType(vb_type, 0); |
| } |
| |
| |
| static LLVMTypeRef |
| get_context_ptr_type(struct draw_llvm_variant *variant) |
| { |
| if (!variant->context_ptr_type) |
| create_jit_types(variant); |
| return variant->context_ptr_type; |
| } |
| |
| |
| static LLVMTypeRef |
| get_buffer_ptr_type(struct draw_llvm_variant *variant) |
| { |
| if (!variant->buffer_ptr_type) |
| create_jit_types(variant); |
| return variant->buffer_ptr_type; |
| } |
| |
| |
| static LLVMTypeRef |
| get_vb_ptr_type(struct draw_llvm_variant *variant) |
| { |
| if (!variant->vb_ptr_type) |
| create_jit_types(variant); |
| return variant->vb_ptr_type; |
| } |
| |
| static LLVMTypeRef |
| get_vertex_header_ptr_type(struct draw_llvm_variant *variant) |
| { |
| if (!variant->vertex_header_ptr_type) |
| create_jit_types(variant); |
| return variant->vertex_header_ptr_type; |
| } |
| |
| |
| /** |
| * Create per-context LLVM info. |
| */ |
| struct draw_llvm * |
| draw_llvm_create(struct draw_context *draw) |
| { |
| struct draw_llvm *llvm; |
| |
| llvm = CALLOC_STRUCT( draw_llvm ); |
| if (!llvm) |
| return NULL; |
| |
| lp_build_init(); |
| |
| llvm->draw = draw; |
| |
| llvm->nr_variants = 0; |
| make_empty_list(&llvm->vs_variants_list); |
| |
| return llvm; |
| } |
| |
| |
| /** |
| * Free per-context LLVM info. |
| */ |
| void |
| draw_llvm_destroy(struct draw_llvm *llvm) |
| { |
| /* XXX free other draw_llvm data? */ |
| FREE(llvm); |
| } |
| |
| |
| /** |
| * Create LLVM-generated code for a vertex shader. |
| */ |
| struct draw_llvm_variant * |
| draw_llvm_create_variant(struct draw_llvm *llvm, |
| unsigned num_inputs, |
| const struct draw_llvm_variant_key *key) |
| { |
| struct draw_llvm_variant *variant; |
| struct llvm_vertex_shader *shader = |
| llvm_vertex_shader(llvm->draw->vs.vertex_shader); |
| LLVMTypeRef vertex_header; |
| |
| variant = MALLOC(sizeof *variant + |
| shader->variant_key_size - |
| sizeof variant->key); |
| if (variant == NULL) |
| return NULL; |
| |
| variant->llvm = llvm; |
| |
| variant->gallivm = gallivm_create(); |
| |
| create_jit_types(variant); |
| |
| memcpy(&variant->key, key, shader->variant_key_size); |
| |
| vertex_header = create_jit_vertex_header(variant->gallivm, num_inputs); |
| |
| variant->vertex_header_ptr_type = LLVMPointerType(vertex_header, 0); |
| |
| draw_llvm_generate(llvm, variant, FALSE); /* linear */ |
| draw_llvm_generate(llvm, variant, TRUE); /* elts */ |
| |
| gallivm_compile_module(variant->gallivm); |
| |
| variant->jit_func = (draw_jit_vert_func) |
| gallivm_jit_function(variant->gallivm, variant->function); |
| |
| variant->jit_func_elts = (draw_jit_vert_func_elts) |
| gallivm_jit_function(variant->gallivm, variant->function_elts); |
| |
| variant->shader = shader; |
| variant->list_item_global.base = variant; |
| variant->list_item_local.base = variant; |
| /*variant->no = */shader->variants_created++; |
| variant->list_item_global.base = variant; |
| |
| return variant; |
| } |
| |
| |
| static void |
| generate_vs(struct draw_llvm_variant *variant, |
| LLVMBuilderRef builder, |
| struct lp_type vs_type, |
| LLVMValueRef (*outputs)[TGSI_NUM_CHANNELS], |
| const LLVMValueRef (*inputs)[TGSI_NUM_CHANNELS], |
| const struct lp_bld_tgsi_system_values *system_values, |
| LLVMValueRef context_ptr, |
| struct lp_build_sampler_soa *draw_sampler, |
| boolean clamp_vertex_color) |
| { |
| struct draw_llvm *llvm = variant->llvm; |
| const struct tgsi_token *tokens = llvm->draw->vs.vertex_shader->state.tokens; |
| LLVMValueRef consts_ptr = draw_jit_context_vs_constants(variant->gallivm, context_ptr); |
| struct lp_build_sampler_soa *sampler = 0; |
| |
| if (gallivm_debug & GALLIVM_DEBUG_IR) { |
| tgsi_dump(tokens, 0); |
| } |
| |
| if (llvm->draw->num_sampler_views && llvm->draw->num_samplers) |
| sampler = draw_sampler; |
| |
| lp_build_tgsi_soa(variant->gallivm, |
| tokens, |
| vs_type, |
| NULL /*struct lp_build_mask_context *mask*/, |
| consts_ptr, |
| system_values, |
| NULL /*pos*/, |
| inputs, |
| outputs, |
| sampler, |
| &llvm->draw->vs.vertex_shader->info); |
| |
| { |
| LLVMValueRef out; |
| unsigned chan, attrib; |
| struct lp_build_context bld; |
| struct tgsi_shader_info* info = &llvm->draw->vs.vertex_shader->info; |
| lp_build_context_init(&bld, variant->gallivm, vs_type); |
| |
| for (attrib = 0; attrib < info->num_outputs; ++attrib) { |
| for (chan = 0; chan < TGSI_NUM_CHANNELS; ++chan) { |
| if (outputs[attrib][chan]) { |
| switch (info->output_semantic_name[attrib]) { |
| case TGSI_SEMANTIC_COLOR: |
| case TGSI_SEMANTIC_BCOLOR: |
| if (clamp_vertex_color) { |
| out = LLVMBuildLoad(builder, outputs[attrib][chan], ""); |
| out = lp_build_clamp(&bld, out, bld.zero, bld.one); |
| LLVMBuildStore(builder, out, outputs[attrib][chan]); |
| } |
| break; |
| case TGSI_SEMANTIC_FOG: |
| if (chan == 1 || chan == 2) |
| LLVMBuildStore(builder, bld.zero, outputs[attrib][chan]); |
| else if (chan == 3) |
| LLVMBuildStore(builder, bld.one, outputs[attrib][chan]); |
| break; |
| } |
| } |
| } |
| } |
| } |
| } |
| |
| |
| static void |
| generate_fetch(struct gallivm_state *gallivm, |
| LLVMValueRef vbuffers_ptr, |
| LLVMValueRef *res, |
| struct pipe_vertex_element *velem, |
| LLVMValueRef vbuf, |
| LLVMValueRef index, |
| LLVMValueRef instance_id) |
| { |
| const struct util_format_description *format_desc = util_format_description(velem->src_format); |
| LLVMValueRef zero = LLVMConstNull(LLVMInt32TypeInContext(gallivm->context)); |
| LLVMBuilderRef builder = gallivm->builder; |
| LLVMValueRef indices = |
| LLVMConstInt(LLVMInt64TypeInContext(gallivm->context), |
| velem->vertex_buffer_index, 0); |
| LLVMValueRef vbuffer_ptr = LLVMBuildGEP(builder, vbuffers_ptr, |
| &indices, 1, ""); |
| LLVMValueRef vb_stride = draw_jit_vbuffer_stride(gallivm, vbuf); |
| LLVMValueRef vb_buffer_offset = draw_jit_vbuffer_offset(gallivm, vbuf); |
| LLVMValueRef stride; |
| |
| if (velem->instance_divisor) { |
| /* array index = instance_id / instance_divisor */ |
| index = LLVMBuildUDiv(builder, instance_id, |
| lp_build_const_int32(gallivm, velem->instance_divisor), |
| "instance_divisor"); |
| } |
| |
| stride = LLVMBuildMul(builder, vb_stride, index, ""); |
| |
| vbuffer_ptr = LLVMBuildLoad(builder, vbuffer_ptr, "vbuffer"); |
| |
| stride = LLVMBuildAdd(builder, stride, |
| vb_buffer_offset, |
| ""); |
| stride = LLVMBuildAdd(builder, stride, |
| lp_build_const_int32(gallivm, velem->src_offset), |
| ""); |
| |
| /* lp_build_printf(gallivm, "vbuf index = %d, stride is %d\n", indices, stride);*/ |
| vbuffer_ptr = LLVMBuildGEP(builder, vbuffer_ptr, &stride, 1, ""); |
| |
| *res = lp_build_fetch_rgba_aos(gallivm, |
| format_desc, |
| lp_float32_vec4_type(), |
| vbuffer_ptr, |
| zero, zero, zero); |
| } |
| |
| static void |
| convert_to_soa(struct gallivm_state *gallivm, |
| LLVMValueRef (*src_aos)[LP_MAX_VECTOR_WIDTH / 32], |
| LLVMValueRef (*dst_soa)[TGSI_NUM_CHANNELS], |
| unsigned num_attribs, const struct lp_type soa_type) |
| { |
| unsigned i, j, k; |
| struct lp_type aos_channel_type = soa_type; |
| |
| debug_assert(TGSI_NUM_CHANNELS == 4); |
| debug_assert((soa_type.length % TGSI_NUM_CHANNELS) == 0); |
| |
| aos_channel_type.length >>= 1; |
| |
| for (i = 0; i < num_attribs; ++i) { |
| LLVMValueRef aos_channels[TGSI_NUM_CHANNELS]; |
| unsigned pixels_per_channel = soa_type.length / TGSI_NUM_CHANNELS; |
| |
| for (j = 0; j < TGSI_NUM_CHANNELS; ++j) { |
| LLVMValueRef channel[LP_MAX_VECTOR_LENGTH] = { 0 }; |
| |
| assert(pixels_per_channel <= LP_MAX_VECTOR_LENGTH); |
| |
| for (k = 0; k < pixels_per_channel; ++k) { |
| channel[k] = src_aos[i][j + TGSI_NUM_CHANNELS * k]; |
| } |
| |
| aos_channels[j] = lp_build_concat(gallivm, channel, aos_channel_type, pixels_per_channel); |
| } |
| |
| lp_build_transpose_aos(gallivm, soa_type, aos_channels, dst_soa[i]); |
| } |
| } |
| |
| |
| static void |
| store_aos(struct gallivm_state *gallivm, |
| LLVMValueRef io_ptr, |
| LLVMValueRef index, |
| LLVMValueRef value) |
| { |
| LLVMTypeRef data_ptr_type = LLVMPointerType(lp_build_vec_type(gallivm, lp_float32_vec4_type()), 0); |
| LLVMBuilderRef builder = gallivm->builder; |
| LLVMValueRef data_ptr = draw_jit_header_data(gallivm, io_ptr); |
| LLVMValueRef indices[3]; |
| |
| indices[0] = lp_build_const_int32(gallivm, 0); |
| indices[1] = index; |
| indices[2] = lp_build_const_int32(gallivm, 0); |
| |
| #if DEBUG_STORE |
| lp_build_printf(gallivm, " ---- %p storing attribute %d (io = %p)\n", data_ptr, index, io_ptr); |
| #endif |
| |
| data_ptr = LLVMBuildGEP(builder, data_ptr, indices, 3, ""); |
| data_ptr = LLVMBuildPointerCast(builder, data_ptr, data_ptr_type, ""); |
| |
| /* Unaligned store due to the vertex header */ |
| lp_set_store_alignment(LLVMBuildStore(builder, value, data_ptr), sizeof(float)); |
| } |
| |
| |
| static void |
| store_aos_array(struct gallivm_state *gallivm, |
| struct lp_type soa_type, |
| LLVMValueRef io_ptr, |
| LLVMValueRef* aos, |
| int attrib, |
| int num_outputs, |
| LLVMValueRef clipmask, |
| boolean have_clipdist) |
| { |
| LLVMBuilderRef builder = gallivm->builder; |
| LLVMValueRef attr_index = lp_build_const_int32(gallivm, attrib); |
| LLVMValueRef inds[LP_MAX_VECTOR_WIDTH / 32]; |
| LLVMValueRef io_ptrs[LP_MAX_VECTOR_WIDTH / 32]; |
| int vector_length = soa_type.length; |
| int i; |
| |
| debug_assert(TGSI_NUM_CHANNELS == 4); |
| |
| for (i = 0; i < vector_length; i++) { |
| inds[i] = lp_build_const_int32(gallivm, i); |
| io_ptrs[i] = LLVMBuildGEP(builder, io_ptr, &inds[i], 1, ""); |
| } |
| |
| if (attrib == 0) { |
| /* store vertex header for each of the n vertices */ |
| LLVMValueRef val, cliptmp; |
| int vertex_id_pad_edgeflag; |
| |
| /* If this assertion fails, it means we need to update the bit twidding |
| * code here. See struct vertex_header in draw_private.h. |
| */ |
| assert(DRAW_TOTAL_CLIP_PLANES==14); |
| /* initialize vertex id:16 = 0xffff, have_clipdist:1 = 0, edgeflag:1 = 1 */ |
| vertex_id_pad_edgeflag = (0xffff << 16) | (1 << DRAW_TOTAL_CLIP_PLANES); |
| if (have_clipdist) |
| vertex_id_pad_edgeflag |= 1 << (DRAW_TOTAL_CLIP_PLANES+1); |
| val = lp_build_const_int_vec(gallivm, lp_int_type(soa_type), vertex_id_pad_edgeflag); |
| /* OR with the clipmask */ |
| cliptmp = LLVMBuildOr(builder, val, clipmask, ""); |
| for (i = 0; i < vector_length; i++) { |
| LLVMValueRef id_ptr = draw_jit_header_id(gallivm, io_ptrs[i]); |
| val = LLVMBuildExtractElement(builder, cliptmp, inds[i], ""); |
| LLVMBuildStore(builder, val, id_ptr); |
| #if DEBUG_STORE |
| lp_build_printf(gallivm, "io = %p, index %d\n, clipmask = %x\n", |
| io_ptrs[i], inds[i], val); |
| #endif |
| } |
| } |
| |
| /* store for each of the n vertices */ |
| for (i = 0; i < vector_length; i++) { |
| store_aos(gallivm, io_ptrs[i], attr_index, aos[i]); |
| } |
| } |
| |
| |
| static void |
| convert_to_aos(struct gallivm_state *gallivm, |
| LLVMValueRef io, |
| LLVMValueRef (*outputs)[TGSI_NUM_CHANNELS], |
| LLVMValueRef clipmask, |
| int num_outputs, |
| struct lp_type soa_type, |
| boolean have_clipdist) |
| { |
| LLVMBuilderRef builder = gallivm->builder; |
| unsigned chan, attrib, i; |
| |
| #if DEBUG_STORE |
| lp_build_printf(gallivm, " # storing begin\n"); |
| #endif |
| for (attrib = 0; attrib < num_outputs; ++attrib) { |
| LLVMValueRef soa[TGSI_NUM_CHANNELS]; |
| LLVMValueRef aos[LP_MAX_VECTOR_WIDTH / 32]; |
| for (chan = 0; chan < TGSI_NUM_CHANNELS; ++chan) { |
| if (outputs[attrib][chan]) { |
| LLVMValueRef out = LLVMBuildLoad(builder, outputs[attrib][chan], ""); |
| lp_build_name(out, "output%u.%c", attrib, "xyzw"[chan]); |
| #if DEBUG_STORE |
| lp_build_printf(gallivm, "output %d : %d ", |
| LLVMConstInt(LLVMInt32TypeInContext(gallivm->context), |
| attrib, 0), |
| LLVMConstInt(LLVMInt32TypeInContext(gallivm->context), |
| chan, 0)); |
| lp_build_print_value(gallivm, "val = ", out); |
| #endif |
| soa[chan] = out; |
| } |
| else { |
| soa[chan] = 0; |
| } |
| } |
| |
| |
| if (soa_type.length == TGSI_NUM_CHANNELS) { |
| lp_build_transpose_aos(gallivm, soa_type, soa, aos); |
| } else { |
| lp_build_transpose_aos(gallivm, soa_type, soa, soa); |
| |
| for (i = 0; i < soa_type.length; ++i) { |
| aos[i] = lp_build_extract_range(gallivm, |
| soa[i % TGSI_NUM_CHANNELS], |
| (i / TGSI_NUM_CHANNELS) * TGSI_NUM_CHANNELS, |
| TGSI_NUM_CHANNELS); |
| } |
| } |
| |
| store_aos_array(gallivm, |
| soa_type, |
| io, |
| aos, |
| attrib, |
| num_outputs, |
| clipmask, have_clipdist); |
| } |
| #if DEBUG_STORE |
| lp_build_printf(gallivm, " # storing end\n"); |
| #endif |
| } |
| |
| |
| /** |
| * Stores original vertex positions in clip coordinates |
| */ |
| static void |
| store_clip(struct gallivm_state *gallivm, |
| const struct lp_type vs_type, |
| LLVMValueRef io_ptr, |
| LLVMValueRef (*outputs)[TGSI_NUM_CHANNELS], |
| boolean pre_clip_pos, int idx) |
| { |
| LLVMBuilderRef builder = gallivm->builder; |
| LLVMValueRef soa[4]; |
| LLVMValueRef aos[LP_MAX_VECTOR_LENGTH]; |
| LLVMValueRef indices[2]; |
| LLVMValueRef io_ptrs[LP_MAX_VECTOR_WIDTH / 32]; |
| LLVMValueRef inds[LP_MAX_VECTOR_WIDTH / 32]; |
| LLVMValueRef clip_ptrs[LP_MAX_VECTOR_WIDTH / 32]; |
| int i, j; |
| |
| indices[0] = |
| indices[1] = lp_build_const_int32(gallivm, 0); |
| |
| for (i = 0; i < vs_type.length; i++) { |
| inds[i] = lp_build_const_int32(gallivm, i); |
| io_ptrs[i] = LLVMBuildGEP(builder, io_ptr, &inds[i], 1, ""); |
| } |
| |
| soa[0] = LLVMBuildLoad(builder, outputs[idx][0], ""); /*x0 x1 .. xn*/ |
| soa[1] = LLVMBuildLoad(builder, outputs[idx][1], ""); /*y0 y1 .. yn*/ |
| soa[2] = LLVMBuildLoad(builder, outputs[idx][2], ""); /*z0 z1 .. zn*/ |
| soa[3] = LLVMBuildLoad(builder, outputs[idx][3], ""); /*w0 w1 .. wn*/ |
| |
| if (!pre_clip_pos) { |
| for (i = 0; i < vs_type.length; i++) { |
| clip_ptrs[i] = draw_jit_header_clip(gallivm, io_ptrs[i]); |
| } |
| } else { |
| for (i = 0; i < vs_type.length; i++) { |
| clip_ptrs[i] = draw_jit_header_pre_clip_pos(gallivm, io_ptrs[i]); |
| } |
| } |
| |
| lp_build_transpose_aos(gallivm, vs_type, soa, soa); |
| for (i = 0; i < vs_type.length; ++i) { |
| aos[i] = lp_build_extract_range(gallivm, |
| soa[i % TGSI_NUM_CHANNELS], |
| (i / TGSI_NUM_CHANNELS) * TGSI_NUM_CHANNELS, |
| TGSI_NUM_CHANNELS); |
| } |
| |
| for (j = 0; j < vs_type.length; j++) { |
| LLVMTypeRef clip_ptr_type = LLVMPointerType(LLVMVectorType(LLVMFloatTypeInContext(gallivm->context), 4), 0); |
| LLVMValueRef clip_ptr; |
| |
| clip_ptr = LLVMBuildGEP(builder, clip_ptrs[j], indices, 2, "clipo"); |
| clip_ptr = LLVMBuildPointerCast(builder, clip_ptr, clip_ptr_type, ""); |
| |
| /* Unaligned store */ |
| lp_set_store_alignment(LLVMBuildStore(builder, aos[j], clip_ptr), sizeof(float)); |
| } |
| } |
| |
| |
| /** |
| * Transforms the outputs for viewport mapping |
| */ |
| static void |
| generate_viewport(struct draw_llvm_variant *variant, |
| LLVMBuilderRef builder, |
| struct lp_type vs_type, |
| LLVMValueRef (*outputs)[TGSI_NUM_CHANNELS], |
| LLVMValueRef context_ptr) |
| { |
| int i; |
| struct gallivm_state *gallivm = variant->gallivm; |
| struct lp_type f32_type = vs_type; |
| LLVMTypeRef vs_type_llvm = lp_build_vec_type(gallivm, vs_type); |
| LLVMValueRef out3 = LLVMBuildLoad(builder, outputs[0][3], ""); /*w0 w1 .. wn*/ |
| LLVMValueRef const1 = lp_build_const_vec(gallivm, f32_type, 1.0); /*1.0 1.0 1.0 1.0*/ |
| LLVMValueRef vp_ptr = draw_jit_context_viewport(gallivm, context_ptr); |
| |
| /* for 1/w convention*/ |
| out3 = LLVMBuildFDiv(builder, const1, out3, ""); |
| LLVMBuildStore(builder, out3, outputs[0][3]); |
| |
| /* Viewport Mapping */ |
| for (i=0; i<3; i++) { |
| LLVMValueRef out = LLVMBuildLoad(builder, outputs[0][i], ""); /*x0 x1 .. xn*/ |
| LLVMValueRef scale; |
| LLVMValueRef trans; |
| LLVMValueRef scale_i; |
| LLVMValueRef trans_i; |
| LLVMValueRef index; |
| |
| index = lp_build_const_int32(gallivm, i); |
| scale_i = LLVMBuildGEP(builder, vp_ptr, &index, 1, ""); |
| |
| index = lp_build_const_int32(gallivm, i+4); |
| trans_i = LLVMBuildGEP(builder, vp_ptr, &index, 1, ""); |
| |
| scale = lp_build_broadcast(gallivm, vs_type_llvm, |
| LLVMBuildLoad(builder, scale_i, "scale")); |
| trans = lp_build_broadcast(gallivm, vs_type_llvm, |
| LLVMBuildLoad(builder, trans_i, "trans")); |
| |
| /* divide by w */ |
| out = LLVMBuildFMul(builder, out, out3, ""); |
| /* mult by scale */ |
| out = LLVMBuildFMul(builder, out, scale, ""); |
| /* add translation */ |
| out = LLVMBuildFAdd(builder, out, trans, ""); |
| |
| /* store transformed outputs */ |
| LLVMBuildStore(builder, out, outputs[0][i]); |
| } |
| |
| } |
| |
| |
| /** |
| * Returns clipmask as nxi32 bitmask for the n vertices |
| */ |
| static LLVMValueRef |
| generate_clipmask(struct draw_llvm *llvm, |
| struct gallivm_state *gallivm, |
| struct lp_type vs_type, |
| LLVMValueRef (*outputs)[TGSI_NUM_CHANNELS], |
| boolean clip_xy, |
| boolean clip_z, |
| boolean clip_user, |
| boolean clip_halfz, |
| unsigned ucp_enable, |
| LLVMValueRef context_ptr, |
| boolean *have_clipdist) |
| { |
| LLVMBuilderRef builder = gallivm->builder; |
| LLVMValueRef mask; /* stores the <nxi32> clipmasks */ |
| LLVMValueRef test, temp; |
| LLVMValueRef zero, shift; |
| LLVMValueRef pos_x, pos_y, pos_z, pos_w; |
| LLVMValueRef cv_x, cv_y, cv_z, cv_w; |
| LLVMValueRef plane1, planes, plane_ptr, sum; |
| struct lp_type f32_type = vs_type; |
| struct lp_type i32_type = lp_int_type(vs_type); |
| const unsigned pos = draw_current_shader_position_output(llvm->draw); |
| const unsigned cv = draw_current_shader_clipvertex_output(llvm->draw); |
| int num_written_clipdistance = llvm->draw->vs.vertex_shader->info.num_written_clipdistance; |
| bool have_cd = false; |
| unsigned cd[2]; |
| |
| cd[0] = draw_current_shader_clipdistance_output(llvm->draw, 0); |
| cd[1] = draw_current_shader_clipdistance_output(llvm->draw, 1); |
| |
| if (cd[0] != pos || cd[1] != pos) |
| have_cd = true; |
| |
| mask = lp_build_const_int_vec(gallivm, i32_type, 0); |
| temp = lp_build_const_int_vec(gallivm, i32_type, 0); |
| zero = lp_build_const_vec(gallivm, f32_type, 0); /* 0.0f 0.0f 0.0f 0.0f */ |
| shift = lp_build_const_int_vec(gallivm, i32_type, 1); /* 1 1 1 1 */ |
| |
| /* |
| * load clipvertex and position from correct locations. |
| * if they are the same just load them once. |
| */ |
| pos_x = LLVMBuildLoad(builder, outputs[pos][0], ""); /*x0 x1 .. xn */ |
| pos_y = LLVMBuildLoad(builder, outputs[pos][1], ""); /*y0 y1 .. yn */ |
| pos_z = LLVMBuildLoad(builder, outputs[pos][2], ""); /*z0 z1 .. zn */ |
| pos_w = LLVMBuildLoad(builder, outputs[pos][3], ""); /*w0 w1 .. wn */ |
| |
| if (clip_user && cv != pos) { |
| cv_x = LLVMBuildLoad(builder, outputs[cv][0], ""); /*x0 x1 .. xn */ |
| cv_y = LLVMBuildLoad(builder, outputs[cv][1], ""); /*y0 y1 .. yn */ |
| cv_z = LLVMBuildLoad(builder, outputs[cv][2], ""); /*z0 z1 .. zn */ |
| cv_w = LLVMBuildLoad(builder, outputs[cv][3], ""); /*w0 w1 .. wn */ |
| } else { |
| cv_x = pos_x; |
| cv_y = pos_y; |
| cv_z = pos_z; |
| cv_w = pos_w; |
| } |
| |
| /* Cliptest, for hardwired planes */ |
| if (clip_xy) { |
| /* plane 1 */ |
| test = lp_build_compare(gallivm, f32_type, PIPE_FUNC_GREATER, pos_x , pos_w); |
| temp = shift; |
| test = LLVMBuildAnd(builder, test, temp, ""); |
| mask = test; |
| |
| /* plane 2 */ |
| test = LLVMBuildFAdd(builder, pos_x, pos_w, ""); |
| test = lp_build_compare(gallivm, f32_type, PIPE_FUNC_GREATER, zero, test); |
| temp = LLVMBuildShl(builder, temp, shift, ""); |
| test = LLVMBuildAnd(builder, test, temp, ""); |
| mask = LLVMBuildOr(builder, mask, test, ""); |
| |
| /* plane 3 */ |
| test = lp_build_compare(gallivm, f32_type, PIPE_FUNC_GREATER, pos_y, pos_w); |
| temp = LLVMBuildShl(builder, temp, shift, ""); |
| test = LLVMBuildAnd(builder, test, temp, ""); |
| mask = LLVMBuildOr(builder, mask, test, ""); |
| |
| /* plane 4 */ |
| test = LLVMBuildFAdd(builder, pos_y, pos_w, ""); |
| test = lp_build_compare(gallivm, f32_type, PIPE_FUNC_GREATER, zero, test); |
| temp = LLVMBuildShl(builder, temp, shift, ""); |
| test = LLVMBuildAnd(builder, test, temp, ""); |
| mask = LLVMBuildOr(builder, mask, test, ""); |
| } |
| |
| if (clip_z) { |
| temp = lp_build_const_int_vec(gallivm, i32_type, 16); |
| if (clip_halfz) { |
| /* plane 5 */ |
| test = lp_build_compare(gallivm, f32_type, PIPE_FUNC_GREATER, zero, pos_z); |
| test = LLVMBuildAnd(builder, test, temp, ""); |
| mask = LLVMBuildOr(builder, mask, test, ""); |
| } |
| else { |
| /* plane 5 */ |
| test = LLVMBuildFAdd(builder, pos_z, pos_w, ""); |
| test = lp_build_compare(gallivm, f32_type, PIPE_FUNC_GREATER, zero, test); |
| test = LLVMBuildAnd(builder, test, temp, ""); |
| mask = LLVMBuildOr(builder, mask, test, ""); |
| } |
| /* plane 6 */ |
| test = lp_build_compare(gallivm, f32_type, PIPE_FUNC_GREATER, pos_z, pos_w); |
| temp = LLVMBuildShl(builder, temp, shift, ""); |
| test = LLVMBuildAnd(builder, test, temp, ""); |
| mask = LLVMBuildOr(builder, mask, test, ""); |
| } |
| |
| if (clip_user) { |
| LLVMValueRef planes_ptr = draw_jit_context_planes(gallivm, context_ptr); |
| LLVMValueRef indices[3]; |
| |
| /* userclip planes */ |
| while (ucp_enable) { |
| unsigned plane_idx = ffs(ucp_enable)-1; |
| ucp_enable &= ~(1 << plane_idx); |
| plane_idx += 6; |
| |
| if (have_cd && num_written_clipdistance) { |
| LLVMValueRef clipdist; |
| int i; |
| i = plane_idx - 6; |
| |
| *have_clipdist = TRUE; |
| if (i < 4) { |
| clipdist = LLVMBuildLoad(builder, outputs[cd[0]][i], ""); |
| } else { |
| clipdist = LLVMBuildLoad(builder, outputs[cd[1]][i-4], ""); |
| } |
| test = lp_build_compare(gallivm, f32_type, PIPE_FUNC_GREATER, zero, clipdist); |
| temp = lp_build_const_int_vec(gallivm, i32_type, 1 << plane_idx); |
| test = LLVMBuildAnd(builder, test, temp, ""); |
| mask = LLVMBuildOr(builder, mask, test, ""); |
| } else { |
| LLVMTypeRef vs_type_llvm = lp_build_vec_type(gallivm, vs_type); |
| indices[0] = lp_build_const_int32(gallivm, 0); |
| indices[1] = lp_build_const_int32(gallivm, plane_idx); |
| |
| indices[2] = lp_build_const_int32(gallivm, 0); |
| plane_ptr = LLVMBuildGEP(builder, planes_ptr, indices, 3, ""); |
| plane1 = LLVMBuildLoad(builder, plane_ptr, "plane_x"); |
| planes = lp_build_broadcast(gallivm, vs_type_llvm, plane1); |
| sum = LLVMBuildFMul(builder, planes, cv_x, ""); |
| |
| indices[2] = lp_build_const_int32(gallivm, 1); |
| plane_ptr = LLVMBuildGEP(builder, planes_ptr, indices, 3, ""); |
| plane1 = LLVMBuildLoad(builder, plane_ptr, "plane_y"); |
| planes = lp_build_broadcast(gallivm, vs_type_llvm, plane1); |
| test = LLVMBuildFMul(builder, planes, cv_y, ""); |
| sum = LLVMBuildFAdd(builder, sum, test, ""); |
| |
| indices[2] = lp_build_const_int32(gallivm, 2); |
| plane_ptr = LLVMBuildGEP(builder, planes_ptr, indices, 3, ""); |
| plane1 = LLVMBuildLoad(builder, plane_ptr, "plane_z"); |
| planes = lp_build_broadcast(gallivm, vs_type_llvm, plane1); |
| test = LLVMBuildFMul(builder, planes, cv_z, ""); |
| sum = LLVMBuildFAdd(builder, sum, test, ""); |
| |
| indices[2] = lp_build_const_int32(gallivm, 3); |
| plane_ptr = LLVMBuildGEP(builder, planes_ptr, indices, 3, ""); |
| plane1 = LLVMBuildLoad(builder, plane_ptr, "plane_w"); |
| planes = lp_build_broadcast(gallivm, vs_type_llvm, plane1); |
| test = LLVMBuildFMul(builder, planes, cv_w, ""); |
| sum = LLVMBuildFAdd(builder, sum, test, ""); |
| |
| test = lp_build_compare(gallivm, f32_type, PIPE_FUNC_GREATER, zero, sum); |
| temp = lp_build_const_int_vec(gallivm, i32_type, 1 << plane_idx); |
| test = LLVMBuildAnd(builder, test, temp, ""); |
| mask = LLVMBuildOr(builder, mask, test, ""); |
| } |
| } |
| } |
| return mask; |
| } |
| |
| |
| /** |
| * Returns boolean if any clipping has occurred |
| * Used zero/non-zero i32 value to represent boolean |
| */ |
| static LLVMValueRef |
| clipmask_booli32(struct gallivm_state *gallivm, |
| const struct lp_type vs_type, |
| LLVMValueRef clipmask_bool_ptr) |
| { |
| LLVMBuilderRef builder = gallivm->builder; |
| LLVMTypeRef int32_type = LLVMInt32TypeInContext(gallivm->context); |
| LLVMValueRef clipmask_bool = LLVMBuildLoad(builder, clipmask_bool_ptr, ""); |
| LLVMValueRef ret = LLVMConstNull(int32_type); |
| LLVMValueRef temp; |
| int i; |
| |
| /* |
| * Can do this with log2(vector length) pack instructions and one extract |
| * (as we don't actually need a or) with sse2 which would be way better. |
| */ |
| for (i=0; i < vs_type.length; i++) { |
| temp = LLVMBuildExtractElement(builder, clipmask_bool, |
| lp_build_const_int32(gallivm, i) , ""); |
| ret = LLVMBuildOr(builder, ret, temp, ""); |
| } |
| return ret; |
| } |
| |
| |
| static void |
| draw_llvm_generate(struct draw_llvm *llvm, struct draw_llvm_variant *variant, |
| boolean elts) |
| { |
| struct gallivm_state *gallivm = variant->gallivm; |
| LLVMContextRef context = gallivm->context; |
| LLVMTypeRef int32_type = LLVMInt32TypeInContext(context); |
| LLVMTypeRef arg_types[8]; |
| LLVMTypeRef func_type; |
| LLVMValueRef context_ptr; |
| LLVMBasicBlockRef block; |
| LLVMBuilderRef builder; |
| struct lp_type vs_type; |
| LLVMValueRef end, start; |
| LLVMValueRef count, fetch_elts, fetch_count; |
| LLVMValueRef stride, step, io_itr; |
| LLVMValueRef io_ptr, vbuffers_ptr, vb_ptr; |
| LLVMValueRef zero = lp_build_const_int32(gallivm, 0); |
| LLVMValueRef one = lp_build_const_int32(gallivm, 1); |
| struct draw_context *draw = llvm->draw; |
| const struct tgsi_shader_info *vs_info = &draw->vs.vertex_shader->info; |
| unsigned i, j; |
| struct lp_build_context bld; |
| struct lp_build_loop_state lp_loop; |
| const int vector_length = lp_native_vector_width / 32; |
| LLVMValueRef outputs[PIPE_MAX_SHADER_OUTPUTS][TGSI_NUM_CHANNELS]; |
| LLVMValueRef fetch_max; |
| struct lp_build_sampler_soa *sampler = 0; |
| LLVMValueRef ret, clipmask_bool_ptr; |
| const boolean bypass_viewport = variant->key.bypass_viewport; |
| const boolean enable_cliptest = variant->key.clip_xy || |
| variant->key.clip_z || |
| variant->key.clip_user; |
| LLVMValueRef variant_func; |
| const unsigned pos = draw_current_shader_position_output(llvm->draw); |
| const unsigned cv = draw_current_shader_clipvertex_output(llvm->draw); |
| boolean have_clipdist = FALSE; |
| struct lp_bld_tgsi_system_values system_values; |
| |
| memset(&system_values, 0, sizeof(system_values)); |
| |
| arg_types[0] = get_context_ptr_type(variant); /* context */ |
| arg_types[1] = get_vertex_header_ptr_type(variant); /* vertex_header */ |
| arg_types[2] = get_buffer_ptr_type(variant); /* vbuffers */ |
| if (elts) |
| arg_types[3] = LLVMPointerType(int32_type, 0);/* fetch_elts * */ |
| else |
| arg_types[3] = int32_type; /* start */ |
| arg_types[4] = int32_type; /* fetch_count / count */ |
| arg_types[5] = int32_type; /* stride */ |
| arg_types[6] = get_vb_ptr_type(variant); /* pipe_vertex_buffer's */ |
| arg_types[7] = int32_type; /* instance_id */ |
| |
| func_type = LLVMFunctionType(int32_type, arg_types, Elements(arg_types), 0); |
| |
| variant_func = LLVMAddFunction(gallivm->module, |
| elts ? "draw_llvm_shader_elts" : "draw_llvm_shader", |
| func_type); |
| |
| if (elts) |
| variant->function_elts = variant_func; |
| else |
| variant->function = variant_func; |
| |
| LLVMSetFunctionCallConv(variant_func, LLVMCCallConv); |
| for (i = 0; i < Elements(arg_types); ++i) |
| if (LLVMGetTypeKind(arg_types[i]) == LLVMPointerTypeKind) |
| LLVMAddAttribute(LLVMGetParam(variant_func, i), |
| LLVMNoAliasAttribute); |
| |
| context_ptr = LLVMGetParam(variant_func, 0); |
| io_ptr = LLVMGetParam(variant_func, 1); |
| vbuffers_ptr = LLVMGetParam(variant_func, 2); |
| stride = LLVMGetParam(variant_func, 5); |
| vb_ptr = LLVMGetParam(variant_func, 6); |
| system_values.instance_id = LLVMGetParam(variant_func, 7); |
| |
| lp_build_name(context_ptr, "context"); |
| lp_build_name(io_ptr, "io"); |
| lp_build_name(vbuffers_ptr, "vbuffers"); |
| lp_build_name(stride, "stride"); |
| lp_build_name(vb_ptr, "vb"); |
| lp_build_name(system_values.instance_id, "instance_id"); |
| |
| if (elts) { |
| fetch_elts = LLVMGetParam(variant_func, 3); |
| fetch_count = LLVMGetParam(variant_func, 4); |
| lp_build_name(fetch_elts, "fetch_elts"); |
| lp_build_name(fetch_count, "fetch_count"); |
| start = count = NULL; |
| } |
| else { |
| start = LLVMGetParam(variant_func, 3); |
| count = LLVMGetParam(variant_func, 4); |
| lp_build_name(start, "start"); |
| lp_build_name(count, "count"); |
| fetch_elts = fetch_count = NULL; |
| } |
| |
| /* |
| * Function body |
| */ |
| |
| block = LLVMAppendBasicBlockInContext(gallivm->context, variant_func, "entry"); |
| builder = gallivm->builder; |
| LLVMPositionBuilderAtEnd(builder, block); |
| |
| lp_build_context_init(&bld, gallivm, lp_type_int(32)); |
| |
| memset(&vs_type, 0, sizeof vs_type); |
| vs_type.floating = TRUE; /* floating point values */ |
| vs_type.sign = TRUE; /* values are signed */ |
| vs_type.norm = FALSE; /* values are not limited to [0,1] or [-1,1] */ |
| vs_type.width = 32; /* 32-bit float */ |
| vs_type.length = vector_length; |
| |
| /* hold temporary "bool" clipmask */ |
| clipmask_bool_ptr = lp_build_alloca(gallivm, lp_build_int_vec_type(gallivm, vs_type), ""); |
| LLVMBuildStore(builder, lp_build_zero(gallivm, lp_int_type(vs_type)), clipmask_bool_ptr); |
| |
| /* code generated texture sampling */ |
| sampler = draw_llvm_sampler_soa_create( |
| draw_llvm_variant_key_samplers(&variant->key), |
| context_ptr); |
| |
| if (elts) { |
| start = zero; |
| end = fetch_count; |
| } |
| else { |
| end = lp_build_add(&bld, start, count); |
| } |
| |
| step = lp_build_const_int32(gallivm, vector_length); |
| |
| fetch_max = LLVMBuildSub(builder, end, one, "fetch_max"); |
| |
| lp_build_loop_begin(&lp_loop, gallivm, start); |
| { |
| LLVMValueRef inputs[PIPE_MAX_SHADER_INPUTS][TGSI_NUM_CHANNELS]; |
| LLVMValueRef aos_attribs[PIPE_MAX_SHADER_INPUTS][LP_MAX_VECTOR_WIDTH / 32] = { { 0 } }; |
| LLVMValueRef io; |
| LLVMValueRef clipmask; /* holds the clipmask value */ |
| const LLVMValueRef (*ptr_aos)[TGSI_NUM_CHANNELS]; |
| |
| if (elts) |
| io_itr = lp_loop.counter; |
| else |
| io_itr = LLVMBuildSub(builder, lp_loop.counter, start, ""); |
| |
| io = LLVMBuildGEP(builder, io_ptr, &io_itr, 1, ""); |
| #if DEBUG_STORE |
| lp_build_printf(gallivm, " --- io %d = %p, loop counter %d\n", |
| io_itr, io, lp_loop.counter); |
| #endif |
| system_values.vertex_id = lp_build_zero(gallivm, lp_type_uint_vec(32, 32*vector_length)); |
| for (i = 0; i < vector_length; ++i) { |
| LLVMValueRef true_index = |
| LLVMBuildAdd(builder, |
| lp_loop.counter, |
| lp_build_const_int32(gallivm, i), ""); |
| |
| /* make sure we're not out of bounds which can happen |
| * if fetch_count % 4 != 0, because on the last iteration |
| * a few of the 4 vertex fetches will be out of bounds */ |
| true_index = lp_build_min(&bld, true_index, fetch_max); |
| |
| if (elts) { |
| LLVMValueRef fetch_ptr; |
| fetch_ptr = LLVMBuildGEP(builder, fetch_elts, |
| &true_index, 1, ""); |
| true_index = LLVMBuildLoad(builder, fetch_ptr, "fetch_elt"); |
| } |
| |
| system_values.vertex_id = LLVMBuildInsertElement(gallivm->builder, |
| system_values.vertex_id, true_index, |
| lp_build_const_int32(gallivm, i), ""); |
| for (j = 0; j < draw->pt.nr_vertex_elements; ++j) { |
| struct pipe_vertex_element *velem = &draw->pt.vertex_element[j]; |
| LLVMValueRef vb_index = |
| lp_build_const_int32(gallivm, velem->vertex_buffer_index); |
| LLVMValueRef vb = LLVMBuildGEP(builder, vb_ptr, &vb_index, 1, ""); |
| generate_fetch(gallivm, vbuffers_ptr, |
| &aos_attribs[j][i], velem, vb, true_index, |
| system_values.instance_id); |
| } |
| } |
| convert_to_soa(gallivm, aos_attribs, inputs, |
| draw->pt.nr_vertex_elements, vs_type); |
| |
| ptr_aos = (const LLVMValueRef (*)[TGSI_NUM_CHANNELS]) inputs; |
| generate_vs(variant, |
| builder, |
| vs_type, |
| outputs, |
| ptr_aos, |
| &system_values, |
| context_ptr, |
| sampler, |
| variant->key.clamp_vertex_color); |
| |
| /* store original positions in clip before further manipulation */ |
| store_clip(gallivm, vs_type, io, outputs, 0, cv); |
| store_clip(gallivm, vs_type, io, outputs, 1, pos); |
| |
| /* do cliptest */ |
| if (enable_cliptest) { |
| LLVMValueRef temp = LLVMBuildLoad(builder, clipmask_bool_ptr, ""); |
| /* allocate clipmask, assign it integer type */ |
| clipmask = generate_clipmask(llvm, |
| gallivm, |
| vs_type, |
| outputs, |
| variant->key.clip_xy, |
| variant->key.clip_z, |
| variant->key.clip_user, |
| variant->key.clip_halfz, |
| variant->key.ucp_enable, |
| context_ptr, &have_clipdist); |
| temp = LLVMBuildOr(builder, clipmask, temp, ""); |
| /* store temporary clipping boolean value */ |
| LLVMBuildStore(builder, temp, clipmask_bool_ptr); |
| } |
| else { |
| clipmask = lp_build_const_int_vec(gallivm, lp_int_type(vs_type), 0); |
| } |
| |
| /* do viewport mapping */ |
| if (!bypass_viewport) { |
| generate_viewport(variant, builder, vs_type, outputs, context_ptr); |
| } |
| |
| /* store clipmask in vertex header, |
| * original positions in clip |
| * and transformed positions in data |
| */ |
| convert_to_aos(gallivm, io, outputs, clipmask, |
| vs_info->num_outputs, vs_type, |
| have_clipdist); |
| } |
| |
| lp_build_loop_end_cond(&lp_loop, end, step, LLVMIntUGE); |
| |
| sampler->destroy(sampler); |
| |
| /* return clipping boolean value for function */ |
| ret = clipmask_booli32(gallivm, vs_type, clipmask_bool_ptr); |
| |
| LLVMBuildRet(builder, ret); |
| |
| gallivm_verify_function(gallivm, variant_func); |
| } |
| |
| |
| struct draw_llvm_variant_key * |
| draw_llvm_make_variant_key(struct draw_llvm *llvm, char *store) |
| { |
| unsigned i; |
| struct draw_llvm_variant_key *key; |
| struct lp_sampler_static_state *sampler; |
| |
| key = (struct draw_llvm_variant_key *)store; |
| |
| key->clamp_vertex_color = llvm->draw->rasterizer->clamp_vertex_color; /**/ |
| |
| /* Presumably all variants of the shader should have the same |
| * number of vertex elements - ie the number of shader inputs. |
| */ |
| key->nr_vertex_elements = llvm->draw->pt.nr_vertex_elements; |
| |
| /* will have to rig this up properly later */ |
| key->clip_xy = llvm->draw->clip_xy; |
| key->clip_z = llvm->draw->clip_z; |
| key->clip_user = llvm->draw->clip_user; |
| key->bypass_viewport = llvm->draw->identity_viewport; |
| key->clip_halfz = !llvm->draw->rasterizer->gl_rasterization_rules; |
| key->need_edgeflags = (llvm->draw->vs.edgeflag_output ? TRUE : FALSE); |
| key->ucp_enable = llvm->draw->rasterizer->clip_plane_enable; |
| key->pad = 0; |
| |
| /* All variants of this shader will have the same value for |
| * nr_samplers. Not yet trying to compact away holes in the |
| * sampler array. |
| */ |
| key->nr_samplers = llvm->draw->vs.vertex_shader->info.file_max[TGSI_FILE_SAMPLER] + 1; |
| |
| sampler = draw_llvm_variant_key_samplers(key); |
| |
| memcpy(key->vertex_element, |
| llvm->draw->pt.vertex_element, |
| sizeof(struct pipe_vertex_element) * key->nr_vertex_elements); |
| |
| memset(sampler, 0, key->nr_samplers * sizeof *sampler); |
| |
| for (i = 0 ; i < key->nr_samplers; i++) { |
| lp_sampler_static_state(&sampler[i], |
| llvm->draw->sampler_views[PIPE_SHADER_VERTEX][i], |
| llvm->draw->samplers[PIPE_SHADER_VERTEX][i]); |
| } |
| |
| return key; |
| } |
| |
| |
| void |
| draw_llvm_set_mapped_texture(struct draw_context *draw, |
| unsigned sampler_idx, |
| uint32_t width, uint32_t height, uint32_t depth, |
| uint32_t first_level, uint32_t last_level, |
| uint32_t row_stride[PIPE_MAX_TEXTURE_LEVELS], |
| uint32_t img_stride[PIPE_MAX_TEXTURE_LEVELS], |
| const void *data[PIPE_MAX_TEXTURE_LEVELS]) |
| { |
| unsigned j; |
| struct draw_jit_texture *jit_tex; |
| |
| assert(sampler_idx < Elements(draw->llvm->jit_context.textures)); |
| |
| jit_tex = &draw->llvm->jit_context.textures[sampler_idx]; |
| |
| jit_tex->width = width; |
| jit_tex->height = height; |
| jit_tex->depth = depth; |
| jit_tex->first_level = first_level; |
| jit_tex->last_level = last_level; |
| |
| for (j = first_level; j <= last_level; j++) { |
| jit_tex->data[j] = data[j]; |
| jit_tex->row_stride[j] = row_stride[j]; |
| jit_tex->img_stride[j] = img_stride[j]; |
| } |
| } |
| |
| |
| void |
| draw_llvm_set_sampler_state(struct draw_context *draw) |
| { |
| unsigned i; |
| |
| for (i = 0; i < draw->num_samplers[PIPE_SHADER_VERTEX]; i++) { |
| struct draw_jit_texture *jit_tex = &draw->llvm->jit_context.textures[i]; |
| |
| if (draw->samplers[i]) { |
| const struct pipe_sampler_state *s |
| = draw->samplers[PIPE_SHADER_VERTEX][i]; |
| jit_tex->min_lod = s->min_lod; |
| jit_tex->max_lod = s->max_lod; |
| jit_tex->lod_bias = s->lod_bias; |
| COPY_4V(jit_tex->border_color, s->border_color.f); |
| } |
| } |
| } |
| |
| |
| void |
| draw_llvm_destroy_variant(struct draw_llvm_variant *variant) |
| { |
| struct draw_llvm *llvm = variant->llvm; |
| |
| if (variant->function_elts) { |
| gallivm_free_function(variant->gallivm, |
| variant->function_elts, variant->jit_func_elts); |
| } |
| |
| if (variant->function) { |
| gallivm_free_function(variant->gallivm, |
| variant->function, variant->jit_func); |
| } |
| |
| gallivm_destroy(variant->gallivm); |
| |
| remove_from_list(&variant->list_item_local); |
| variant->shader->variants_cached--; |
| remove_from_list(&variant->list_item_global); |
| llvm->nr_variants--; |
| FREE(variant); |
| } |