vl_vertex_buffers.c revision 794cde3f5ef59cf603be284fbc8de33d2cda7d2c
1/************************************************************************** 2 * 3 * Copyright 2010 Christian König 4 * All Rights Reserved. 5 * 6 * Permission is hereby granted, free of charge, to any person obtaining a 7 * copy of this software and associated documentation files (the 8 * "Software"), to deal in the Software without restriction, including 9 * without limitation the rights to use, copy, modify, merge, publish, 10 * distribute, sub license, and/or sell copies of the Software, and to 11 * permit persons to whom the Software is furnished to do so, subject to 12 * the following conditions: 13 * 14 * The above copyright notice and this permission notice (including the 15 * next paragraph) shall be included in all copies or substantial portions 16 * of the Software. 17 * 18 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS 19 * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF 20 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT. 21 * IN NO EVENT SHALL TUNGSTEN GRAPHICS AND/OR ITS SUPPLIERS BE LIABLE FOR 22 * ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, 23 * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE 24 * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. 25 * 26 **************************************************************************/ 27 28#include <assert.h> 29#include <util/u_format.h> 30#include "vl_vertex_buffers.h" 31#include "vl_types.h" 32 33struct vl_vertex_stream 34{ 35 struct vertex2s pos; 36 int8_t eb[3][2][2]; 37 int8_t dct_type_field; 38 int8_t mo_type_frame; 39 int8_t mb_type_intra; 40 int8_t mv_wheights; 41 struct vertex2s mv[4]; 42}; 43 44/* vertices for a quad covering a block */ 45static const struct vertex2f block_quad[4] = { 46 {0.0f, 0.0f}, {1.0f, 0.0f}, {1.0f, 1.0f}, {0.0f, 1.0f} 47}; 48 49struct pipe_vertex_buffer 50vl_vb_upload_quads(struct pipe_context *pipe, unsigned blocks_x, unsigned blocks_y) 51{ 52 struct pipe_vertex_buffer quad; 53 struct pipe_transfer *buf_transfer; 54 struct vertex4f *v; 55 56 unsigned x, y, i; 57 58 assert(pipe); 59 60 /* create buffer */ 61 quad.stride = sizeof(struct vertex4f); 62 quad.buffer_offset = 0; 63 quad.buffer = pipe_buffer_create 64 ( 65 pipe->screen, 66 PIPE_BIND_VERTEX_BUFFER, 67 PIPE_USAGE_STATIC, 68 sizeof(struct vertex4f) * 4 * blocks_x * blocks_y 69 ); 70 71 if(!quad.buffer) 72 return quad; 73 74 /* and fill it */ 75 v = pipe_buffer_map 76 ( 77 pipe, 78 quad.buffer, 79 PIPE_TRANSFER_WRITE | PIPE_TRANSFER_DISCARD, 80 &buf_transfer 81 ); 82 83 for ( y = 0; y < blocks_y; ++y) { 84 for ( x = 0; x < blocks_x; ++x) { 85 for (i = 0; i < 4; ++i, ++v) { 86 v->x = block_quad[i].x; 87 v->y = block_quad[i].y; 88 89 v->z = x; 90 v->w = y; 91 } 92 } 93 } 94 95 pipe_buffer_unmap(pipe, buf_transfer); 96 97 return quad; 98} 99 100static struct pipe_vertex_element 101vl_vb_get_quad_vertex_element(void) 102{ 103 struct pipe_vertex_element element; 104 105 /* setup rectangle element */ 106 element.src_offset = 0; 107 element.instance_divisor = 0; 108 element.vertex_buffer_index = 0; 109 element.src_format = PIPE_FORMAT_R32G32B32A32_FLOAT; 110 111 return element; 112} 113 114static void 115vl_vb_element_helper(struct pipe_vertex_element* elements, unsigned num_elements, 116 unsigned vertex_buffer_index) 117{ 118 unsigned i, offset = 0; 119 120 assert(elements && num_elements); 121 122 for ( i = 0; i < num_elements; ++i ) { 123 if (elements[i].src_offset) 124 offset = elements[i].src_offset; 125 else 126 elements[i].src_offset = offset; 127 elements[i].instance_divisor = 1; 128 elements[i].vertex_buffer_index = vertex_buffer_index; 129 offset += util_format_get_blocksize(elements[i].src_format); 130 } 131} 132 133void * 134vl_vb_get_elems_state(struct pipe_context *pipe, int component) 135{ 136 struct pipe_vertex_element vertex_elems[NUM_VS_INPUTS]; 137 138 memset(&vertex_elems, 0, sizeof(vertex_elems)); 139 vertex_elems[VS_I_RECT] = vl_vb_get_quad_vertex_element(); 140 141 /* Position element */ 142 vertex_elems[VS_I_VPOS].src_format = PIPE_FORMAT_R16G16_SSCALED; 143 144 /* empty block element of selected component */ 145 vertex_elems[VS_I_EB].src_offset = 4 + component * 4; 146 vertex_elems[VS_I_EB].src_format = PIPE_FORMAT_R8G8B8A8_SSCALED; 147 148 /* flags */ 149 vertex_elems[VS_I_FLAGS].src_offset = 16; 150 vertex_elems[VS_I_FLAGS].src_format = PIPE_FORMAT_R8G8B8A8_SSCALED; 151 152 /* motion vector 0 TOP element */ 153 vertex_elems[VS_I_MV0_TOP].src_format = PIPE_FORMAT_R16G16_SSCALED; 154 155 /* motion vector 0 BOTTOM element */ 156 vertex_elems[VS_I_MV0_BOTTOM].src_format = PIPE_FORMAT_R16G16_SSCALED; 157 158 /* motion vector 1 TOP element */ 159 vertex_elems[VS_I_MV1_TOP].src_format = PIPE_FORMAT_R16G16_SSCALED; 160 161 /* motion vector 1 BOTTOM element */ 162 vertex_elems[VS_I_MV1_BOTTOM].src_format = PIPE_FORMAT_R16G16_SSCALED; 163 164 vl_vb_element_helper(&vertex_elems[VS_I_VPOS], NUM_VS_INPUTS - 1, 1); 165 166 return pipe->create_vertex_elements_state(pipe, NUM_VS_INPUTS, vertex_elems); 167} 168 169struct pipe_vertex_buffer 170vl_vb_init(struct vl_vertex_buffer *buffer, struct pipe_context *pipe, unsigned size) 171{ 172 struct pipe_vertex_buffer buf; 173 174 assert(buffer); 175 176 buffer->size = size; 177 buffer->num_not_empty = 0; 178 buffer->num_empty = 0; 179 180 buf.stride = sizeof(struct vl_vertex_stream); 181 buf.buffer_offset = 0; 182 buf.buffer = pipe_buffer_create 183 ( 184 pipe->screen, 185 PIPE_BIND_VERTEX_BUFFER, 186 PIPE_USAGE_STREAM, 187 sizeof(struct vl_vertex_stream) * size 188 ); 189 190 pipe_resource_reference(&buffer->resource, buf.buffer); 191 192 vl_vb_map(buffer, pipe); 193 194 return buf; 195} 196 197void 198vl_vb_map(struct vl_vertex_buffer *buffer, struct pipe_context *pipe) 199{ 200 assert(buffer && pipe); 201 202 buffer->start = pipe_buffer_map 203 ( 204 pipe, 205 buffer->resource, 206 PIPE_TRANSFER_WRITE | PIPE_TRANSFER_DISCARD, 207 &buffer->transfer 208 ); 209 buffer->end = buffer->start + buffer->resource->width0 / sizeof(struct vl_vertex_stream); 210} 211 212static void 213get_motion_vectors(struct pipe_mpeg12_macroblock *mb, struct vertex2s mv[4]) 214{ 215 if (mb->mb_type == PIPE_MPEG12_MACROBLOCK_TYPE_BI || 216 mb->mb_type == PIPE_MPEG12_MACROBLOCK_TYPE_FWD) { 217 218 if (mb->mo_type == PIPE_MPEG12_MOTION_TYPE_FRAME) { 219 mv[0].x = mv[1].x = mb->mv[0].top.x; 220 mv[0].y = mv[1].y = mb->mv[0].top.y; 221 222 } else { 223 mv[0].x = mb->mv[0].top.x; 224 mv[0].y = mb->mv[0].top.y - (mb->mv[0].top.y % 4); 225 226 mv[1].x = mb->mv[0].bottom.x; 227 mv[1].y = mb->mv[0].bottom.y - (mb->mv[0].bottom.y % 4); 228 229 if (mb->mv[0].top.field_select) mv[0].y += 2; 230 if (!mb->mv[0].bottom.field_select) mv[1].y -= 2; 231 } 232 } else { 233 mv[0].x = mv[0].y = mv[1].x = mv[1].y = 0x8000; 234 } 235 236 if (mb->mb_type == PIPE_MPEG12_MACROBLOCK_TYPE_BI || 237 mb->mb_type == PIPE_MPEG12_MACROBLOCK_TYPE_BKWD) { 238 239 if (mb->mo_type == PIPE_MPEG12_MOTION_TYPE_FRAME) { 240 mv[2].x = mv[3].x = mb->mv[1].top.x; 241 mv[2].y = mv[3].y = mb->mv[1].top.y; 242 243 } else { 244 mv[2].x = mb->mv[1].top.x; 245 mv[2].y = mb->mv[1].top.y - (mb->mv[1].top.y % 4); 246 247 mv[3].x = mb->mv[1].bottom.x; 248 mv[3].y = mb->mv[1].bottom.y - (mb->mv[1].bottom.y % 4); 249 250 if (mb->mv[1].top.field_select) mv[2].y += 2; 251 if (!mb->mv[1].bottom.field_select) mv[3].y -= 2; 252 } 253 } else { 254 mv[2].x = mv[2].y = mv[3].x = mv[3].y = 0x8000; 255 } 256} 257 258void 259vl_vb_add_block(struct vl_vertex_buffer *buffer, struct pipe_mpeg12_macroblock *mb, 260 const unsigned (*empty_block_mask)[3][2][2]) 261{ 262 struct vl_vertex_stream *stream; 263 unsigned i, j, k; 264 265 assert(buffer); 266 assert(mb); 267 268 if(mb->cbp) 269 stream = buffer->start + buffer->num_not_empty++; 270 else 271 stream = buffer->end - ++buffer->num_empty; 272 273 stream->pos.x = mb->mbx; 274 stream->pos.y = mb->mby; 275 276 for ( i = 0; i < 3; ++i) 277 for ( j = 0; j < 2; ++j) 278 for ( k = 0; k < 2; ++k) 279 stream->eb[i][j][k] = !(mb->cbp & (*empty_block_mask)[i][j][k]); 280 281 stream->dct_type_field = mb->dct_type == PIPE_MPEG12_DCT_TYPE_FIELD; 282 stream->mo_type_frame = mb->mo_type == PIPE_MPEG12_MOTION_TYPE_FRAME; 283 stream->mb_type_intra = mb->mb_type != PIPE_MPEG12_MACROBLOCK_TYPE_INTRA; 284 switch (mb->mb_type) { 285 case PIPE_MPEG12_MACROBLOCK_TYPE_FWD: 286 stream->mv_wheights = 0; 287 break; 288 289 case PIPE_MPEG12_MACROBLOCK_TYPE_BI: 290 stream->mv_wheights = 1; 291 break; 292 293 case PIPE_MPEG12_MACROBLOCK_TYPE_BKWD: 294 stream->mv_wheights = 2; 295 break; 296 297 default: 298 stream->mv_wheights = 0; 299 } 300 301 get_motion_vectors(mb, stream->mv); 302} 303 304void 305vl_vb_unmap(struct vl_vertex_buffer *buffer, struct pipe_context *pipe) 306{ 307 assert(buffer && pipe); 308 309 pipe_buffer_unmap(pipe, buffer->transfer); 310} 311 312void 313vl_vb_restart(struct vl_vertex_buffer *buffer, 314 unsigned *not_empty_start_instance, unsigned *not_empty_num_instances, 315 unsigned *empty_start_instance, unsigned *empty_num_instances) 316{ 317 assert(buffer); 318 319 *not_empty_start_instance = 0; 320 *not_empty_num_instances = buffer->num_not_empty; 321 *empty_start_instance = buffer->size - buffer->num_empty; 322 *empty_num_instances = buffer->num_empty; 323 324 buffer->num_not_empty = 0; 325 buffer->num_empty = 0; 326} 327 328void 329vl_vb_cleanup(struct vl_vertex_buffer *buffer) 330{ 331 assert(buffer); 332 333 pipe_resource_reference(&buffer->resource, NULL); 334} 335