2014-04-05 06:21:19 +00:00
|
|
|
#pragma once
|
|
|
|
|
2021-11-06 22:57:35 +00:00
|
|
|
#include <libavcodec/avcodec.h>
|
2022-11-20 21:48:42 +00:00
|
|
|
#include <libavutil/pixdesc.h>
|
2021-11-06 22:57:35 +00:00
|
|
|
|
2014-04-05 08:13:11 +00:00
|
|
|
static inline int64_t rescale_ts(int64_t val, AVCodecContext *context,
|
|
|
|
AVRational new_base)
|
|
|
|
{
|
|
|
|
return av_rescale_q_rnd(val, context->time_base, new_base,
|
|
|
|
AV_ROUND_NEAR_INF | AV_ROUND_PASS_MINMAX);
|
|
|
|
}
|
|
|
|
|
2014-04-05 06:21:19 +00:00
|
|
|
static inline enum AVPixelFormat
|
|
|
|
obs_to_ffmpeg_video_format(enum video_format format)
|
|
|
|
{
|
|
|
|
switch (format) {
|
2015-04-17 05:52:44 +00:00
|
|
|
case VIDEO_FORMAT_I444:
|
|
|
|
return AV_PIX_FMT_YUV444P;
|
2022-04-19 17:38:28 +00:00
|
|
|
case VIDEO_FORMAT_I412:
|
|
|
|
return AV_PIX_FMT_YUV444P12LE;
|
2014-04-05 06:21:19 +00:00
|
|
|
case VIDEO_FORMAT_I420:
|
|
|
|
return AV_PIX_FMT_YUV420P;
|
|
|
|
case VIDEO_FORMAT_NV12:
|
|
|
|
return AV_PIX_FMT_NV12;
|
|
|
|
case VIDEO_FORMAT_YUY2:
|
|
|
|
return AV_PIX_FMT_YUYV422;
|
|
|
|
case VIDEO_FORMAT_UYVY:
|
|
|
|
return AV_PIX_FMT_UYVY422;
|
2022-04-19 17:39:22 +00:00
|
|
|
case VIDEO_FORMAT_YVYU:
|
|
|
|
return AV_PIX_FMT_YVYU422;
|
2014-04-05 06:21:19 +00:00
|
|
|
case VIDEO_FORMAT_RGBA:
|
|
|
|
return AV_PIX_FMT_RGBA;
|
|
|
|
case VIDEO_FORMAT_BGRA:
|
|
|
|
return AV_PIX_FMT_BGRA;
|
|
|
|
case VIDEO_FORMAT_BGRX:
|
|
|
|
return AV_PIX_FMT_BGRA;
|
2016-02-25 04:02:11 +00:00
|
|
|
case VIDEO_FORMAT_Y800:
|
|
|
|
return AV_PIX_FMT_GRAY8;
|
2019-05-30 13:05:53 +00:00
|
|
|
case VIDEO_FORMAT_BGR3:
|
|
|
|
return AV_PIX_FMT_BGR24;
|
2019-06-18 05:25:18 +00:00
|
|
|
case VIDEO_FORMAT_I422:
|
|
|
|
return AV_PIX_FMT_YUV422P;
|
2022-04-19 17:38:28 +00:00
|
|
|
case VIDEO_FORMAT_I210:
|
|
|
|
return AV_PIX_FMT_YUV422P10LE;
|
2019-08-11 18:26:32 +00:00
|
|
|
case VIDEO_FORMAT_I40A:
|
|
|
|
return AV_PIX_FMT_YUVA420P;
|
|
|
|
case VIDEO_FORMAT_I42A:
|
|
|
|
return AV_PIX_FMT_YUVA422P;
|
|
|
|
case VIDEO_FORMAT_YUVA:
|
|
|
|
return AV_PIX_FMT_YUVA444P;
|
2022-04-19 17:38:28 +00:00
|
|
|
#if LIBAVUTIL_BUILD >= AV_VERSION_INT(56, 31, 100)
|
2022-11-20 21:48:42 +00:00
|
|
|
case VIDEO_FORMAT_YA2L:
|
2022-04-19 17:38:28 +00:00
|
|
|
return AV_PIX_FMT_YUVA444P12LE;
|
|
|
|
#endif
|
2022-03-16 08:32:03 +00:00
|
|
|
case VIDEO_FORMAT_I010:
|
|
|
|
return AV_PIX_FMT_YUV420P10LE;
|
|
|
|
case VIDEO_FORMAT_P010:
|
|
|
|
return AV_PIX_FMT_P010LE;
|
2022-11-21 03:11:27 +00:00
|
|
|
#if LIBAVUTIL_BUILD >= AV_VERSION_INT(57, 17, 100)
|
|
|
|
case VIDEO_FORMAT_P216:
|
|
|
|
return AV_PIX_FMT_P216LE;
|
|
|
|
case VIDEO_FORMAT_P416:
|
|
|
|
return AV_PIX_FMT_P416LE;
|
|
|
|
#endif
|
2019-08-31 05:13:11 +00:00
|
|
|
case VIDEO_FORMAT_NONE:
|
|
|
|
case VIDEO_FORMAT_AYUV:
|
2022-11-20 21:48:42 +00:00
|
|
|
default:
|
2019-08-31 05:13:11 +00:00
|
|
|
return AV_PIX_FMT_NONE;
|
2014-04-05 06:21:19 +00:00
|
|
|
}
|
2022-11-20 21:48:42 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
static enum AVChromaLocation
|
|
|
|
determine_chroma_location(enum AVPixelFormat pix_fmt,
|
|
|
|
enum AVColorSpace colorspace)
|
|
|
|
{
|
|
|
|
const AVPixFmtDescriptor *const desc = av_pix_fmt_desc_get(pix_fmt);
|
|
|
|
if (desc) {
|
|
|
|
const unsigned log_chroma_w = desc->log2_chroma_w;
|
|
|
|
const unsigned log_chroma_h = desc->log2_chroma_h;
|
|
|
|
switch (log_chroma_h) {
|
|
|
|
case 0:
|
|
|
|
switch (log_chroma_w) {
|
|
|
|
case 0:
|
|
|
|
/* 4:4:4 */
|
|
|
|
return AVCHROMA_LOC_CENTER;
|
|
|
|
case 1:
|
|
|
|
/* 4:2:2 */
|
|
|
|
return AVCHROMA_LOC_LEFT;
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
case 1:
|
|
|
|
if (log_chroma_w == 1) {
|
|
|
|
/* 4:2:0 */
|
|
|
|
return (colorspace == AVCOL_SPC_BT2020_NCL)
|
|
|
|
? AVCHROMA_LOC_TOPLEFT
|
|
|
|
: AVCHROMA_LOC_LEFT;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2014-04-05 06:21:19 +00:00
|
|
|
|
2022-11-20 21:48:42 +00:00
|
|
|
return AVCHROMA_LOC_UNSPECIFIED;
|
2014-04-05 06:21:19 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
static inline enum audio_format
|
|
|
|
convert_ffmpeg_sample_format(enum AVSampleFormat format)
|
|
|
|
{
|
2022-08-05 12:25:57 +00:00
|
|
|
switch (format) {
|
2014-04-05 06:21:19 +00:00
|
|
|
case AV_SAMPLE_FMT_U8:
|
|
|
|
return AUDIO_FORMAT_U8BIT;
|
|
|
|
case AV_SAMPLE_FMT_S16:
|
|
|
|
return AUDIO_FORMAT_16BIT;
|
|
|
|
case AV_SAMPLE_FMT_S32:
|
|
|
|
return AUDIO_FORMAT_32BIT;
|
|
|
|
case AV_SAMPLE_FMT_FLT:
|
|
|
|
return AUDIO_FORMAT_FLOAT;
|
|
|
|
case AV_SAMPLE_FMT_U8P:
|
|
|
|
return AUDIO_FORMAT_U8BIT_PLANAR;
|
|
|
|
case AV_SAMPLE_FMT_S16P:
|
|
|
|
return AUDIO_FORMAT_16BIT_PLANAR;
|
|
|
|
case AV_SAMPLE_FMT_S32P:
|
|
|
|
return AUDIO_FORMAT_32BIT_PLANAR;
|
|
|
|
case AV_SAMPLE_FMT_FLTP:
|
|
|
|
return AUDIO_FORMAT_FLOAT_PLANAR;
|
2023-02-16 19:57:15 +00:00
|
|
|
default:
|
|
|
|
return AUDIO_FORMAT_16BIT;
|
2014-04-05 06:21:19 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
/* shouldn't get here */
|
|
|
|
return AUDIO_FORMAT_16BIT;
|
|
|
|
}
|