content: /* -------------------
   * SelectionDemo2.java
   * -------------------
   * (C) Copyright 2009, by Object Refinery Limited.
   *
   */
  
  package org.jfree.chart.demo.selection;
  
  import java.awt.BasicStroke;
  import java.awt.BorderLayout;
  import java.awt.Color;
  import java.util.Random;
  
  import javax.swing.BorderFactory;
  import javax.swing.JPanel;
  import javax.swing.JScrollPane;
  import javax.swing.JSplitPane;
  import javax.swing.JTable;
  import javax.swing.border.EmptyBorder;
  import javax.swing.border.TitledBorder;
  import javax.swing.table.DefaultTableModel;
  import javax.swing.table.TableColumnModel;
  
  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.panel.selectionhandler.CircularRegionSelectionHandler;
  import org.jfree.chart.panel.selectionhandler.EntitySelectionManager;
  import org.jfree.chart.panel.selectionhandler.MouseClickSelectionHandler;
  import org.jfree.chart.panel.selectionhandler.RegionSelectionHandler;
  import org.jfree.chart.plot.XYPlot;
  import org.jfree.chart.renderer.item.IRSUtilities;
  import org.jfree.chart.renderer.xy.XYLineAndShapeRenderer;
  import org.jfree.chart.ui.ApplicationFrame;
  import org.jfree.chart.ui.NumberCellRenderer;
  import org.jfree.chart.ui.RefineryUtilities;
  import org.jfree.data.extension.DatasetIterator;
  import org.jfree.data.extension.DatasetSelectionExtension;
  import org.jfree.data.extension.impl.DatasetExtensionManager;
  import org.jfree.data.extension.impl.XYCursor;
  import org.jfree.data.extension.impl.XYDatasetSelectionExtension;
  import org.jfree.data.general.Dataset;
  import org.jfree.data.general.SelectionChangeEvent;
  import org.jfree.data.general.SelectionChangeListener;
  import org.jfree.data.xy.XYDataset;
  import org.jfree.data.xy.XYSeries;
  import org.jfree.data.xy.XYSeriesCollection;
  
  /**
   * A demo scatter plot.
   */
  public class SelectionDemo2 extends ApplicationFrame 
          implements SelectionChangeListener<XYCursor> {
  
      private JTable table;
  
      private DefaultTableModel model;
  
      private XYSeriesCollection dataset;
  
      /**
       * A demonstration application showing a scatter plot.
       *
       * @param title  the frame title.
       */
      public SelectionDemo2(String title) {
          super(title);
          ChartPanel chartPanel = (ChartPanel) createDemoPanel();
          chartPanel.setPreferredSize(new java.awt.Dimension(500, 270));
  
          JFreeChart chart = chartPanel.getChart();
          XYPlot plot = (XYPlot) chart.getPlot();
          this.dataset = (XYSeriesCollection) plot.getDataset();
          JSplitPane split = new JSplitPane(JSplitPane.HORIZONTAL_SPLIT);
          split.add(chartPanel);
  
          this.model = new DefaultTableModel(new String[] {"Series:", "Item:", "X:", "Y:"}, 0);
          this.table = new JTable(this.model);
          TableColumnModel tcm = this.table.getColumnModel();
          tcm.getColumn(2).setCellRenderer(new NumberCellRenderer());
          tcm.getColumn(3).setCellRenderer(new NumberCellRenderer());
          JPanel p = new JPanel(new BorderLayout());
          JScrollPane scroller = new JScrollPane(this.table);
          p.add(scroller);
          p.setBorder(BorderFactory.createCompoundBorder(
                  new TitledBorder("Selected Items: "),
                  new EmptyBorder(4, 4, 4, 4)));
          split.add(p);
          setContentPane(split);
      }
  
       /**
        * The selection changed, so we change the table model
        * 
        * @param event
        */
       public void selectionChanged(SelectionChangeEvent<XYCursor> event) {
           while (this.model.getRowCount() > 0) {
               this.model.removeRow(0);
           }
  
           XYDatasetSelectionExtension ext = (XYDatasetSelectionExtension) 
                   event.getSelectionExtension(); 
           DatasetIterator<XYCursor> iter = ext.getSelectionIterator(true);
           while (iter.hasNext()) {
               XYCursor dc = iter.next();
               Comparable seriesKey = this.dataset.getSeriesKey(dc.series);
               Number x = this.dataset.getX(dc.series, dc.item);
               Number y = this.dataset.getX(dc.series, dc.item);  
               this.model.addRow(new Object[] { seriesKey, new Integer(dc.item), 
                   x,  y});
           }
      }
  
      private static JFreeChart createChart(XYDataset dataset, 
              DatasetSelectionExtension<XYCursor> ext) {
          JFreeChart chart = ChartFactory.createScatterPlot("SelectionDemo2",
                  "X", "Y", dataset);
  
          XYPlot plot = (XYPlot) chart.getPlot();
          plot.setNoDataMessage("NO DATA");
  
          plot.setDomainPannable(true);
          plot.setRangePannable(true);
          plot.setDomainZeroBaselineVisible(true);
          plot.setRangeZeroBaselineVisible(true);
  
          plot.setDomainGridlineStroke(new BasicStroke(0.0f));
          plot.setRangeGridlineStroke(new BasicStroke(0.0f));
  
          plot.setDomainMinorGridlinesVisible(true);
          plot.setRangeMinorGridlinesVisible(true);
  
          XYLineAndShapeRenderer r = (XYLineAndShapeRenderer) plot.getRenderer();
          r.setSeriesFillPaint(0, r.lookupSeriesPaint(0));
          r.setSeriesFillPaint(1, r.lookupSeriesPaint(1));
          r.setSeriesFillPaint(2, r.lookupSeriesPaint(2));
          r.setUseFillPaint(true);
  
          NumberAxis domainAxis = (NumberAxis) plot.getDomainAxis();
          domainAxis.setAutoRangeIncludesZero(false);
  
          domainAxis.setTickMarkInsideLength(2.0f);
          domainAxis.setTickMarkOutsideLength(2.0f);
  
          domainAxis.setMinorTickCount(2);
          domainAxis.setMinorTickMarksVisible(true);
  
          NumberAxis rangeAxis = (NumberAxis) plot.getRangeAxis();
          rangeAxis.setTickMarkInsideLength(2.0f);
          rangeAxis.setTickMarkOutsideLength(2.0f);
          rangeAxis.setMinorTickCount(2);
          rangeAxis.setMinorTickMarksVisible(true);
          
          //add selection specific rendering
          IRSUtilities.setSelectedItemFillPaint(r, ext, Color.white);
            
          //register plot as selection change listener
          ext.addChangeListener(plot);
            
          return chart;
      }
  
      public static XYDataset createDataset() {
          Random rgen = new Random();
          XYSeriesCollection dataset = new XYSeriesCollection();
          for (int s = 0; s < 3; s++) {
              XYSeries series = new XYSeries("S" + s);
              for (int i = 0; i < 100; i++) {
                  double x = rgen.nextGaussian() * 200;
                  double y = rgen.nextGaussian() * 200;
                  series.add(x, y);
              }
              dataset.addSeries(series);
          }
          return dataset;
      }
  
      /**
       * Creates a panel for the demo (used by SuperDemo.java).
       *
       * @return A panel.
       */
      public JPanel createDemoPanel() {
          XYDataset dataset = createDataset();
          //extend dataset and add selection change listener for the demo
          DatasetSelectionExtension<XYCursor> datasetExtension 
                  = new XYDatasetSelectionExtension(dataset);     
          datasetExtension.addChangeListener(this);
            
          //standard setup
          JFreeChart chart = createChart(dataset, datasetExtension);
          ChartPanel panel = new ChartPanel(chart);
          panel.setMouseWheelEnabled(true);
  
          // add a selection handler
          RegionSelectionHandler selectionHandler 
                  = new CircularRegionSelectionHandler();
          panel.addMouseHandler(selectionHandler);
          panel.addMouseHandler(new MouseClickSelectionHandler());
          panel.removeMouseHandler(panel.getZoomHandler());
            
          // add a selection manager
          DatasetExtensionManager dExManager = new DatasetExtensionManager();
          dExManager.registerDatasetExtension(datasetExtension);
          panel.setSelectionManager(new EntitySelectionManager(panel, 
                  new Dataset[] { dataset }, dExManager));
            
          return panel;
      }
  
      /**
       * Starting point for the demonstration application.
       *
       * @param args  ignored.
       */
      public static void main(String[] args) {
          SelectionDemo2 demo = new SelectionDemo2(
                  "JFreeChart: SelectionDemo2.java");
          demo.pack();
          RefineryUtilities.centerFrameOnScreen(demo);
          demo.setVisible(true);
      }
  
  }
  
  