FFmpeg: libavcodec/nvdec_mpeg12.c Source File

FFmpeg

libavcodec

nvdec_mpeg12.c

Go to the documentation of this file.

1 /*

2 * MPEG-1/2 HW decode acceleration through NVDEC

3 *

4 * Copyright (c) 2017 Philip Langdale

5 *

6 * This file is part of FFmpeg.

7 *

8 * FFmpeg is free software; you can redistribute it and/or

9 * modify it under the terms of the GNU Lesser General Public

10 * License as published by the Free Software Foundation; either

11 * version 2.1 of the License, or (at your option) any later version.

12 *

13 * FFmpeg is distributed in the hope that it will be useful,

14 * but WITHOUT ANY WARRANTY; without even the implied warranty of

15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU

16 * Lesser General Public License for more details.

17 *

18 * You should have received a copy of the GNU Lesser General Public

19 * License along with FFmpeg; if not, write to the Free Software

20 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA

21 */

22

23 #include "config_components.h"

24

25 #include "avcodec.h"

26 #include "hwaccel_internal.h"

27 #include "internal.h"

28 #include "mpegutils.h"

29 #include "mpegvideo.h"

30 #include "nvdec.h"

31 #include "decode.h"

32

33 static int nvdec_mpeg12_start_frame(AVCodecContext *avctx, const uint8_t *buffer, uint32_t size)

34 {

35 MpegEncContext *s = avctx->priv_data;

36

37 NVDECContext *ctx = avctx->internal->hwaccel_priv_data;

38 CUVIDPICPARAMS *pp = &ctx->pic_params;

39 CUVIDMPEG2PICPARAMS *ppc = &pp->CodecSpecific.mpeg2;

40 FrameDecodeData *fdd;

41 NVDECFrame *cf;

42 AVFrame *cur_frame = s->cur_pic.ptr->f;

43

44 int ret, i;

45

46 ret = ff_nvdec_start_frame(avctx, cur_frame);

47 if (ret < 0)

48 return ret;

49

50 fdd = (FrameDecodeData*)cur_frame->private_ref->data;

51 cf = (NVDECFrame*)fdd->hwaccel_priv;

52

53 *pp = (CUVIDPICPARAMS) {

54 .PicWidthInMbs = (cur_frame->width + 15) / 16,

55 .FrameHeightInMbs = (cur_frame->height + 15) / 16,

56 .CurrPicIdx = cf->idx,

57

58 .field_pic_flag = s->picture_structure != PICT_FRAME,

59 .bottom_field_flag = s->picture_structure == PICT_BOTTOM_FIELD,

60 .second_field = s->picture_structure != PICT_FRAME && !s->first_field,

61

62 .intra_pic_flag = s->pict_type == AV_PICTURE_TYPE_I,

63 .ref_pic_flag = s->pict_type == AV_PICTURE_TYPE_I ||

64 s->pict_type == AV_PICTURE_TYPE_P,

65

66 .CodecSpecific.mpeg2 = {

67 .ForwardRefIdx = ff_nvdec_get_ref_idx(s->last_pic.ptr ? s->last_pic.ptr->f : NULL),

68 .BackwardRefIdx = ff_nvdec_get_ref_idx(s->next_pic.ptr ? s->next_pic.ptr->f : NULL),

69

70 .picture_coding_type = s->pict_type,

71 .full_pel_forward_vector = s->full_pel[0],

72 .full_pel_backward_vector = s->full_pel[1],

73 .f_code = { { s->mpeg_f_code[0][0],

74 s->mpeg_f_code[0][1] },

75 { s->mpeg_f_code[1][0],

76 s->mpeg_f_code[1][1] } },

77 .intra_dc_precision = s->intra_dc_precision,

78 .frame_pred_frame_dct = s->frame_pred_frame_dct,

79 .concealment_motion_vectors = s->concealment_motion_vectors,

80 .q_scale_type = s->q_scale_type,

81 .intra_vlc_format = s->intra_vlc_format,

82 .alternate_scan = s->alternate_scan,

83 .top_field_first = s->top_field_first,

84 }

85 };

86

87 for (i = 0; i < 64; ++i) {

88 int n = s->idsp.idct_permutation[i];

89 ppc->QuantMatrixIntra[i] = s->intra_matrix[n];

90 ppc->QuantMatrixInter[i] = s->inter_matrix[n];

91 }

92

93 return 0;

94 }

95

96 static int nvdec_mpeg12_frame_params(AVCodecContext *avctx,

97 AVBufferRef *hw_frames_ctx)

98 {

99 // Each frame can at most have one P and one B reference

100 return ff_nvdec_frame_params(avctx, hw_frames_ctx, 2, 0);

101 }

102

103 #if CONFIG_MPEG2_NVDEC_HWACCEL

104 const FFHWAccel ff_mpeg2_nvdec_hwaccel = {

105 .p.name = "mpeg2_nvdec",

106 .p.type = AVMEDIA_TYPE_VIDEO,

107 .p.id = AV_CODEC_ID_MPEG2VIDEO,

108 .p.pix_fmt = AV_PIX_FMT_CUDA,

109 .start_frame = nvdec_mpeg12_start_frame,

110 .end_frame = ff_nvdec_simple_end_frame,

111 .decode_slice = ff_nvdec_simple_decode_slice,

112 .frame_params = nvdec_mpeg12_frame_params,

113 .init = ff_nvdec_decode_init,

114 .uninit = ff_nvdec_decode_uninit,

115 .priv_data_size = sizeof(NVDECContext),

116 };

117 #endif

118

119 #if CONFIG_MPEG1_NVDEC_HWACCEL

120 const FFHWAccel ff_mpeg1_nvdec_hwaccel = {

121 .p.name = "mpeg1_nvdec",

122 .p.type = AVMEDIA_TYPE_VIDEO,

123 .p.id = AV_CODEC_ID_MPEG1VIDEO,

124 .p.pix_fmt = AV_PIX_FMT_CUDA,

125 .start_frame = nvdec_mpeg12_start_frame,

126 .end_frame = ff_nvdec_simple_end_frame,

127 .decode_slice = ff_nvdec_simple_decode_slice,

128 .frame_params = nvdec_mpeg12_frame_params,

129 .init = ff_nvdec_decode_init,

130 .uninit = ff_nvdec_decode_uninit,

131 .priv_data_size = sizeof(NVDECContext),

132 };

133 #endif

#define PICT_FRAME

Definition: mpegutils.h:33

AV_PIX_FMT_CUDA

@ AV_PIX_FMT_CUDA

HW acceleration through CUDA.

Definition: pixfmt.h:260

AVBufferRef::data

uint8_t * data

The data buffer.

Definition: buffer.h:90

ff_nvdec_get_ref_idx

int ff_nvdec_get_ref_idx(AVFrame *frame)

Definition: nvdec.c:747

AVHWAccel p

The public AVHWAccel.

Definition: hwaccel_internal.h:38

FrameDecodeData

This struct stores per-frame lavc-internal data and is attached to it via private_ref.

Definition: decode.h:33

This structure describes decoded (raw) audio or video data.

Definition: frame.h:389

int width

Definition: frame.h:461

PICT_BOTTOM_FIELD

#define PICT_BOTTOM_FIELD

Definition: mpegutils.h:32

Definition: hwaccel_internal.h:34

ff_nvdec_start_frame

int ff_nvdec_start_frame(AVCodecContext *avctx, AVFrame *frame)

Definition: nvdec.c:561

nvdec_mpeg12_frame_params

static int nvdec_mpeg12_frame_params(AVCodecContext *avctx, AVBufferRef *hw_frames_ctx)

Definition: nvdec_mpeg12.c:96

ff_mpeg1_nvdec_hwaccel

const struct FFHWAccel ff_mpeg1_nvdec_hwaccel

s

#define s(width, name)

Definition: cbs_vp9.c:198

Definition: nvdec.h:44

AVFormatContext * ctx

Definition: movenc.c:49

ff_nvdec_simple_end_frame

int ff_nvdec_simple_end_frame(AVCodecContext *avctx)

Definition: nvdec.c:663

#define NULL

Definition: coverity.c:32

hwaccel_internal.h

ff_nvdec_decode_init

int ff_nvdec_decode_init(AVCodecContext *avctx)

Definition: nvdec.c:327

AVCodecContext::internal

struct AVCodecInternal * internal

Private context used for internal data.

Definition: avcodec.h:486

AV_PICTURE_TYPE_I

@ AV_PICTURE_TYPE_I

Intra.

Definition: avutil.h:279

AV_CODEC_ID_MPEG1VIDEO

@ AV_CODEC_ID_MPEG1VIDEO

Definition: codec_id.h:53

AVCodecInternal::hwaccel_priv_data

void * hwaccel_priv_data

hwaccel-specific private data

Definition: internal.h:130

NVDECFrame::idx

unsigned int idx

Definition: nvdec.h:45

int size

Definition: twinvq_data.h:10344

ff_nvdec_decode_uninit

int ff_nvdec_decode_uninit(AVCodecContext *avctx)

Definition: nvdec.c:259

AVFrame::private_ref

AVBufferRef * private_ref

AVBufferRef for internal use by a single libav* library.

Definition: frame.h:785

AVHWAccel::name

const char * name

Name of the hardware accelerated codec.

Definition: avcodec.h:2105

i

#define i(width, name, range_min, range_max)

Definition: cbs_h2645.c:256

ff_mpeg2_nvdec_hwaccel

const struct FFHWAccel ff_mpeg2_nvdec_hwaccel

ret

Definition: filter_design.txt:187

main external API structure.

Definition: avcodec.h:451

AVFrame::height

int height

Definition: frame.h:461

ff_nvdec_simple_decode_slice

int ff_nvdec_simple_decode_slice(AVCodecContext *avctx, const uint8_t *buffer, uint32_t size)

Definition: nvdec.c:673

the frame and frame reference mechanism is intended to as much as expensive copies of that data while still allowing the filters to produce correct results The data is stored in buffers represented by AVFrame structures Several references can point to the same frame buffer

Definition: filter_design.txt:49

ff_nvdec_frame_params

int ff_nvdec_frame_params(AVCodecContext *avctx, AVBufferRef *hw_frames_ctx, int dpb_size, int supports_444)

Definition: nvdec.c:695

AV_PICTURE_TYPE_P

@ AV_PICTURE_TYPE_P

Predicted.

Definition: avutil.h:280

AVMEDIA_TYPE_VIDEO

@ AVMEDIA_TYPE_VIDEO

Definition: avutil.h:201

A reference to a data buffer.

Definition: buffer.h:82

nvdec_mpeg12_start_frame

static int nvdec_mpeg12_start_frame(AVCodecContext *avctx, const uint8_t *buffer, uint32_t size)

Definition: nvdec_mpeg12.c:33

AVCodecContext::priv_data

void * priv_data

Definition: avcodec.h:478

FrameDecodeData::hwaccel_priv

void * hwaccel_priv

Per-frame private data for hwaccels.

Definition: decode.h:51

Definition: nvdec.h:52

AV_CODEC_ID_MPEG2VIDEO

@ AV_CODEC_ID_MPEG2VIDEO

preferred ID for MPEG-1/2 video decoding

Definition: codec_id.h:54

MpegEncContext.

Definition: mpegvideo.h:73

Generated on Fri Aug 22 2025 13:58:43 for FFmpeg by doxygen 1.8.17