summaryrefslogtreecommitdiff
path: root/src/va.h
diff options
context:
space:
mode:
authorAustin Yuan <shengquan.yuan@intel.com>2008-04-29 18:00:52 -0400
committerAustin Yuan <shengquan.yuan@intel.com>2008-04-29 18:00:52 -0400
commitb55eb7baf850eeb1c4a1f235fe050cd0e2818e3c (patch)
tree930971f069e9e08e1286da5507e3ba673ad9b6f5 /src/va.h
parente9a72caabb78327fb39b2be09824fbf78a357a69 (diff)
downloadlibva-b55eb7baf850eeb1c4a1f235fe050cd0e2818e3c.tar.gz
Add data structures of encode into va.h
Diffstat (limited to 'src/va.h')
-rwxr-xr-xsrc/va.h268
1 files changed, 267 insertions, 1 deletions
diff --git a/src/va.h b/src/va.h
index 739af99..0384f16 100755
--- a/src/va.h
+++ b/src/va.h
@@ -189,6 +189,7 @@ typedef enum
VAProfileVC1Simple = 8,
VAProfileVC1Main = 9,
VAProfileVC1Advanced = 10,
+ VAProfileH263Baseline = 11
} VAProfile;
/*
@@ -201,6 +202,8 @@ typedef enum
VAEntrypointIDCT = 3,
VAEntrypointMoComp = 4,
VAEntrypointDeblocking = 5,
+ /* Encode entrypoints */
+ VAEntrypointVLC = 6, /* slice level encode */
} VAEntrypoint;
/* Currently defined configuration attribute types */
@@ -211,6 +214,7 @@ typedef enum
VAConfigAttribSpatialClipping = 2,
VAConfigAttribIntraResidual = 3,
VAConfigAttribEncryption = 4,
+ VAConfigAttribRateControl = 5,
} VAConfigAttribType;
/*
@@ -229,6 +233,11 @@ typedef struct _VAConfigAttrib {
#define VA_RT_FORMAT_YUV422 0x00000002
#define VA_RT_FORMAT_YUV444 0x00000004
+/* attribute value for VAConfigAttribRateControl */
+#define VA_RC_NONE 0x00000001
+#define VA_RC_CBR 0x00000002
+#define VA_RC_VBR 0x00000004
+
/*
* if an attribute is not applicable for a given
* profile/entrypoint pair, then set the value to the following
@@ -445,7 +454,11 @@ typedef enum
VAMacroblockParameterBufferType = 6,
VAResidualDataBufferType = 7,
VADeblockingParameterBufferType = 8,
- VAImageBufferType = 9
+ VAImageBufferType = 9,
+ VAEncCodedBufferType = 10,
+ VAEncSequenceParameterBufferType = 11,
+ VAEncPictureParameterBufferType = 12,
+ VAEncSliceParameterBufferType = 13,
} VABufferType;
/****************************
@@ -957,6 +970,103 @@ typedef struct _VASliceParameterBufferH264
short chroma_offset_l1[32][2];
} VASliceParameterBufferH264;
+/*****************************************
+ Encode Data Structures
+ ****************************************/
+typedef struct _VAEncSequenceParameterBufferH264
+{
+ unsigned char level_idc;
+ unsigned int picture_width_in_mbs;
+ unsigned int picture_height_in_mbs;
+ unsigned int bits_per_second;
+ unsigned int frame_rate;
+ int initial_qp;
+ int min_qp;
+ unsigned int basic_unit_size;
+ unsigned char vui_flag;
+} VAEncSequenceParameterBufferH264;
+
+typedef struct _VAEncVUIParameterBufferH264
+{
+ unsigned int time_scale;
+ unsigned char initial_cpb_removeal_delay_length_minus1;
+ unsigned char dpb_output_delay_length_minus1;
+ unsigned char time_offset_length;
+} VAEncVUIParameterBufferH264;
+
+typedef struct _VAEncSequenceParameterBufferMPEG4
+{
+ unsigned char profile_and_level_indication;
+ unsigned int video_object_layer_width;
+ unsigned int video_object_layer_height;
+ unsigned int vop_time_increment_resolution;
+ unsigned int fixed_vop_time_increment;
+ unsigned int bits_per_second;
+ int initial_qp;
+ int min_qp;
+} VAEncSequenceParameterBufferMPEG4;
+
+typedef struct _VAEncSequenceParameterBufferH263
+{
+ unsigned int bits_per_second;
+ unsigned int frame_rate;
+ int initial_qp;
+ int min_qp;
+} VAEncSequenceParameterBufferH263;
+
+typedef struct _VAEncPictureParameterBufferH264
+{
+ VASurfaceID reference_picture;
+ VASurfaceID reconstructed_picture;
+ VABufferID coded_buf;
+ unsigned short picture_width;
+ unsigned short picture_height;
+ unsigned int frame_num;
+} VAEncPictureParameterBufferH264;
+
+typedef enum
+{
+ VAEncPictureTypeIntra = 0,
+ VAEncPictureTypePredictive = 1,
+ VAEncPictureTypeBidirectional = 2,
+} VAEncPictureType;
+
+typedef struct _VAEncPictureParameterBufferMPEG4
+{
+ VASurfaceID reference_picture;
+ VASurfaceID reconstructed_picture;
+ VABufferID coded_buf;
+ unsigned short picture_width;
+ unsigned short picture_height;
+ unsigned int vop_time_increment;
+ VAEncPictureType picture_type;
+} VAEncPictureParameterBufferMPEG4;
+
+typedef struct _VAEncPictureParameterBufferH263
+{
+ VASurfaceID reference_picture;
+ VASurfaceID reconstructed_picture;
+ VABufferID coded_buf;
+ unsigned short picture_width;
+ unsigned short picture_height;
+ VAEncPictureType picture_type;
+} VAEncPictureParameterBufferH263;
+
+/* Encode Slice Parameter Buffer */
+typedef struct _VAEncSliceParameterBuffer
+{
+ unsigned int start_row_number; /* starting MB row number for this slice */
+ unsigned int slice_height; /* slice height measured in MB */
+ union {
+ struct {
+ unsigned char is_intra : 1;
+ unsigned char deblock_on : 1;
+ };
+ unsigned char flags;
+ };
+} VAEncSliceParameterBuffer;
+
+
/* Buffer functions */
/*
@@ -1706,4 +1816,160 @@ Mostly to demonstrate program flow with no error handling ...
vaSetSubpicturePalette(dpy, subpicture, palette);
vaAssociateSubpicture(dpy, subpicture, surfaces, 1, 0, 0, 296, 400, 128, 16);
vaPutSurface(dpy, surfaces, win, 0, 0, 720, 480, 100, 100, 640, 480, NULL, 0, 0);
+
+/*****************************************************************************/
+
+Sample Encode Program (w/ pseudo code)
+
+Mostly to demonstrate program flow with no error handling ...
+
+/*****************************************************************************/
+
+ /* H.264 VLC encode for a 640x480 frame */
+ int major_ver, minor_ver;
+ vaInitialize(dpy, &major_ver, &minor_ver);
+
+ int max_num_profiles, max_num_entrypoints, max_num_attribs;
+ max_num_profiles = vaMaxNumProfiles(dpy);
+ max_num_entrypoints = vaMaxNumEntrypoints(dpy);
+ max_num_attribs = vaMaxNumAttributes(dpy);
+
+ /* find out whether H.264 BP encode is supported */
+ VAProfile *profiles = malloc(sizeof(VAProfile)*max_num_profiles);
+ int num_profiles;
+ vaQueryConfigProfiles(dpy, profiles, &profiles);
+ /*
+ * traverse "profiles" to locate the one that matches VAProfileH264BP
+ */
+
+ /* now get the available entrypoints for H.264 BP */
+ VAEntrypoint *entrypoints = malloc(sizeof(VAEntrypoint)*max_num_entrypoints);
+ int num_entrypoints;
+ vaQueryConfigEntrypoints(dpy, VAProfileH264Baseline, entrypoints, &num_entrypoints);
+
+ /* traverse "entrypoints" to see whether VLC is there */
+
+ /* Assuming finding VLC, find out the format and rate control mode for the source */
+ VAConfigAttrib attrib;
+ attrib.type = VAConfigAttribRTFormat;
+ vaGetConfigAttributes(dpy, VAProfileH264Baseline, VAEntrypointVLC,
+ &attrib, 1);
+
+ if (attrib[0].value & VA_RT_FORMAT_YUV420)
+ /* Found desired format, keep going */
+
+ if (attrib[1].value & VA_RC_CBR)
+ /* Found desired rate control mode, keep going */
+
+ VAConfigID config_id;
+ vaCreateConfig(dpy, VAProfileMPEG4Simple, VAEntrypointVLC, &attrib[0], 2,
+ &config_id);
+
+ /*
+ * create surfaces for the current target as well as reference frames
+ * we can get by with 3 surfaces for H.264 BP
+ */
+ VASurfaceID surfaces[3];
+ vaCreateSurfaces(dpy, 640, 480, VA_RT_FORMAT_YUV420, 3, surfaces);
+
+ /* Create a context for this encode pipe */
+ VAContextID context;
+ vaCreateContext(dpy, config_id, 640, 480, VA_PROGRESSIVE, surfaces,
+ 3, &context);
+
+ /*
+ * Create a coded buffer
+ */
+ VABufferID coded_buf;
+ int coded_buf_size = 640*480*400/(16*16);
+ vaCreateBuffer(dpy, context, VAEncCodedBufferType, coded_buf_size, 1, NULL, &coded_buf);
+
+
+ /*
+ * Create header buffer for sequence header, picture header, and slice header
+ */
+ VABufferID sequence_param_buf, picture_param_buf, slice_param_buf;
+ vaCreateBuffer(dpy, context, VAEncSequenceParameterBufferType, sizof(VAEncSequenceParameterBufferH264), 1, NULL, &sequence_param_buf);
+ vaCreateBuffer(dpy, context, VAEncPictureParameterBufferType, sizof(VAEncPictureParameterBufferH264), 1, NULL, &picture_param_buf);
+ vaCreateBuffer(dpy, context, VAEncSliceParameterBufferType, sizof(VAEncSliceParameterBuffer), 1, NULL, &slice_param_buf);
+
+ /* Create sequence parameter */
+ VAEncSequenceParameterBufferH264 sequence_param;
+ vaMapBuffer(dpy, sequence_param_buf, &sequence_param);
+ sequence_param.picture_width_in_mbs = 40;
+ sequence_param.picture_width_in_mbs = 30;
+ sequence_param.bits_per_second = 3000000;
+ sequence_param.basic_unit_size = 5;
+ sequence_param.frame_rate = 30;
+ sequence_param.initial_qp = 25;
+ sequence_param.min_qp = 0;
+ vaUnmapBuffer(dpy,sequence_param_buf);
+
+ /*
+ * Derive 3 image from the surface to upload RAW source YUV data
+ */
+ int src_srf=0,rec_srf=1,ref_srf=2;
+ VAImage *source_image[3];
+
+ for (int i=0; i<3;i++)
+ vaDeriveImage(dpy,surface[i], source_image[i]);
+
+ /* encoding 30 frames */
+ for (int i = 0; i <= 30; i++) {
+ /*
+ * Progess the buffers
+ * src_srf rec_srf ref_srf
+ * <--------------------------
+ */
+ int temp = rec_srf;
+ rec_srf = src_srf;
+ src_srf = ref_src;
+ ref_srf = temp;
+
+ void *source_buf;
+ vaMapBuffer(dpy, source_image[src_srf].buf, source_buf);
+ /* put RAW YUV data into source_buf here */
+ vaUnmapBuffer(dpy, source_image[src_srf].buf);
+
+ vaBeginPicture(dpy, context, surfaces[src_srf]);
+
+ if (i==0) {
+ /* Fill sequence header into sequence_header_buf here */
+ vaRenderPicture(dpy,context, &sequence_param, 1);
+ }
+
+ vaMapBuffer(dpy, picture_param_buf, &picture_param);
+ if (i==0) /* first frame refer itself */
+ picture->reference_picture = surfaces[src_srf];
+ else
+ picture_param->reference_picture = surfaces[ref_srf];
+ picture_param->reconstructed_picture = surfaces[rec_srf];
+ picture_param->coded_buf = coded_buf;
+ picture_param->picture_width= 640;
+ picture_param->picture_height = 480;
+ if ((i % 10) == 0) { /* one I frame within 10 frames */
+ picture_param->picture_type = VAEncPictureTypeIntra;
+ } else {
+ picture_param->picture_type = VAEncPictureTypePredictive;
+ }
+ vaUnmapBuffer(dpy,picture_param_buf);
+
+ vaRenderPicture(dpy, context, &picture_param_buf, 1);
+
+ /* Render slice */
+ vaMapBuffer(dpy, slice_param_buf, &slice_param);
+ slice_param->start_row_number = 0;
+ slice_param->slice_height = 480;
+ slice_param->is_intra = 1;
+ vaUnmapBuffer(dpy,slice_buf);
+
+ vaRenderPicture(dpy,context,&slice_param_buf, 1);
+ vaEndPicture(dpy, context);
+
+ void *coded_data;
+ vaMapBuffer(dpy, coded_buf, &coded_data)
+ /* output coded buf from coded_data here */
+ vaUnmapBuffer(dpy, coded_buf);
+ }
+
#endif