require("../env");

require("../../d3");

require("../../d3.layout");

var vows = require("vows"),

    assert = require("assert");

var suite = vows.describe("d3.layout.treemap");

suite.addBatch({

  "treemap": {

    topic: function() {

      return d3.layout.treemap;

    },

    "outputs a squarified treemap": function(treemap) {

      var t = treemap().size([1000, 1000]).sort(null);

      assert.deepEqual(t.nodes({children: [{value: 1}, {value: 2}, {children: [{value: 1}, {value: 2}]}]}).map(layout), [

        {x: 0, y: 0, dx: 1000, dy: 1000},

        {x: 0, y: 833, dx: 1000, dy: 167},

        {x: 600, y: 0, dx: 400, dy: 833},

        {x: 0, y: 0, dx: 600, dy: 833},

        {x: 0, y: 555, dx: 600, dy: 278},

        {x: 0, y: 0, dx: 600, dy: 555}

      ]);

    },

    "sorts by value by default": function(treemap) {

      var t = treemap().size([1000, 1000]);

      assert.deepEqual(t.nodes({children: [{value: 1}, {value: 2}, {children: [{value: 1}, {value: 2}]}]}).map(layout), [

        {x: 0, y: 0, dx: 1000, dy: 1000},

        {x: 0, y: 0, dx: 333, dy: 500},

        {x: 333, y: 0, dx: 667, dy: 500},

        {x: 0, y: 500, dx: 1000, dy: 500},

        {x: 0, y: 500, dx: 333, dy: 500},

        {x: 333, y: 500, dx: 667, dy: 500}

      ]);

    },

    "ignores zero values": function(treemap) {

      var t = treemap().size([1000, 1000]).sort(null);

      assert.deepEqual(t.nodes({children: [{value: 1}, {value: 0}, {value: 2}, {children: [{value: 1}, {value: 2}]}]}).map(layout), [

        {x: 0, y: 0, dx: 1000, dy: 1000},

        {x: 0, y: 833, dx: 1000, dy: 167},

        {x: 1000, y: 0, dx: 0, dy: 833},

        {x: 600, y: 0, dx: 400, dy: 833},

        {x: 0, y: 0, dx: 600, dy: 833},

        {x: 0, y: 555, dx: 600, dy: 278},

        {x: 0, y: 0, dx: 600, dy: 555}

      ]);

    },

    "ignores NaN values": function(treemap) {

      var t = treemap().size([1000, 1000]).sort(null);

      assert.deepEqual(t.nodes({children: [{value: 1}, {value: NaN}, {value: 2}, {children: [{value: 1}, {value: 2}]}]}).map(layout), [

        {x: 0, y: 0, dx: 1000, dy: 1000},

        {x: 0, y: 833, dx: 1000, dy: 167},

        {x: 1000, y: 0, dx: 0, dy: 833},

        {x: 600, y: 0, dx: 400, dy: 833},

        {x: 0, y: 0, dx: 600, dy: 833},

        {x: 0, y: 555, dx: 600, dy: 278},

        {x: 0, y: 0, dx: 600, dy: 555}

      ]);

    },

    "does not overflow empty size": function(treemap) {

      var t = treemap().size([0, 0]).sort(null);

      assert.deepEqual(t.nodes({children: [{value: 1}, {value: 2}, {children: [{value: 1}, {value: 2}]}]}).map(layout), [

        {x: 0, y: 0, dx: 0, dy: 0},

        {x: 0, y: 0, dx: 0, dy: 0},

        {x: 0, y: 0, dx: 0, dy: 0},

        {x: 0, y: 0, dx: 0, dy: 0},

        {x: 0, y: 0, dx: 0, dy: 0},

        {x: 0, y: 0, dx: 0, dy: 0}

      ]);

    },

    "can specify padding as a number": function(treemap) {

      var t = treemap().size([1000, 1000]).sort(null).padding(1);

      assert.deepEqual(t.nodes({children: [{value: 1}, {value: 2}, {children: [{value: 1}, {value: 2}]}]}).map(layout), [

        {x: 0, y: 0, dx: 1000, dy: 1000},

        {x: 1, y: 833, dx: 998, dy: 166},

        {x: 600, y: 1, dx: 399, dy: 832},

        {x: 1, y: 1, dx: 599, dy: 832},

        {x: 2, y: 555, dx: 597, dy: 277},

        {x: 2, y: 2, dx: 597, dy: 553}

      ]);

    },

    "can specify padding as an array": function(treemap) {

      var t = treemap().size([1000, 1000]).sort(null).padding([1,2,3,4]);

      assert.deepEqual(t.nodes({children: [{value: 1}, {value: 2}, {children: [{value: 1}, {value: 2}]}]}).map(layout), [

        {x: 0, y: 0, dx: 1000, dy: 1000},

        {x: 4, y: 831, dx: 994, dy: 166},

        {x: 600, y: 1, dx: 398, dy: 830},

        {x: 4, y: 1, dx: 596, dy: 830},

        {x: 8, y: 553, dx: 590, dy: 275},

        {x: 8, y: 2, dx: 590, dy: 551}

      ]);

    },

    "can specify padding as null": function(treemap) {

      var t = treemap().size([1000, 1000]).sort(null).padding(null);

      assert.deepEqual(t.nodes({children: [{value: 1}, {value: 2}, {children: [{value: 1}, {value: 2}]}]}).map(layout), [

        {x: 0, y: 0, dx: 1000, dy: 1000},

        {x: 0, y: 833, dx: 1000, dy: 167},

        {x: 600, y: 0, dx: 400, dy: 833},

        {x: 0, y: 0, dx: 600, dy: 833},

        {x: 0, y: 555, dx: 600, dy: 278},

        {x: 0, y: 0, dx: 600, dy: 555}

      ]);

    },

    "can specify padding as a function that returns a number": function(treemap) {

      var t = treemap().size([1000, 1000]).sort(null).padding(function(d) { return d.depth; });

      assert.deepEqual(t.nodes({children: [{value: 1}, {value: 2}, {children: [{value: 1}, {value: 2}]}]}).map(layout), [

        {x: 0, y: 0, dx: 1000, dy: 1000},

        {x: 0, y: 833, dx: 1000, dy: 167},

        {x: 600, y: 0, dx: 400, dy: 833},

        {x: 0, y: 0, dx: 600, dy: 833},

        {x: 1, y: 555, dx: 598, dy: 277},

        {x: 1, y: 1, dx: 598, dy: 554}

      ]);

    },

    "can specify padding as a function that returns an array": function(treemap) {

      var t = treemap().size([1000, 1000]).sort(null).padding(function(d) { return [d.depth,2,3,4]; });

      assert.deepEqual(t.nodes({children: [{value: 1}, {value: 2}, {children: [{value: 1}, {value: 2}]}]}).map(layout), [

        {x: 0, y: 0, dx: 1000, dy: 1000},

        {x: 4, y: 831, dx: 994, dy: 166},

        {x: 600, y: 0, dx: 398, dy: 831},

        {x: 4, y: 0, dx: 596, dy: 831},

        {x: 8, y: 552, dx: 590, dy: 276},

        {x: 8, y: 1, dx: 590, dy: 551}

      ]);

    },

    "can specify padding as a function that returns null": function(treemap) {

      var t = treemap().size([1000, 1000]).sort(null).padding(function(d) { return d.depth & 1 ? null : 1; });

      assert.deepEqual(t.nodes({children: [{value: 1}, {value: 2}, {children: [{value: 1}, {value: 2}]}]}).map(layout), [

        {x: 0, y: 0, dx: 1000, dy: 1000},

        {x: 1, y: 833, dx: 998, dy: 166},

        {x: 600, y: 1, dx: 399, dy: 832},

        {x: 1, y: 1, dx: 599, dy: 832},

        {x: 1, y: 556, dx: 599, dy: 277},

        {x: 1, y: 1, dx: 599, dy: 555}

      ]);

    },

    "no negatively sized rectangles": function(treemap) {

      var t = treemap().size([615, 500]).sort(function(a, b) { return a.value - b.value; }).padding(29),

          nodes = t.nodes({"children": [

            {"value": 1},

            {"value": 9},

            {"value": 3},

            {"value": 15},

            {"value": 44},

            {"value": 28},

            {"value": 32},

            {"value": 41},

            {"value": 50},

            {"value": 60},

            {"value": 64},

            {"value": 75},

            {"value": 76},

            {"value": 84},

            {"value": 88},

            {"value": 100},

            {"value": 140},

            {"value": 142},

            {"value": 363},

            {"value": 657},

            {"value": 670},

            {"value": 822},

            {"value": 1173},

            {"value": 1189}

          ]}).map(layout);

      assert.equal(nodes.filter(function(n) { return n.dx < 0 || n.dy < 0; }).length, 0);

    },

    "can handle an empty children array": function(treemap) {

      assert.deepEqual(treemap().nodes({children: []}).map(layout), [

        {x: 0, y: 0, dx: 1, dy: 1}

      ]);

      assert.deepEqual(treemap().nodes({children: [{children: []}, {value: 1}]}).map(layout), [

        {x: 0, y: 0, dx: 1, dy: 1},

        {x: 0, y: 0, dx: 0, dy: 1},

        {x: 0, y: 0, dx: 1, dy: 1}

      ]);

    }

  }

});

function layout(node) {

  return {

    x: node.x,

    y: node.y,

    dx: node.dx,

    dy: node.dy

  };

}

suite.export(module);