kibana-with-account/src/ui/public/vislib/lib/chart_title.js

kibana-with-account is develop based kibana project

define(function (require) {
  return function ChartTitleFactory(Private) {
    var d3 = require('d3');
    var $ = require('jquery');
    var _ = require('lodash');

    var ErrorHandler = Private(require('ui/vislib/lib/_error_handler'));
    var Tooltip = Private(require('ui/vislib/components/Tooltip'));

    /**
     * Appends chart titles to the visualization
     *
     * @class ChartTitle
     * @constructor
     * @param el {HTMLElement} Reference to DOM element
     */
    _.class(ChartTitle).inherits(ErrorHandler);
    function ChartTitle(el) {
      if (!(this instanceof ChartTitle)) {
        return new ChartTitle(el);
      }

      this.el = el;
      this.tooltip = new Tooltip('chart-title', el, function (d) {
        return '<p>' + _.escape(d.label) + '</p>';
      });
    }

    /**
     * Renders chart titles
     *
     * @method render
     * @returns {D3.Selection|D3.Transition.Transition} DOM element with chart titles
     */
    ChartTitle.prototype.render = function () {
      var el = d3.select(this.el).select('.chart-title').node();
      var width = el ? el.clientWidth : 0;
      var height = el ? el.clientHeight : 0;

      return d3.select(this.el).selectAll('.chart-title').call(this.draw(width, height));
    };

    /**
     * Truncates chart title text
     *
     * @method truncate
     * @param size {Number} Height or width of the HTML Element
     * @returns {Function} Truncates text
     */
    ChartTitle.prototype.truncate = function (size) {
      var self = this;

      return function (selection) {
        selection.each(function () {
          var text = d3.select(this);
          var n = text[0].length;
          var maxWidth = size / n * 0.9;
          var length = this.getComputedTextLength();
          var str;
          var avg;
          var end;

          if (length > maxWidth) {
            str = text.text();
            avg = length / str.length;
            end = Math.floor(maxWidth / avg) - 5;
            str = str.substr(0, end) + '...';
            self.addMouseEvents(text);

            return text.text(str);
          }

          return text.text();
        });
      };
    };

    /**
     * Adds tooltip events on truncated chart titles
     *
     * @method addMouseEvents
     * @param target {HTMLElement} DOM element to attach event listeners
     * @returns {*} DOM element with event listeners attached
     */
    ChartTitle.prototype.addMouseEvents = function (target) {
      if (this.tooltip) {
        return target.call(this.tooltip.render());
      }
    };

    /**
     * Appends chart titles to the visualization
     *
     * @method draw
     * @returns {Function} Appends chart titles to a D3 selection
     */
    ChartTitle.prototype.draw = function (width, height) {
      var self = this;

      return function (selection) {
        selection.each(function () {
          var div = d3.select(this);
          var dataType = this.parentNode.__data__.rows ? 'rows' : 'columns';
          var size = dataType === 'rows' ? height : width;
          var txtHtOffset = 11;

          self.validateWidthandHeight(width, height);

          div.append('svg')
          .attr('width', width)
          .attr('height', height)
          .append('text')
          .attr('transform', function () {
            if (dataType === 'rows') {
              return 'translate(' + txtHtOffset + ',' + height / 2 + ')rotate(270)';
            }
            return 'translate(' + width / 2 + ',' + txtHtOffset + ')';
          })
          .attr('text-anchor', 'middle')
          .text(function (d) { return d.label; });

          // truncate long chart titles
          div.selectAll('text')
          .call(self.truncate(size));
        });
      };
    };

    return ChartTitle;
  };
});