mirror of https://github.com/FFmpeg/FFmpeg.git
On M1 Max, this supports profiles 0 and 2, but not 1 and 3.pull/375/head
rcombs
3 years ago
parent
e3c6cc96cf
commit
a41a2efc85
8 changed files with 171 additions and 3 deletions
@ -0,0 +1,140 @@ |
||||
/*
|
||||
* Videotoolbox hardware acceleration for VP9 |
||||
* |
||||
* copyright (c) 2021 rcombs |
||||
* |
||||
* This file is part of FFmpeg. |
||||
* |
||||
* FFmpeg is free software; you can redistribute it and/or |
||||
* modify it under the terms of the GNU Lesser General Public |
||||
* License as published by the Free Software Foundation; either |
||||
* version 2.1 of the License, or (at your option) any later version. |
||||
* |
||||
* FFmpeg is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
||||
* Lesser General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU Lesser General Public |
||||
* License along with FFmpeg; if not, write to the Free Software |
||||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
||||
*/ |
||||
|
||||
#include "config.h" |
||||
#include "videotoolbox.h" |
||||
#include "libavutil/hwcontext_videotoolbox.h" |
||||
#include "vt_internal.h" |
||||
#include "libavutil/avutil.h" |
||||
#include "libavutil/frame.h" |
||||
#include "libavutil/hwcontext.h" |
||||
#include "libavutil/intreadwrite.h" |
||||
#include "libavutil/pixdesc.h" |
||||
#include "decode.h" |
||||
#include "internal.h" |
||||
#include "vp9shared.h" |
||||
|
||||
enum VPX_CHROMA_SUBSAMPLING |
||||
{ |
||||
VPX_SUBSAMPLING_420_VERTICAL = 0, |
||||
VPX_SUBSAMPLING_420_COLLOCATED_WITH_LUMA = 1, |
||||
VPX_SUBSAMPLING_422 = 2, |
||||
VPX_SUBSAMPLING_444 = 3, |
||||
}; |
||||
|
||||
static int get_vpx_chroma_subsampling(enum AVPixelFormat pixel_format, |
||||
enum AVChromaLocation chroma_location) |
||||
{ |
||||
int chroma_w, chroma_h; |
||||
if (av_pix_fmt_get_chroma_sub_sample(pixel_format, &chroma_w, &chroma_h) == 0) { |
||||
if (chroma_w == 1 && chroma_h == 1) { |
||||
return (chroma_location == AVCHROMA_LOC_LEFT) |
||||
? VPX_SUBSAMPLING_420_VERTICAL |
||||
: VPX_SUBSAMPLING_420_COLLOCATED_WITH_LUMA; |
||||
} else if (chroma_w == 1 && chroma_h == 0) { |
||||
return VPX_SUBSAMPLING_422; |
||||
} else if (chroma_w == 0 && chroma_h == 0) { |
||||
return VPX_SUBSAMPLING_444; |
||||
} |
||||
} |
||||
return -1; |
||||
} |
||||
|
||||
CFDataRef ff_videotoolbox_vpcc_extradata_create(AVCodecContext *avctx) |
||||
{ |
||||
const VP9SharedContext *h = avctx->priv_data; |
||||
CFDataRef data = NULL; |
||||
uint8_t *p; |
||||
int vt_extradata_size; |
||||
uint8_t *vt_extradata; |
||||
int subsampling = get_vpx_chroma_subsampling(avctx->sw_pix_fmt, avctx->chroma_sample_location); |
||||
|
||||
vt_extradata_size = 1 + 3 + 6 + 2; |
||||
vt_extradata = av_malloc(vt_extradata_size); |
||||
|
||||
if (subsampling < 0) |
||||
return NULL; |
||||
|
||||
if (!vt_extradata) |
||||
return NULL; |
||||
|
||||
p = vt_extradata; |
||||
|
||||
*p++ = 1; /* version */ |
||||
AV_WB24(p + 1, 0); /* flags */ |
||||
p += 3; |
||||
|
||||
*p++ = h->h.profile; |
||||
*p++ = avctx->level; |
||||
*p++ = (h->h.bpp << 4) | (subsampling << 1) | (avctx->color_range == AVCOL_RANGE_JPEG); |
||||
*p++ = avctx->color_primaries; |
||||
*p++ = avctx->color_trc; |
||||
*p++ = avctx->colorspace; |
||||
|
||||
AV_WB16(p + 0, 0); |
||||
p += 2; |
||||
|
||||
av_assert0(p - vt_extradata == vt_extradata_size); |
||||
|
||||
data = CFDataCreate(kCFAllocatorDefault, vt_extradata, vt_extradata_size); |
||||
av_free(vt_extradata); |
||||
return data; |
||||
} |
||||
|
||||
static int videotoolbox_vp9_start_frame(AVCodecContext *avctx, |
||||
const uint8_t *buffer, |
||||
uint32_t size) |
||||
{ |
||||
return 0; |
||||
} |
||||
|
||||
static int videotoolbox_vp9_decode_slice(AVCodecContext *avctx, |
||||
const uint8_t *buffer, |
||||
uint32_t size) |
||||
{ |
||||
VTContext *vtctx = avctx->internal->hwaccel_priv_data; |
||||
|
||||
return ff_videotoolbox_buffer_copy(vtctx, buffer, size); |
||||
} |
||||
|
||||
static int videotoolbox_vp9_end_frame(AVCodecContext *avctx) |
||||
{ |
||||
const VP9SharedContext *h = avctx->priv_data; |
||||
AVFrame *frame = h->frames[CUR_FRAME].tf.f; |
||||
|
||||
return ff_videotoolbox_common_end_frame(avctx, frame); |
||||
} |
||||
|
||||
const AVHWAccel ff_vp9_videotoolbox_hwaccel = { |
||||
.name = "vp9_videotoolbox", |
||||
.type = AVMEDIA_TYPE_VIDEO, |
||||
.id = AV_CODEC_ID_VP9, |
||||
.pix_fmt = AV_PIX_FMT_VIDEOTOOLBOX, |
||||
.alloc_frame = ff_videotoolbox_alloc_frame, |
||||
.start_frame = videotoolbox_vp9_start_frame, |
||||
.decode_slice = videotoolbox_vp9_decode_slice, |
||||
.end_frame = videotoolbox_vp9_end_frame, |
||||
.frame_params = ff_videotoolbox_frame_params, |
||||
.init = ff_videotoolbox_common_init, |
||||
.uninit = ff_videotoolbox_uninit, |
||||
.priv_data_size = sizeof(VTContext), |
||||
}; |
||||
Loading…
Reference in new issue