From 015307cbece289fd07ef32e312c9165bedd4bbd3 Mon Sep 17 00:00:00 2001
From: lisyarus <lisyarus@gmail.com>
Date: Fri, 15 Oct 2021 21:37:13 +0300
Subject: [PATCH] Support ui::element::post_draw invoked after children draw

---
 libs/ui/include/psemek/ui/element.hpp | 1 +
 libs/ui/source/controller.cpp         | 1 +
 2 files changed, 2 insertions(+)

diff --git a/libs/ui/include/psemek/ui/element.hpp b/libs/ui/include/psemek/ui/element.hpp
index eae0ef23..319df991 100644
--- a/libs/ui/include/psemek/ui/element.hpp
+++ b/libs/ui/include/psemek/ui/element.hpp
@@ -59,6 +59,7 @@ namespace psemek::ui
 		virtual void update(float /* dt */) {}
 
 		virtual void draw(painter & p) const = 0;
+		virtual void post_draw(painter & /* p */) const {}
 
 		virtual ~element() {}
 
diff --git a/libs/ui/source/controller.cpp b/libs/ui/source/controller.cpp
index 83e53394..7e2554ba 100644
--- a/libs/ui/source/controller.cpp
+++ b/libs/ui/source/controller.cpp
@@ -178,6 +178,7 @@ namespace psemek::ui
 			elem->draw(impl().painter);
 			for (auto c : elem->children())
 				if (c) self(c);
+			elem->post_draw(impl().painter);
 		});
 
 		visitor(impl().root.get());