1 files changed, 16 insertions, 4 deletions

diff --git a/pictureviewer2.h b/pictureviewer2.h
index e18fc18..d31780e 100644
--- a/pictureviewer2.h
+++ b/pictureviewer2.h
@@ -19,7 +19,11 @@ class QGraphicsPixmapItem;
 class QWheelEvent;
 class QContextMenuEvent;
 class QTimer;
-class PictureViewer2FileinfoItem;
+class QTextDocument;
+class QTextCursor;
+class PictureViewer2Item;
+class SmTreeItem;
+class MappingTreeModel;
 
 typedef QList<QVariant> PicData;
 typedef QList<QList<QVariant> > PicDataList;
@@ -51,10 +55,13 @@ class PictureViewer2 : public QGraphicsView {
     private slots:
         void showFile(const PicData &data);
         void constructInfoItem(const PicData &file, QSize picSize);
+        void constructMappingItem(const PicData &file);
 
     private:
         void setupDialog();
         void setGradient(const QPixmap &pic);
+        QTextDocument *treeToString(const SmTreeItem *root) const;
+        void treeToStringRecursive(const SmTreeItem *parent, QTextCursor *cursor) const;
         QString constructWindowTitle() const;
         PicDataList mFiles;
         QGraphicsScene *mScene;
@@ -63,22 +70,27 @@ class PictureViewer2 : public QGraphicsView {
         int mCurPos;
         QTimer *mSlideTimer;
         const QString mDefaultFile;
-        PictureViewer2FileinfoItem *mFnItem;
+        PictureViewer2Item *mFnItem;
+        PictureViewer2Item *mMappingItem;
         bool mUseGradient;
         QColor mBgColor;
         QPointF mInfoPos;
+        QPointF mMappingPos;
         QAction *mSlideA;
         QAction *mHideA;
 };
 
-class PictureViewer2FileinfoItem : public QGraphicsItem {
+class PictureViewer2Item : public QGraphicsItem {
     public:
-        explicit PictureViewer2FileinfoItem(const PicData &data, const int numSelected, const QSize &picSize = QSize(), QGraphicsItem *parent = 0);
+        explicit PictureViewer2Item(const PicData &data, const int numSelected, const QSize &picSize = QSize(), QGraphicsItem *parent = 0);
+        explicit PictureViewer2Item(QTextDocument *doc, QGraphicsItem *parent = 0);
+        ~PictureViewer2Item();
         QRectF boundingRect() const;
         void paint(QPainter *painter, const QStyleOptionGraphicsItem *option, QWidget *widget);
 
     private:
         QStaticText mText;
+        QTextDocument *mDoc;
 };
 
 #endif // PICTUREVIEWER2_H