xuanluo / flot-axislabels Goto Github PK

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>.

function init(plot) {
    // This is kind of a hack. There are no hooks in Flot between
    // the creation and measuring of the ticks (setTicks, measureTickLabels
    // in setupGrid() ) and the drawing of the ticks and plot box
    // (insertAxisLabels in setupGrid() ).
    //
    // Therefore, we use a trick where we run the draw routine twice:
    // the first time to get the tick measurements, so that we can change
    // them, and then have it draw it again.
    var secondPass = false;
    plot.hooks.draw.push(function (plot, ctx) {
        if (!secondPass) {
            // MEASURE AND SET OPTIONS
            $.each(plot.getAxes(), function(axisName, axis) {
                var opts = axis.options // Flot 0.7
                    || plot.getOptions()[axisName]; // Flot 0.6
                if (!opts || !opts.axisLabel)
                    return;

                var w, h;
                if (opts.axisLabelUseCanvas != false)
                    opts.axisLabelUseCanvas = true;

                if (opts.axisLabelUseCanvas) {
                    // canvas text
                    if (!opts.axisLabelFontSizePixels)
                        opts.axisLabelFontSizePixels = 14;
                    if (!opts.axisLabelFontFamily)
                        opts.axisLabelFontFamily = 'sans-serif';
                    // since we currently always display x as horiz.
                    // and y as vertical, we only care about the height
                    |*w = opts.axisLabelFontSizePixels + 3; *|
                    h = opts.axisLabelFontSizePixels;

                } else {
                    // HTML text
                    var elem = $('<div class="axisLabels" style="position:absolute;">' + opts.axisLabel + '</div>');
                    plot.getPlaceholder().append(elem);
                    w = elem.outerWidth(true);
                    h = elem.outerHeight(true);
                    elem.remove();
                }

                if (axisName.charAt(0) == 'x')
                    axis.labelHeight += h;
                else
                    axis.labelWidth += w;
                opts.labelHeight = axis.labelHeight;
                opts.labelWidth = axis.labelWidth;
            });
            // re-draw with new label widths and heights
            secondPass = true;
            plot.setupGrid();
            plot.draw();


        } else {
            // DRAW
            $.each(plot.getAxes(), function(axisName, axis) {
                var opts = axis.options // Flot 0.7
                    || plot.getOptions()[axisName]; // Flot 0.6
                if (!opts || !opts.axisLabel)
                    return;

                if (opts.axisLabelUseCanvas) {
                    // canvas text
                    var ctx = plot.getCanvas().getContext('2d');
                    ctx.save();
                    ctx.font = opts.axisLabelFontSizePixels + 'px ' +
                            opts.axisLabelFontFamily;
                    var width = ctx.measureText(opts.axisLabel).width;
                    var height = opts.axisLabelFontSizePixels;
                    var x, y;
                    if (axisName.charAt(0) == 'x') {
                        x = plot.getPlotOffset().left + plot.width()/2 - width/2;
                        y = plot.getCanvas().height;
                    } else {
                        |*height += 3;*|
                        x = height * 0.72;
                        y = plot.getPlotOffset().top + plot.height()/2 - width/2;
                    }
                    ctx.translate(x, y);
                    ctx.rotate((axisName.charAt(0) == 'x') ? 0 : -Math.PI/2);
                    ctx.fillText(opts.axisLabel, 0, 0);
                    ctx.restore();

                } else {
                    // HTML text
                    plot.getPlaceholder().find('#' + axisName + 'Label').remove();
                    var elem = $('<div id="' + axisName + 'Label" " class="axisLabels" style="position:absolute;">' + opts.axisLabel + '</div>');
                    if (axisName.charAt(0) == 'x') {
                        elem.css('left', plot.getPlotOffset().left + plot.width()/2 - elem.outerWidth()/2 + 'px');
                        elem.css('bottom', '0px');
                    } else {
                        elem.css('top', plot.getPlotOffset().top + plot.height()/2 - elem.outerHeight()/2 + 'px');
                        elem.css('left', '0px');
                    }
                    plot.getPlaceholder().append(elem);
                }
            });
            secondPass = false;
        }
    });
}



$.plot.plugins.push({
    init: init,
    options: options,
    name: 'axisLabels',
    version: '1.0'
});