reuse flip tree in icicle

Author: archmoj

src/traces/icicle/partition.js

@@ -1,6 +1,7 @@
 'use strict';
 
 var d3Hierarchy = require('d3-hierarchy');
+var flipTree = require('../treemap/flip_tree');
 
 module.exports = function partition(entry, size, opts) {
     var flipX = opts.flipX;
@@ -31,38 +32,3 @@ module.exports = function partition(entry, size, opts) {
     }
     return result;
 };
-
-function flipTree(node, size, opts) {
-    var tmp;
-
-    if(opts.swapXY) {
-        // swap x0 and y0
-        tmp = node.x0;
-        node.x0 = node.y0;
-        node.y0 = tmp;
-
-        // swap x1 and y1
-        tmp = node.x1;
-        node.x1 = node.y1;
-        node.y1 = tmp;
-    }
-
-    if(opts.flipX) {
-        tmp = node.x0;
-        node.x0 = size[0] - node.x1;
-        node.x1 = size[0] - tmp;
-    }
-
-    if(opts.flipY) {
-        tmp = node.y0;
-        node.y0 = size[1] - node.y1;
-        node.y1 = size[1] - tmp;
-    }
-
-    var children = node.children;
-    if(children) {
-        for(var i = 0; i < children.length; i++) {
-            flipTree(children[i], size, opts);
-        }
-    }
-}

src/traces/treemap/flip_tree.js

@@ -0,0 +1,36 @@
+'use strict';
+
+module.exports = function flipTree(node, size, opts) {
+    var tmp;
+
+    if(opts.swapXY) {
+        // swap x0 and y0
+        tmp = node.x0;
+        node.x0 = node.y0;
+        node.y0 = tmp;
+
+        // swap x1 and y1
+        tmp = node.x1;
+        node.x1 = node.y1;
+        node.y1 = tmp;
+    }
+
+    if(opts.flipX) {
+        tmp = node.x0;
+        node.x0 = size[0] - node.x1;
+        node.x1 = size[0] - tmp;
+    }
+
+    if(opts.flipY) {
+        tmp = node.y0;
+        node.y0 = size[1] - node.y1;
+        node.y1 = size[1] - tmp;
+    }
+
+    var children = node.children;
+    if(children) {
+        for(var i = 0; i < children.length; i++) {
+            flipTree(children[i], size, opts);
+        }
+    }
+};

src/traces/treemap/partition.js

@@ -1,6 +1,7 @@
 'use strict';
 
 var d3Hierarchy = require('d3-hierarchy');
+var flipTree = require('./flip_tree');
 
 module.exports = function partition(entry, size, opts) {
     var flipX = opts.flipX;
@@ -59,38 +60,3 @@ function getTilingMethod(key, squarifyratio) {
             return d3Hierarchy.treemapSliceDice;
     }
 }
-
-function flipTree(node, size, opts) {
-    var tmp;
-
-    if(opts.swapXY) {
-        // swap x0 and y0
-        tmp = node.x0;
-        node.x0 = node.y0;
-        node.y0 = tmp;
-
-        // swap x1 and y1
-        tmp = node.x1;
-        node.x1 = node.y1;
-        node.y1 = tmp;
-    }
-
-    if(opts.flipX) {
-        tmp = node.x0;
-        node.x0 = size[0] - node.x1;
-        node.x1 = size[0] - tmp;
-    }
-
-    if(opts.flipY) {
-        tmp = node.y0;
-        node.y0 = size[1] - node.y1;
-        node.y1 = size[1] - tmp;
-    }
-
-    var children = node.children;
-    if(children) {
-        for(var i = 0; i < children.length; i++) {
-            flipTree(children[i], size, opts);
-        }
-    }
-}