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());