|
|
|
@ -38,6 +38,28 @@ |
|
|
|
|
#include "libavutil/imgutils.h" |
|
|
|
|
#include "libavutil/samplefmt.h" |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static const enum AVPixelFormat *get_compliance_unofficial_pix_fmts(enum AVCodecID codec_id, const enum AVPixelFormat default_formats[]) |
|
|
|
|
{ |
|
|
|
|
static const enum AVPixelFormat mjpeg_formats[] = |
|
|
|
|
{ AV_PIX_FMT_YUVJ420P, AV_PIX_FMT_YUVJ422P, AV_PIX_FMT_YUVJ444P, |
|
|
|
|
AV_PIX_FMT_YUV420P, AV_PIX_FMT_YUV422P, AV_PIX_FMT_YUV444P, |
|
|
|
|
AV_PIX_FMT_NONE }; |
|
|
|
|
static const enum AVPixelFormat ljpeg_formats[] = |
|
|
|
|
{ AV_PIX_FMT_BGR24 , AV_PIX_FMT_BGRA , AV_PIX_FMT_BGR0, |
|
|
|
|
AV_PIX_FMT_YUVJ420P, AV_PIX_FMT_YUVJ444P, AV_PIX_FMT_YUVJ422P, |
|
|
|
|
AV_PIX_FMT_YUV420P , AV_PIX_FMT_YUV444P , AV_PIX_FMT_YUV422P, |
|
|
|
|
AV_PIX_FMT_NONE}; |
|
|
|
|
|
|
|
|
|
if (codec_id == AV_CODEC_ID_MJPEG) { |
|
|
|
|
return mjpeg_formats; |
|
|
|
|
} else if (codec_id == AV_CODEC_ID_LJPEG) { |
|
|
|
|
return ljpeg_formats; |
|
|
|
|
} else { |
|
|
|
|
return default_formats; |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
enum AVPixelFormat choose_pixel_fmt(AVStream *st, AVCodecContext *enc_ctx, AVCodec *codec, enum AVPixelFormat target) |
|
|
|
|
{ |
|
|
|
|
if (codec && codec->pix_fmts) { |
|
|
|
@ -45,22 +67,9 @@ enum AVPixelFormat choose_pixel_fmt(AVStream *st, AVCodecContext *enc_ctx, AVCod |
|
|
|
|
const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(target); |
|
|
|
|
int has_alpha = desc ? desc->nb_components % 2 == 0 : 0; |
|
|
|
|
enum AVPixelFormat best= AV_PIX_FMT_NONE; |
|
|
|
|
static const enum AVPixelFormat mjpeg_formats[] = |
|
|
|
|
{ AV_PIX_FMT_YUVJ420P, AV_PIX_FMT_YUVJ422P, AV_PIX_FMT_YUVJ444P, |
|
|
|
|
AV_PIX_FMT_YUV420P, AV_PIX_FMT_YUV422P, AV_PIX_FMT_YUV444P, |
|
|
|
|
AV_PIX_FMT_NONE }; |
|
|
|
|
static const enum AVPixelFormat ljpeg_formats[] = |
|
|
|
|
{ AV_PIX_FMT_BGR24 , AV_PIX_FMT_BGRA , AV_PIX_FMT_BGR0, |
|
|
|
|
AV_PIX_FMT_YUVJ420P, AV_PIX_FMT_YUVJ444P, AV_PIX_FMT_YUVJ422P, |
|
|
|
|
AV_PIX_FMT_YUV420P , AV_PIX_FMT_YUV444P , AV_PIX_FMT_YUV422P, |
|
|
|
|
AV_PIX_FMT_NONE}; |
|
|
|
|
|
|
|
|
|
if (enc_ctx->strict_std_compliance <= FF_COMPLIANCE_UNOFFICIAL) { |
|
|
|
|
if (enc_ctx->codec_id == AV_CODEC_ID_MJPEG) { |
|
|
|
|
p = mjpeg_formats; |
|
|
|
|
} else if (enc_ctx->codec_id == AV_CODEC_ID_LJPEG) { |
|
|
|
|
p =ljpeg_formats; |
|
|
|
|
} |
|
|
|
|
p = get_compliance_unofficial_pix_fmts(enc_ctx->codec_id, p); |
|
|
|
|
} |
|
|
|
|
for (; *p != AV_PIX_FMT_NONE; p++) { |
|
|
|
|
best= avcodec_find_best_pix_fmt_of_2(best, *p, target, has_alpha, NULL); |
|
|
|
@ -130,12 +139,7 @@ static char *choose_pix_fmts(OutputStream *ost) |
|
|
|
|
|
|
|
|
|
p = ost->enc->pix_fmts; |
|
|
|
|
if (ost->enc_ctx->strict_std_compliance <= FF_COMPLIANCE_UNOFFICIAL) { |
|
|
|
|
if (ost->enc_ctx->codec_id == AV_CODEC_ID_MJPEG) { |
|
|
|
|
p = (const enum AVPixelFormat[]) { AV_PIX_FMT_YUVJ420P, AV_PIX_FMT_YUVJ422P, AV_PIX_FMT_YUV420P, AV_PIX_FMT_YUV422P, AV_PIX_FMT_NONE }; |
|
|
|
|
} else if (ost->enc_ctx->codec_id == AV_CODEC_ID_LJPEG) { |
|
|
|
|
p = (const enum AVPixelFormat[]) { AV_PIX_FMT_YUVJ420P, AV_PIX_FMT_YUVJ422P, AV_PIX_FMT_YUVJ444P, AV_PIX_FMT_YUV420P, |
|
|
|
|
AV_PIX_FMT_YUV422P, AV_PIX_FMT_YUV444P, AV_PIX_FMT_BGRA, AV_PIX_FMT_NONE }; |
|
|
|
|
} |
|
|
|
|
p = get_compliance_unofficial_pix_fmts(ost->enc_ctx->codec_id, p); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
for (; *p != AV_PIX_FMT_NONE; p++) { |
|
|
|
|