**************************************************************************/
#include <assert.h>
-#include <pipe/p_context.h>
-#include <pipe/p_screen.h>
-#include <util/u_memory.h>
-#include <util/u_inlines.h>
-#include <util/u_format.h>
+#include "util/u_format.h"
#include "vl_vertex_buffers.h"
#include "vl_types.h"
-struct vl_vertex_stream
-{
- struct vertex2s pos;
- struct {
- int8_t y;
- int8_t cr;
- int8_t cb;
- int8_t flag;
- } eb[2][2];
- struct vertex2s mv[4];
-};
-
/* vertices for a quad covering a block */
static const struct vertex2f block_quad[4] = {
{0.0f, 0.0f}, {1.0f, 0.0f}, {1.0f, 1.0f}, {0.0f, 1.0f}
};
struct pipe_vertex_buffer
-vl_vb_upload_quads(struct pipe_context *pipe, unsigned blocks_x, unsigned blocks_y)
+vl_vb_upload_quads(struct pipe_context *pipe)
{
struct pipe_vertex_buffer quad;
struct pipe_transfer *buf_transfer;
- struct vertex4f *v;
+ struct vertex2f *v;
- unsigned x, y, i;
+ unsigned i;
assert(pipe);
/* create buffer */
- quad.stride = sizeof(struct vertex4f);
+ quad.stride = sizeof(struct vertex2f);
quad.buffer_offset = 0;
quad.buffer = pipe_buffer_create
(
pipe->screen,
PIPE_BIND_VERTEX_BUFFER,
PIPE_USAGE_STATIC,
- sizeof(struct vertex4f) * 4 * blocks_x * blocks_y
+ sizeof(struct vertex2f) * 4
);
if(!quad.buffer)
(
pipe,
quad.buffer,
- PIPE_TRANSFER_WRITE | PIPE_TRANSFER_DISCARD,
+ PIPE_TRANSFER_WRITE | PIPE_TRANSFER_DISCARD_RANGE,
&buf_transfer
);
- for ( y = 0; y < blocks_y; ++y) {
- for ( x = 0; x < blocks_x; ++x) {
- for (i = 0; i < 4; ++i, ++v) {
- v->x = block_quad[i].x;
- v->y = block_quad[i].y;
+ for (i = 0; i < 4; ++i, ++v) {
+ v->x = block_quad[i].x;
+ v->y = block_quad[i].y;
+ }
+
+ pipe_buffer_unmap(pipe, buf_transfer);
+
+ return quad;
+}
+
+struct pipe_vertex_buffer
+vl_vb_upload_pos(struct pipe_context *pipe, unsigned width, unsigned height)
+{
+ struct pipe_vertex_buffer pos;
+ struct pipe_transfer *buf_transfer;
+ struct vertex2s *v;
+
+ unsigned x, y;
+
+ assert(pipe);
+
+ /* create buffer */
+ pos.stride = sizeof(struct vertex2s);
+ pos.buffer_offset = 0;
+ pos.buffer = pipe_buffer_create
+ (
+ pipe->screen,
+ PIPE_BIND_VERTEX_BUFFER,
+ PIPE_USAGE_STATIC,
+ sizeof(struct vertex2s) * width * height
+ );
+
+ if(!pos.buffer)
+ return pos;
+
+ /* and fill it */
+ v = pipe_buffer_map
+ (
+ pipe,
+ pos.buffer,
+ PIPE_TRANSFER_WRITE | PIPE_TRANSFER_DISCARD_RANGE,
+ &buf_transfer
+ );
- v->z = x;
- v->w = y;
- }
+ for ( y = 0; y < height; ++y) {
+ for ( x = 0; x < width; ++x, ++v) {
+ v->x = x;
+ v->y = y;
}
}
pipe_buffer_unmap(pipe, buf_transfer);
- return quad;
+ return pos;
}
static struct pipe_vertex_element
element.src_offset = 0;
element.instance_divisor = 0;
element.vertex_buffer_index = 0;
- element.src_format = PIPE_FORMAT_R32G32B32A32_FLOAT;
+ element.src_format = PIPE_FORMAT_R32G32_FLOAT;
return element;
}
static void
vl_vb_element_helper(struct pipe_vertex_element* elements, unsigned num_elements,
- unsigned vertex_buffer_index)
+ unsigned vertex_buffer_index)
{
unsigned i, offset = 0;
}
void *
-vl_vb_get_elems_state(struct pipe_context *pipe, bool include_mvs)
+vl_vb_get_ves_ycbcr(struct pipe_context *pipe)
{
struct pipe_vertex_element vertex_elems[NUM_VS_INPUTS];
- unsigned i;
+ assert(pipe);
memset(&vertex_elems, 0, sizeof(vertex_elems));
vertex_elems[VS_I_RECT] = vl_vb_get_quad_vertex_element();
/* Position element */
- vertex_elems[VS_I_VPOS].src_format = PIPE_FORMAT_R16G16_SSCALED;
+ vertex_elems[VS_I_VPOS].src_format = PIPE_FORMAT_R8G8B8A8_USCALED;
+
+ /* block num element */
+ vertex_elems[VS_I_BLOCK_NUM].src_format = PIPE_FORMAT_R32_FLOAT;
+
+ vl_vb_element_helper(&vertex_elems[VS_I_VPOS], 2, 1);
+
+ return pipe->create_vertex_elements_state(pipe, 3, vertex_elems);
+}
+
+void *
+vl_vb_get_ves_mv(struct pipe_context *pipe)
+{
+ struct pipe_vertex_element vertex_elems[NUM_VS_INPUTS];
+
+ assert(pipe);
- /* y, cr, cb empty block element top left block */
- vertex_elems[VS_I_EB_0_0].src_format = PIPE_FORMAT_R8G8B8A8_SSCALED;
+ memset(&vertex_elems, 0, sizeof(vertex_elems));
+ vertex_elems[VS_I_RECT] = vl_vb_get_quad_vertex_element();
- /* y, cr, cb empty block element top right block */
- vertex_elems[VS_I_EB_0_1].src_format = PIPE_FORMAT_R8G8B8A8_SSCALED;
+ /* Position element */
+ vertex_elems[VS_I_VPOS].src_format = PIPE_FORMAT_R16G16_SSCALED;
- /* y, cr, cb empty block element bottom left block */
- vertex_elems[VS_I_EB_1_0].src_format = PIPE_FORMAT_R8G8B8A8_SSCALED;
+ vl_vb_element_helper(&vertex_elems[VS_I_VPOS], 1, 1);
- /* y, cr, cb empty block element bottom right block */
- vertex_elems[VS_I_EB_1_1].src_format = PIPE_FORMAT_R8G8B8A8_SSCALED;
+ /* motion vector TOP element */
+ vertex_elems[VS_I_MV_TOP].src_format = PIPE_FORMAT_R16G16B16A16_SSCALED;
- for (i = 0; i < 4; ++i)
- /* motion vector 0..4 element */
- vertex_elems[VS_I_MV0 + i].src_format = PIPE_FORMAT_R16G16_SSCALED;
+ /* motion vector BOTTOM element */
+ vertex_elems[VS_I_MV_BOTTOM].src_format = PIPE_FORMAT_R16G16B16A16_SSCALED;
- vl_vb_element_helper(&vertex_elems[VS_I_VPOS], NUM_VS_INPUTS - (include_mvs ? 1 : 5), 1);
+ vl_vb_element_helper(&vertex_elems[VS_I_MV_TOP], 2, 2);
- return pipe->create_vertex_elements_state(pipe, NUM_VS_INPUTS - (include_mvs ? 0 : 4), vertex_elems);
+ return pipe->create_vertex_elements_state(pipe, NUM_VS_INPUTS, vertex_elems);
}
-struct pipe_vertex_buffer
-vl_vb_init(struct vl_vertex_buffer *buffer, struct pipe_context *pipe, unsigned size)
+bool
+vl_vb_init(struct vl_vertex_buffer *buffer, struct pipe_context *pipe,
+ unsigned width, unsigned height)
{
- struct pipe_vertex_buffer buf;
+ unsigned i, size;
assert(buffer);
- buffer->size = size;
- buffer->num_not_empty = 0;
- buffer->num_empty = 0;
-
- buf.stride = sizeof(struct vl_vertex_stream);
- buf.buffer_offset = 0;
- buf.buffer = pipe_buffer_create
- (
- pipe->screen,
- PIPE_BIND_VERTEX_BUFFER,
- PIPE_USAGE_STREAM,
- sizeof(struct vl_vertex_stream) * size
- );
+ buffer->width = width;
+ buffer->height = height;
+
+ size = width * height;
+
+ for (i = 0; i < VL_MAX_PLANES; ++i) {
+ buffer->ycbcr[i].resource = pipe_buffer_create
+ (
+ pipe->screen,
+ PIPE_BIND_VERTEX_BUFFER,
+ PIPE_USAGE_STREAM,
+ sizeof(struct vl_ycbcr_block) * size * 4
+ );
+ if (!buffer->ycbcr[i].resource)
+ goto error_ycbcr;
+ }
- pipe_resource_reference(&buffer->resource, buf.buffer);
+ for (i = 0; i < VL_MAX_REF_FRAMES; ++i) {
+ buffer->mv[i].resource = pipe_buffer_create
+ (
+ pipe->screen,
+ PIPE_BIND_VERTEX_BUFFER,
+ PIPE_USAGE_STREAM,
+ sizeof(struct vl_motionvector) * size
+ );
+ if (!buffer->mv[i].resource)
+ goto error_mv;
+ }
vl_vb_map(buffer, pipe);
+ return true;
- return buf;
+error_mv:
+ for (i = 0; i < VL_MAX_PLANES; ++i)
+ pipe_resource_reference(&buffer->mv[i].resource, NULL);
+
+error_ycbcr:
+ for (i = 0; i < VL_MAX_PLANES; ++i)
+ pipe_resource_reference(&buffer->ycbcr[i].resource, NULL);
+ return false;
}
-void
-vl_vb_map(struct vl_vertex_buffer *buffer, struct pipe_context *pipe)
+unsigned
+vl_vb_attributes_per_plock(struct vl_vertex_buffer *buffer)
{
- assert(buffer && pipe);
-
- buffer->start = pipe_buffer_map
- (
- pipe,
- buffer->resource,
- PIPE_TRANSFER_WRITE | PIPE_TRANSFER_DISCARD,
- &buffer->transfer
- );
- buffer->end = buffer->start + buffer->resource->width0 / sizeof(struct vl_vertex_stream);
+ return 1;
}
-static void
-get_motion_vectors(struct pipe_mpeg12_macroblock *mb, struct vertex2s mv[4])
+struct pipe_vertex_buffer
+vl_vb_get_ycbcr(struct vl_vertex_buffer *buffer, int component)
{
- switch (mb->mb_type) {
- case PIPE_MPEG12_MACROBLOCK_TYPE_BI:
- {
- if (mb->mo_type == PIPE_MPEG12_MOTION_TYPE_FRAME) {
- mv[2].x = mb->pmv[0][1][0];
- mv[2].y = mb->pmv[0][1][1];
-
- } else {
- mv[2].x = mb->pmv[0][1][0];
- mv[2].y = mb->pmv[0][1][1] - (mb->pmv[0][1][1] % 4);
-
- mv[3].x = mb->pmv[1][1][0];
- mv[3].y = mb->pmv[1][1][1] - (mb->pmv[1][1][1] % 4);
+ struct pipe_vertex_buffer buf;
- if(mb->mvfs[0][1]) mv[2].y += 2;
- if(!mb->mvfs[1][1]) mv[3].y -= 2;
- }
+ assert(buffer);
- /* fall-through */
- }
- case PIPE_MPEG12_MACROBLOCK_TYPE_FWD:
- case PIPE_MPEG12_MACROBLOCK_TYPE_BKWD:
- {
- if (mb->mb_type == PIPE_MPEG12_MACROBLOCK_TYPE_BKWD) {
+ buf.stride = sizeof(struct vl_ycbcr_block);
+ buf.buffer_offset = 0;
+ buf.buffer = buffer->ycbcr[component].resource;
- if (mb->mo_type == PIPE_MPEG12_MOTION_TYPE_FRAME) {
- mv[0].x = mb->pmv[0][1][0];
- mv[0].y = mb->pmv[0][1][1];
+ return buf;
+}
- } else {
- mv[0].x = mb->pmv[0][1][0];
- mv[0].y = mb->pmv[0][1][1] - (mb->pmv[0][1][1] % 4);
+struct pipe_vertex_buffer
+vl_vb_get_mv(struct vl_vertex_buffer *buffer, int motionvector)
+{
+ struct pipe_vertex_buffer buf;
- mv[1].x = mb->pmv[1][1][0];
- mv[1].y = mb->pmv[1][1][1] - (mb->pmv[1][1][1] % 4);
+ assert(buffer);
- if(mb->mvfs[0][1]) mv[0].y += 2;
- if(!mb->mvfs[1][1]) mv[1].y -= 2;
- }
+ buf.stride = sizeof(struct vl_motionvector);
+ buf.buffer_offset = 0;
+ buf.buffer = buffer->mv[motionvector].resource;
- } else {
+ return buf;
+}
- if (mb->mo_type == PIPE_MPEG12_MOTION_TYPE_FRAME) {
- mv[0].x = mb->pmv[0][0][0];
- mv[0].y = mb->pmv[0][0][1];
+void
+vl_vb_map(struct vl_vertex_buffer *buffer, struct pipe_context *pipe)
+{
+ unsigned i;
- } else {
- mv[0].x = mb->pmv[0][0][0];
- mv[0].y = mb->pmv[0][0][1] - (mb->pmv[0][0][1] % 4);
+ assert(buffer && pipe);
- mv[1].x = mb->pmv[1][0][0];
- mv[1].y = mb->pmv[1][0][1] - (mb->pmv[1][0][1] % 4);
+ for (i = 0; i < VL_MAX_PLANES; ++i) {
+ buffer->ycbcr[i].vertex_stream = pipe_buffer_map
+ (
+ pipe,
+ buffer->ycbcr[i].resource,
+ PIPE_TRANSFER_WRITE | PIPE_TRANSFER_DISCARD_RANGE,
+ &buffer->ycbcr[i].transfer
+ );
+ }
- if(mb->mvfs[0][0]) mv[0].y += 2;
- if(!mb->mvfs[1][0]) mv[1].y -= 2;
- }
- }
- }
- default:
- break;
+ for (i = 0; i < VL_MAX_REF_FRAMES; ++i) {
+ buffer->mv[i].vertex_stream = pipe_buffer_map
+ (
+ pipe,
+ buffer->mv[i].resource,
+ PIPE_TRANSFER_WRITE | PIPE_TRANSFER_DISCARD_RANGE,
+ &buffer->mv[i].transfer
+ );
}
+
}
-void
-vl_vb_add_block(struct vl_vertex_buffer *buffer, struct pipe_mpeg12_macroblock *mb,
- const unsigned (*empty_block_mask)[3][2][2])
+struct vl_ycbcr_block *
+vl_vb_get_ycbcr_stream(struct vl_vertex_buffer *buffer, int component)
{
- struct vl_vertex_stream *stream;
- unsigned i, j;
-
assert(buffer);
- assert(mb);
+ assert(component < VL_MAX_PLANES);
- if(mb->cbp)
- stream = buffer->start + buffer->num_not_empty++;
- else
- stream = buffer->end - ++buffer->num_empty;
-
- stream->pos.x = mb->mbx;
- stream->pos.y = mb->mby;
+ return buffer->ycbcr[component].vertex_stream;
+}
- for ( i = 0; i < 2; ++i) {
- for ( j = 0; j < 2; ++j) {
- stream->eb[i][j].y = !(mb->cbp & (*empty_block_mask)[0][i][j]);
- stream->eb[i][j].cr = !(mb->cbp & (*empty_block_mask)[1][i][j]);
- stream->eb[i][j].cb = !(mb->cbp & (*empty_block_mask)[2][i][j]);
- }
- }
- stream->eb[0][0].flag = mb->dct_type == PIPE_MPEG12_DCT_TYPE_FIELD;
- stream->eb[0][1].flag = mb->mo_type == PIPE_MPEG12_MOTION_TYPE_FRAME;
- stream->eb[1][0].flag = mb->mb_type == PIPE_MPEG12_MACROBLOCK_TYPE_BKWD;
- switch (mb->mb_type) {
- case PIPE_MPEG12_MACROBLOCK_TYPE_INTRA:
- stream->eb[1][1].flag = -1;
- break;
-
- case PIPE_MPEG12_MACROBLOCK_TYPE_FWD:
- case PIPE_MPEG12_MACROBLOCK_TYPE_BKWD:
- stream->eb[1][1].flag = 1;
- break;
-
- case PIPE_MPEG12_MACROBLOCK_TYPE_BI:
- stream->eb[1][1].flag = 0;
- break;
-
- default:
- assert(0);
- }
+unsigned
+vl_vb_get_mv_stream_stride(struct vl_vertex_buffer *buffer)
+{
+ assert(buffer);
- get_motion_vectors(mb, stream->mv);
+ return buffer->width;
}
-void
-vl_vb_unmap(struct vl_vertex_buffer *buffer, struct pipe_context *pipe)
+struct vl_motionvector *
+vl_vb_get_mv_stream(struct vl_vertex_buffer *buffer, int ref_frame)
{
- assert(buffer && pipe);
+ assert(buffer);
+ assert(ref_frame < VL_MAX_REF_FRAMES);
- pipe_buffer_unmap(pipe, buffer->transfer);
+ return buffer->mv[ref_frame].vertex_stream;
}
void
-vl_vb_restart(struct vl_vertex_buffer *buffer,
- unsigned *not_empty_start_instance, unsigned *not_empty_num_instances,
- unsigned *empty_start_instance, unsigned *empty_num_instances)
+vl_vb_unmap(struct vl_vertex_buffer *buffer, struct pipe_context *pipe)
{
- assert(buffer);
+ unsigned i;
+
+ assert(buffer && pipe);
- *not_empty_start_instance = 0;
- *not_empty_num_instances = buffer->num_not_empty;
- *empty_start_instance = buffer->size - buffer->num_empty;
- *empty_num_instances = buffer->num_empty;
+ for (i = 0; i < VL_MAX_PLANES; ++i) {
+ pipe_buffer_unmap(pipe, buffer->ycbcr[i].transfer);
+ }
- buffer->num_not_empty = 0;
- buffer->num_empty = 0;
+ for (i = 0; i < VL_MAX_REF_FRAMES; ++i) {
+ pipe_buffer_unmap(pipe, buffer->mv[i].transfer);
+ }
}
void
vl_vb_cleanup(struct vl_vertex_buffer *buffer)
{
+ unsigned i;
+
assert(buffer);
- pipe_resource_reference(&buffer->resource, NULL);
+ for (i = 0; i < VL_MAX_PLANES; ++i) {
+ pipe_resource_reference(&buffer->ycbcr[i].resource, NULL);
+ }
+
+ for (i = 0; i < VL_MAX_REF_FRAMES; ++i) {
+ pipe_resource_reference(&buffer->mv[i].resource, NULL);
+ }
}