1 files changed, 19 insertions, 6 deletions
diff --git a/Swift/QtUI/Whiteboard/WhiteboardElementDrawingVisitor.h b/Swift/QtUI/Whiteboard/WhiteboardElementDrawingVisitor.h
index f3d2526..38ed676 100644
--- a/Swift/QtUI/Whiteboard/WhiteboardElementDrawingVisitor.h
+++ b/Swift/QtUI/Whiteboard/WhiteboardElementDrawingVisitor.h
@@ -27,7 +27,9 @@ namespace Swift {
 			pen.setColor(QColor(color.getRed(), color.getGreen(), color.getBlue(), color.getAlpha()));
 			pen.setWidth(element.getPenWidth());
 			item->setPen(pen);
-			graphicsView_->scene()->addItem(item);
+			QString id = P2QSTRING(element.getID());
+			item->setData(100, id);
+			graphicsView_->addItem(item, id);
 		}
 
 		void visit(WhiteboardFreehandPathElement& element) {
@@ -44,7 +46,9 @@ namespace Swift {
 				item->lineTo(QPointF(it->first, it->second));
 			}
 
-			graphicsView_->scene()->addItem(item);
+			QString id = P2QSTRING(element.getID());
+			item->setData(100, id);
+			graphicsView_->addItem(item, id);
 		}
 
 		void visit(WhiteboardRectElement& element) {
@@ -58,7 +62,9 @@ namespace Swift {
 			brush.setColor(QColor(brushColor.getRed(), brushColor.getGreen(), brushColor.getBlue(), brushColor.getAlpha()));
 			item->setPen(pen);
 			item->setBrush(brush);
-			graphicsView_->scene()->addItem(item);
+			QString id = P2QSTRING(element.getID());
+			item->setData(100, id);
+			graphicsView_->addItem(item, id);
 		}
 
 		void visit(WhiteboardPolygonElement& element) {
@@ -74,7 +80,9 @@ namespace Swift {
 				brush.setColor(QColor(brushColor.getRed(), brushColor.getGreen(), brushColor.getBlue(), brushColor.getAlpha()));
 				item->setPen(pen);
 				item->setBrush(brush);
-				graphicsView_->addItem(item, P2QSTRING(element.getID()));
+				QString id = P2QSTRING(element.getID());
+				item->setData(100, id);
+				graphicsView_->addItem(item, id);
 			}
 			QPolygonF polygon;
 			std::vector<std::pair<int, int> >::const_iterator it = element.getPoints().begin();
@@ -92,7 +100,10 @@ namespace Swift {
 			item->setFont(font);
 			Color color = element.getColor();
 			item->setDefaultTextColor(QColor(color.getRed(), color.getGreen(), color.getBlue(), color.getAlpha()));
-			graphicsView_->scene()->addItem(item);
+
+			QString id = P2QSTRING(element.getID());
+			item->setData(100, id);
+			graphicsView_->addItem(item, id);
 		}
 
 		void visit(WhiteboardEllipseElement& element) {
@@ -111,7 +122,9 @@ namespace Swift {
 			brush.setColor(QColor(brushColor.getRed(), brushColor.getGreen(), brushColor.getBlue(), brushColor.getAlpha()));
 			item->setPen(pen);
 			item->setBrush(brush);
-			graphicsView_->scene()->addItem(item);
+			QString id = P2QSTRING(element.getID());
+			item->setData(100, id);
+			graphicsView_->addItem(item, id);
 		}
 
 	private: