init

Telegram/SourceFiles/editor/scene/scene.cpp

@@ -0,0 +1,220 @@
+/*
+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/scene/scene.h"
+
+#include "editor/scene/scene_item_canvas.h"
+#include "editor/scene/scene_item_line.h"
+#include "editor/scene/scene_item_sticker.h"
+#include "ui/rp_widget.h"
+
+#include <QGraphicsSceneMouseEvent>
+
+namespace Editor {
+namespace {
+
+using ItemPtr = std::shared_ptr<NumberedItem>;
+
+bool SkipMouseEvent(not_null<QGraphicsSceneMouseEvent*> event) {
+	return event->isAccepted() || (event->button() == Qt::RightButton);
+}
+
+} // namespace
+
+Scene::Scene(const QRectF &rect)
+: QGraphicsScene(rect)
+, _canvas(std::make_shared<ItemCanvas>())
+, _lastZ(std::make_shared<float64>(9000.)) {
+	QGraphicsScene::addItem(_canvas.get());
+	_canvas->clearPixmap();
+
+	_canvas->grabContentRequests(
+	) | rpl::on_next([=](ItemCanvas::Content &&content) {
+		const auto item = std::make_shared<ItemLine>(
+			std::move(content.pixmap));
+		item->setPos(content.position);
+		addItem(item);
+		_canvas->setZValue(++_lastLineZ);
+	}, _lifetime);
+}
+
+void Scene::cancelDrawing() {
+	_canvas->cancelDrawing();
+}
+
+void Scene::addItem(ItemPtr item) {
+	if (!item) {
+		return;
+	}
+	item->setNumber(_itemNumber++);
+	QGraphicsScene::addItem(item.get());
+	_items.push_back(std::move(item));
+	_addsItem.fire({});
+}
+
+void Scene::removeItem(not_null<QGraphicsItem*> item) {
+	const auto it = ranges::find_if(_items, [&](const ItemPtr &i) {
+		return i.get() == item;
+	});
+	if (it == end(_items)) {
+		return;
+	}
+	removeItem(*it);
+}
+
+void Scene::removeItem(const ItemPtr &item) {
+	item->setStatus(NumberedItem::Status::Removed);
+	_removesItem.fire({});
+}
+
+void Scene::mousePressEvent(QGraphicsSceneMouseEvent *event) {
+	QGraphicsScene::mousePressEvent(event);
+	if (SkipMouseEvent(event)) {
+		return;
+	}
+	_canvas->handleMousePressEvent(event);
+}
+
+void Scene::mouseReleaseEvent(QGraphicsSceneMouseEvent *event) {
+	QGraphicsScene::mouseReleaseEvent(event);
+	if (SkipMouseEvent(event)) {
+		return;
+	}
+	_canvas->handleMouseReleaseEvent(event);
+}
+
+void Scene::mouseMoveEvent(QGraphicsSceneMouseEvent *event) {
+	QGraphicsScene::mouseMoveEvent(event);
+	if (SkipMouseEvent(event)) {
+		return;
+	}
+	_canvas->handleMouseMoveEvent(event);
+}
+
+void Scene::applyBrush(const QColor &color, float size) {
+	_canvas->applyBrush(color, size);
+}
+
+rpl::producer<> Scene::addsItem() const {
+	return _addsItem.events();
+}
+
+rpl::producer<> Scene::removesItem() const {
+	return _removesItem.events();
+}
+
+std::vector<ItemPtr> Scene::items(
+		Qt::SortOrder order) const {
+	auto copyItems = _items;
+
+	ranges::sort(copyItems, [&](ItemPtr a, ItemPtr b) {
+		const auto numA = a->number();
+		const auto numB = b->number();
+		return (order == Qt::AscendingOrder) ? (numA < numB) : (numA > numB);
+	});
+
+	return copyItems;
+}
+
+std::shared_ptr<float64> Scene::lastZ() const {
+	return _lastZ;
+}
+
+void Scene::updateZoom(float64 zoom) {
+	_canvas->updateZoom(zoom);
+	for (const auto &item : items()) {
+		if (item->type() >= ItemBase::Type) {
+			static_cast<ItemBase*>(item.get())->updateZoom(zoom);
+		}
+	}
+}
+
+bool Scene::hasUndo() const {
+	return ranges::any_of(_items, &NumberedItem::isNormalStatus);
+}
+
+bool Scene::hasRedo() const {
+	return ranges::any_of(_items, &NumberedItem::isUndidStatus);
+}
+
+void Scene::performUndo() {
+	const auto filtered = items(Qt::DescendingOrder);
+
+	const auto it = ranges::find_if(filtered, &NumberedItem::isNormalStatus);
+	if (it != filtered.end()) {
+		(*it)->setStatus(NumberedItem::Status::Undid);
+	}
+}
+
+void Scene::performRedo() {
+	const auto filtered = items(Qt::AscendingOrder);
+
+	const auto it = ranges::find_if(filtered, &NumberedItem::isUndidStatus);
+	if (it != filtered.end()) {
+		(*it)->setStatus(NumberedItem::Status::Normal);
+	}
+}
+
+void Scene::removeIf(Fn<bool(const ItemPtr &)> proj) {
+	auto copy = std::vector<ItemPtr>();
+	for (const auto &item : _items) {
+		const auto toRemove = proj(item);
+		if (toRemove) {
+			// Scene loses ownership of an item.
+			// It seems for some reason this line causes a crash. =(
+			// QGraphicsScene::removeItem(item.get());
+		} else {
+			copy.push_back(item);
+		}
+	}
+	_items = std::move(copy);
+}
+
+void Scene::clearRedoList() {
+	for (const auto &item : _items) {
+		if (item->isUndidStatus()) {
+			item->setStatus(NumberedItem::Status::Removed);
+		}
+	}
+}
+
+void Scene::save(SaveState state) {
+	removeIf([](const ItemPtr &item) {
+		return item->isRemovedStatus()
+			&& !item->hasState(SaveState::Keep)
+			&& !item->hasState(SaveState::Save);
+	});
+
+	for (const auto &item : _items) {
+		item->save(state);
+	}
+	clearSelection();
+	cancelDrawing();
+}
+
+void Scene::restore(SaveState state) {
+	removeIf([=](const ItemPtr &item) {
+		return !item->hasState(state);
+	});
+
+	for (const auto &item : _items) {
+		item->restore(state);
+	}
+	clearSelection();
+	cancelDrawing();
+}
+
+Scene::~Scene() {
+	// Prevent destroying by scene of all items.
+	QGraphicsScene::removeItem(_canvas.get());
+	for (const auto &item : items()) {
+		// Scene loses ownership of an item.
+		QGraphicsScene::removeItem(item.get());
+	}
+}
+
+} // namespace Editor