1 files changed, 16 insertions, 0 deletions
diff --git a/src/menu/container.cpp b/src/menu/container.cpp
index a5519f0d..379dd7c1 100644
--- a/src/menu/container.cpp
+++ b/src/menu/container.cpp
@@ -25,6 +25,7 @@
  */
 
 #include "container.hpp"
+#include "frame.hpp"
 
 namespace usdx
 {
@@ -43,6 +44,10 @@ namespace usdx
 
 	void Container::draw(void)
 	{
+		if (frame) {
+			frame->repaint();
+		}
+
 		for (std::list<DrawableControl*>::iterator it =
 			     controls.begin(); it != controls.end(); it++) {
 			(*it)->repaint();
@@ -58,4 +63,15 @@ namespace usdx
 	{
 		controls.remove(child);
 	}
+
+	void Container::setFrame(Frame *f)
+	{
+		f->set_size(get_size());
+		frame = f;
+	}
+
+	void Container::removeFrame(void)
+	{
+		frame = NULL;
+	}
 };