1409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan/* 2409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan * Copyright (c) 2012 Intel Corporation. All Rights Reserved. 3409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan * 4409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan * Permission is hereby granted, free of charge, to any person obtaining a 5409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan * copy of this software and associated documentation files (the 6409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan * "Software"), to deal in the Software without restriction, including 7409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan * without limitation the rights to use, copy, modify, merge, publish, 8409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan * distribute, sub license, and/or sell copies of the Software, and to 9409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan * permit persons to whom the Software is furnished to do so, subject to 10409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan * the following conditions: 11409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan * 12409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan * The above copyright notice and this permission notice (including the 13409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan * next paragraph) shall be included in all copies or substantial portions 14409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan * of the Software. 15409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan * 16409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS 17409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF 18409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT. 19409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan * IN NO EVENT SHALL PRECISION INSIGHT AND/OR ITS SUPPLIERS BE LIABLE FOR 20409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan * ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, 21409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE 22409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. 23409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan */ 24409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan/* 25409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan * Simple MPEG-2 encoder based on libVA. 26409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan * 27409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan */ 28409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 29409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan#include "sysdeps.h" 30409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 31409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan#include <getopt.h> 32409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan#include <unistd.h> 33409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 34409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan#include <sys/time.h> 35409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan#include <sys/types.h> 36409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan#include <fcntl.h> 37409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan#include <time.h> 38409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan#include <pthread.h> 39409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 40409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan#include <va/va.h> 41409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan#include <va/va_enc_mpeg2.h> 42409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 43409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan#include "va_display.h" 44409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 45409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan#define START_CODE_PICUTRE 0x00000100 46409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan#define START_CODE_SLICE 0x00000101 47409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan#define START_CODE_USER 0x000001B2 48409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan#define START_CODE_SEQ 0x000001B3 49409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan#define START_CODE_EXT 0x000001B5 50409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan#define START_CODE_GOP 0x000001B8 51409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 52409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan#define CHROMA_FORMAT_RESERVED 0 53409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan#define CHROMA_FORMAT_420 1 54409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan#define CHROMA_FORMAT_422 2 55409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan#define CHROMA_FORMAT_444 3 56409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 57409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan#define MAX_SLICES 128 58409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 59409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanenum { 60409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan MPEG2_MODE_I = 0, 61409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan MPEG2_MODE_IP, 62409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan MPEG2_MODE_IPB, 63409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan}; 64409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 65409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanenum { 66409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan MPEG2_LEVEL_LOW = 0, 67409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan MPEG2_LEVEL_MAIN, 68409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan MPEG2_LEVEL_HIGH, 69409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan}; 70409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 71409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan#define CHECK_VASTATUS(va_status, func) \ 72409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (va_status != VA_STATUS_SUCCESS) { \ 73409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan fprintf(stderr, "%s:%s (%d) failed, exit\n", __func__, func, __LINE__); \ 74409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan exit(1); \ 75409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 76409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 77409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanstatic VAProfile mpeg2_va_profiles[] = { 78409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VAProfileMPEG2Simple, 79409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VAProfileMPEG2Main 80409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan}; 81409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 82409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanstatic struct _mpeg2_sampling_density 83409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan{ 84409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int samplers_per_line; 85409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int line_per_frame; 86409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int frame_per_sec; 87409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan} mpeg2_upper_samplings[2][3] = { 88409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan { { 0, 0, 0 }, 89409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan { 720, 576, 30 }, 90409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan { 0, 0, 0 }, 91409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan }, 92409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 93409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan { { 352, 288, 30 }, 94409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan { 720, 576, 30 }, 95409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan { 1920, 1152, 60 }, 96409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 97409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan}; 98409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 99409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanstruct mpeg2enc_context { 100409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan /* args */ 101409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int rate_control_mode; 102409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int fps; 103409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int mode; /* 0:I, 1:I/P, 2:I/P/B */ 104409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VAProfile profile; 105409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int level; 106409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int width; 107409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int height; 108409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int frame_size; 109409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int num_pictures; 110409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int qp; 111409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan FILE *ifp; 112409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan FILE *ofp; 113409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan unsigned char *frame_data_buffer; 114409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int intra_period; 115409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int ip_period; 116409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int bit_rate; /* in kbps */ 117409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VAEncPictureType next_type; 118409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int next_display_order; 119409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int next_bframes; 120409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int new_sequence; 121409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int new_gop_header; 122409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int gop_header_in_display_order; 123409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 124409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan /* VA resource */ 125409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VADisplay va_dpy; 126409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VAEncSequenceParameterBufferMPEG2 seq_param; 127409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VAEncPictureParameterBufferMPEG2 pic_param; 128409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VAEncSliceParameterBufferMPEG2 slice_param[MAX_SLICES]; 129409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VAContextID context_id; 130409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VAConfigID config_id; 131409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VABufferID seq_param_buf_id; /* Sequence level parameter */ 132409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VABufferID pic_param_buf_id; /* Picture level parameter */ 133409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VABufferID slice_param_buf_id[MAX_SLICES]; /* Slice level parameter, multil slices */ 134409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VABufferID codedbuf_buf_id; /* Output buffer, compressed data */ 135409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VABufferID packed_seq_header_param_buf_id; 136409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VABufferID packed_seq_buf_id; 137409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VABufferID packed_pic_header_param_buf_id; 138409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VABufferID packed_pic_buf_id; 139409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int num_slice_groups; 140409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int codedbuf_i_size; 141409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int codedbuf_pb_size; 142409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 143409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan /* thread */ 144409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pthread_t upload_thread_id; 145409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int upload_thread_value; 146409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int current_input_surface; 147409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int current_upload_surface; 148409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan}; 149409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 150409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan/* 151409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan * mpeg2enc helpers 152409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan */ 153409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan#define BITSTREAM_ALLOCATE_STEPPING 4096 154409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 155409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanstruct __bitstream { 156409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan unsigned int *buffer; 157409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int bit_offset; 158409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int max_size_in_dword; 159409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan}; 160409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 161409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuantypedef struct __bitstream bitstream; 162409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 163409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanstatic unsigned int 164409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanswap32(unsigned int val) 165409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan{ 166409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan unsigned char *pval = (unsigned char *)&val; 167409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 168409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan return ((pval[0] << 24) | 169409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan (pval[1] << 16) | 170409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan (pval[2] << 8) | 171409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan (pval[3] << 0)); 172409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan} 173409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 174409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanstatic void 175409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanbitstream_start(bitstream *bs) 176409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan{ 177409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bs->max_size_in_dword = BITSTREAM_ALLOCATE_STEPPING; 178409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bs->buffer = calloc(bs->max_size_in_dword * sizeof(int), 1); 179409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bs->bit_offset = 0; 180409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan} 181409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 182409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanstatic void 183409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanbitstream_end(bitstream *bs) 184409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan{ 185409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int pos = (bs->bit_offset >> 5); 186409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int bit_offset = (bs->bit_offset & 0x1f); 187409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int bit_left = 32 - bit_offset; 188409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 189409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (bit_offset) { 190409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bs->buffer[pos] = swap32((bs->buffer[pos] << bit_left)); 191409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 192409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan} 193409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 194409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanstatic void 195409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanbitstream_put_ui(bitstream *bs, unsigned int val, int size_in_bits) 196409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan{ 197409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int pos = (bs->bit_offset >> 5); 198409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int bit_offset = (bs->bit_offset & 0x1f); 199409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int bit_left = 32 - bit_offset; 200409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 201409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (!size_in_bits) 202409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan return; 203409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 204409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (size_in_bits < 32) 205409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan val &= ((1 << size_in_bits) - 1); 206409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 207409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bs->bit_offset += size_in_bits; 208409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 209409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (bit_left > size_in_bits) { 210409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bs->buffer[pos] = (bs->buffer[pos] << size_in_bits | val); 211409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } else { 212409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan size_in_bits -= bit_left; 213409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bs->buffer[pos] = (bs->buffer[pos] << bit_left) | (val >> size_in_bits); 214409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bs->buffer[pos] = swap32(bs->buffer[pos]); 215409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 216409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (pos + 1 == bs->max_size_in_dword) { 217409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bs->max_size_in_dword += BITSTREAM_ALLOCATE_STEPPING; 218409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bs->buffer = realloc(bs->buffer, bs->max_size_in_dword * sizeof(unsigned int)); 219409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 220409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 221409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bs->buffer[pos + 1] = val; 222409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 223409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan} 224409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 225409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanstatic void 226409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanbitstream_byte_aligning(bitstream *bs, int bit) 227409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan{ 228409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int bit_offset = (bs->bit_offset & 0x7); 229409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int bit_left = 8 - bit_offset; 230409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int new_val; 231409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 232409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (!bit_offset) 233409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan return; 234409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 235409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan assert(bit == 0 || bit == 1); 236409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 237409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (bit) 238409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan new_val = (1 << bit_left) - 1; 239409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan else 240409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan new_val = 0; 241409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 242409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, new_val, bit_left); 243409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan} 244409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 245409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanstatic struct mpeg2_frame_rate { 246409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int code; 247409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan float value; 248409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan} frame_rate_tab[] = { 249409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan {1, 23.976}, 250409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan {2, 24.0}, 251409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan {3, 25.0}, 252409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan {4, 29.97}, 253409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan {5, 30}, 254409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan {6, 50}, 255409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan {7, 59.94}, 256409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan {8, 60} 257409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan}; 258409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 259409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanstatic int 260409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanfind_frame_rate_code(const VAEncSequenceParameterBufferMPEG2 *seq_param) 261409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan{ 262409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan unsigned int delta = -1; 263409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int code = 1, i; 264409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan float frame_rate_value = seq_param->frame_rate * 265409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan (seq_param->sequence_extension.bits.frame_rate_extension_d + 1) / 266409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan (seq_param->sequence_extension.bits.frame_rate_extension_n + 1); 267409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 268409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan for (i = 0; i < sizeof(frame_rate_tab) / sizeof(frame_rate_tab[0]); i++) { 269409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 270409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (abs(1000 * frame_rate_tab[i].value - 1000 * frame_rate_value) < delta) { 271409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan code = frame_rate_tab[i].code; 272409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan delta = abs(1000 * frame_rate_tab[i].value - 1000 * frame_rate_value); 273409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 274409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 275409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 276409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan return code; 277409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan} 278409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 279409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanstatic void 280409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuansps_rbsp(struct mpeg2enc_context *ctx, 281409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan const VAEncSequenceParameterBufferMPEG2 *seq_param, 282409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream *bs) 283409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan{ 284409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int frame_rate_code = find_frame_rate_code(seq_param); 285409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 286409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (ctx->new_sequence) { 287409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, START_CODE_SEQ, 32); 288409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, seq_param->picture_width, 12); 289409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, seq_param->picture_height, 12); 290409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, seq_param->aspect_ratio_information, 4); 291409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, frame_rate_code, 4); /* frame_rate_code */ 292409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, (seq_param->bits_per_second + 399) / 400, 18); /* the low 18 bits of bit_rate */ 293409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, 1, 1); /* marker_bit */ 294409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, seq_param->vbv_buffer_size, 10); 295409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, 0, 1); /* constraint_parameter_flag, always 0 for MPEG-2 */ 296409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, 0, 1); /* load_intra_quantiser_matrix */ 297409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, 0, 1); /* load_non_intra_quantiser_matrix */ 298409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 299409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_byte_aligning(bs, 0); 300409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 301409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, START_CODE_EXT, 32); 302409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, 1, 4); /* sequence_extension id */ 303409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, seq_param->sequence_extension.bits.profile_and_level_indication, 8); 304409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, seq_param->sequence_extension.bits.progressive_sequence, 1); 305409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, seq_param->sequence_extension.bits.chroma_format, 2); 306409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, seq_param->picture_width >> 12, 2); 307409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, seq_param->picture_height >> 12, 2); 308409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, ((seq_param->bits_per_second + 399) / 400) >> 18, 12); /* bit_rate_extension */ 309409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, 1, 1); /* marker_bit */ 310409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, seq_param->vbv_buffer_size >> 10, 8); 311409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, seq_param->sequence_extension.bits.low_delay, 1); 312409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, seq_param->sequence_extension.bits.frame_rate_extension_n, 2); 313409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, seq_param->sequence_extension.bits.frame_rate_extension_d, 5); 314409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 315409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_byte_aligning(bs, 0); 316409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 317409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 318409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (ctx->new_gop_header) { 319409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, START_CODE_GOP, 32); 320409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, seq_param->gop_header.bits.time_code, 25); 321409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, seq_param->gop_header.bits.closed_gop, 1); 322409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, seq_param->gop_header.bits.broken_link, 1); 323409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 324409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_byte_aligning(bs, 0); 325409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 326409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan} 327409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 328409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanstatic void 329409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanpps_rbsp(const VAEncSequenceParameterBufferMPEG2 *seq_param, 330409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan const VAEncPictureParameterBufferMPEG2 *pic_param, 331409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream *bs) 332409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan{ 333409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int chroma_420_type; 334409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 335409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (seq_param->sequence_extension.bits.chroma_format == CHROMA_FORMAT_420) 336409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan chroma_420_type = pic_param->picture_coding_extension.bits.progressive_frame; 337409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan else 338409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan chroma_420_type = 0; 339409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 340409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, START_CODE_PICUTRE, 32); 341409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, pic_param->temporal_reference, 10); 342409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, 343409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param->picture_type == VAEncPictureTypeIntra ? 1 : 344409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param->picture_type == VAEncPictureTypePredictive ? 2 : 3, 345409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 3); 346409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, 0xFFFF, 16); /* vbv_delay, always 0xFFFF */ 347409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 348409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (pic_param->picture_type == VAEncPictureTypePredictive || 349409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param->picture_type == VAEncPictureTypeBidirectional) { 350409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, 0, 1); /* full_pel_forward_vector, always 0 for MPEG-2 */ 351409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, 7, 3); /* forward_f_code, always 7 for MPEG-2 */ 352409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 353409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 354409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (pic_param->picture_type == VAEncPictureTypeBidirectional) { 355409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, 0, 1); /* full_pel_backward_vector, always 0 for MPEG-2 */ 356409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, 7, 3); /* backward_f_code, always 7 for MPEG-2 */ 357409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 358409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 359409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, 0, 1); /* extra_bit_picture, 0 */ 360409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 361409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_byte_aligning(bs, 0); 362409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 363409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, START_CODE_EXT, 32); 364409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, 8, 4); /* Picture Coding Extension ID: 8 */ 365409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, pic_param->f_code[0][0], 4); 366409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, pic_param->f_code[0][1], 4); 367409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, pic_param->f_code[1][0], 4); 368409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, pic_param->f_code[1][1], 4); 369409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 370409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, pic_param->picture_coding_extension.bits.intra_dc_precision, 2); 371409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, pic_param->picture_coding_extension.bits.picture_structure, 2); 372409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, pic_param->picture_coding_extension.bits.top_field_first, 1); 373409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, pic_param->picture_coding_extension.bits.frame_pred_frame_dct, 1); 374409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, pic_param->picture_coding_extension.bits.concealment_motion_vectors, 1); 375409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, pic_param->picture_coding_extension.bits.q_scale_type, 1); 376409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, pic_param->picture_coding_extension.bits.intra_vlc_format, 1); 377409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, pic_param->picture_coding_extension.bits.alternate_scan, 1); 378409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, pic_param->picture_coding_extension.bits.repeat_first_field, 1); 379409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, chroma_420_type, 1); 380409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, pic_param->picture_coding_extension.bits.progressive_frame, 1); 381409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_put_ui(bs, pic_param->picture_coding_extension.bits.composite_display_flag, 1); 382409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 383409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_byte_aligning(bs, 0); 384409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan} 385409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 386409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanstatic int 387409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanbuild_packed_pic_buffer(const VAEncSequenceParameterBufferMPEG2 *seq_param, 388409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan const VAEncPictureParameterBufferMPEG2 *pic_param, 389409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan unsigned char **header_buffer) 390409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan{ 391409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream bs; 392409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 393409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_start(&bs); 394409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pps_rbsp(seq_param, pic_param, &bs); 395409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_end(&bs); 396409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 397409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan *header_buffer = (unsigned char *)bs.buffer; 398409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan return bs.bit_offset; 399409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan} 400409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 401409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanstatic int 402409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanbuild_packed_seq_buffer(struct mpeg2enc_context *ctx, 403409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan const VAEncSequenceParameterBufferMPEG2 *seq_param, 404409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan unsigned char **header_buffer) 405409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan{ 406409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream bs; 407409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 408409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_start(&bs); 409409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan sps_rbsp(ctx, seq_param, &bs); 410409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan bitstream_end(&bs); 411409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 412409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan *header_buffer = (unsigned char *)bs.buffer; 413409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan return bs.bit_offset; 414409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan} 415409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 416409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan/* 417409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan * mpeg2enc 418409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan */ 419409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan#define SID_INPUT_PICTURE_0 0 420409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan#define SID_INPUT_PICTURE_1 1 421409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan#define SID_REFERENCE_PICTURE_L0 2 422409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan#define SID_REFERENCE_PICTURE_L1 3 423409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan#define SID_RECON_PICTURE 4 424409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan#define SID_NUMBER SID_RECON_PICTURE + 1 425409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 426409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanstatic VASurfaceID surface_ids[SID_NUMBER]; 427409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 428409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan/* 429409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan * upload thread function 430409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan */ 431409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanstatic void * 432409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanupload_yuv_to_surface(void *data) 433409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan{ 434409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan struct mpeg2enc_context *ctx = data; 435409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VAImage surface_image; 436409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VAStatus va_status; 437409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan void *surface_p = NULL; 438409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan unsigned char *y_src, *u_src, *v_src; 439409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan unsigned char *y_dst, *u_dst, *v_dst; 440409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int y_size = ctx->width * ctx->height; 441409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int u_size = (ctx->width >> 1) * (ctx->height >> 1); 442409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int row, col; 443409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan size_t n_items; 444409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 445409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan do { 446409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan n_items = fread(ctx->frame_data_buffer, ctx->frame_size, 1, ctx->ifp); 447409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } while (n_items != 1); 448409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 449409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan va_status = vaDeriveImage(ctx->va_dpy, surface_ids[ctx->current_upload_surface], &surface_image); 450409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan CHECK_VASTATUS(va_status,"vaDeriveImage"); 451409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 452409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan vaMapBuffer(ctx->va_dpy, surface_image.buf, &surface_p); 453409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan assert(VA_STATUS_SUCCESS == va_status); 454409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 455409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan y_src = ctx->frame_data_buffer; 456409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan u_src = ctx->frame_data_buffer + y_size; /* UV offset for NV12 */ 457409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan v_src = ctx->frame_data_buffer + y_size + u_size; 458409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 459409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan y_dst = surface_p + surface_image.offsets[0]; 460409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan u_dst = surface_p + surface_image.offsets[1]; /* UV offset for NV12 */ 461409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan v_dst = surface_p + surface_image.offsets[2]; 462409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 463409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan /* Y plane */ 464409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan for (row = 0; row < surface_image.height; row++) { 465409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan memcpy(y_dst, y_src, surface_image.width); 466409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan y_dst += surface_image.pitches[0]; 467409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan y_src += ctx->width; 468409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 469409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 470409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (surface_image.format.fourcc == VA_FOURCC_NV12) { /* UV plane */ 471409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan for (row = 0; row < surface_image.height / 2; row++) { 472409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan for (col = 0; col < surface_image.width / 2; col++) { 473409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan u_dst[col * 2] = u_src[col]; 474409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan u_dst[col * 2 + 1] = v_src[col]; 475409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 476409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 477409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan u_dst += surface_image.pitches[1]; 478409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan u_src += (ctx->width / 2); 479409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan v_src += (ctx->width / 2); 480409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 481409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } else { 482409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan for (row = 0; row < surface_image.height / 2; row++) { 483409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan for (col = 0; col < surface_image.width / 2; col++) { 484409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan u_dst[col] = u_src[col]; 485409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan v_dst[col] = v_src[col]; 486409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 487409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 488409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan u_dst += surface_image.pitches[1]; 489409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan v_dst += surface_image.pitches[2]; 490409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan u_src += (ctx->width / 2); 491409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan v_src += (ctx->width / 2); 492409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 493409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 494409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 495409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan vaUnmapBuffer(ctx->va_dpy, surface_image.buf); 496409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan vaDestroyImage(ctx->va_dpy, surface_image.image_id); 497409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 498409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan return NULL; 499409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan} 500409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 501409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanstatic void 502409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanmpeg2enc_exit(struct mpeg2enc_context *ctx, int exit_code) 503409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan{ 504409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (ctx->frame_data_buffer) { 505409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan free(ctx->frame_data_buffer); 506409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->frame_data_buffer = NULL; 507409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 508409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 509409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (ctx->ifp) { 510409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan fclose(ctx->ifp); 511409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->ifp = NULL; 512409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 513409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 514409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (ctx->ofp) { 515409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan fclose(ctx->ofp); 516409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->ofp = NULL; 517409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 518409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 519409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan exit(exit_code); 520409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan} 521409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 522409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanstatic void 523409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanusage(char *program) 524409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan{ 525409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan fprintf(stderr, "Usage: %s --help\n", program); 526409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan fprintf(stderr, "\t--help print this message\n"); 527409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan fprintf(stderr, "Usage: %s <width> <height> <ifile> <ofile> [options]\n", program); 528409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan fprintf(stderr, "\t<width> specifies the frame width\n"); 529409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan fprintf(stderr, "\t<height> specifies the frame height\n"); 530409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan fprintf(stderr, "\t<ifile> specifies the I420/IYUV YUV file\n"); 531409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan fprintf(stderr, "\t<ofile> specifies the encoded MPEG-2 file\n"); 532409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan fprintf(stderr, "where options include:\n"); 533409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan fprintf(stderr, "\t--cqp <QP> const qp mode with specified <QP>\n"); 534409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan fprintf(stderr, "\t--fps <FPS> specify the frame rate\n"); 535409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan fprintf(stderr, "\t--mode <MODE> specify the mode 0 (I), 1 (I/P) and 2 (I/P/B)\n"); 536409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan fprintf(stderr, "\t--profile <PROFILE> specify the profile 0(Simple), or 1(Main, default)\n"); 537409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan fprintf(stderr, "\t--level <LEVEL> specify the level 0(Low), 1(Main, default) or 2(High)\n"); 538409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan} 539409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 540409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanvoid 541409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanmpeg2_profile_level(struct mpeg2enc_context *ctx, 542409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int profile, 543409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int level) 544409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan{ 545409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int l = 2, p; 546409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 547409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan for (p = profile; p < 2; p++) { 548409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan for (l = level; l < 3; l++) { 549409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (ctx->width <= mpeg2_upper_samplings[p][l].samplers_per_line && 550409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->height <= mpeg2_upper_samplings[p][l].line_per_frame && 551409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->fps <= mpeg2_upper_samplings[p][l].frame_per_sec) { 552409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 553409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan goto __find; 554409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan break; 555409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 556409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 557409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 558409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 559409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (p == 2) { 560409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan fprintf(stderr, "Warning: can't find a proper profile and level for the specified width/height/fps\n"); 561409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan p = 1; 562409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan l = 2; 563409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 564409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 565409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan__find: 566409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->profile = mpeg2_va_profiles[p]; 567409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->level = l; 568409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan} 569409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 570409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanstatic void 571409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanparse_args(struct mpeg2enc_context *ctx, int argc, char **argv) 572409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan{ 573409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int c, tmp; 574409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int option_index = 0; 575409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan long file_size; 576409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int profile = 1, level = 1; 577409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 578409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan static struct option long_options[] = { 579409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan {"help", no_argument, 0, 'h'}, 580409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan {"cqp", required_argument, 0, 'c'}, 581409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan {"fps", required_argument, 0, 'f'}, 582409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan {"mode", required_argument, 0, 'm'}, 583409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan {"profile", required_argument, 0, 'p'}, 584409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan {"level", required_argument, 0, 'l'}, 585409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan { NULL, 0, NULL, 0 } 586409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan }; 587409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 588409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if ((argc == 2 && strcmp(argv[1], "--help") == 0) || 589409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan (argc < 5)) 590409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan goto print_usage; 591409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 592409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->width = atoi(argv[1]); 593409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->height = atoi(argv[2]); 594409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 595409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (ctx->width <= 0 || ctx->height <= 0) { 596409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan fprintf(stderr, "<width> and <height> must be greater than 0\n"); 597409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan goto err_exit; 598409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 599409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 600409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->ifp = fopen(argv[3], "rb"); 601409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 602409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (ctx->ifp == NULL) { 603409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan fprintf(stderr, "Can't open the input file\n"); 604409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan goto err_exit; 605409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 606409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 607409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan fseek(ctx->ifp, 0l, SEEK_END); 608409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan file_size = ftell(ctx->ifp); 609409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->frame_size = ctx->width * ctx->height * 3 / 2; 610409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 611409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if ((file_size < ctx->frame_size) || 612409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan (file_size % ctx->frame_size)) { 613409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan fprintf(stderr, "The input file size %ld isn't a multiple of the frame size %d\n", file_size, ctx->frame_size); 614409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan goto err_exit; 615409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 616409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 617409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->num_pictures = file_size / ctx->frame_size; 618409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan fseek(ctx->ifp, 0l, SEEK_SET); 619409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 620409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->ofp = fopen(argv[4], "wb"); 621409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 622409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (ctx->ofp == NULL) { 623409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan fprintf(stderr, "Can't create the output file\n"); 624409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan goto err_exit; 625409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 626409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 627409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan opterr = 0; 628409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->fps = 30; 629409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->qp = 8; 630409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->rate_control_mode = VA_RC_CQP; 631409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->mode = MPEG2_MODE_IP; 632409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->profile = VAProfileMPEG2Main; 633409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->level = MPEG2_LEVEL_MAIN; 634409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 635409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan optind = 5; 636409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 637409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan while((c = getopt_long(argc, argv, 638409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan "", 639409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan long_options, 640409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan &option_index)) != -1) { 641409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan switch(c) { 642409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan case 'c': 643409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan tmp = atoi(optarg); 644409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 645409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan /* only support q_scale_type = 0 */ 646409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (tmp > 62 || tmp < 2) { 647409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan fprintf(stderr, "Warning: QP must be in [2, 62]\n"); 648409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 649409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (tmp > 62) 650409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan tmp = 62; 651409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 652409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (tmp < 2) 653409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan tmp = 2; 654409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 655409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 656409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->qp = tmp & 0xFE; 657409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->rate_control_mode = VA_RC_CQP; 658409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 659409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan break; 660409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 661409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan case 'f': 662409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan tmp = atoi(optarg); 663409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 664409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (tmp <= 0) 665409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan fprintf(stderr, "Warning: FPS must be greater than 0\n"); 666409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan else 667409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->fps = tmp; 668409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 669409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->rate_control_mode = VA_RC_CBR; 670409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 671409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan break; 672409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 673409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan case 'm': 674409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan tmp = atoi(optarg); 675409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 676409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (tmp < MPEG2_MODE_I || tmp > MPEG2_MODE_IPB) 677409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan fprintf(stderr, "Waning: MODE must be 0, 1, or 2\n"); 678409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan else 679409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->mode = tmp; 680409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 681409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan break; 682409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 683409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan case 'p': 684409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan tmp = atoi(optarg); 685409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 686409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (tmp < 0 || tmp > 1) 687409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan fprintf(stderr, "Waning: PROFILE must be 0 or 1\n"); 688409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan else 689409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan profile = tmp; 690409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 691409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan break; 692409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 693409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan case 'l': 694409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan tmp = atoi(optarg); 695409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 696409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (tmp < MPEG2_LEVEL_LOW || tmp > MPEG2_LEVEL_HIGH) 697409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan fprintf(stderr, "Waning: LEVEL must be 0, 1, or 2\n"); 698409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan else 699409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan level = tmp; 700409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 701409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan break; 702409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 703409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan case '?': 704409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan fprintf(stderr, "Error: unkown command options\n"); 705409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 706409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan case 'h': 707409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan goto print_usage; 708409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 709409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 710409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 711409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan mpeg2_profile_level(ctx, profile, level); 712409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 713409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan return; 714409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 715409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanprint_usage: 716409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan usage(argv[0]); 717409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanerr_exit: 718409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan mpeg2enc_exit(ctx, 1); 719409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan} 720409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 721409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan/* 722409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan * init 723409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan */ 724409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanvoid 725409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanmpeg2enc_init_sequence_parameter(struct mpeg2enc_context *ctx, 726409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VAEncSequenceParameterBufferMPEG2 *seq_param) 727409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan{ 728409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int profile = 4, level = 8; 729409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 730409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan switch (ctx->profile) { 731409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan case VAProfileMPEG2Simple: 732409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan profile = 5; 733409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan break; 734409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 735409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan case VAProfileMPEG2Main: 736409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan profile = 4; 737409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan break; 738409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 739409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan default: 740409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan assert(0); 741409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan break; 742409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 743409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 744409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan switch (ctx->level) { 745409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan case MPEG2_LEVEL_LOW: 746409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan level = 10; 747409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan break; 748409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 749409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan case MPEG2_LEVEL_MAIN: 750409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan level = 8; 751409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan break; 752409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 753409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan case MPEG2_LEVEL_HIGH: 754409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan level = 4; 755409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan break; 756409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 757409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan default: 758409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan assert(0); 759409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan break; 760409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 761409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 762409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan seq_param->intra_period = ctx->intra_period; 763409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan seq_param->ip_period = ctx->ip_period; /* FIXME: ??? */ 764409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan seq_param->picture_width = ctx->width; 765409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan seq_param->picture_height = ctx->height; 766409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 767409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (ctx->bit_rate > 0) 768409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan seq_param->bits_per_second = 1024 * ctx->bit_rate; /* use kbps as input */ 769409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan else 770409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan seq_param->bits_per_second = 0x3FFFF * 400; 771409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 772409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan seq_param->frame_rate = ctx->fps; 773409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan seq_param->aspect_ratio_information = 1; 774409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan seq_param->vbv_buffer_size = 3; /* B = 16 * 1024 * vbv_buffer_size */ 775409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 776409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan seq_param->sequence_extension.bits.profile_and_level_indication = profile << 4 | level; 777409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan seq_param->sequence_extension.bits.progressive_sequence = 1; /* progressive frame-pictures */ 778409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan seq_param->sequence_extension.bits.chroma_format = CHROMA_FORMAT_420; /* 4:2:0 */ 779409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan seq_param->sequence_extension.bits.low_delay = 0; /* FIXME */ 780409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan seq_param->sequence_extension.bits.frame_rate_extension_n = 0; 781409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan seq_param->sequence_extension.bits.frame_rate_extension_d = 0; 782409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 783409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan seq_param->gop_header.bits.time_code = (1 << 12); /* bit12: marker_bit */ 784409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan seq_param->gop_header.bits.closed_gop = 0; 785409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan seq_param->gop_header.bits.broken_link = 0; 786409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan} 787409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 788409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanstatic void 789409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanmpeg2enc_init_picture_parameter(struct mpeg2enc_context *ctx, 790409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VAEncPictureParameterBufferMPEG2 *pic_param) 791409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan{ 792409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param->forward_reference_picture = VA_INVALID_ID; 793409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param->backward_reference_picture = VA_INVALID_ID; 794409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param->reconstructed_picture = VA_INVALID_ID; 795409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param->coded_buf = VA_INVALID_ID; 796409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param->picture_type = VAEncPictureTypeIntra; 797409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 798409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param->temporal_reference = 0; 799409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param->f_code[0][0] = 0xf; 800409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param->f_code[0][1] = 0xf; 801409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param->f_code[1][0] = 0xf; 802409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param->f_code[1][1] = 0xf; 803409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 804409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param->picture_coding_extension.bits.intra_dc_precision = 0; /* 8bits */ 805409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param->picture_coding_extension.bits.picture_structure = 3; /* frame picture */ 806409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param->picture_coding_extension.bits.top_field_first = 0; 807409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param->picture_coding_extension.bits.frame_pred_frame_dct = 1; /* FIXME */ 808409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param->picture_coding_extension.bits.concealment_motion_vectors = 0; 809409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param->picture_coding_extension.bits.q_scale_type = 0; 810409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param->picture_coding_extension.bits.intra_vlc_format = 0; 811409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param->picture_coding_extension.bits.alternate_scan = 0; 812409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param->picture_coding_extension.bits.repeat_first_field = 0; 813409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param->picture_coding_extension.bits.progressive_frame = 1; 814409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param->picture_coding_extension.bits.composite_display_flag = 0; 815409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan} 816409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 817409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanstatic void 818409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanmpeg2enc_alloc_va_resources(struct mpeg2enc_context *ctx) 819409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan{ 820409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VAEntrypoint *entrypoint_list; 821409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VAConfigAttrib attrib_list[2]; 822409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VAStatus va_status; 823409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int max_entrypoints, num_entrypoints, entrypoint; 824409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int major_ver, minor_ver; 825409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 826409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->va_dpy = va_open_display(); 827409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan va_status = vaInitialize(ctx->va_dpy, 828409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan &major_ver, 829409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan &minor_ver); 830409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan CHECK_VASTATUS(va_status, "vaInitialize"); 831409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 832409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan max_entrypoints = vaMaxNumEntrypoints(ctx->va_dpy); 833409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan entrypoint_list = malloc(max_entrypoints * sizeof(VAEntrypoint)); 834409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan vaQueryConfigEntrypoints(ctx->va_dpy, 835409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->profile, 836409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan entrypoint_list, 837409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan &num_entrypoints); 838409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 839409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan for (entrypoint = 0; entrypoint < num_entrypoints; entrypoint++) { 840409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (entrypoint_list[entrypoint] == VAEntrypointEncSlice) 841409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan break; 842409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 843409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 844409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan free(entrypoint_list); 845409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 846409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (entrypoint == num_entrypoints) { 847409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan /* not find Slice entry point */ 848409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan assert(0); 849409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 850409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 851409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan /* find out the format for the render target, and rate control mode */ 852409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan attrib_list[0].type = VAConfigAttribRTFormat; 853409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan attrib_list[1].type = VAConfigAttribRateControl; 854409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan vaGetConfigAttributes(ctx->va_dpy, 855409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->profile, 856409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VAEntrypointEncSlice, 857409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan &attrib_list[0], 858409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 2); 859409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 860409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if ((attrib_list[0].value & VA_RT_FORMAT_YUV420) == 0) { 861409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan /* not find desired YUV420 RT format */ 862409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan assert(0); 863409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 864409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 865409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if ((attrib_list[1].value & ctx->rate_control_mode) == 0) { 866409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan /* Can't find matched RC mode */ 867409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan fprintf(stderr, "RC mode %d isn't found, exit\n", ctx->rate_control_mode); 868409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan assert(0); 869409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 870409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 871409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan attrib_list[0].value = VA_RT_FORMAT_YUV420; /* set to desired RT format */ 872409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan attrib_list[1].value = ctx->rate_control_mode; /* set to desired RC mode */ 873409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 874409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan va_status = vaCreateConfig(ctx->va_dpy, 875409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->profile, 876409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VAEntrypointEncSlice, 877409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan attrib_list, 878409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 2, 879409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan &ctx->config_id); 880409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan CHECK_VASTATUS(va_status, "vaCreateConfig"); 881409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 882409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan /* Create a context for this decode pipe */ 883409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan va_status = vaCreateContext(ctx->va_dpy, 884409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->config_id, 885409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->width, 886409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->height, 887409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VA_PROGRESSIVE, 888409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 0, 889409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 0, 890409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan &ctx->context_id); 891409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan CHECK_VASTATUS(va_status, "vaCreateContext"); 892409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 893409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan va_status = vaCreateSurfaces(ctx->va_dpy, 894409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VA_RT_FORMAT_YUV420, 895409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->width, 896409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->height, 897409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan surface_ids, 898409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan SID_NUMBER, 899409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan NULL, 900409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 0); 901409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan CHECK_VASTATUS(va_status, "vaCreateSurfaces"); 902409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan} 903409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 904409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanstatic void 905409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanmpeg2enc_init(struct mpeg2enc_context *ctx) 906409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan{ 907409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int i; 908409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 909409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->frame_data_buffer = (unsigned char *)malloc(ctx->frame_size); 910409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->seq_param_buf_id = VA_INVALID_ID; 911409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->pic_param_buf_id = VA_INVALID_ID; 912409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->packed_seq_header_param_buf_id = VA_INVALID_ID; 913409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->packed_seq_buf_id = VA_INVALID_ID; 914409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->packed_pic_header_param_buf_id = VA_INVALID_ID; 915409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->packed_pic_buf_id = VA_INVALID_ID; 916409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->codedbuf_buf_id = VA_INVALID_ID; 917409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->codedbuf_i_size = ctx->frame_size; 918409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->codedbuf_pb_size = 0; 919409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->next_display_order = 0; 920409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->next_type = VAEncPictureTypeIntra; 921409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 922409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (ctx->mode == MPEG2_MODE_I) { 923409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->intra_period = 1; 924409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->ip_period = 0; 925409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } else if (ctx->mode == MPEG2_MODE_IP) { 926409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->intra_period = 16; 927409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->ip_period = 0; 928409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } else { 929409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->intra_period = 16; 930409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->ip_period = 2; 931409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 932409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 933409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->next_bframes = ctx->ip_period; 934409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 935409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->new_sequence = 1; 936409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->new_gop_header = 1; 937409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->gop_header_in_display_order = 0; 938409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 939409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->bit_rate = -1; 940409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 941409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan for (i = 0; i < MAX_SLICES; i++) { 942409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->slice_param_buf_id[i] = VA_INVALID_ID; 943409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 944409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 945409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan mpeg2enc_init_sequence_parameter(ctx, &ctx->seq_param); 946409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan mpeg2enc_init_picture_parameter(ctx, &ctx->pic_param); 947409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan mpeg2enc_alloc_va_resources(ctx); 948409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 949409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan /* thread */ 950409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->current_input_surface = SID_INPUT_PICTURE_0; 951409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->current_upload_surface = SID_INPUT_PICTURE_1; 952409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->upload_thread_value = pthread_create(&ctx->upload_thread_id, 953409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan NULL, 954409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan upload_yuv_to_surface, 955409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx); 956409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan} 957409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 958409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanstatic int 959409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanmpeg2enc_time_code(VAEncSequenceParameterBufferMPEG2 *seq_param, 960409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int num_frames) 961409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan{ 962409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int fps = (int)(seq_param->frame_rate + 0.5); 963409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int time_code = 0; 964409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int time_code_pictures, time_code_seconds, time_code_minutes, time_code_hours; 965409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int drop_frame_flag = 0; 966409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 967409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan assert(fps <= 60); 968409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 969409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan time_code_seconds = num_frames / fps; 970409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan time_code_pictures = num_frames % fps; 971409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan time_code |= time_code_pictures; 972409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 973409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan time_code_minutes = time_code_seconds / 60; 974409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan time_code_seconds = time_code_seconds % 60; 975409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan time_code |= (time_code_seconds << 6); 976409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 977409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan time_code_hours = time_code_minutes / 60; 978409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan time_code_minutes = time_code_minutes % 60; 979409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 980409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan time_code |= (1 << 12); /* marker_bit */ 981409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan time_code |= (time_code_minutes << 13); 982409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 983409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan time_code_hours = time_code_hours % 24; 984409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan time_code |= (time_code_hours << 19); 985409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 986409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan time_code |= (drop_frame_flag << 24); 987409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 988409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan return time_code; 989409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan} 990409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 991409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan/* 992409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan * run 993409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan */ 994409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanstatic void 995409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanmpeg2enc_update_sequence_parameter(struct mpeg2enc_context *ctx, 996409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VAEncPictureType picture_type, 997409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int coded_order, 998409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int display_order) 999409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan{ 1000409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VAEncSequenceParameterBufferMPEG2 *seq_param = &ctx->seq_param; 1001409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1002409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan /* update the time_code info for the new GOP */ 1003409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (ctx->new_gop_header) { 1004409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan seq_param->gop_header.bits.time_code = mpeg2enc_time_code(seq_param, display_order); 1005409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 1006409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan} 1007409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1008409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanstatic void 1009409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanmpeg2enc_update_picture_parameter(struct mpeg2enc_context *ctx, 1010409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VAEncPictureType picture_type, 1011409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int coded_order, 1012409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int display_order) 1013409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan{ 1014409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VAEncPictureParameterBufferMPEG2 *pic_param = &ctx->pic_param; 1015409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1016409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param->picture_type = picture_type; 1017409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param->temporal_reference = (display_order - ctx->gop_header_in_display_order) & 0x3FF; 1018409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param->reconstructed_picture = surface_ids[SID_RECON_PICTURE]; 1019409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param->forward_reference_picture = surface_ids[SID_REFERENCE_PICTURE_L0]; 1020409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param->backward_reference_picture = surface_ids[SID_REFERENCE_PICTURE_L1]; 1021409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1022409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (pic_param->picture_type == VAEncPictureTypeIntra) { 1023409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param->f_code[0][0] = 0xf; 1024409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param->f_code[0][1] = 0xf; 1025409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param->f_code[1][0] = 0xf; 1026409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param->f_code[1][1] = 0xf; 1027409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param->forward_reference_picture = VA_INVALID_SURFACE; 1028409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param->backward_reference_picture = VA_INVALID_SURFACE; 1029409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1030409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } else if (pic_param->picture_type == VAEncPictureTypePredictive) { 1031409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param->f_code[0][0] = 0x1; 1032409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param->f_code[0][1] = 0x1; 1033409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param->f_code[1][0] = 0xf; 1034409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param->f_code[1][1] = 0xf; 1035409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param->forward_reference_picture = surface_ids[SID_REFERENCE_PICTURE_L0]; 1036409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param->backward_reference_picture = VA_INVALID_SURFACE; 1037409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } else if (pic_param->picture_type == VAEncPictureTypeBidirectional) { 1038409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param->f_code[0][0] = 0x1; 1039409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param->f_code[0][1] = 0x1; 1040409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param->f_code[1][0] = 0x1; 1041409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param->f_code[1][1] = 0x1; 1042409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param->forward_reference_picture = surface_ids[SID_REFERENCE_PICTURE_L0]; 1043409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param->backward_reference_picture = surface_ids[SID_REFERENCE_PICTURE_L1]; 1044409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } else { 1045409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan assert(0); 1046409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 1047409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan} 1048409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1049409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanstatic void 1050409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanmpeg2enc_update_picture_parameter_buffer(struct mpeg2enc_context *ctx, 1051409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VAEncPictureType picture_type, 1052409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int coded_order, 1053409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int display_order) 1054409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan{ 1055409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VAEncPictureParameterBufferMPEG2 *pic_param = &ctx->pic_param; 1056409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VAStatus va_status; 1057409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1058409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan /* update the coded buffer id */ 1059409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param->coded_buf = ctx->codedbuf_buf_id; 1060409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan va_status = vaCreateBuffer(ctx->va_dpy, 1061409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->context_id, 1062409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VAEncPictureParameterBufferType, 1063409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan sizeof(*pic_param), 1064409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1, 1065409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param, 1066409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan &ctx->pic_param_buf_id); 1067409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan CHECK_VASTATUS(va_status, "vaCreateBuffer"); 1068409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan} 1069409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1070409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanstatic void 1071409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanmpeg2enc_update_slice_parameter(struct mpeg2enc_context *ctx, VAEncPictureType picture_type) 1072409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan{ 1073409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VAEncSequenceParameterBufferMPEG2 *seq_param; 1074409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VAEncPictureParameterBufferMPEG2 *pic_param; 1075409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VAEncSliceParameterBufferMPEG2 *slice_param; 1076409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VAStatus va_status; 1077409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int i, width_in_mbs, height_in_mbs; 1078409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1079409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pic_param = &ctx->pic_param; 1080409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan assert(pic_param->picture_coding_extension.bits.q_scale_type == 0); 1081409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1082409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan seq_param = &ctx->seq_param; 1083409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan width_in_mbs = (seq_param->picture_width + 15) / 16; 1084409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan height_in_mbs = (seq_param->picture_height + 15) / 16; 1085409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->num_slice_groups = 1; 1086409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1087409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan for (i = 0; i < height_in_mbs; i++) { 1088409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan slice_param = &ctx->slice_param[i]; 1089409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan slice_param->macroblock_address = i * width_in_mbs; 1090409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan slice_param->num_macroblocks = width_in_mbs; 1091409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan slice_param->is_intra_slice = (picture_type == VAEncPictureTypeIntra); 1092409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan slice_param->quantiser_scale_code = ctx->qp / 2; 1093409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 1094409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1095409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan va_status = vaCreateBuffer(ctx->va_dpy, 1096409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->context_id, 1097409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VAEncSliceParameterBufferType, 1098409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan sizeof(*slice_param), 1099409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan height_in_mbs, 1100409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->slice_param, 1101409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->slice_param_buf_id); 1102409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan CHECK_VASTATUS(va_status, "vaCreateBuffer");; 1103409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan} 1104409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1105409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanstatic int 1106409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanbegin_picture(struct mpeg2enc_context *ctx, 1107409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int coded_order, 1108409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int display_order, 1109409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VAEncPictureType picture_type) 1110409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan{ 1111409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VAStatus va_status; 1112409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int tmp; 1113409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VAEncPackedHeaderParameterBuffer packed_header_param_buffer; 1114409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan unsigned int length_in_bits; 1115409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan unsigned char *packed_seq_buffer = NULL, *packed_pic_buffer = NULL; 1116409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1117409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (ctx->upload_thread_value != 0) { 1118409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan fprintf(stderr, "FATAL error!!!\n"); 1119409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan exit(1); 1120409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 1121409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1122409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pthread_join(ctx->upload_thread_id, NULL); 1123409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1124409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->upload_thread_value = -1; 1125409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan tmp = ctx->current_input_surface; 1126409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->current_input_surface = ctx->current_upload_surface; 1127409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->current_upload_surface = tmp; 1128409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1129409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan mpeg2enc_update_sequence_parameter(ctx, picture_type, coded_order, display_order); 1130409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan mpeg2enc_update_picture_parameter(ctx, picture_type, coded_order, display_order); 1131409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1132409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (ctx->new_sequence || ctx->new_gop_header) { 1133409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan assert(picture_type == VAEncPictureTypeIntra); 1134409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan length_in_bits = build_packed_seq_buffer(ctx, &ctx->seq_param, &packed_seq_buffer); 1135409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan packed_header_param_buffer.type = VAEncPackedHeaderMPEG2_SPS; 1136409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan packed_header_param_buffer.has_emulation_bytes = 0; 1137409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan packed_header_param_buffer.bit_length = length_in_bits; 1138409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan va_status = vaCreateBuffer(ctx->va_dpy, 1139409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->context_id, 1140409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VAEncPackedHeaderParameterBufferType, 1141409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan sizeof(packed_header_param_buffer), 1, &packed_header_param_buffer, 1142409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan &ctx->packed_seq_header_param_buf_id); 1143409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan CHECK_VASTATUS(va_status,"vaCreateBuffer"); 1144409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1145409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan va_status = vaCreateBuffer(ctx->va_dpy, 1146409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->context_id, 1147409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VAEncPackedHeaderDataBufferType, 1148409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan (length_in_bits + 7) / 8, 1, packed_seq_buffer, 1149409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan &ctx->packed_seq_buf_id); 1150409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan CHECK_VASTATUS(va_status,"vaCreateBuffer"); 1151409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1152409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan free(packed_seq_buffer); 1153409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 1154409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1155409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan length_in_bits = build_packed_pic_buffer(&ctx->seq_param, &ctx->pic_param, &packed_pic_buffer); 1156409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan packed_header_param_buffer.type = VAEncPackedHeaderMPEG2_PPS; 1157409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan packed_header_param_buffer.has_emulation_bytes = 0; 1158409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan packed_header_param_buffer.bit_length = length_in_bits; 1159409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1160409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan va_status = vaCreateBuffer(ctx->va_dpy, 1161409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->context_id, 1162409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VAEncPackedHeaderParameterBufferType, 1163409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan sizeof(packed_header_param_buffer), 1, &packed_header_param_buffer, 1164409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan &ctx->packed_pic_header_param_buf_id); 1165409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan CHECK_VASTATUS(va_status,"vaCreateBuffer"); 1166409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1167409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan va_status = vaCreateBuffer(ctx->va_dpy, 1168409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->context_id, 1169409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VAEncPackedHeaderDataBufferType, 1170409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan (length_in_bits + 7) / 8, 1, packed_pic_buffer, 1171409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan &ctx->packed_pic_buf_id); 1172409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan CHECK_VASTATUS(va_status,"vaCreateBuffer"); 1173409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1174409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan free(packed_pic_buffer); 1175409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1176409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan /* sequence parameter set */ 1177409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VAEncSequenceParameterBufferMPEG2 *seq_param = &ctx->seq_param; 1178409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan va_status = vaCreateBuffer(ctx->va_dpy, 1179409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->context_id, 1180409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VAEncSequenceParameterBufferType, 1181409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan sizeof(*seq_param), 1182409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1, 1183409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan seq_param, 1184409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan &ctx->seq_param_buf_id); 1185409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan CHECK_VASTATUS(va_status,"vaCreateBuffer");; 1186409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1187409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan /* slice parameter */ 1188409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan mpeg2enc_update_slice_parameter(ctx, picture_type); 1189409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1190409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan return 0; 1191409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan} 1192409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1193409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanstatic int 1194409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanmpeg2enc_render_picture(struct mpeg2enc_context *ctx) 1195409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan{ 1196409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VAStatus va_status; 1197409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VABufferID va_buffers[16]; 1198409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan unsigned int num_va_buffers = 0; 1199409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1200409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan va_buffers[num_va_buffers++] = ctx->seq_param_buf_id; 1201409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan va_buffers[num_va_buffers++] = ctx->pic_param_buf_id; 1202409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1203409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (ctx->packed_seq_header_param_buf_id != VA_INVALID_ID) 1204409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan va_buffers[num_va_buffers++] = ctx->packed_seq_header_param_buf_id; 1205409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1206409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (ctx->packed_seq_buf_id != VA_INVALID_ID) 1207409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan va_buffers[num_va_buffers++] = ctx->packed_seq_buf_id; 1208409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1209409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (ctx->packed_pic_header_param_buf_id != VA_INVALID_ID) 1210409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan va_buffers[num_va_buffers++] = ctx->packed_pic_header_param_buf_id; 1211409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1212409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (ctx->packed_pic_buf_id != VA_INVALID_ID) 1213409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan va_buffers[num_va_buffers++] = ctx->packed_pic_buf_id; 1214409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1215409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan va_status = vaBeginPicture(ctx->va_dpy, 1216409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->context_id, 1217409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan surface_ids[ctx->current_input_surface]); 1218409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan CHECK_VASTATUS(va_status,"vaBeginPicture"); 1219409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1220409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan va_status = vaRenderPicture(ctx->va_dpy, 1221409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->context_id, 1222409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan va_buffers, 1223409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan num_va_buffers); 1224409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan CHECK_VASTATUS(va_status,"vaRenderPicture"); 1225409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1226409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan va_status = vaRenderPicture(ctx->va_dpy, 1227409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->context_id, 1228409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan &ctx->slice_param_buf_id[0], 1229409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->num_slice_groups); 1230409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan CHECK_VASTATUS(va_status,"vaRenderPicture"); 1231409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1232409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan va_status = vaEndPicture(ctx->va_dpy, ctx->context_id); 1233409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan CHECK_VASTATUS(va_status,"vaEndPicture"); 1234409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1235409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan return 0; 1236409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan} 1237409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1238409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanstatic int 1239409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanmpeg2enc_destroy_buffers(struct mpeg2enc_context *ctx, VABufferID *va_buffers, unsigned int num_va_buffers) 1240409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan{ 1241409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VAStatus va_status; 1242409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan unsigned int i; 1243409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1244409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan for (i = 0; i < num_va_buffers; i++) { 1245409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (va_buffers[i] != VA_INVALID_ID) { 1246409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan va_status = vaDestroyBuffer(ctx->va_dpy, va_buffers[i]); 1247409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan CHECK_VASTATUS(va_status,"vaDestroyBuffer"); 1248409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan va_buffers[i] = VA_INVALID_ID; 1249409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 1250409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 1251409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1252409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan return 0; 1253409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan} 1254409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1255409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanstatic void 1256409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanend_picture(struct mpeg2enc_context *ctx, VAEncPictureType picture_type, int next_is_bpic) 1257409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan{ 1258409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VABufferID tempID; 1259409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1260409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan /* Prepare for next picture */ 1261409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan tempID = surface_ids[SID_RECON_PICTURE]; 1262409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1263409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (picture_type != VAEncPictureTypeBidirectional) { 1264409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (next_is_bpic) { 1265409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan surface_ids[SID_RECON_PICTURE] = surface_ids[SID_REFERENCE_PICTURE_L1]; 1266409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan surface_ids[SID_REFERENCE_PICTURE_L1] = tempID; 1267409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } else { 1268409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan surface_ids[SID_RECON_PICTURE] = surface_ids[SID_REFERENCE_PICTURE_L0]; 1269409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan surface_ids[SID_REFERENCE_PICTURE_L0] = tempID; 1270409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 1271409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } else { 1272409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (!next_is_bpic) { 1273409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan surface_ids[SID_RECON_PICTURE] = surface_ids[SID_REFERENCE_PICTURE_L0]; 1274409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan surface_ids[SID_REFERENCE_PICTURE_L0] = surface_ids[SID_REFERENCE_PICTURE_L1]; 1275409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan surface_ids[SID_REFERENCE_PICTURE_L1] = tempID; 1276409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 1277409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 1278409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1279409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan mpeg2enc_destroy_buffers(ctx, &ctx->seq_param_buf_id, 1); 1280409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan mpeg2enc_destroy_buffers(ctx, &ctx->pic_param_buf_id, 1); 1281409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan mpeg2enc_destroy_buffers(ctx, &ctx->packed_seq_header_param_buf_id, 1); 1282409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan mpeg2enc_destroy_buffers(ctx, &ctx->packed_seq_buf_id, 1); 1283409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan mpeg2enc_destroy_buffers(ctx, &ctx->packed_pic_header_param_buf_id, 1); 1284409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan mpeg2enc_destroy_buffers(ctx, &ctx->packed_pic_buf_id, 1); 1285409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan mpeg2enc_destroy_buffers(ctx, &ctx->slice_param_buf_id[0], ctx->num_slice_groups); 1286409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan mpeg2enc_destroy_buffers(ctx, &ctx->codedbuf_buf_id, 1); 1287409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan memset(ctx->slice_param, 0, sizeof(ctx->slice_param)); 1288409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->num_slice_groups = 0; 1289409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan} 1290409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1291409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanstatic int 1292409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanstore_coded_buffer(struct mpeg2enc_context *ctx, VAEncPictureType picture_type) 1293409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan{ 1294409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VACodedBufferSegment *coded_buffer_segment; 1295409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan unsigned char *coded_mem; 1296409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int slice_data_length; 1297409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VAStatus va_status; 1298409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VASurfaceStatus surface_status; 1299409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan size_t w_items; 1300409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1301409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan va_status = vaSyncSurface(ctx->va_dpy, surface_ids[ctx->current_input_surface]); 1302409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan CHECK_VASTATUS(va_status,"vaSyncSurface"); 1303409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1304409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan surface_status = 0; 1305409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan va_status = vaQuerySurfaceStatus(ctx->va_dpy, surface_ids[ctx->current_input_surface], &surface_status); 1306409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan CHECK_VASTATUS(va_status,"vaQuerySurfaceStatus"); 1307409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1308409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan va_status = vaMapBuffer(ctx->va_dpy, ctx->codedbuf_buf_id, (void **)(&coded_buffer_segment)); 1309409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan CHECK_VASTATUS(va_status,"vaMapBuffer"); 1310409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan coded_mem = coded_buffer_segment->buf; 1311409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1312409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (coded_buffer_segment->status & VA_CODED_BUF_STATUS_SLICE_OVERFLOW_MASK) { 1313409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (picture_type == VAEncPictureTypeIntra) 1314409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->codedbuf_i_size *= 2; 1315409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan else 1316409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->codedbuf_pb_size *= 2; 1317409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1318409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan vaUnmapBuffer(ctx->va_dpy, ctx->codedbuf_buf_id); 1319409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan return -1; 1320409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 1321409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1322409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan slice_data_length = coded_buffer_segment->size; 1323409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1324409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan do { 1325409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan w_items = fwrite(coded_mem, slice_data_length, 1, ctx->ofp); 1326409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } while (w_items != 1); 1327409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1328409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (picture_type == VAEncPictureTypeIntra) { 1329409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (ctx->codedbuf_i_size > slice_data_length * 3 / 2) { 1330409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->codedbuf_i_size = slice_data_length * 3 / 2; 1331409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 1332409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1333409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (ctx->codedbuf_pb_size < slice_data_length) { 1334409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->codedbuf_pb_size = slice_data_length; 1335409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 1336409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } else { 1337409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (ctx->codedbuf_pb_size > slice_data_length * 3 / 2) { 1338409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->codedbuf_pb_size = slice_data_length * 3 / 2; 1339409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 1340409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 1341409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1342409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan vaUnmapBuffer(ctx->va_dpy, ctx->codedbuf_buf_id); 1343409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1344409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan return 0; 1345409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan} 1346409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1347409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanstatic void 1348409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanencode_picture(struct mpeg2enc_context *ctx, 1349409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int coded_order, 1350409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int display_order, 1351409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VAEncPictureType picture_type, 1352409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int next_is_bpic, 1353409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int next_display_order) 1354409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan{ 1355409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VAStatus va_status; 1356409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int ret = 0, codedbuf_size; 1357409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1358409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan begin_picture(ctx, coded_order, display_order, picture_type); 1359409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1360409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (1) { 1361409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan /* upload YUV data to VA surface for next frame */ 1362409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (next_display_order >= ctx->num_pictures) 1363409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan next_display_order = ctx->num_pictures - 1; 1364409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1365409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan fseek(ctx->ifp, ctx->frame_size * next_display_order, SEEK_SET); 1366409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->upload_thread_value = pthread_create(&ctx->upload_thread_id, 1367409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan NULL, 1368409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan upload_yuv_to_surface, 1369409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx); 1370409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 1371409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1372409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan do { 1373409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan mpeg2enc_destroy_buffers(ctx, &ctx->codedbuf_buf_id, 1); 1374409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan mpeg2enc_destroy_buffers(ctx, &ctx->pic_param_buf_id, 1); 1375409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1376409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1377409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (VAEncPictureTypeIntra == picture_type) { 1378409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan codedbuf_size = ctx->codedbuf_i_size; 1379409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } else { 1380409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan codedbuf_size = ctx->codedbuf_pb_size; 1381409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 1382409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1383409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan /* coded buffer */ 1384409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan va_status = vaCreateBuffer(ctx->va_dpy, 1385409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->context_id, 1386409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VAEncCodedBufferType, 1387409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan codedbuf_size, 1, NULL, 1388409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan &ctx->codedbuf_buf_id); 1389409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan CHECK_VASTATUS(va_status,"vaCreateBuffer"); 1390409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1391409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan /* picture parameter set */ 1392409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan mpeg2enc_update_picture_parameter_buffer(ctx, picture_type, coded_order, display_order); 1393409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1394409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan mpeg2enc_render_picture(ctx); 1395409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1396409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ret = store_coded_buffer(ctx, picture_type); 1397409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } while (ret); 1398409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1399409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan end_picture(ctx, picture_type, next_is_bpic); 1400409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan} 1401409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1402409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanstatic void 1403409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanupdate_next_frame_info(struct mpeg2enc_context *ctx, 1404409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VAEncPictureType curr_type, 1405409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int curr_coded_order, 1406409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int curr_display_order) 1407409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan{ 1408409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (((curr_coded_order + 1) % ctx->intra_period) == 0) { 1409409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->next_type = VAEncPictureTypeIntra; 1410409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->next_display_order = curr_coded_order + 1; 1411409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1412409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan return; 1413409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 1414409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1415409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (curr_type == VAEncPictureTypeIntra) { 1416409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan assert(curr_display_order == curr_coded_order); 1417409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->next_type = VAEncPictureTypePredictive; 1418409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->next_bframes = ctx->ip_period; 1419409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->next_display_order = curr_display_order + ctx->next_bframes + 1; 1420409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } else if (curr_type == VAEncPictureTypePredictive) { 1421409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (ctx->ip_period == 0) { 1422409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan assert(curr_display_order == curr_coded_order); 1423409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->next_type = VAEncPictureTypePredictive; 1424409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->next_display_order = curr_display_order + 1; 1425409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } else { 1426409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->next_type = VAEncPictureTypeBidirectional; 1427409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->next_display_order = curr_display_order - ctx->next_bframes; 1428409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->next_bframes--; 1429409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 1430409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } else if (curr_type == VAEncPictureTypeBidirectional) { 1431409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (ctx->next_bframes == 0) { 1432409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->next_type = VAEncPictureTypePredictive; 1433409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->next_bframes = ctx->ip_period; 1434409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->next_display_order = curr_display_order + ctx->next_bframes + 2; 1435409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } else { 1436409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->next_type = VAEncPictureTypeBidirectional; 1437409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->next_display_order = curr_display_order + 1; 1438409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->next_bframes--; 1439409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 1440409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 1441409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1442409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (ctx->next_display_order >= ctx->num_pictures) { 1443409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int rtmp = ctx->next_display_order - (ctx->num_pictures - 1); 1444409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->next_display_order = ctx->num_pictures - 1; 1445409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->next_bframes -= rtmp; 1446409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 1447409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan} 1448409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1449409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanstatic void 1450409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanmpeg2enc_run(struct mpeg2enc_context *ctx) 1451409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan{ 1452409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan int display_order = 0, coded_order = 0; 1453409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan VAEncPictureType type; 1454409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1455409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->new_sequence = 1; 1456409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->new_gop_header = 1; 1457409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->gop_header_in_display_order = display_order; 1458409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1459409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan while (coded_order < ctx->num_pictures) { 1460409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan type = ctx->next_type; 1461409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan display_order = ctx->next_display_order; 1462409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan /* follow the IPBxxBPBxxB mode */ 1463409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan update_next_frame_info(ctx, type, coded_order, display_order); 1464409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan encode_picture(ctx, 1465409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan coded_order, 1466409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan display_order, 1467409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan type, 1468409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->next_type == VAEncPictureTypeBidirectional, 1469409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->next_display_order); 1470409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1471409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan /* update gop_header */ 1472409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->new_sequence = 0; 1473409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->new_gop_header = ctx->next_type == VAEncPictureTypeIntra; 1474409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1475409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan if (ctx->new_gop_header) 1476409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan ctx->gop_header_in_display_order += ctx->intra_period; 1477409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1478409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan coded_order++; 1479409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1480409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan fprintf(stderr, "\r %d/%d ...", coded_order, ctx->num_pictures); 1481409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan fflush(stdout); 1482409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan } 1483409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan} 1484409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1485409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan/* 1486409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan * end 1487409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan */ 1488409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanstatic void 1489409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanmpeg2enc_release_va_resources(struct mpeg2enc_context *ctx) 1490409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan{ 1491409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan vaDestroySurfaces(ctx->va_dpy, surface_ids, SID_NUMBER); 1492409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan vaDestroyContext(ctx->va_dpy, ctx->context_id); 1493409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan vaDestroyConfig(ctx->va_dpy, ctx->config_id); 1494409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan vaTerminate(ctx->va_dpy); 1495409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan va_close_display(ctx->va_dpy); 1496409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan} 1497409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1498409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanstatic void 1499409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanmpeg2enc_end(struct mpeg2enc_context *ctx) 1500409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan{ 1501409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan pthread_join(ctx->upload_thread_id, NULL); 1502409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan mpeg2enc_release_va_resources(ctx); 1503409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan} 1504409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1505409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanint 1506409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuanmain(int argc, char *argv[]) 1507409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan{ 1508409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan struct mpeg2enc_context ctx; 1509409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan struct timeval tpstart, tpend; 1510409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan float timeuse; 1511409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1512409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan gettimeofday(&tpstart, NULL); 1513409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1514409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan memset(&ctx, 0, sizeof(ctx)); 1515409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan parse_args(&ctx, argc, argv); 1516409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan mpeg2enc_init(&ctx); 1517409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan mpeg2enc_run(&ctx); 1518409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan mpeg2enc_end(&ctx); 1519409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1520409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan gettimeofday(&tpend, NULL); 1521409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan timeuse = 1000000 * (tpend.tv_sec - tpstart.tv_sec) + tpend.tv_usec - tpstart.tv_usec; 1522409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan timeuse /= 1000000; 1523409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan fprintf(stderr, "\ndone!\n"); 1524409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan fprintf(stderr, "encode %d frames in %f secondes, FPS is %.1f\n", ctx.num_pictures, timeuse, ctx.num_pictures / timeuse); 1525409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1526409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan mpeg2enc_exit(&ctx, 0); 1527409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan 1528409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan return 0; 1529409de6acb4473c973ed2532e340831dc582e5e0eAustin Yuan} 1530