tdesktop/Telegram/SourceFiles/data/data_document_media.h

/*
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/flags.h"

class Image;
class FileLoader;

namespace Media {
namespace Clip {
enum Notification : int;
class ReaderPointer;
} // namespace Clip
} // namespace Media

namespace Data {

class DocumentMedia;

class VideoPreviewState final {
public:
	explicit VideoPreviewState(DocumentMedia *media);

	void automaticLoad(Data::FileOrigin origin) const;
	[[nodiscard]] ::Media::Clip::ReaderPointer makeAnimation(
		Fn<void(::Media::Clip::Notification)> callback) const;
	[[nodiscard]] bool usingThumbnail() const;
	[[nodiscard]] bool loading() const;
	[[nodiscard]] bool loaded() const;

private:
	DocumentMedia *_media = nullptr;
	bool _usingThumbnail = false;

};

class DocumentMedia final {
public:
	explicit DocumentMedia(not_null<DocumentData*> owner);
	~DocumentMedia();

	[[nodiscard]] not_null<DocumentData*> owner() const;

	void goodThumbnailWanted();
	[[nodiscard]] Image *goodThumbnail() const;
	void setGoodThumbnail(QImage thumbnail);

	[[nodiscard]] Image *thumbnailInline() const;

	[[nodiscard]] Image *thumbnail() const;
	[[nodiscard]] QSize thumbnailSize() const;
	void thumbnailWanted(Data::FileOrigin origin);
	void setThumbnail(QImage thumbnail);

	[[nodiscard]] QByteArray videoThumbnailContent() const;
	[[nodiscard]] QSize videoThumbnailSize() const;
	void videoThumbnailWanted(Data::FileOrigin origin);
	void setVideoThumbnail(QByteArray content);

	void checkStickerLarge();
	void checkStickerSmall();
	[[nodiscard]] Image *getStickerSmall();
	[[nodiscard]] Image *getStickerLarge();
	void checkStickerLarge(not_null<FileLoader*> loader);

	void setBytes(const QByteArray &bytes);
	[[nodiscard]] QByteArray bytes() const;
	[[nodiscard]] bool loaded(bool check = false) const;
	[[nodiscard]] float64 progress() const;
	[[nodiscard]] bool canBePlayed() const;

	void automaticLoad(Data::FileOrigin origin, const HistoryItem *item);

	void collectLocalData(not_null<DocumentMedia*> local);

	// For DocumentData.
	static void CheckGoodThumbnail(not_null<DocumentData*> document);

private:
	enum class Flag : uchar {
		GoodThumbnailWanted = 0x01,
	};
	inline constexpr bool is_flag_type(Flag) { return true; };
	using Flags = base::flags<Flag>;

	static void ReadOrGenerateThumbnail(not_null<DocumentData*> document);
	static void GenerateGoodThumbnail(
		not_null<DocumentData*> document,
		QByteArray data);

	[[nodiscard]] bool thumbnailEnoughForSticker() const;

	// NB! Right now DocumentMedia can outlive Main::Session!
	// In DocumentData::collectLocalData a shared_ptr is sent on_main.
	// In case this is a problem the ~Gif code should be rewritten.
	const not_null<DocumentData*> _owner;
	std::unique_ptr<Image> _goodThumbnail;
	mutable std::unique_ptr<Image> _inlineThumbnail;
	std::unique_ptr<Image> _thumbnail;
	std::unique_ptr<Image> _sticker;
	QByteArray _bytes;
	QByteArray _videoThumbnailBytes;
	Flags _flags;

};

} // namespace Data