2019-06-26 15:04:38 +00:00
|
|
|
/*
|
|
|
|
This file is part of Telegram Desktop,
|
|
|
|
the official desktop application for the Telegram messaging service.
|
|
|
|
|
|
|
|
For license and copyright information please follow this link:
|
|
|
|
https://github.com/telegramdesktop/tdesktop/blob/master/LEGAL
|
|
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
|
|
|
|
#include "base/bytes.h"
|
2021-05-26 18:53:54 +00:00
|
|
|
#include "base/algorithm.h"
|
2019-06-26 15:04:38 +00:00
|
|
|
|
|
|
|
#include <crl/crl_time.h>
|
|
|
|
|
|
|
|
#include <QSize>
|
2019-07-05 21:13:31 +00:00
|
|
|
#include <QString>
|
2019-06-26 15:04:38 +00:00
|
|
|
|
|
|
|
extern "C" {
|
|
|
|
#include <libavcodec/avcodec.h>
|
|
|
|
#include <libavformat/avformat.h>
|
|
|
|
#include <libswscale/swscale.h>
|
|
|
|
} // extern "C"
|
|
|
|
|
|
|
|
class QImage;
|
|
|
|
|
|
|
|
namespace FFmpeg {
|
|
|
|
|
|
|
|
inline constexpr auto kPixelBytesSize = 4;
|
2020-10-13 16:43:18 +00:00
|
|
|
inline constexpr auto kAVBlockSize = 4096; // 4Kb for ffmpeg blocksize
|
2019-06-26 15:04:38 +00:00
|
|
|
|
|
|
|
constexpr auto kUniversalTimeBase = AVRational{ 1, AV_TIME_BASE };
|
|
|
|
constexpr auto kNormalAspect = AVRational{ 1, 1 };
|
|
|
|
|
|
|
|
class AvErrorWrap {
|
|
|
|
public:
|
|
|
|
AvErrorWrap(int code = 0) : _code(code) {
|
|
|
|
}
|
|
|
|
|
|
|
|
[[nodiscard]] bool failed() const {
|
|
|
|
return (_code < 0);
|
|
|
|
}
|
|
|
|
[[nodiscard]] explicit operator bool() const {
|
|
|
|
return failed();
|
|
|
|
}
|
|
|
|
|
|
|
|
[[nodiscard]] int code() const {
|
|
|
|
return _code;
|
|
|
|
}
|
|
|
|
|
|
|
|
[[nodiscard]] QString text() const {
|
|
|
|
char string[AV_ERROR_MAX_STRING_SIZE] = { 0 };
|
|
|
|
return QString::fromUtf8(av_make_error_string(
|
|
|
|
string,
|
|
|
|
sizeof(string),
|
|
|
|
_code));
|
|
|
|
}
|
|
|
|
|
|
|
|
private:
|
|
|
|
int _code = 0;
|
|
|
|
|
|
|
|
};
|
|
|
|
|
|
|
|
class Packet {
|
|
|
|
public:
|
2021-05-26 18:53:54 +00:00
|
|
|
Packet() = default;
|
|
|
|
Packet(Packet &&other) : _data(base::take(other._data)) {
|
2019-06-26 15:04:38 +00:00
|
|
|
}
|
|
|
|
Packet &operator=(Packet &&other) {
|
|
|
|
if (this != &other) {
|
2021-05-26 18:53:54 +00:00
|
|
|
release();
|
|
|
|
_data = base::take(other._data);
|
2019-06-26 15:04:38 +00:00
|
|
|
}
|
|
|
|
return *this;
|
|
|
|
}
|
|
|
|
~Packet() {
|
2021-05-26 18:53:54 +00:00
|
|
|
release();
|
2019-06-26 15:04:38 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
[[nodiscard]] AVPacket &fields() {
|
2021-05-26 18:53:54 +00:00
|
|
|
if (!_data) {
|
|
|
|
_data = av_packet_alloc();
|
|
|
|
}
|
|
|
|
return *_data;
|
2019-06-26 15:04:38 +00:00
|
|
|
}
|
|
|
|
[[nodiscard]] const AVPacket &fields() const {
|
2021-05-26 18:53:54 +00:00
|
|
|
if (!_data) {
|
|
|
|
_data = av_packet_alloc();
|
|
|
|
}
|
|
|
|
return *_data;
|
2019-06-26 15:04:38 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
[[nodiscard]] bool empty() const {
|
2021-05-26 18:53:54 +00:00
|
|
|
return !_data || !fields().data;
|
2019-06-26 15:04:38 +00:00
|
|
|
}
|
|
|
|
void release() {
|
2021-05-26 18:53:54 +00:00
|
|
|
av_packet_free(&_data);
|
2019-06-26 15:04:38 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
private:
|
2021-05-26 18:53:54 +00:00
|
|
|
mutable AVPacket *_data = nullptr;
|
2019-06-26 15:04:38 +00:00
|
|
|
|
|
|
|
};
|
|
|
|
|
|
|
|
struct IODeleter {
|
|
|
|
void operator()(AVIOContext *value);
|
|
|
|
};
|
|
|
|
using IOPointer = std::unique_ptr<AVIOContext, IODeleter>;
|
|
|
|
[[nodiscard]] IOPointer MakeIOPointer(
|
|
|
|
void *opaque,
|
|
|
|
int(*read)(void *opaque, uint8_t *buffer, int bufferSize),
|
|
|
|
int(*write)(void *opaque, uint8_t *buffer, int bufferSize),
|
|
|
|
int64_t(*seek)(void *opaque, int64_t offset, int whence));
|
|
|
|
|
|
|
|
struct FormatDeleter {
|
|
|
|
void operator()(AVFormatContext *value);
|
|
|
|
};
|
|
|
|
using FormatPointer = std::unique_ptr<AVFormatContext, FormatDeleter>;
|
|
|
|
[[nodiscard]] FormatPointer MakeFormatPointer(
|
|
|
|
void *opaque,
|
|
|
|
int(*read)(void *opaque, uint8_t *buffer, int bufferSize),
|
|
|
|
int(*write)(void *opaque, uint8_t *buffer, int bufferSize),
|
|
|
|
int64_t(*seek)(void *opaque, int64_t offset, int whence));
|
|
|
|
|
|
|
|
struct CodecDeleter {
|
|
|
|
void operator()(AVCodecContext *value);
|
|
|
|
};
|
|
|
|
using CodecPointer = std::unique_ptr<AVCodecContext, CodecDeleter>;
|
2022-03-22 13:56:43 +00:00
|
|
|
|
|
|
|
struct CodecDescriptor {
|
|
|
|
not_null<AVStream*> stream;
|
|
|
|
AVHWDeviceType type = AV_HWDEVICE_TYPE_NONE;
|
|
|
|
};
|
|
|
|
[[nodiscard]] CodecPointer MakeCodecPointer(CodecDescriptor descriptor);
|
2019-06-26 15:04:38 +00:00
|
|
|
|
|
|
|
struct FrameDeleter {
|
|
|
|
void operator()(AVFrame *value);
|
|
|
|
};
|
|
|
|
using FramePointer = std::unique_ptr<AVFrame, FrameDeleter>;
|
|
|
|
[[nodiscard]] FramePointer MakeFramePointer();
|
|
|
|
[[nodiscard]] bool FrameHasData(AVFrame *frame);
|
|
|
|
void ClearFrameMemory(AVFrame *frame);
|
|
|
|
|
|
|
|
struct SwscaleDeleter {
|
2019-06-27 13:27:01 +00:00
|
|
|
QSize srcSize;
|
|
|
|
int srcFormat = int(AV_PIX_FMT_NONE);
|
|
|
|
QSize dstSize;
|
|
|
|
int dstFormat = int(AV_PIX_FMT_NONE);
|
2019-06-26 15:04:38 +00:00
|
|
|
|
|
|
|
void operator()(SwsContext *value);
|
|
|
|
};
|
|
|
|
using SwscalePointer = std::unique_ptr<SwsContext, SwscaleDeleter>;
|
2019-06-27 13:27:01 +00:00
|
|
|
[[nodiscard]] SwscalePointer MakeSwscalePointer(
|
|
|
|
QSize srcSize,
|
|
|
|
int srcFormat,
|
|
|
|
QSize dstSize,
|
|
|
|
int dstFormat, // This field doesn't take part in caching!
|
|
|
|
SwscalePointer *existing = nullptr);
|
2019-06-26 15:04:38 +00:00
|
|
|
[[nodiscard]] SwscalePointer MakeSwscalePointer(
|
|
|
|
not_null<AVFrame*> frame,
|
|
|
|
QSize resize,
|
|
|
|
SwscalePointer *existing = nullptr);
|
|
|
|
|
|
|
|
void LogError(QLatin1String method);
|
|
|
|
void LogError(QLatin1String method, FFmpeg::AvErrorWrap error);
|
|
|
|
|
2022-02-08 15:20:27 +00:00
|
|
|
[[nodiscard]] const AVCodec *FindDecoder(not_null<AVCodecContext*> context);
|
2019-06-26 15:04:38 +00:00
|
|
|
[[nodiscard]] crl::time PtsToTime(int64_t pts, AVRational timeBase);
|
|
|
|
// Used for full duration conversion.
|
|
|
|
[[nodiscard]] crl::time PtsToTimeCeil(int64_t pts, AVRational timeBase);
|
|
|
|
[[nodiscard]] int64_t TimeToPts(crl::time time, AVRational timeBase);
|
|
|
|
[[nodiscard]] crl::time PacketPosition(
|
|
|
|
const FFmpeg::Packet &packet,
|
|
|
|
AVRational timeBase);
|
|
|
|
[[nodiscard]] crl::time PacketDuration(
|
|
|
|
const FFmpeg::Packet &packet,
|
|
|
|
AVRational timeBase);
|
|
|
|
[[nodiscard]] int DurationByPacket(
|
|
|
|
const FFmpeg::Packet &packet,
|
|
|
|
AVRational timeBase);
|
|
|
|
[[nodiscard]] int ReadRotationFromMetadata(not_null<AVStream*> stream);
|
|
|
|
[[nodiscard]] AVRational ValidateAspectRatio(AVRational aspect);
|
|
|
|
[[nodiscard]] bool RotationSwapWidthHeight(int rotation);
|
|
|
|
[[nodiscard]] QSize CorrectByAspect(QSize size, AVRational aspect);
|
|
|
|
|
|
|
|
[[nodiscard]] bool GoodStorageForFrame(const QImage &storage, QSize size);
|
|
|
|
[[nodiscard]] QImage CreateFrameStorage(QSize size);
|
|
|
|
|
2019-07-04 11:12:58 +00:00
|
|
|
void UnPremultiply(QImage &to, const QImage &from);
|
|
|
|
void PremultiplyInplace(QImage &image);
|
|
|
|
|
2019-06-26 15:04:38 +00:00
|
|
|
} // namespace FFmpeg
|