qosd: исправлен комментарий

Примеров итератора с доступом на чтение я много оставил, а пример
модификации дерева только один. И тот я сразу забыл перевести на
итераторы (range-based for loop).

Т.к. проход теперь не по вектору (он давал экземпляр умного указателя),
а по обычному указателю (собственному прокси, если точнее), то узел
разрушался уже в процессе его отключения от дерева. Добавил удержание
ссылки в сам `unlinkParent()` (ну, нам действительно может быть нужно
только безвозвратное удаление поддерева) и пример в `main.cpp` расширил
и прокомментировал.

BFSIterator.h

@@ -0,0 +1,21 @@
+#pragma once
+
+#include "iterators/BaseIterator.h"
+
+template <typename T>
+struct IteratorForward : public BaseIterator<T> {
+	explicit IteratorForward(T* ptr) : Iterator<T>(first(ptr)) {}
+
+	static T* first(T* node) {
+		return node;
+	}
+
+	static T* next(T* node) {
+		return node ? node->get_next() : nullptr;
+	}
+
+	Iterator<T>& operator++() override {
+		this->current = next(this->current);
+		return *this;
+	}
+};

include/ifaces/IIterator.h

@@ -0,0 +1,14 @@
+#pragma once
+
+/// \brief Интерфейс для классов-итераторов.
+/// \tparam T Тип итерируемого элемента.
+template <typename T>
+class IIterator {
+public:
+	virtual T* operator->() const = 0;
+	virtual operator const T*() const = 0;
+
+	virtual bool operator!=(const IIterator& other) const = 0;
+
+	virtual IIterator& operator++() = 0;
+};

include/ifaces/ILinkMixin.h

@@ -11,8 +11,10 @@ public:
 	using ElemPtr = std::shared_ptr<TElem>;
 
 	virtual ~ILinkMixin() = default;
+	virtual operator std::shared_ptr<TElem>() = 0;
 	virtual void linkChild(const ElemPtr& child) = 0;
 	virtual void unlinkParent() = 0;
 	virtual const std::vector<ElemPtr>& children() = 0;
+	virtual ElemPtr parent() = 0;
 	virtual LinkPtr getLink() = 0;
 };

include/iterators/BFSIterator.h

@@ -0,0 +1,40 @@
+#pragma once
+#include <queue>
+
+#include "iterators/BaseIterator.h"
+
+template <typename T>
+class BFSIterator : public BaseIterator<T> {
+public:
+	explicit BFSIterator(T* ptr) : BaseIterator<T>(ptr) {
+		if (!this->current())
+			return;
+
+		for (auto& it : this->current()->children())
+			q.push({it.get(), 1});
+	}
+
+	IIterator<T>& operator++() override {
+		advance();
+		return *this;
+	}
+
+protected:
+	std::queue<std::pair<T*, size_t>> q;
+
+	void advance() {
+		if (q.empty()) {
+			this->set_current(nullptr);
+			return;
+		}
+
+		auto& [node, lvl] = q.front();
+		q.pop();
+
+		this->set_current(node);
+		this->set_level(lvl);
+
+		for (auto& it : this->current()->children())
+			q.push({it.get(), lvl + 1});
+	}
+};

include/iterators/BaseIterator.h

@@ -0,0 +1,42 @@
+#pragma once
+
+#include "ifaces/IIterator.h"
+
+/// \brief Базовый итератор.
+/// \tparam T Тип итерируемого элемента.
+/// Предоставляет базовую реализацию методов для работы с итерируемыми объектами.
+/// Возвращает объект-прокси, мимикрирующий под итерируемый элемент. Позволяет использовать
+/// распаковку структуры для получения уровня элемента в поддереве.
+template <typename T>
+class BaseIterator : public IIterator<T> {
+public:
+	explicit BaseIterator(T* ptr) : proxy{ptr, 0} {}
+
+	struct Proxy {
+		T* e;
+		size_t level;
+
+		T& operator*() const { return *e; }
+		T* operator->() const { return e; }
+	};
+	Proxy& operator*() { return proxy; }
+
+	T* operator->() const override { return current(); }
+	operator const T*() const { return current(); }
+	operator T&() const { return *current(); }
+	operator T*() const { return current(); }
+
+	bool operator!=(const IIterator<T>& other) const override {
+		const T* other_current = other;
+		return current() != other_current;
+	}
+
+protected:
+	T* current() const { return proxy.e; }
+	void set_current(T* e) { proxy.e = e; }
+
+	size_t level() const { return proxy.level; }
+	void set_level(size_t level) { proxy.level = level; }
+private:
+	Proxy proxy;
+};

include/iterators/DFSIterator.h

@@ -0,0 +1,43 @@
+#pragma once
+#include <stack>
+#include <iterator>
+
+#include "iterators/BaseIterator.h"
+
+template <typename T>
+class DFSIterator : public BaseIterator<T> {
+public:
+	explicit DFSIterator(T* ptr) : BaseIterator<T>(ptr) {
+		if (!this->current())
+			return;
+
+		for (auto it = this->current()->children().rbegin();
+		     it != this->current()->children().rend(); ++it)
+			s.push({it->get(), 1});
+	}
+
+	IIterator<T>& operator++() override {
+		advance();
+		return *this;
+	}
+
+protected:
+	std::stack<std::pair<T*, size_t>> s;
+
+	void advance() {
+		if (s.empty()) {
+			this->set_current(nullptr);
+			return;
+		}
+
+		auto [node, lvl] = s.top();
+		s.pop();
+
+		this->set_current(node);
+		this->set_level(lvl);
+
+		for (auto it = this->current()->children().rbegin();
+		     it != this->current()->children().rend(); ++it)
+			s.push({it->get(), lvl + 1});
+	}
+};

include/iterators/Traversal.h

@@ -0,0 +1,27 @@
+#pragma once
+
+#include "ifaces/INode.h"
+
+#include "iterators/DFSIterator.h"
+#include "iterators/BFSIterator.h"
+
+namespace traversal {
+	/// \brief Фабрика итераторов узлов дерева.
+	/// \tparam T Тип алгоритма для обхода дерева.
+	/// Конструирует итерируемый объект с заданным алгоритмом обхода. Например, в ширину
+	/// или глубину.
+	template<typename Policy>
+	class Traversal {
+		INode* start;
+
+	public:
+		explicit Traversal(INode* start) : start(start) {}
+		explicit Traversal(NodePtr start) : start(start.get()) {}
+
+		Policy begin() const { return Policy(start); }
+		Policy end() const { return Policy(nullptr); }
+	};
+
+	using DFS = Traversal<DFSIterator<INode>>;
+	using BFS = Traversal<BFSIterator<INode>>;
+}

include/links/BaseLink.h

@@ -34,7 +34,7 @@ public:
 	}
 
 	~BaseLink() override {
-		Logger::get("Link").dbg("--- Destructor called for: BaseLink");
+		Logger::get("ConDes").dbg("--- Destructor called for: BaseLink");
 	}
 protected:
 	std::vector<ElemPtr> children_;

include/links/NotImplementedLink.h

@@ -20,7 +20,7 @@ public:
 	void removeChild(const ElemPtr& child) override { }
 
 	~NotImplementedLink() override {
-		Logger::get("Link").dbg("--- Destructor called for: NotImplementedLink");
+		Logger::get("ConDes").dbg("--- Destructor called for: NotImplementedLink");
 	}
 private:
 	std::vector<ElemPtr> empty_;

include/mixins/BaseLinkMixin.h

@@ -14,7 +14,11 @@ class BaseLinkMixin : public virtual ILinkMixin<INode>,
 
 public:
 	~BaseLinkMixin() override {
-		Logger::get("Mixin").dbg("--- Destructor called for: BaseLinkMixin");
+		Logger::get("ConDes").dbg("--- Destructor called for: BaseLinkMixin");
+	}
+
+	operator std::shared_ptr<INode>() override {
+		return this->getNode();
 	}
 
 	void linkChild(const ElemPtr& child) override {
@@ -33,10 +37,25 @@ public:
 
 		auto parentLink = parent->getLink();
 
-		parentLink->removeChild(getNode());
+		/* NOTE:
+		 *
+		 * Keep a reference to the node we gonna to unlink.
+		 * Otherwise, we'll disappear between `removeChild` and
+		 * `setParent`. Do not rearrange these calls, because
+		 * we want to modify the tree top down.
+		 */
+		auto node = getNode();
+		parentLink->removeChild(node);
 		getLink()->setParent(nullptr);
 	}
 
+	ElemPtr parent() override {
+		auto link = getLink();
+
+		ElemPtr parent = link->getParent();
+
+		return parent;
+	}
 
 	const std::vector<ElemPtr>& children() override {
 		auto link = getLink();

include/mixins/FabricMixin.h

@@ -0,0 +1,11 @@
+#pragma once
+#include <memory>
+
+template <typename T>
+class FabricMixin {
+public:
+	template <typename... Args>
+	static std::shared_ptr<T> create(Args&&... args) {
+		return std::shared_ptr<T>(new T(std::forward<Args>(args)...));
+	}
+};

include/mixins/HierarchicalLinkMixin.h

@@ -16,7 +16,7 @@ class HierarchicalLinkMixin : public LazyLinkMixin<OneToOneLink<TElem>> {
 
 public:
 	~HierarchicalLinkMixin() override {
-		Logger::get("Mixin").dbg("--- Destructor called for: HierarchicalLinkMixin");
+		Logger::get("ConDes").dbg("--- Destructor called for: HierarchicalLinkMixin");
 	}
 
 	void linkChild(const ElemPtr& child) override {
@@ -34,10 +34,10 @@ protected:
 
 		if (typeid(*child) == typeid(*this)) {
 			Logger::get("Mixin").dbg("---  Mutate to OneToMany");
-			newLink = std::make_shared<OneToManyLink<TElem>>(this->getNode());
+			newLink = std::make_shared<OneToManyLink<TElem>>(*this);
 		} else {
 			Logger::get("Mixin").dbg("---  Mutate to OneToOne");
-			newLink = std::make_shared<OneToOneLink<TElem>>(this->getNode());
+			newLink = std::make_shared<OneToOneLink<TElem>>(*this);
 		}
 
 		if (newLink && this->link_)

include/mixins/LazyLinkMixin.h

@@ -25,14 +25,13 @@ public:
 	LazyLinkMixin() {}
 
 	~LazyLinkMixin() override {
-		Logger::get("Mixin").dbg("--- Destructor called for: LazyLinkMixin");
+		Logger::get("ConDes").info("--- Destructor called for: LazyLinkMixin");
 	}
 
 protected:
 	void lazyInit() {
 		if (!link_) {
-			link_ = std::make_shared<TLink>(
+			link_ = std::make_shared<TLink>(*this);
-					BaseLinkMixin::getNode());
 		}
 	}
 	LinkPtr link_;

include/nodes/BaseNode.h

@@ -1,5 +1,4 @@
 #pragma once
-#include <iostream>
 
 #include "ifaces/INode.h"
 #include "Logger.h"
@@ -9,12 +8,12 @@
 class BaseNode : public virtual INode {
 public:
 	BaseNode(std::string name) : name_(std::move(name)) {
-		Logger::get("Node").dbg(std::string("--- Base constructor called for: ") + name_);
+		Logger::get("ConDes").dbg(std::string("--- Base constructor called for: ") + name_);
 	}
 	const std::string& name() const override { return name_; }
 	const std::string& kind() const override { return kind_; }
 	~BaseNode() {
-		Logger::get("Node").dbg(std::string("--- Base destructor called for: ") + name_);
+		Logger::get("ConDes").info(std::string("--- Base destructor called for: ") + name_);
 	}
 protected:
 	std::string name_;

include/nodes/ComplexNode.h

@@ -1,19 +1,22 @@
 #pragma once
-#include <iostream>
 
 #include "nodes/BaseNode.h"
 #include "mixins/HierarchicalLinkMixin.h"
+#include "mixins/FabricMixin.h"
 #include "Logger.h"
 
 /// \brief Класс сложного (составного) узла дерева.
-/// Может содержать несколько дочерних ComplexNode и один SimpleNode.
+/// Может содержать несколько дочерних ComplexNode или один SimpleNode.
 class ComplexNode : public BaseNode,
-		    virtual public HierarchicalLinkMixin<INode> {
+		    virtual public HierarchicalLinkMixin<INode>,
+		    public FabricMixin<ComplexNode> {
 public:
 	~ComplexNode() {
-		Logger::get("Node").dbg(std::string("--- Complex destructor called for: ") + name_);
+		Logger::get("ConDes").dbg(std::string("--- Complex destructor called for: ") + name_);
 	}
+private:
+	friend class FabricMixin<ComplexNode>;
 	ComplexNode(std::string name) : BaseNode(std::move(name)) {
-		Logger::get("Node").dbg(std::string("--- Complex constructor called for: ") + name_);
+		Logger::get("ConDes").dbg(std::string("--- Complex constructor called for: ") + name_);
 	}
 };

include/nodes/SimpleNode.h

@@ -1,20 +1,23 @@
 #pragma once
-#include <iostream>
 
 #include "nodes/BaseNode.h"
 #include "mixins/LazyLinkMixin.h"
+#include "mixins/FabricMixin.h"
 #include "links/OneToOneLink.h"
 #include "Logger.h"
 
 /// \brief Класс простого (листового) узла дерева.
 /// Может содержать только одного дочернего ComplexNode.
 class SimpleNode : public BaseNode,
-		   virtual public LazyLinkMixin<OneToOneLink<INode>> {
+		   virtual public LazyLinkMixin<OneToOneLink<INode>>,
+		   public FabricMixin<SimpleNode> {
 public:
 	~SimpleNode() {
-		Logger::get("Node").dbg(std::string("--- Simple destructor called for: ") + name_);
+		Logger::get("ConDes").dbg(std::string("--- Simple destructor called for: ") + name_);
 	}
+private:
+	friend class FabricMixin<SimpleNode>;
 	SimpleNode(std::string name) : BaseNode(std::move(name)) {
-		Logger::get("Node").dbg(std::string("--- Simple constructor called for: ") + name_);
+		Logger::get("ConDes").dbg(std::string("--- Simple constructor called for: ") + name_);
 	}
 };

src/main.cpp

@@ -1,19 +1,29 @@
-#include "nodes/SimpleNode.h"
-#include "nodes/ComplexNode.h"
 #include <iostream>
+
 #include "Logger.h"
 
-void printTree(const NodePtr& node, int indent = 0) {
+#include "nodes/SimpleNode.h"
-	if (!node) {
+#include "nodes/ComplexNode.h"
-		Logger::get("MAIN").err("No node");
+#include "iterators/Traversal.h"
-		return;
+
+
+void printNode(INode& node, size_t level) {
+	for (size_t i = level; i > 0; i--)
+		std::cout << "  ";
+	std::cout << node.name() << "\n";
 }
 
-	for (int i = 0; i < indent; ++i) std::cout << "  ";
+void printTreeBFS(INode& node) {
-	std::cout << node->name() << "\n";
+	for (auto& child : traversal::BFS(&node))
-
+		printNode(*child, 0);
-	for (const auto& child : node->children())
+}
-		printTree(child, indent + 1);
+void printTreeList(INode& node) {
+	for (auto& child : traversal::DFS(&node))
+		printNode(*child, 0);
+}
+void printTreeLadder(INode& node) {
+	for (auto& [child, level] : traversal::DFS(&node))
+		printNode(*child, level);
 }
 
 int main() {
@@ -21,37 +31,39 @@ int main() {
 	Logger::suppressCategory("Node");
 	Logger::suppressCategory("Link");
 	Logger::suppressCategory("Mixin");
+
+	Logger::suppressCategory("ConDes");
+	Logger::setMinSeverity("ConDes", Logger::Severity::Info);
+
 	auto& logger = Logger::get("MAIN");
 	logger.info("Entering main scope...");
 
 	{
-		auto root = std::make_shared<ComplexNode>("ComplexRoot");
+		auto root = ComplexNode::create("ComplexRoot");
-		auto child1 = std::make_shared<ComplexNode>("ComplexChild1");
+		auto child1 = ComplexNode::create("ComplexChild1");
 
 		root->linkChild(child1);
 
 		std::cout << "\nInit tree:\n";
-		printTree(root);
+		printTreeLadder(*root);
 		std::cout << "\n";
 
 		{
-			auto child2 = std::make_shared<ComplexNode>("ComplexChild2");
+			auto child2 = ComplexNode::create("ComplexChild2");
 			root->linkChild(child2);
 
-			auto subchild2 = std::make_shared<SimpleNode>("SimpleSubChild2");
+			auto subchild2 = SimpleNode::create("SimpleSubChild2");
 			child2->linkChild(subchild2);
 
-			auto child3 = std::make_shared<ComplexNode>("ComplexChild3");
+			auto child3 = ComplexNode::create("ComplexChild3");
 			root->linkChild(child3);
 
-			auto subchild3 = std::make_shared<SimpleNode>("SimpleSubChild3");
+			child3->linkChild(SimpleNode::create("SimpleSubChild3"));
-			child3->linkChild(subchild3);
 
 			{
 				// Негативный сценарий 1: попытка добавить второй SimpleNode к ComplexNode
 				try {
-					auto anotherSimple = std::make_shared<SimpleNode>("ShouldFail");
+					child2->linkChild(SimpleNode::create("ShouldFail"));
-					child2->linkChild(anotherSimple);
 					logger.err("[ERROR] Не должно было получиться добавить второй SimpleNode к ComplexNode!");
 				} catch (const std::logic_error& e) {
 					logger.warn(std::string("[Ожидаемое исключение] ") + e.what());
@@ -59,14 +71,12 @@ int main() {
 
 				// Негативный сценарий 2: попытка добавить ComplexNode к SimpleNode
 				// Это допустимо: SimpleNode может иметь ComplexNode в качестве единственного ребёнка
-				auto goodComplex = std::make_shared<ComplexNode>("GoodComplex");
+				subchild2->linkChild(ComplexNode::create("GoodComplex"));
-				subchild2->linkChild(goodComplex);
 				logger.info("[OK] ComplexNode успешно добавлен к SimpleNode как единственный ребёнок.");
 
 				// Негативный сценарий: попытка добавить второго ребёнка к SimpleNode
 				try {
-					auto anotherSimple = std::make_shared<SimpleNode>("ShouldFail2");
+					subchild2->linkChild(SimpleNode::create("ShouldFail2"));
-					subchild2->linkChild(anotherSimple);
 					logger.err("[ERROR] Не должно было получиться добавить второго ребёнка к SimpleNode!");
 				} catch (const std::logic_error& e) {
 					logger.warn(std::string("[Ожидаемое исключение] ") + e.what());
@@ -74,8 +84,7 @@ int main() {
 
 				// Негативный сценарий: попытка добавить SimpleNode в ComplexNode, который уже содержит несколько ComplexNode-дочерних
 				try {
-					auto badSimple = std::make_shared<SimpleNode>("BadSimple");
+					root->linkChild(SimpleNode::create("BadSimple"));
-					root->linkChild(badSimple);
 					logger.err("[ERROR] Не должно было получиться добавить SimpleNode в ComplexNode с несколькими ComplexNode-дочерними!");
 				} catch (const std::logic_error& e) {
 					logger.warn(std::string("[Ожидаемое исключение] ") + e.what());
@@ -83,42 +92,71 @@ int main() {
 			}
 
 			std::cout << "\nAnother tree:\n";
-			printTree(root);
+			printTreeLadder(*root);
+			std::cout << "\n";
+
+			std::cout << "\nList:\n";
+			printTreeList(*root);
+			std::cout << "\n";
+
+			std::cout << "\nBFS:\n";
+			printTreeBFS(*root);
 			std::cout << "\n";
 
 			logger.info("Unlinking ComplexChild2...\n");
 			child2->unlinkParent();
 
 			std::cout << "\nTree after unlink:\n";
-			printTree(root);
+			printTreeLadder(*root);
 			std::cout << "\n";
 
 			logger.info("Put refs of ComplexChild2, ComplexChild3 and its children");
 		}
 
 		std::cout << "\nTree after scope out:\n";
-		printTree(root);
+		printTreeLadder(*root);
 		std::cout << "\n";
 
 		logger.info("Unlinking ComplexChild1 and ComplexChild3...\n");
 		child1->unlinkParent();
-		for (auto child : root->getLink()->getChildren()) {
+		for (auto child : traversal::BFS(root)) {
-			NodePtr childNode = std::dynamic_pointer_cast<INode>(child);
+			if (child->name() == "ComplexChild3") {
-			if (childNode->name() == "ComplexChild3") {
+				/* Avoid of disconnecting a node of the tree
-				childNode->unlinkParent();
+				 * we're traversing. Method holds reference
+				 * to it's object itself. But iterator ignores
+				 * smart pointers, feeling free to dereference
+				 * the destroyed object.
+				 */
+#if 1
+				/* Keep in mind to keep your reference until
+				 * the cycle ends, if you want more than just
+				 * remove subtree.
+				 */
+				NodePtr node = *child;
+				node->unlinkParent();
+				std::cout << "Unlinked: "
+					  << node->name() << std::endl;
 				break;
+#else
+				/* Otherwise, just break immediately you
+				 * unlinked a node. The current pointer is not
+				 * a valid pointer anymore.
+				 */
+				child->unlinkParent();
+				break;
+#endif
 			}
 		}
 
-		auto child4 = std::make_shared<SimpleNode>("SimpleChild4");
+		root->linkChild(SimpleNode::create("SimpleChild4"));
-		root->linkChild(child4);
 
 		std::cout << "\nTree flush and link SimpleChild4:\n";
-		printTree(root);
+		printTreeLadder(*root);
 		std::cout << "\n";
 	}
 
 	logger.info("Exited main scope. All smart pointers destroyed.");
+	logger.info("(It don't? Check ConDes logger)");
 
 	return 0;
 }