From 5fa863d28c15aa739557b8f3fa54b6fa5434404c Mon Sep 17 00:00:00 2001
From: Ian Gulliver <ian@flamingcow.tv>
Date: Mon, 15 Jul 2019 23:58:23 +0000
Subject: [PATCH] Reduce affinity confusion

---
 LayoutNode.js | 7 ++-----
 1 file changed, 2 insertions(+), 5 deletions(-)

diff --git a/LayoutNode.js b/LayoutNode.js
index 667a644..a125d3c 100644
--- a/LayoutNode.js
+++ b/LayoutNode.js
@@ -44,7 +44,7 @@ class LayoutNode {
       // If so, preserve one space above the group
       let labeled = new Set(Array.from(this.groups).filter(g => !!g.label));
       if (asymDifference(labeled, node.groups).size) {
-        node.addAffinity(this, (d, v) =>
+        this.addAffinity(node, (d, v) =>
                          (v[0] == 0 && v[1] > 0 && v[1] < 2) ? -INF : 0);
       }
 
@@ -65,10 +65,7 @@ class LayoutNode {
         if (group.nodes.has(node)) {
           continue;
         }
-        this.affinity_.push({
-          node: node,
-          distanceToWeight: (d, v, p) => group.isContained(p) ? -INF : 0,
-        });
+        this.addAffinity(node, (d, v, p) => group.isContained(p) ? -INF : 0);
       }
     }