diff --git a/architype.js b/architype.js index f9d7d52..f15d3c3 100644 --- a/architype.js +++ b/architype.js @@ -420,10 +420,9 @@ class Architype { iterate(graph) { this.sortByMostTension(graph.nodes); + let newPos = null; + let newTension = this.getTotalTension(graph.nodes); for (let node of graph.nodes) { - let newPos = null; - let newTension = node.tension; - let origPos = node.pos; let offsets = new Map(); let addOffset = (x, y) => { @@ -440,8 +439,9 @@ class Architype { if (graph.nodesByPos.has(testPos.toString())) { continue; } - let testVec = this.findVec(testPos, node.affinity); - let testTension = this.findTension(testVec); + node.pos = testPos; + let testTension = this.getTotalTension(graph.nodes); + node.pos = origPos; if (testTension < newTension) { newPos = testPos; newTension = testTension;