lifegpc/eh-downloader
1
0
Fork 0
mirror of https://github.com/lifegpc/eh-downloader.git synced 2026-06-06 13:48:51 +08:00
Code Issues Packages Projects Releases Wiki Activity
Files
ebd90ca352eeebec434ef50e5cc3e377fbc7ff0c
eh-downloader/extensions/thumbnail/thumbnail.c
lifegpc cf728267f1 Fix fucking warning message
2024-06-08 15:41:11 +08:00

449 lines
16 KiB
C
Raw Blame History

#include "thumbnail.h"
#include "cfileop.h"
#include "libavformat/avformat.h"
#include "libavcodec/avcodec.h"
#include "libavutil/opt.h"
#include "libswscale/swscale.h"
#include <string.h>
static int hooked_ffmpeg_log = 0;
void ffmpeg_log(void* avcl, int level, const char* fmt, va_list vl) {
if (level == AV_LOG_WARNING) {
if (!strcmp(fmt, "deprecated pixel format used, make sure you did set range correctly\n")) {
return;
}
}
av_log_default_callback(avcl, level, fmt, vl);
}
void hook_ffmpeg_log() {
if (!hooked_ffmpeg_log) {
hooked_ffmpeg_log = 1;
av_log_set_callback(ffmpeg_log);
}
}
void thumbnail_fferror(int e, char* buf, size_t bufsize) {
if (!buf) return;
av_make_error_string(buf, bufsize, e);
}
int thumbnail_convert_frame(THUMBNAIL_ERROR* err, AVFrame* ifr, AVFrame** ofr, AVCodecContext* occ, THUMBNAIL_METHOD method, ALIGN_METHOD align) {
if (!err || !ifr || !ofr) return 1;
int re = 0;
struct SwsContext* sws = NULL;
AVFrame* fr = NULL, * fr2 = NULL;
if (!(fr = av_frame_alloc())) {
err->e = THUMBNAIL_OOM;
re = 1;
goto end;
}
if (method == THUMBNAIL_FILL) {
fr->width = occ->width;
fr->height = occ->height;
} else if (method == THUMBNAIL_COVER) {
int twidth = ifr->width * occ->height / ifr->height;
if (twidth > occ->width) {
fr->width = twidth;
fr->height = occ->height;
} else {
fr->width = occ->width;
fr->height = ifr->height * occ->width / ifr->width;
}
} else if (method == THUMBNAIL_CONTAIN) {
int twidth = ifr->width * occ->height / ifr->height;
if (twidth > occ->width) {
fr->width = occ->width;
fr->height = ifr->height * occ->width / ifr->width;
} else {
fr->width = twidth;
fr->height = occ->height;
}
}
fr->format = occ->pix_fmt;
fr->sample_aspect_ratio = ifr->sample_aspect_ratio;
if ((err->fferr = av_frame_get_buffer(fr, 0)) < 0) {
err->e = THUMBNAIL_FFMPEG_ERROR;
re = 1;
av_log(NULL, AV_LOG_ERROR, "Failed to get buffer for output frame: %s\n", av_err2str(err->fferr));
goto end;
}
if ((err->fferr = av_frame_make_writable(fr)) < 0) {
err->e = THUMBNAIL_FFMPEG_ERROR;
re = 1;
av_log(NULL, AV_LOG_ERROR, "Failed to make writeable for output frame: %s\n", av_err2str(err->fferr));
goto end;
}
if (!(sws = sws_getContext(ifr->width, ifr->height, (enum AVPixelFormat)ifr->format, fr->width, fr->height, (enum AVPixelFormat)fr->format, SWS_BILINEAR, NULL, NULL, NULL))) {
err->e = THUMBNAIL_UNABLE_SCALE;
re = 1;
goto end;
}
if ((err->fferr = sws_scale(sws, (const uint8_t* const*)ifr->data, ifr->linesize, 0, ifr->height, fr->data, fr->linesize)) < 0) {
err->e = THUMBNAIL_UNABLE_SCALE;
re = 1;
goto end;
}
if (method != THUMBNAIL_FILL) {
if (!(fr2 = av_frame_alloc())) {
err->e = THUMBNAIL_OOM;
re = 1;
goto end;
}
fr2->width = occ->width;
fr2->height = occ->height;
fr2->format = occ->pix_fmt;
fr2->sample_aspect_ratio = occ->sample_aspect_ratio;
if ((err->fferr = av_frame_get_buffer(fr2, 0)) < 0) {
err->e = THUMBNAIL_FFMPEG_ERROR;
av_log(NULL, AV_LOG_ERROR, "Failed to get buffer for output frame: %s\n", av_err2str(err->fferr));
re = 1;
goto end;
}
if ((err->fferr = av_frame_make_writable(fr2)) < 0) {
err->e = THUMBNAIL_FFMPEG_ERROR;
av_log(NULL, AV_LOG_ERROR, "Failed to make writeable for output frame: %s\n", av_err2str(err->fferr));
re = 1;
goto end;
}
int width = 0, height = 0, x1 = 0, y1 = 0, i = 0, j = 0;
if (method == THUMBNAIL_COVER) {
width = occ->width;
height = occ->height;
if (fr->width == occ->width) {
if (align == ALIGN_CENTER) {
y1 = (fr->height - occ->height) / 2;
} else if (align == ALIGN_BOTTOM) {
y1 = fr->height - occ->height;
}
} else if (fr->height == occ->height) {
if (align == ALIGN_CENTER) {
x1 = (fr->width - occ->width) / 2;
} else if (align == ALIGN_RIGHT) {
x1 = fr->width - occ->width;
}
}
for (i = 0; i < height; i++) {
memcpy(fr2->data[0] + i * fr2->linesize[0], fr->data[0] + (i + y1) * fr->linesize[0] + x1, width);
}
for (j = 1; j < 3; j++) {
for (i = 0; i < height / 2; i++) {
memcpy(fr2->data[j] + i * fr2->linesize[j], fr->data[j] + (i + y1 / 2) * fr->linesize[j] + x1 / 2, width / 2);
}
}
} else if (method == THUMBNAIL_CONTAIN) {
width = fr->width;
height = fr->height;
if (fr->width == occ->width) {
if (align == ALIGN_CENTER) {
y1 = (occ->height - fr->height) / 2;
} else if (align == ALIGN_BOTTOM) {
y1 = occ->height - fr->height;
}
} else {
if (align == ALIGN_CENTER) {
x1 = (occ->width - fr->width) / 2;
} else if (align == ALIGN_RIGHT) {
x1 = occ->width - fr->width;
}
}
memset(fr2->data[0], 255, fr2->height * fr2->linesize[0]);
memset(fr2->data[1], 128, fr2->height / 2 * fr2->linesize[1]);
memset(fr2->data[2], 128, fr2->height / 2 * fr2->linesize[2]);
for (i = 0; i < height; i++) {
memcpy(fr2->data[0] + (i + y1) * fr2->linesize[0] + x1, fr->data[0] + i * fr->linesize[0], width);
}
for (j = 1; j < 3; j++) {
for (i = 0; i < height / 2; i++) {
memcpy(fr2->data[j] + (i + y1 / 2) * fr2->linesize[j] + x1 / 2, fr->data[j] + i * fr->linesize[j], width / 2);
}
}
}
if (fr) av_frame_free(&fr);
fr = fr2;
fr2 = NULL;
}
end:
if (re == 1 && fr) av_frame_free(&fr);
else if (re == 0) *ofr = fr;
if (fr2) av_frame_free(&fr2);
if (sws) sws_freeContext(sws);
return re;
}
int thumbnail_encode_video(THUMBNAIL_ERROR* err, AVFrame* ofr, AVFormatContext* oc, AVCodecContext* occ, char* writed_data) {
if (!err || !oc || !occ || !writed_data) return 1;
int re = 0;
AVPacket* pkt = av_packet_alloc();
if (!pkt) {
err->e = THUMBNAIL_OOM;
re = 1;
goto end;
}
*writed_data = 0;
if (ofr) {
ofr->pts = 0;
ofr->pkt_dts = 0;
}
if ((err->fferr = avcodec_send_frame(occ, ofr)) < 0) {
if (err->fferr == AVERROR_EOF) {
err->fferr = 0;
} else {
av_log(NULL, AV_LOG_ERROR, "Failed to send frame to encoder: %s\n", av_err2str(err->fferr));
err->e = THUMBNAIL_FFMPEG_ERROR;
re = 1;
goto end;
}
}
err->fferr = avcodec_receive_packet(occ, pkt);
if (err->fferr >= 0) {
*writed_data = 1;
} else if (err->fferr == AVERROR_EOF || err->fferr == AVERROR(EAGAIN)) {
err->fferr = 0;
goto end;
} else {
av_log(NULL, AV_LOG_ERROR, "Failed to recive data from encoder: %s\n", av_err2str(err->fferr));
err->e = THUMBNAIL_FFMPEG_ERROR;
re = 1;
goto end;
}
if (*writed_data && pkt) {
pkt->stream_index = 0;
if ((err->fferr = av_write_frame(oc, pkt)) < 0) {
err->e = THUMBNAIL_FFMPEG_ERROR;
av_log(NULL, AV_LOG_ERROR, "Failed to write data to muxer: %s\n", av_err2str(err->fferr));
re = 1;
goto end;
}
}
end:
if (pkt) av_packet_free(&pkt);
return re;
}
THUMBNAIL_ERROR gen_thumbnail(const char* src, const char* dest, int width, int height, THUMBNAIL_METHOD method, ALIGN_METHOD align, int quality) {
hook_ffmpeg_log();
THUMBNAIL_ERROR re = { THUMBNAIL_OK, 0 };
AVFormatContext* ic = NULL, * oc = NULL;
AVStream* is = NULL, * os = NULL;
const AVCodec* input_codec = NULL, * output_codec = NULL;
AVCodecContext* icc = NULL, * occ = NULL;
AVPacket pkt;
AVFrame* ifr = NULL, * ofr = NULL;
if (method < THUMBNAIL_COVER || method > THUMBNAIL_FILL) {
re.e = THUMBNAIL_UNKNOWN_METHOD;
goto end;
}
if (align < ALIGN_LEFT || align > ALIGN_RIGHT) {
re.e = THUMBNAIL_UNKNOWN_ALIGN;
goto end;
}
if (!src || !dest) {
re.e = THUMBNAIL_NULL_POINTER;
goto end;
}
if (fileop_exists(dest)) {
goto end;
}
if ((re.fferr = avformat_open_input(&ic, src, NULL, NULL)) < 0) {
re.e = THUMBNAIL_FFMPEG_ERROR;
av_log(NULL, AV_LOG_ERROR, "Failed to open file: %s\n", av_err2str(re.fferr));
goto end;
}
if ((re.fferr = avformat_find_stream_info(ic, NULL)) < 0) {
re.e = THUMBNAIL_FFMPEG_ERROR;
av_log(NULL, AV_LOG_ERROR, "Failed to find stream info in file: %s\n", av_err2str(re.fferr));
goto end;
}
for (unsigned int i = 0; i < ic->nb_streams; i++) {
AVStream* s = ic->streams[i];
if (s->codecpar && s->codecpar->codec_type == AVMEDIA_TYPE_VIDEO) {
is = s;
break;
}
}
if (!is) {
re.e = THUMBNAIL_NO_VIDEO_STREAM;
goto end;
}
if ((re.fferr = avformat_alloc_output_context2(&oc, NULL, "image2", dest)) < 0) {
av_log(NULL, AV_LOG_ERROR, "Failed to allocate output context: %s\n", av_err2str(re.fferr));
re.e = THUMBNAIL_FFMPEG_ERROR;
goto end;
}
av_opt_set_int(oc, "update", 1, AV_OPT_SEARCH_CHILDREN);
if (!(input_codec = avcodec_find_decoder(is->codecpar->codec_id))) {
re.e = THUMBNAIL_NO_DECODER;
goto end;
}
if (!(icc = avcodec_alloc_context3(input_codec))) {
re.e = THUMBNAIL_OOM;
goto end;
}
if ((re.fferr = avcodec_parameters_to_context(icc, is->codecpar)) < 0) {
av_log(NULL, AV_LOG_ERROR, "Failed to copy decode parameters: %s\n", av_err2str(re.fferr));
re.e = THUMBNAIL_FFMPEG_ERROR;
goto end;
}
if ((re.fferr = avcodec_open2(icc, input_codec, NULL)) < 0) {
av_log(NULL, AV_LOG_ERROR, "Failed to open decoder: %s\n", av_err2str(re.fferr));
re.e = THUMBNAIL_FFMPEG_ERROR;
goto end;
}
output_codec = avcodec_find_encoder(AV_CODEC_ID_MJPEG);
if (!output_codec) {
re.e = THUMBNAIL_NO_ENCODER;
goto end;
}
if (!(occ = avcodec_alloc_context3(output_codec))) {
re.e = THUMBNAIL_OOM;
goto end;
}
occ->width = width;
occ->height = height;
occ->pix_fmt = AV_PIX_FMT_YUVJ420P;
occ->sample_aspect_ratio = icc->sample_aspect_ratio;
occ->time_base = AV_TIME_BASE_Q;
occ->color_range = AVCOL_RANGE_JPEG;
occ->global_quality = quality;
occ->qmax = quality;
occ->qmin = quality;
if ((re.fferr = avcodec_open2(occ, output_codec, NULL)) < 0) {
av_log(NULL, AV_LOG_ERROR, "Failed to open encoder: %s\n", av_err2str(re.fferr));
re.e = THUMBNAIL_FFMPEG_ERROR;
goto end;
}
if (!(os = avformat_new_stream(oc, NULL))) {
re.e = THUMBNAIL_OOM;
goto end;
}
if ((re.fferr = avcodec_parameters_from_context(os->codecpar, occ)) < 0) {
av_log(NULL, AV_LOG_ERROR, "Failed to copy encoder's params to muxer: %s\n", av_err2str(re.fferr));
re.e = THUMBNAIL_FFMPEG_ERROR;
goto end;
}
if (!(oc->oformat->flags & AVFMT_NOFILE)) {
if ((re.fferr = avio_open(&oc->pb, dest, AVIO_FLAG_WRITE)) < 0) {
av_log(NULL, AV_LOG_ERROR, "Failed to open output file: %s\n", av_err2str(re.fferr));
re.e = THUMBNAIL_FFMPEG_ERROR;
goto end;
}
}
if ((re.fferr = avformat_write_header(oc, NULL)) < 0) {
av_log(NULL, AV_LOG_ERROR, "Failed to write headers to file: %s\n", av_err2str(re.fferr));
re.e = THUMBNAIL_FFMPEG_ERROR;
goto end;
}
while (1) {
if ((re.fferr = av_read_frame(ic, &pkt)) < 0) {
av_log(NULL, AV_LOG_ERROR, "Failed to read frames from file: %s\n", av_err2str(re.fferr));
re.e = THUMBNAIL_FFMPEG_ERROR;
goto end;
}
if (pkt.data == NULL) {
av_packet_unref(&pkt);
av_log(NULL, AV_LOG_ERROR, "No data find in frames: %s\n", av_err2str(re.fferr));
re.e = THUMBNAIL_FFMPEG_ERROR;
goto end;
}
if (pkt.stream_index != is->index) {
av_packet_unref(&pkt);
continue;
}
if (!(ifr = av_frame_alloc())) {
av_packet_unref(&pkt);
re.e = THUMBNAIL_OOM;
goto end;
}
if ((re.fferr = avcodec_send_packet(icc, &pkt)) < 0) {
av_packet_unref(&pkt);
av_log(NULL, AV_LOG_ERROR, "Failed to send packet to decoder: %s\n", av_err2str(re.fferr));
re.e = THUMBNAIL_FFMPEG_ERROR;
goto end;
}
if ((re.fferr = avcodec_receive_frame(icc, ifr)) < 0) {
if (re.fferr == AVERROR(EAGAIN)) {
av_packet_unref(&pkt);
re.fferr = 0;
continue;
}
av_packet_unref(&pkt);
av_log(NULL, AV_LOG_ERROR, "Failed to receive frame from decoder: %s\n", av_err2str(re.fferr));
re.e = THUMBNAIL_FFMPEG_ERROR;
goto end;
}
if (thumbnail_convert_frame(&re, ifr, &ofr, occ, method, align)) {
av_packet_unref(&pkt);
re.e = THUMBNAIL_FFMPEG_ERROR;
goto end;
}
char writed = 0;
if (thumbnail_encode_video(&re, ofr, oc, occ, &writed)) {
av_packet_unref(&pkt);
goto end;
}
while (1) {
if (thumbnail_encode_video(&re, NULL, oc, occ, &writed)) {
av_packet_unref(&pkt);
goto end;
}
if (!writed) break;
}
av_packet_unref(&pkt);
break;
}
av_write_trailer(oc);
end:
if (ifr) av_frame_free(&ifr);
if (ofr) av_frame_free(&ofr);
if (icc) avcodec_free_context(&icc);
if (occ) avcodec_free_context(&occ);
if (oc) {
if (!(oc->oformat->flags & AVFMT_NOFILE)) avio_closep(&oc->pb);
avformat_free_context(oc);
}
if (ic) avformat_close_input(&ic);
return re;
}
const char* thumbnail_berror(THUMBNAIL_ERROR_E e) {
switch (e) {
case THUMBNAIL_OK:
return "OK";
case THUMBNAIL_FFMPEG_ERROR:
return "A error occured in ffmpeg code.";
case THUMBNAIL_NULL_POINTER:
return "Arguments have null pointers.";
case THUMBNAIL_REMOVE_OUTPUT_FILE_FAILED:
return "Can not remove output file.";
case THUMBNAIL_NO_VIDEO_STREAM:
return "Can not find video stream in source file.";
case THUMBNAIL_OOM:
return "Out of memory.";
case THUMBNAIL_NO_DECODER:
return "No available decoder.";
case THUMBNAIL_NO_ENCODER:
return "No available encoder.";
case THUMBNAIL_UNABLE_SCALE:
return "Unable to scale image.";
case THUMBNAIL_UNKNOWN_METHOD:
return "Unknown method.";
case THUMBNAIL_UNKNOWN_ALIGN:
return "Unknown align.";
default:
return "Unknown error.";
}
}
void thumbnail_error(THUMBNAIL_ERROR e, char* buf, size_t bufsize) {
switch (e.e) {
case THUMBNAIL_FFMPEG_ERROR:
thumbnail_fferror(e.fferr, buf, bufsize);
break;
default:
strncpy(buf, thumbnail_berror(e.e), bufsize);
break;
}
}
Reference in New Issue View Git Blame Copy Permalink