1 files changed, 4 insertions, 2 deletions
diff --git a/src/diagramdocument.h b/src/diagramdocument.h
index 2c87fe0..79d38d9 100644
--- a/src/diagramdocument.h
+++ b/src/diagramdocument.h
@@ -26,7 +26,9 @@
 
 class DatabaseTable;
 class DatabaseRelationship;
+class DiagramConnection;
 class DiagramItem;
+class DiagramObject;
 
 class DiagramDocument : public QGraphicsScene
 {
@@ -46,7 +48,6 @@ public:
 
 	DatabaseTable *selectedTable();
 	void deleteSelectedItems();
-	QList<DatabaseRelationship *> findTableRelations(DatabaseTable *table);
 
 	void save(const QString &fileName);
 	void load(const QString &fileName);
@@ -59,6 +60,7 @@ public:
 	void itemMoved(DiagramItem *table);
 	void itemHasMoved(DiagramItem *table);
 
+	QList<DiagramConnection *> findConnections(DiagramObject *object);
 	template <class T> QList<T *> itemsByType();
 
 signals:
@@ -67,7 +69,7 @@ signals:
 	friend class DatabaseTable;
 
 protected slots:
-	void updatePositions(DatabaseTable *table);
+	void updatePositions(DiagramObject *object);
 
 protected:
 	void mousePressEvent(QGraphicsSceneMouseEvent *event);