138 lines
3.5 KiB
C++
138 lines
3.5 KiB
C++
/*
|
|
This file is part of Telegram Desktop,
|
|
the official desktop version of Telegram messaging app, see https://telegram.org
|
|
|
|
Telegram Desktop is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
It 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 General Public License for more details.
|
|
|
|
In addition, as a special exception, the copyright holders give permission
|
|
to link the code of portions of this program with the OpenSSL library.
|
|
|
|
Full license: https://github.com/telegramdesktop/tdesktop/blob/master/LICENSE
|
|
Copyright (c) 2014-2017 John Preston, https://desktop.telegram.org
|
|
*/
|
|
#pragma once
|
|
|
|
extern "C" {
|
|
|
|
#include <libswscale/swscale.h>
|
|
|
|
} // extern "C"
|
|
|
|
#include "media/media_clip_implementation.h"
|
|
#include "media/media_child_ffmpeg_loader.h"
|
|
|
|
namespace Media {
|
|
namespace Clip {
|
|
namespace internal {
|
|
|
|
class FFMpegReaderImplementation : public ReaderImplementation {
|
|
public:
|
|
FFMpegReaderImplementation(FileLocation *location, QByteArray *data, const AudioMsgId &audio);
|
|
|
|
ReadResult readFramesTill(TimeMs frameMs, TimeMs systemMs) override;
|
|
|
|
TimeMs frameRealTime() const override;
|
|
TimeMs framePresentationTime() const override;
|
|
|
|
bool renderFrame(QImage &to, bool &hasAlpha, const QSize &size) override;
|
|
|
|
TimeMs durationMs() const override;
|
|
bool hasAudio() const override {
|
|
return (_audioStreamId >= 0);
|
|
}
|
|
|
|
bool start(Mode mode, TimeMs &positionMs) override;
|
|
bool inspectAt(TimeMs &positionMs);
|
|
|
|
QString logData() const;
|
|
|
|
bool isGifv() const;
|
|
|
|
~FFMpegReaderImplementation();
|
|
|
|
private:
|
|
ReadResult readNextFrame();
|
|
void processReadFrame();
|
|
|
|
enum class PacketResult {
|
|
Ok,
|
|
EndOfFile,
|
|
Error,
|
|
};
|
|
PacketResult readPacket(AVPacket *packet);
|
|
void processPacket(AVPacket *packet);
|
|
TimeMs countPacketMs(AVPacket *packet) const;
|
|
PacketResult readAndProcessPacket();
|
|
|
|
enum class Rotation {
|
|
None,
|
|
Degrees90,
|
|
Degrees180,
|
|
Degrees270,
|
|
};
|
|
Rotation rotationFromDegrees(int degrees) const;
|
|
bool rotationSwapWidthHeight() const {
|
|
return (_rotation == Rotation::Degrees90) || (_rotation == Rotation::Degrees270);
|
|
}
|
|
|
|
void startPacket();
|
|
void finishPacket();
|
|
void clearPacketQueue();
|
|
|
|
static int _read(void *opaque, uint8_t *buf, int buf_size);
|
|
static int64_t _seek(void *opaque, int64_t offset, int whence);
|
|
|
|
Mode _mode = Mode::Normal;
|
|
|
|
Rotation _rotation = Rotation::None;
|
|
|
|
uchar *_ioBuffer = nullptr;
|
|
AVIOContext *_ioContext = nullptr;
|
|
AVFormatContext *_fmtContext = nullptr;
|
|
AVCodec *_codec = nullptr;
|
|
AVCodecContext *_codecContext = nullptr;
|
|
int _streamId = 0;
|
|
AVFrame *_frame = nullptr;
|
|
bool _opened = false;
|
|
bool _hadFrame = false;
|
|
bool _frameRead = false;
|
|
int _skippedInvalidDataPackets = 0;
|
|
|
|
bool _hasAudioStream = false;
|
|
int _audioStreamId = -1;
|
|
AudioMsgId _audioMsgId;
|
|
TimeMs _lastReadVideoMs = 0;
|
|
TimeMs _lastReadAudioMs = 0;
|
|
|
|
QQueue<FFMpeg::AVPacketDataWrap> _packetQueue;
|
|
AVPacket _packetNull; // for final decoding
|
|
int _packetStartedSize = 0;
|
|
uint8_t *_packetStartedData = nullptr;
|
|
bool _packetStarted = false;
|
|
|
|
int _width = 0;
|
|
int _height = 0;
|
|
SwsContext *_swsContext = nullptr;
|
|
QSize _swsSize;
|
|
|
|
TimeMs _frameMs = 0;
|
|
int _nextFrameDelay = 0;
|
|
int _currentFrameDelay = 0;
|
|
|
|
TimeMs _frameTime = 0;
|
|
TimeMs _frameTimeCorrection = 0;
|
|
|
|
};
|
|
|
|
} // namespace internal
|
|
} // namespace Clip
|
|
} // namespace Media
|