Make diagram notation configurable

2 files changed, 12 insertions, 12 deletions

diff --git a/src/items/database/databaserelationship.cpp b/src/items/database/databaserelationship.cpp
index d908409..6efe07b 100644
--- a/src/items/database/databaserelationship.cpp
+++ b/src/items/database/databaserelationship.cpp
@@ -329,17 +329,17 @@ DatabaseRelationship::updateLayout()
 
 	Q_ASSERT(haveLine);
 
-	// "Relational"
-	if (1) {
-		d->fillEnds = true;
-		d->targetEnd = d->arrowHeadPath();
-		d->sourceEnd = QPainterPath();
-	}
-	// "Crow's Foot"
-	else {
-		d->fillEnds = false;
-		d->targetEnd = d->crowsFootPath(d->cardinality);
-		d->sourceEnd = d->crowsFootPath(isRequired() ? One : ZeroOrOne);
+	switch (document()->notation()) {
+		case DiagramDocument::Relational:
+			d->fillEnds = true;
+			d->targetEnd = d->arrowHeadPath();
+			d->sourceEnd = QPainterPath();
+			break;
+		case DiagramDocument::CrowsFoot:
+			d->fillEnds = false;
+			d->targetEnd = d->crowsFootPath(d->cardinality);
+			d->sourceEnd = d->crowsFootPath(isRequired() ? One : ZeroOrOne);
+			break;
 	}
 
 	if (!d->sourceEnd.isEmpty()) {
diff --git a/src/items/database/databaserelationship.h b/src/items/database/databaserelationship.h
index d8762cf..9b5f0af 100644
--- a/src/items/database/databaserelationship.h
+++ b/src/items/database/databaserelationship.h
@@ -90,7 +90,7 @@ public:
 
 	static DiagramItemProperties *createPropertiesEditor(QWidget *parent = 0);
 
-protected slots:
+public slots:
 	void updateLayout();
 
 protected: