org.schreibubi.JCombinations.logic.visitors.ChartNodesVisitor.java Source code

Introduction

Here is the source code for org.schreibubi.JCombinations.logic.visitors.ChartNodesVisitor.java

Source

/**
 * Copyright (C) 2009 Jrg Werner schreibubi@gmail.com
 *
 * 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/>.
 */
package org.schreibubi.JCombinations.logic.visitors;

import java.util.ArrayList;

import javax.swing.tree.TreePath;

import org.jfree.chart.JFreeChart;
import org.jfree.chart.axis.NumberAxis;
import org.jfree.chart.labels.StandardXYToolTipGenerator;
import org.jfree.chart.plot.Marker;
import org.jfree.chart.plot.PlotOrientation;
import org.jfree.chart.plot.XYPlot;
import org.jfree.chart.renderer.xy.XYLineAndShapeRenderer;
import org.jfree.chart.title.TextTitle;
import org.jfree.data.xy.XYSeries;
import org.jfree.data.xy.XYSeriesCollection;
import org.jfree.ui.Layer;
import org.schreibubi.JCombinations.FileFormat.Alternative;
import org.schreibubi.JCombinations.FileFormat.Asdap;
import org.schreibubi.JCombinations.FileFormat.OurTreeNode;
import org.schreibubi.JCombinations.FileFormat.Shmoo;
import org.schreibubi.JCombinations.FileFormat.TreeVisitor;
import org.schreibubi.JCombinations.FileFormat.Ydata;
import org.schreibubi.JCombinations.jfreechart.XYLineAndShapeRendererExtended;
import org.schreibubi.JCombinations.logic.ExtendedJFreeChart;

/**
 * Creates the charts from all selected nodes
 * 
 * @author Jrg Werner
 * 
 */
public class ChartNodesVisitor implements TreeVisitor {

    ArrayList<TreePath> treePaths = null;

    ArrayList<ExtendedJFreeChart> charts = null;

    ArrayList<ArrayList<Double>> dutData = null;

    ArrayList<String> dutName = null;

    ArrayList<String> seriesMask = null;

    /**
     * Constructor
     * 
     * @param treePaths
     *            selected paths
     * @param seriesMask
     *            series Mask
     * @param charts
     *            generated charts
     */
    public ChartNodesVisitor(ArrayList<TreePath> treePaths, ArrayList<String> seriesMask,
            ArrayList<ExtendedJFreeChart> charts) {
        this.charts = charts;
        this.seriesMask = seriesMask;
        this.treePaths = treePaths;
    }

    /*
     * (non-Javadoc)
     * 
     * @see org.schreibubi.JCombinations.FileFormat.TreeVisitor#visit(org.schreibubi.JCombinations.FileFormat.Alternative)
     */
    public void visit(Alternative a) throws Exception {
        for (int i = 0; i < a.getChildCount(); i++)
            (a.getChildAt(i)).accept(this);
    }

    public void visit(Asdap r) throws Exception {
        for (int i = 0; i < r.getChildCount(); i++)
            (r.getChildAt(i)).accept(this);
    }

    /*
     * (non-Javadoc)
     * 
     * @see org.schreibubi.JCombinations.FileFormat.TreeVisitor#visit(org.schreibubi.JCombinations.FileFormat.Shmoo)
     */
    public void visit(Shmoo s) throws Exception {
        if (s.componentSelected(this.treePaths, OurTreeNode.MYSELF | OurTreeNode.PARENTS | OurTreeNode.CHILDS)) {
            NumberAxis xAxis = new NumberAxis(s.getTrim() + " [" + s.getXdataDefault().getUnit() + "]");
            xAxis.setAutoRangeIncludesZero(false);
            NumberAxis yAxis = new NumberAxis(
                    s.getMeasure() + " [" + ((Ydata) s.getYdata().get(0)).getUnit() + "]");
            yAxis.setAutoRangeIncludesZero(false);
            XYLineAndShapeRenderer renderer = new XYLineAndShapeRendererExtended(true, true);
            renderer.setBaseToolTipGenerator(new StandardXYToolTipGenerator());

            this.dutData = new ArrayList<ArrayList<Double>>();
            this.dutName = new ArrayList<String>();
            for (int i = 0; i < s.getChildCount(); i++)
                (s.getChildAt(i)).accept(this);

            XYSeriesCollection xyseries = new XYSeriesCollection();
            ArrayList<Double> x = s.getXdataDefault().getXPositions();
            for (int j = 0; j < this.dutData.size(); j++) {
                ArrayList<Double> y = this.dutData.get(j);
                XYSeries xy = new XYSeries(this.dutName.get(j));
                for (int i = 0; i < y.size(); i++)
                    xy.add(x.get(i), y.get(i));
                xyseries.addSeries(xy);
            }

            XYPlot plot = new XYPlot(xyseries, xAxis, yAxis, renderer);
            plot.setOrientation(PlotOrientation.VERTICAL);
            plot.setDomainCrosshairVisible(true);
            plot.setRangeCrosshairVisible(true);
            Marker marker = s.getMarker();
            if (marker != null)
                plot.addRangeMarker(marker, Layer.BACKGROUND);
            ExtendedJFreeChart chart = new ExtendedJFreeChart(s.getDescription(), JFreeChart.DEFAULT_TITLE_FONT,
                    plot, false);
            if (marker == null)
                chart.addSubtitle(new TextTitle(s.getSubtitle()));
            else
                chart.addSubtitle(new TextTitle(
                        s.getSubtitle() + " " + s.getValueAt(3) + "/" + s.getValueAt(4) + "/" + s.getValueAt(5)));

            chart.setTreePath(s.getTreePath());
            this.charts.add(chart);
        }

    }

    /*
     * (non-Javadoc)
     * 
     * @see org.schreibubi.JCombinations.FileFormat.TreeVisitor#visit(org.schreibubi.JCombinations.FileFormat.Data)
     */
    public void visit(Ydata d) throws Exception {
        if (d.componentSelected(this.treePaths, OurTreeNode.MYSELF | OurTreeNode.PARENTS))
            if (!this.seriesMask.contains(d.getDescription())
                    | d.componentSelected(this.treePaths, OurTreeNode.MYSELF)) {
                this.dutData.add(d.getDoubleValues());
                this.dutName.add(d.getDescription());
            }
    }

}