add option to change calculation of magnitude of edges

sandsmark · Mar 2, 2016 · 03e70ee · 03e70ee
1 parent f0970f8
commit 03e70ee
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/hivewidget.cpp b/hivewidget.cpp
@@ -6,7 +6,8 @@
 #include <QMouseEvent>
 
 HiveWidget::HiveWidget(QWidget *parent)
-    : QOpenGLWidget(parent)
+    : QOpenGLWidget(parent),
+      m_scaleEdgeMax(true)
 {
     setMouseTracking(true);
 }
@@ -182,8 +183,13 @@ void HiveWidget::calculate()
             double otherMagnitude = hypot(otherX - cx, otherY - cy);
             double averageRadians = atan2(((nodeY - cy) + (otherY - cy))/2, ((nodeX - cx) + (otherX - cx))/2);
 
-//            double averageMagnitude = (magnitude + otherMagnitude) / 2;
-            double averageMagnitude = qMax(magnitude, otherMagnitude);
+            double averageMagnitude;
+            if (m_scaleEdgeMax) {
+                averageMagnitude = qMax(magnitude, otherMagnitude);
+            } else {
+                averageMagnitude = (magnitude + otherMagnitude) / 2;
+            }
+
             QPointF controlPoint(cos(averageRadians) * averageMagnitude + cx, sin(averageRadians) * averageMagnitude + cy);
 
             QLinearGradient gradient(m_positions[node], m_positions[otherNode]);

diff --git a/hivewidget.h b/hivewidget.h
@@ -38,6 +38,7 @@ class HiveWidget : public QOpenGLWidget
     QMap<QString, QColor> m_colors;
     QMap<QString, QPoint> m_positions;
     QString m_closest;
+    bool m_scaleEdgeMax;
 };
 
 #endif // HIVEWIDGET_H