edu.gcsc.vrl.jfreechart.TrajectoryPlotter.java Source code

Introduction

Here is the source code for edu.gcsc.vrl.jfreechart.TrajectoryPlotter.java

Source

/* 
 * TrajectoryPlotter.java
 * 
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
 *
 * Copyright (c) 20092012 Steinbeis Forschungszentrum (STZ lbronn),
 * Copyright (c) 20062012 by Michael Hoffer
 * 
 * This file is part of Visual Reflection Library (VRL).
 *
 * VRL is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License version 3
 * as published by the Free Software Foundation.
 * 
 * see: http://opensource.org/licenses/LGPL-3.0
 *      file://path/to/VRL/src/eu/mihosoft/vrl/resources/license/lgplv3.txt
 *
 * VRL 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 Lesser General Public License for more details.
 *
 * This version of VRL includes copyright notice and attribution requirements.
 * According to the LGPL this information must be displayed even if you modify
 * the source code of VRL. Neither the VRL Canvas attribution icon nor any
 * copyright statement/attribution may be removed.
 *
 * Attribution Requirements:
 *
 * If you create derived work you must do three things regarding copyright
 * notice and author attribution.
 *
 * First, the following text must be displayed on the Canvas:
 * "based on VRL source code". In this case the VRL canvas icon must be removed.
 * 
 * Second, the copyright notice must remain. It must be reproduced in any
 * program that uses VRL.
 *
 * Third, add an additional notice, stating that you modified VRL. In addition
 * you must cite the publications listed below. A suitable notice might read
 * "VRL source code modified by YourName 2012".
 * 
 * Note, that these requirements are in full accordance with the LGPL v3
 * (see 7. Additional Terms, b).
 *
 * Publications:
 *
 * M. Hoffer, C.Poliwoda, G.Wittum. Visual Reflection Library -
 * A Framework for Declarative GUI Programming on the Java Platform.
 * Computing and Visualization in Science, 2011, in press.
 */

package edu.gcsc.vrl.jfreechart;

import eu.mihosoft.vrl.annotation.ComponentInfo;
import eu.mihosoft.vrl.annotation.ObjectInfo;
import eu.mihosoft.vrl.annotation.ParamInfo;
import eu.mihosoft.vrl.math.Trajectory;
import java.io.Serializable;
import org.jfree.chart.ChartFactory;
import org.jfree.chart.JFreeChart;
import org.jfree.chart.axis.LogarithmicAxis;
import org.jfree.chart.plot.PlotOrientation;

@ComponentInfo(name = "TrajectoryPlotter", category = "JFreeChart")
@ObjectInfo(name = "TrajectoryPlotter")
public class TrajectoryPlotter implements Serializable {

    private static final long serialVersionUID = 1;

    public JFreeChart lineChart(
            @ParamInfo(name = "Trajectories", style = "array", options = "serialization=false") Trajectory[] trajectories) {

        TitledXYSeriesCollection dataset = new TitledXYSeriesCollection();

        // set title if available
        if (trajectories.length > 0 && trajectories[0] != null) {
            dataset.setTitle(trajectories[0].getTitle());
            dataset.setxAxisLabel(trajectories[0].getxAxisLabel());
            dataset.setyAxisLabel(trajectories[0].getyAxisLabel());
            dataset.setxAxisLogarithmic(trajectories[0].isxAxisLogarithmic());
            dataset.setyAxisLogarithmic(trajectories[0].isyAxisLogarithmic());
        }

        for (Trajectory t : trajectories) {
            dataset.addSeries(new TitledXYSeries(t));
        }

        final JFreeChart chart = ChartFactory.createXYLineChart(dataset.getTitle(), dataset.getxAxisLabel(),
                dataset.getyAxisLabel(), dataset, PlotOrientation.VERTICAL, true, true, false);

        if (dataset.isxAxisLogarithmic()) {
            chart.getXYPlot().setDomainAxis(new LogarithmicAxis(dataset.getxAxisLabel()));
        }

        if (dataset.isyAxisLogarithmic()) {
            chart.getXYPlot().setRangeAxis(new LogarithmicAxis(dataset.getyAxisLabel()));
        }

        return chart;
    }

    public JFreeChart lineCharts(@ParamInfo(name = "Trajectories", style = "default") Trajectory[] trajectories) {

        return lineChart(trajectories);
    }
}