org.neo4j.bench.chart.AbstractJFreeChart.java Source code

Introduction

Here is the source code for org.neo4j.bench.chart.AbstractJFreeChart.java

Source

/**
 * Copyright (c) 2002-2011 "Neo Technology,"
 * Network Engine for Objects in Lund AB [http://neotechnology.com]
 *
 * This file is part of Neo4j.
 *
 * Neo4j is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero 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 Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */
package org.neo4j.bench.chart;

import java.awt.Dimension;
import java.io.File;
import java.io.IOException;
import java.io.Reader;
import java.text.SimpleDateFormat;
import java.util.Collection;
import java.util.Date;
import java.util.Map;

import org.jfree.chart.ChartPanel;
import org.jfree.chart.ChartUtilities;
import org.jfree.chart.JFreeChart;
import org.jfree.chart.axis.CategoryAxis;
import org.jfree.chart.axis.CategoryLabelPositions;
import org.jfree.chart.plot.CategoryPlot;
import org.neo4j.bench.AggregatedResultHandler;
import org.neo4j.bench.ResultHandler;
import org.neo4j.bench.ResultParser;
import org.neo4j.bench.RunUtil;
import org.neo4j.helpers.Args;

public abstract class AbstractJFreeChart<T> implements Chart {
    public AbstractJFreeChart() {
    }

    protected abstract T instantiateDataset();

    public void open(Reader input, Args options) throws IOException {
        final T dataset = instantiateDataset();
        ResultHandler handler = new ResultHandler() {
            public void newResult(Map<String, String> header) {
            }

            public void value(Map<String, String> header, double value, int numberOfIterations, String benchCase,
                    String timer) {
                addValue(dataset, header, (int) value, numberOfIterations, benchCase, timer);
            }

            public void endResult() {
            }
        };

        Map<String, Collection<String>> aggregations = RunUtil.loadAggregations(options);
        if (aggregations != null) {
            handler = new AggregatedResultHandler(handler, aggregations);
        }

        ResultParser parser = new ResultParser(handler);
        parser.parse(input, options);

        JFreeChart chart = createChart(dataset);
        ChartPanel chartPanel = new ChartPanel(chart);
        Dimension dimensions = getDimensions();
        chartPanel.setPreferredSize(dimensions);
        File chartFile = new File(
                "chart-" + new SimpleDateFormat("yyyy-MM-dd HH:mm:ss").format(new Date()) + ".jpg");
        ChartUtilities.saveChartAsJPEG(chartFile, chart, (int) dimensions.getWidth(), (int) dimensions.getHeight());
    }

    protected void rotateCategoryAxis(JFreeChart chart, double piAngle) {
        CategoryPlot plot = chart.getCategoryPlot();
        CategoryAxis axis = plot.getDomainAxis();
        axis.setCategoryLabelPositions(CategoryLabelPositions.createUpRotationLabelPositions(piAngle));
    }

    protected Dimension getDimensions() {
        return new Dimension(1024, 600);
    }

    protected abstract void addValue(T dataset, Map<String, String> header, double value, int numberOfIterations,
            String benchCase, String timer);

    public void close() {
    }

    protected abstract JFreeChart createChart(T dataset);
}