crosshairdemo2.java

在软件开发中用来绘制各种图表的源码

/* -------------------
 * CrosshairDemo2.java
 * -------------------
 * (C) Copyright 2003-2005, by Object Refinery Limited.
 *
 */

package demo;

import java.awt.BorderLayout;
import java.awt.Color;
import java.awt.Dimension;
import java.text.SimpleDateFormat;
import java.util.Date;

import javax.swing.BorderFactory;
import javax.swing.JPanel;
import javax.swing.JScrollPane;
import javax.swing.JTable;
import javax.swing.border.Border;
import javax.swing.table.AbstractTableModel;
import javax.swing.table.TableCellRenderer;
import javax.swing.table.TableModel;

import org.jfree.chart.ChartFactory;
import org.jfree.chart.ChartPanel;
import org.jfree.chart.JFreeChart;
import org.jfree.chart.axis.NumberAxis;
import org.jfree.chart.event.ChartChangeEvent;
import org.jfree.chart.event.ChartChangeListener;
import org.jfree.chart.event.ChartProgressEvent;
import org.jfree.chart.event.ChartProgressListener;
import org.jfree.chart.plot.PlotOrientation;
import org.jfree.chart.plot.XYPlot;
import org.jfree.chart.renderer.xy.XYItemRenderer;
import org.jfree.chart.renderer.xy.XYLineAndShapeRenderer;
import org.jfree.data.time.Minute;
import org.jfree.data.time.RegularTimePeriod;
import org.jfree.data.time.TimeSeries;
import org.jfree.data.time.TimeSeriesCollection;
import org.jfree.data.time.TimeSeriesDataItem;
import org.jfree.data.xy.XYDataset;
import org.jfree.ui.ApplicationFrame;
import org.jfree.ui.DateCellRenderer;
import org.jfree.ui.NumberCellRenderer;
import org.jfree.ui.RectangleInsets;
import org.jfree.ui.RefineryUtilities;

/**
 * An example of....
 */
public class CrosshairDemo2 extends ApplicationFrame {

    private static class DemoPanel extends JPanel implements ChartChangeListener, ChartProgressListener {

        private static final int SERIES_COUNT = 4;
        
        private TimeSeriesCollection[] datasets;
        
        private TimeSeries[] series;
        
        private ChartPanel chartPanel;
        
        private DemoTableModel model;

		/**
		 * Creates a new demo panel.
		 */
		public DemoPanel() {
			super(new BorderLayout());
            this.datasets = new TimeSeriesCollection[SERIES_COUNT];
            this.series = new TimeSeries[SERIES_COUNT];
            
            JPanel content = new JPanel(new BorderLayout());
            
            JFreeChart chart = createChart();
            this.chartPanel = new ChartPanel(chart);
            this.chartPanel.setPreferredSize(new java.awt.Dimension(600, 270));
            this.chartPanel.setDomainZoomable(true);
            this.chartPanel.setRangeZoomable(true);
            Border border = BorderFactory.createCompoundBorder(
                BorderFactory.createEmptyBorder(4, 4, 4, 4),
                BorderFactory.createEtchedBorder()
            );
            this.chartPanel.setBorder(border);
            content.add(this.chartPanel);
            
            JPanel dashboard = new JPanel(new BorderLayout());
            dashboard.setPreferredSize(new Dimension(400, 120));
            dashboard.setBorder(BorderFactory.createEmptyBorder(0, 4, 4, 4));
            
            this.model = new DemoTableModel(SERIES_COUNT);
            for (int row = 0; row < SERIES_COUNT; row++) {
                this.model.setValueAt(
                    chart.getXYPlot().getDataset(row).getSeriesKey(0), row, 0
                );                
                this.model.setValueAt(new Double("0.00"), row, 1);
                this.model.setValueAt(new Double("0.00"), row, 2);
                this.model.setValueAt(new Double("0.00"), row, 3);
                this.model.setValueAt(new Double("0.00"), row, 4);
                this.model.setValueAt(new Double("0.00"), row, 5);
                this.model.setValueAt(new Double("0.00"), row, 6);
            
            }
            JTable table = new JTable(this.model);
            TableCellRenderer renderer1 = new DateCellRenderer(
                new SimpleDateFormat("HH:mm:ss")
            );
            TableCellRenderer renderer2 = new NumberCellRenderer();
            table.getColumnModel().getColumn(1).setCellRenderer(renderer1);
            table.getColumnModel().getColumn(2).setCellRenderer(renderer2);
            table.getColumnModel().getColumn(3).setCellRenderer(renderer1);
            table.getColumnModel().getColumn(4).setCellRenderer(renderer2);
            table.getColumnModel().getColumn(5).setCellRenderer(renderer1);
            table.getColumnModel().getColumn(6).setCellRenderer(renderer2);
            dashboard.add(new JScrollPane(table));
            content.add(dashboard, BorderLayout.SOUTH);
            add(content);

		}

        
        /**
         * Creates a sample dataset.
         * 
         * @param index  the dataset index.
         * @param name  the dataset name.
         * @param base  the starting value.
         * @param start  the starting period.
         * @param count  the number of values to generate.
         *
         * @return The dataset.
         */
        private XYDataset createDataset(int index, String name,
                                        double base, RegularTimePeriod start, 
                                        int count) {

            this.series[index] = new TimeSeries(name, start.getClass());
            RegularTimePeriod period = start;
            double value = base;
            for (int i = 0; i < count; i++) {
                this.series[index].add(period, value);    
                period = period.next();
                value = value * (1 + (Math.random() - 0.495) / 10.0);
            }

            this.datasets[index] = new TimeSeriesCollection();
            this.datasets[index].addSeries(this.series[index]);

            return this.datasets[index];

        }

        /**
         * Receives notification of a {@link ChartChangeEvent}.
         * 
         * @param event  the event.
         */
        public void chartChanged(ChartChangeEvent event) {
            if (this.chartPanel != null) {
                JFreeChart chart = this.chartPanel.getChart();
                if (chart != null) {
                    XYPlot plot = chart.getXYPlot();
                    XYDataset dataset = plot.getDataset();
                    Comparable seriesKey = dataset.getSeriesKey(0);
                    double xx = plot.getDomainCrosshairValue();
                    this.model.setValueAt(seriesKey, 0, 0);
                    long millis = (long) xx;
                    for (int row = 0; row < SERIES_COUNT; row++) {
                        this.model.setValueAt(new Long(millis), row, 1);
                        int[] bounds 
                            = this.datasets[row].getSurroundingItems(0, millis);
                        long prevX = 0;
                        long nextX = 0;
                        double prevY = 0.0;
                        double nextY = 0.0;
                        if (bounds[0] >= 0) {
                            TimeSeriesDataItem prevItem 
                                = this.series[row].getDataItem(bounds[0]);
                            prevX = prevItem.getPeriod().getMiddleMillisecond();      
                            Number y = prevItem.getValue();
                            if (y != null) {
                                prevY = y.doubleValue();  
                                this.model.setValueAt(new Double(prevY), row, 4);
                            }  
                            else {     
                                this.model.setValueAt(null, row, 4);
                            }
                            this.model.setValueAt(new Long(prevX), row, 3);
                        }
                        else {
                            this.model.setValueAt(new Double(0.00), row, 4);
                            this.model.setValueAt(
                                new Double(this.chartPanel.getChart().getXYPlot().getDomainAxis().getRange().getLowerBound()), row, 3
                            );                        
                        }
                        if (bounds[1] >= 0) {
                            TimeSeriesDataItem nextItem = this.series[row].getDataItem(bounds[1]);
                            nextX = nextItem.getPeriod().getMiddleMillisecond();
                            Number y = nextItem.getValue();
                            if (y != null) {
                                nextY = y.doubleValue();
                                this.model.setValueAt(new Double(nextY), row, 6);                            
                            }
                            else {     
                                this.model.setValueAt(null, row, 6);
                            }
                            this.model.setValueAt(new Long(nextX), row, 5);
                        }
                        else {
                            this.model.setValueAt(new Double(0.00), row, 6);
                            this.model.setValueAt(
                                new Double(this.chartPanel.getChart().getXYPlot().getDomainAxis().getRange().getUpperBound()), row, 5
                            );                        
                        }
                        double interpolatedY = 0.0;
                        if ((nextX - prevX) > 0) {
                            interpolatedY = prevY 
                                + (((double)millis - (double)prevX) 
                                / ((double)nextX - (double)prevX)) * (nextY - prevY); 
                        }
                        else {
                            interpolatedY = prevY;
                        }
                        this.model.setValueAt(new Double(interpolatedY), row, 2);
                    }
                }
            }
        }