1 files changed, 14 insertions, 2 deletions

diff --git a/utils/themeeditor/graphics/rbscreen.h b/utils/themeeditor/graphics/rbscreen.h
index 7b8b83060a..358a49e4a7 100644
--- a/utils/themeeditor/graphics/rbscreen.h
+++ b/utils/themeeditor/graphics/rbscreen.h
@@ -28,8 +28,8 @@
 #include "rbrenderinfo.h"
 #include "rbimage.h"
 #include "rbfont.h"
-
-class RBViewport;
+#include "rbalbumart.h"
+#include "rbviewport.h"
 
 class RBScreen : public QGraphicsItem
 {
@@ -73,6 +73,16 @@ public:
     QColor foreground(){ return fgColor; }
     QColor background(){ return bgColor; }
 
+    void setAlbumArt(RBAlbumArt* art){ albumArt = art; }
+    void showAlbumArt(RBViewport* view)
+    {
+        if(albumArt)
+        {
+            albumArt->setParentItem(view);
+            albumArt->show();
+        }
+    }
+
 
 private:
     int width;
@@ -90,6 +100,8 @@ private:
     QMap<int, RBFont*> fonts;
     QList<QString> displayedViewports;
 
+    RBAlbumArt* albumArt;
+
 };
 
 #endif // RBSCREEN_H