2013-04-25 2 views
1

Я начинаю реализовывать видеодекодер custum, который использует cuda HW-декодер для генерации кадра YUV для последующего кодирования.Видеодекодер на Cuda ffmpeg

Как я могу заполнить «CUVIDPICPARAMS» struc ??? Возможно ли это?

Мой алгоритм являются:

Для ПОЛУЧИТЬ пакета видеопотока я использовать FFmpeg-DEV ЛИЭС avcodec, avformat ...

Мои шаги:

1) Открыть файл ввода:

avformat_open_input(&ff_formatContext,in_filename,nullptr,nullptr); 

2) Получить видеопоток отеле находится:

avformat_find_stream_info(ff_formatContext,nullptr); 

3) Получить видеопоток:

ff_video_stream=ff_formatContext->streams[i]; 

4) Получить CUDA устройство и инициализировать его:

cuDeviceGet(&cu_device,0); 
CUcontext cu_vid_ctx; 

5) Иниц видео CUDA декодер и установить создать Params:

CUVIDDECODECREATEINFO *cu_decoder_info=new CUVIDDECODECREATEINFO; 
memset(cu_decoder_info,0,sizeof(CUVIDDECODECREATEINFO)); 
... 
cuvidCreateDecoder(cu_video_decoder,cu_decoder_info); 

6) Считать данные кадра в AVpacket

av_read_frame(ff_formatContext,ff_packet); 

и теперь я НЕОБХОДИМО декодировать кадр пакет на CUDA видеодекодер, в теоретическом являются:

cuvidDecodePicture(pDecoder,&picParams); 

Но прежде, чем мне нужно заполнить CUVIDPICPARAMS

CUVIDPICPARAMS picParams; // = новые CUVIDPICPARAMS; memset (& picParams, 0, sizeof (CUVIDPICPARAMS));

HOW CAN I FILL "CUVIDPICPARAMS" struc ??? 

typedef struct _CUVIDPICPARAMS 
{ 
    int PicWidthInMbs;  // Coded Frame Size 
    int FrameHeightInMbs; // Coded Frame Height 
    int CurrPicIdx;   // Output index of the current picture 
    int field_pic_flag;  // 0=frame picture, 1=field picture 
    int bottom_field_flag; // 0=top field, 1=bottom field (ignored if field_pic_flag=0) 
    int second_field;  // Second field of a complementary field pair 
    // Bitstream data 
    unsigned int nBitstreamDataLen;  // Number of bytes in bitstream data buffer 
    const unsigned char *pBitstreamData; // Ptr to bitstream data for this picture (slice-layer) 
    unsigned int nNumSlices;    // Number of slices in this picture 
    const unsigned int *pSliceDataOffsets; // nNumSlices entries, contains offset of each slice within the bitstream data buffer 
    int ref_pic_flag;  // This picture is a reference picture 
    int intra_pic_flag;  // This picture is entirely intra coded 
    unsigned int Reserved[30];    // Reserved for future use 
    // Codec-specific data 
    union { 
     CUVIDMPEG2PICPARAMS mpeg2;   // Also used for MPEG-1 
     CUVIDH264PICPARAMS h264; 
     CUVIDVC1PICPARAMS vc1; 
     CUVIDMPEG4PICPARAMS mpeg4; 
     CUVIDJPEGPICPARAMS jpeg; 
     unsigned int CodecReserved[1024]; 
    } CodecSpecific; 
} CUVIDPICPARAMS; 

typedef struct _CUVIDH264PICPARAMS 
{ 
    // SPS 
    int log2_max_frame_num_minus4; 
    int pic_order_cnt_type; 
    int log2_max_pic_order_cnt_lsb_minus4; 
    int delta_pic_order_always_zero_flag; 
    int frame_mbs_only_flag; 
    int direct_8x8_inference_flag; 
    int num_ref_frames;    // NOTE: shall meet level 4.1 restrictions 
    unsigned char residual_colour_transform_flag; 
    unsigned char bit_depth_luma_minus8; // Must be 0 (only 8-bit supported) 
    unsigned char bit_depth_chroma_minus8; // Must be 0 (only 8-bit supported) 
    unsigned char qpprime_y_zero_transform_bypass_flag; 
    // PPS 
    int entropy_coding_mode_flag; 
    int pic_order_present_flag; 
    int num_ref_idx_l0_active_minus1; 
    int num_ref_idx_l1_active_minus1; 
    int weighted_pred_flag; 
    int weighted_bipred_idc; 
    int pic_init_qp_minus26; 
    int deblocking_filter_control_present_flag; 
    int redundant_pic_cnt_present_flag; 
    int transform_8x8_mode_flag; 
    int MbaffFrameFlag; 
    int constrained_intra_pred_flag; 
    int chroma_qp_index_offset; 
    int second_chroma_qp_index_offset; 
    int ref_pic_flag; 
    int frame_num; 
    int CurrFieldOrderCnt[2]; 
    // DPB 
    CUVIDH264DPBENTRY dpb[16];   // List of reference frames within the DPB 
    // Quantization Matrices (raster-order) 
    unsigned char WeightScale4x4[6][16]; 
    unsigned char WeightScale8x8[2][64]; 
    // FMO/ASO 
    unsigned char fmo_aso_enable; 
    unsigned char num_slice_groups_minus1; 
    unsigned char slice_group_map_type; 
    signed char pic_init_qs_minus26; 
    unsigned int slice_group_change_rate_minus1; 
    union 
    { 
     unsigned long long slice_group_map_addr; 
     const unsigned char *pMb2SliceGroupMap; 
    } fmo; 
    unsigned int Reserved[12]; 
    // SVC/MVC 
    union 
    { 
     CUVIDH264MVCEXT mvcext; 
     CUVIDH264SVCEXT svcext; 
    }; 
} CUVIDH264PICPARAMS; 
+0

Возможный дубликат [Декодирование видео с CUDA nccuvid и ffmpeg] (http://stackoverflow.com/questions/16203930/decode-video-with-cuda-nccuvid-and-ffmpeg) – talonmies

+0

Вы уже задали этот вопрос. Пожалуйста, не спрашивайте об этом еще раз. – talonmies

+0

Вы просмотрели декодер видео NVIDIA [образцы] (http://docs.nvidia.com/cuda/cuda-samples/index.html#cuda-video-decoder-d3d9-api)? –

ответ

3

Это цель объекта CUvideoparser. Вы подаете поток данных кадр за кадром через cuvidParseVideoData, и он перезвонит вам с CUVIDPICPARAMS, готовым перейти к декодеру, когда он обнаружит, что он имеет полный фрейм.

Все это и многое другое хорошо иллюстрируется в примере декодирования D3D9, доступном here. Я предлагаю изучить его подробно, потому что для этого API не так много документации.

+0

Да, я знаю и использую образец декодирования D3D9, но я не использую CUvideoparser! Я использую ffmpeg -> AVpacket, чтобы получить потоковые данные пакета из источника видео. Поэтому мне нужно правильно заполнить CUVIDPICPARAMS. Как я узнал спецификацию codec из ffmpeg, чтобы разобрать ее в CUVIDPICPARAMS struc? –

+0

Я не слишком хорошо знаком с FFmpeg, но я бы предположил, что вы можете просто передать данные AVpacket в CUvideoparser, и он заполнит CUVIDPICPARAMS для вас. Есть ли причина, по которой вы ее не используете? – Asik

+0

CUvideoparser не читает файлы видео mkv для меня, разные файлы -различные контейнеры –