DiffDistributionDialog.java

/*
 * Copyright 2005, Sixth and Red River Software
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *  http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 */

package com.sixrr.metrics.ui.charts;

import com.intellij.openapi.project.Project;
import com.intellij.openapi.ui.DialogWrapper;
import com.sixrr.metrics.MetricType;
import com.sixrr.metrics.utils.MetricsReloadedBundle;
import org.jetbrains.annotations.NonNls;
import org.jetbrains.annotations.NotNull;
import org.jfree.chart.ChartPanel;
import org.jfree.chart.JFreeChart;
import org.jfree.chart.JFreeChartConstants;
import org.jfree.chart.axis.NumberAxis;
import org.jfree.chart.labels.StandardXYToolTipGenerator;
import org.jfree.chart.plot.PlotOrientation;
import org.jfree.chart.plot.XYPlot;
import org.jfree.chart.renderer.StandardXYItemRenderer;
import org.jfree.chart.renderer.XYItemRenderer;
import org.jfree.data.XYDataset;
import org.jfree.data.XYSeries;
import org.jfree.data.XYSeriesCollection;

import javax.swing.*;
import java.awt.*;
import java.util.Arrays;

public class DiffDistributionDialog extends DialogWrapper {
    private final ChartPanel chartPanel;
    private final String metricName;
    private final MetricType metricType;
    private final String category;
    private final Double[] datapoints;
    private final Double[] prevDatapoints;

    public DiffDistributionDialog(Project project, String metricName, String category, MetricType metricType,
                                  Double[] datapoints, Double[] prevDatapoints) {
        super(project, false);
        this.category = category;
        this.metricName = metricName;
        this.metricType = metricType;
        this.datapoints = datapoints.clone();
        this.prevDatapoints = prevDatapoints.clone();
        final XYDataset dataset = createDistributionSeries();
        final JFreeChart chart = createChart(dataset);
        chartPanel = new ChartPanel(chart);
        chartPanel.setPreferredSize(new Dimension(500, 270));
        chartPanel.setMouseZoomable(true, false);
        init();
    }

    private XYSeriesCollection createDistributionSeries() {

        final double[] strippedDataPoints = GraphUtils.stripNulls(datapoints);
        final XYSeries series = new XYSeries(metricName);
        Arrays.sort(strippedDataPoints);
        if (strippedDataPoints[0] != 0.0) {
            series.add(0.0, 0.0);
        }
        for (int i = 0; i < strippedDataPoints.length; i++) {
            int j = i + 1;
            while (j < strippedDataPoints.length && strippedDataPoints[i] == strippedDataPoints[j]) {
                i++;
                j++;
            }
            series.add(strippedDataPoints[i], (double) (i + 1) * 100.0 / (double) strippedDataPoints.length);
        }
        final XYSeriesCollection seriesCollection = new XYSeriesCollection();
        seriesCollection.addSeries(series);
        final double[] strippedPrevDataPoints = GraphUtils.stripNulls(prevDatapoints);
        final XYSeries prevSeries = new XYSeries(MetricsReloadedBundle.message("previous") + " " + metricName);
        Arrays.sort(strippedPrevDataPoints);
        if (strippedPrevDataPoints[0] != 0.0) {
            prevSeries.add(0.0, 0.0);
        }
        for (int i = 0; i < strippedPrevDataPoints.length; i++) {
            int j = i + 1;
            while (j < strippedPrevDataPoints.length && strippedPrevDataPoints[i] == strippedPrevDataPoints[j]) {
                i++;
                j++;
            }
            prevSeries
                    .add(strippedPrevDataPoints[i], (double) (i + 1) * 100.0 / (double) strippedPrevDataPoints.length);
        }
        seriesCollection.addSeries(prevSeries);
        return seriesCollection;
    }

    private JFreeChart createChart(XYDataset dataset) {
        final String title = getTitle();

        final NumberAxis xAxis = new NumberAxis(metricName);
        xAxis.setAutoRangeIncludesZero(false);
        if (metricType.equals(MetricType.Ratio) || metricType.equals(MetricType.RecursiveRatio)) {
            xAxis.setNumberFormatOverride(new PercentFormatter());
        }
        final NumberAxis yAxis = new NumberAxis("%");
        final XYItemRenderer renderer = new StandardXYItemRenderer(StandardXYItemRenderer.LINES);
        final XYPlot plot = new XYPlot(dataset, xAxis, yAxis, renderer);
        plot.setOrientation(PlotOrientation.VERTICAL);
        renderer.setToolTipGenerator(new StandardXYToolTipGenerator());
        return new JFreeChart(title, JFreeChartConstants.DEFAULT_TITLE_FONT, plot, true);
    }

    @Override
    public JComponent createCenterPanel() {
        return chartPanel;
    }

    @NotNull
    @Override
    public Action[] createActions() {
        return new Action[0];
    }

    @Override
    public String getTitle() {
        if (metricName.startsWith(MetricsReloadedBundle.message("number.of"))) {
            final String shortName = metricName.substring(MetricsReloadedBundle.message("number.of").length());
            return MetricsReloadedBundle.message("distribution.diff.dialog.title", shortName.toLowerCase(), category);
        } else {
            return MetricsReloadedBundle.message("distribution.diff.dialog.title", metricName.toLowerCase(), category);
        }
    }


    @Override
    @NonNls
    protected String getDimensionServiceKey() {
        return "MetricsReloaded.DiffDistributionDialog";
    }

}