init

Telegram/SourceFiles/data/data_document_media.h

@@ -0,0 +1,126 @@
+/*
+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 Ui {
+class FrameGenerator;
+} // namespace Ui
+
+namespace Media {
+namespace Clip {
+enum class Notification;
+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]] const QPainterPath &thumbnailPath() 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(HistoryItem *item) 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;
+	mutable QPainterPath _pathThumbnail;
+	std::unique_ptr<Image> _thumbnail;
+	std::unique_ptr<Image> _sticker;
+	QByteArray _bytes;
+	QByteArray _videoThumbnailBytes;
+	Flags _flags;
+
+};
+
+[[nodiscard]] auto DocumentIconFrameGenerator(not_null<DocumentMedia*> media)
+-> FnMut<std::unique_ptr<Ui::FrameGenerator>()>;
+
+[[nodiscard]] auto DocumentIconFrameGenerator(
+	const std::shared_ptr<DocumentMedia> &media)
+-> FnMut<std::unique_ptr<Ui::FrameGenerator>()>;
+
+} // namespace Data