init

Telegram/SourceFiles/editor/photo_editor.cpp

@@ -0,0 +1,242 @@
+/*
+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
+*/
+#include "editor/photo_editor.h"
+
+#include "core/application.h"
+#include "core/core_settings.h"
+#include "editor/color_picker.h"
+#include "editor/controllers/controllers.h"
+#include "editor/photo_editor_content.h"
+#include "editor/photo_editor_controls.h"
+#include "window/window_controller.h"
+#include "window/window_session_controller.h"
+#include "ui/layers/layer_widget.h"
+#include "styles/style_editor.h"
+
+namespace Editor {
+namespace {
+
+constexpr auto kPrecision = 100000;
+
+[[nodiscard]] QByteArray Serialize(const Brush &brush) {
+	auto result = QByteArray();
+	auto stream = QDataStream(&result, QIODevice::WriteOnly);
+	stream.setVersion(QDataStream::Qt_5_3);
+	stream << qint32(brush.sizeRatio * kPrecision) << brush.color;
+	stream.device()->close();
+
+	return result;
+}
+
+[[nodiscard]] Brush Deserialize(const QByteArray &data) {
+	auto stream = QDataStream(data);
+	auto result = Brush();
+	auto size = qint32(0);
+	stream >> size >> result.color;
+	result.sizeRatio = size / float(kPrecision);
+	return (stream.status() != QDataStream::Ok)
+		? Brush()
+		: result;
+}
+
+} // namespace
+
+PhotoEditor::PhotoEditor(
+	not_null<QWidget*> parent,
+	not_null<Window::Controller*> controller,
+	std::shared_ptr<Image> photo,
+	PhotoModifications modifications,
+	EditorData data)
+: PhotoEditor(
+	parent,
+	controller->uiShow(),
+	(controller->sessionController()
+		? controller->sessionController()->uiShow()
+		: nullptr),
+	std::move(photo),
+	std::move(modifications),
+	std::move(data)) {
+}
+
+PhotoEditor::PhotoEditor(
+	not_null<QWidget*> parent,
+	std::shared_ptr<Ui::Show> show,
+	std::shared_ptr<ChatHelpers::Show> sessionShow,
+	std::shared_ptr<Image> photo,
+	PhotoModifications modifications,
+	EditorData data)
+: RpWidget(parent)
+, _modifications(std::move(modifications))
+, _controllers(std::make_shared<Controllers>(
+	sessionShow
+		? std::make_unique<StickersPanelController>(
+			this,
+			std::move(sessionShow))
+		: nullptr,
+	std::make_unique<UndoController>(),
+	std::move(show)))
+, _content(base::make_unique_q<PhotoEditorContent>(
+	this,
+	photo,
+	_modifications,
+	_controllers,
+	data))
+, _controls(base::make_unique_q<PhotoEditorControls>(
+	this,
+	_controllers,
+	_modifications,
+	data))
+, _colorPicker(std::make_unique<ColorPicker>(
+	this,
+	Deserialize(Core::App().settings().photoEditorBrush()))) {
+
+	sizeValue(
+	) | rpl::on_next([=](const QSize &size) {
+		if (size.isEmpty()) {
+			return;
+		}
+		_content->setGeometry(rect() - st::photoEditorContentMargins);
+	}, lifetime());
+
+	_content->innerRect(
+	) | rpl::on_next([=](QRect inner) {
+		if (inner.isEmpty()) {
+			return;
+		}
+		const auto innerTop = _content->y() + inner.top();
+		const auto skip = st::photoEditorCropPointSize;
+		const auto controlsRect = rect()
+			- style::margins(0, innerTop + inner.height() + skip, 0, 0);
+		_controls->setGeometry(controlsRect);
+	}, lifetime());
+
+	_controls->colorLinePositionValue(
+	) | rpl::on_next([=](const QPoint &p) {
+		_colorPicker->moveLine(p);
+	}, _controls->lifetime());
+
+	_controls->colorLineShownValue(
+	) | rpl::on_next([=](bool shown) {
+		_colorPicker->setVisible(shown);
+	}, _controls->lifetime());
+
+	_mode.value(
+	) | rpl::on_next([=](const PhotoEditorMode &mode) {
+		_content->applyMode(mode);
+		_controls->applyMode(mode);
+	}, lifetime());
+
+	_controls->rotateRequests(
+	) | rpl::on_next([=](int angle) {
+		_modifications.angle += 90;
+		if (_modifications.angle >= 360) {
+			_modifications.angle -= 360;
+		}
+		_content->applyModifications(_modifications);
+	}, lifetime());
+
+	_controls->flipRequests(
+	) | rpl::on_next([=] {
+		_modifications.flipped = !_modifications.flipped;
+		_content->applyModifications(_modifications);
+	}, lifetime());
+
+	_controls->paintModeRequests(
+	) | rpl::on_next([=] {
+		_mode = PhotoEditorMode{
+			.mode = PhotoEditorMode::Mode::Paint,
+			.action = PhotoEditorMode::Action::None,
+		};
+	}, lifetime());
+
+	_controls->doneRequests(
+	) | rpl::on_next([=] {
+		const auto mode = _mode.current().mode;
+		if (mode == PhotoEditorMode::Mode::Paint) {
+			_mode = PhotoEditorMode{
+				.mode = PhotoEditorMode::Mode::Transform,
+				.action = PhotoEditorMode::Action::Save,
+			};
+		} else if (mode == PhotoEditorMode::Mode::Transform) {
+			_mode = PhotoEditorMode{
+				.mode = PhotoEditorMode::Mode::Out,
+				.action = PhotoEditorMode::Action::Save,
+			};
+			save();
+		}
+	}, lifetime());
+
+	_controls->cancelRequests(
+	) | rpl::on_next([=] {
+		const auto mode = _mode.current().mode;
+		if (mode == PhotoEditorMode::Mode::Paint) {
+			_mode = PhotoEditorMode{
+				.mode = PhotoEditorMode::Mode::Transform,
+				.action = PhotoEditorMode::Action::Discard,
+			};
+		} else if (mode == PhotoEditorMode::Mode::Transform) {
+			_mode = PhotoEditorMode{
+				.mode = PhotoEditorMode::Mode::Out,
+				.action = PhotoEditorMode::Action::Discard,
+			};
+			_cancel.fire({});
+		}
+	}, lifetime());
+
+	_colorPicker->saveBrushRequests(
+	) | rpl::on_next([=](const Brush &brush) {
+		_content->applyBrush(brush);
+
+		const auto serialized = Serialize(brush);
+		if (Core::App().settings().photoEditorBrush() != serialized) {
+			Core::App().settings().setPhotoEditorBrush(serialized);
+			Core::App().saveSettingsDelayed();
+		}
+	}, lifetime());
+}
+
+void PhotoEditor::keyPressEvent(QKeyEvent *e) {
+	if (!_colorPicker->preventHandleKeyPress()) {
+		_content->handleKeyPress(e) || _controls->handleKeyPress(e);
+	}
+}
+
+void PhotoEditor::save() {
+	_content->save(_modifications);
+	_done.fire_copy(_modifications);
+}
+
+rpl::producer<PhotoModifications> PhotoEditor::doneRequests() const {
+	return _done.events();
+}
+
+rpl::producer<> PhotoEditor::cancelRequests() const {
+	return _cancel.events();
+}
+
+void InitEditorLayer(
+		not_null<Ui::LayerWidget*> layer,
+		not_null<PhotoEditor*> editor,
+		Fn<void(PhotoModifications)> doneCallback) {
+	editor->cancelRequests(
+	) | rpl::on_next([=] {
+		layer->closeLayer();
+	}, editor->lifetime());
+
+	const auto weak = base::make_weak(layer.get());
+	editor->doneRequests(
+	) | rpl::on_next([=, done = std::move(doneCallback)](
+			const PhotoModifications &mods) {
+		done(mods);
+		if (const auto strong = weak.get()) {
+			strong->closeLayer();
+		}
+	}, editor->lifetime());
+}
+
+} // namespace Editor