List of usage examples for org.jfree.chart.plot XYPlot getFixedLegendItems
public LegendItemCollection getFixedLegendItems()
From source file:edu.ucla.stat.SOCR.chart.demo.SOCR_EM_MixtureModelChartDemo.java
/** * Creates a chart./*from ww w . j av a 2s . c o m*/ * * @param dataset the data for the chart. * * @return a chart. */ // protected JFreeChart createChart(XYDataset dataset, double[][] polygons, int num_poly) { protected JFreeChart createChart(XYDataset[] dataset) { //System.out.println("createChart get called"); boolean legend = false; /*if (num_series>1&&segment_flag==false) legend = true;*/ // System.out.println("createchart: inde="+independentVarLength); // create the chart... JFreeChart chart = ChartFactory.createXYLineChart(chartTitle, // chart title "X", // x axis label "Y", // y axis label dataset[0], // data PlotOrientation.VERTICAL, legend, // include legend true, // tooltips false // urls ); //printDataset(dataset, num_series); // NOW DO SOME OPTIONAL CUSTOMISATION OF THE CHART... chart.setBackgroundPaint(Color.white); // get a reference to the plot for further customisation... XYPlot subplot1 = (XYPlot) chart.getPlot(); subplot1.setBackgroundPaint(Color.lightGray); subplot1.setAxisOffset(new RectangleInsets(5.0, 5.0, 5.0, 5.0)); subplot1.setDomainGridlinePaint(Color.white); subplot1.setRangeGridlinePaint(Color.white); XYLineAndShapeRenderer renderer1 = (XYLineAndShapeRenderer) subplot1.getRenderer(); renderer1.setBaseLinesVisible(false); renderer1.setBaseShapesVisible(true); renderer1.setBaseShapesFilled(true); int[] num_group = new int[num_series]; Paint color_mainGroup = mEMexperiment[0].getColorOfMainGroup(); // use same color for all series if EM together // System.out.println("dataset0. seriescount="+dataset[0].getSeriesCount()+" itemcount="+dataset[0].getItemCount(0)); if (num_series == 1) for (int s = 0; s < dataset[0].getSeriesCount(); s++) { renderer1.setSeriesLinesVisible(s, false); renderer1.setSeriesShapesVisible(s, true); renderer1.setSeriesShapesFilled(s, true); renderer1.setSeriesShape(s, series_shapes[0]); // for the shape of the dots renderer1.setSeriesPaint(s, color_mainGroup); } // System.out.println("num_series="+num_series); // System.out.println("num_group="+num_group); // System.out.println("segment_flag="+segment_flag); if (legend) { LegendItemCollection legends_old = subplot1.getFixedLegendItems(); LegendItemCollection legends_new = new LegendItemCollection(); for (int s = 0; s < num_series; s++) { legends_new.add(legends_old.get(s)); } subplot1.setFixedLegendItems(legends_new); } //segment for (int s = 0; s < num_series; s++) { num_group[s] = mEMexperiment[s].getNumOfGroup(); } // System.out.println("num_group[0]="+num_group[0]); if (num_group[0] > 0 && segment_flag && (mixSelected == CG_MIX)) { Paint[] color_groups = mEMexperiment[0].getColorOfGroups(); for (int i = 0; i < num_group[0] - 1; i++) { //System.out.println("color "+ i +"=" + color_groups[i].toString()); renderer1.setSeriesPaint(i, color_groups[i]); } //System.out.println("color main ="+ color_mainGroup.toString()); renderer1.setSeriesPaint(nkSelected, color_mainGroup); } else renderer1.setBasePaint(color_mainGroup); // renderer1.setShape(series_shapes[0]); // subplot1.setDataset(0, dataset[0]); //System.out.println("setting renderer for " +0); // subplot1.setRenderer(0, renderer1); // System.out.println("SOCR_EM creatChart: pt0="+dataset.getXValue(0, 0)+ ","+dataset.getYValue(0, 0)); // change the auto tick unit selection to integer units only... NumberAxis rangeAxis = (NumberAxis) subplot1.getRangeAxis(); rangeAxis.setStandardTickUnits(NumberAxis.createIntegerTickUnits()); rangeAxis.setAutoRangeIncludesZero(true); // rangeAxis.setUpperMargin(0); // rangeAxis.setLowerMargin(0); NumberAxis domainAxis = (NumberAxis) subplot1.getDomainAxis(); domainAxis.setStandardTickUnits(NumberAxis.createIntegerTickUnits()); domainAxis.setAutoRangeIncludesZero(true); // domainAxis.setUpperMargin(0); // domainAxis.setLowerMargin(0); XYLineAndShapeRenderer[] renderer1a = new XYLineAndShapeRenderer[num_series]; if (num_series > 1) { for (int s = 0; s < num_series; s++) { // System.out.println("adding pt for series:"+s +": seriesCount "+ dataset[s].getSeriesCount()+"itemcount="+dataset[s].getItemCount(0)); renderer1a[s] = new XYLineAndShapeRenderer(); if (num_group[0] > 0 && segment_flag && (mixSelected == CG_MIX)) { Paint[] color_groups = mEMexperiment[0].getColorOfGroups(); for (int i = 0; i < num_group[0] - 1; i++) { //System.out.println("color "+ i +"=" + color_groups[i].toString()); renderer1a[s].setSeriesPaint(i, color_groups[i]); } //System.out.println("color main ="+ color_mainGroup.toString()); renderer1a[s].setSeriesPaint(nkSelected, color_mainGroup); } else renderer1a[s].setBasePaint(color_mainGroup); renderer1a[s].setBaseLinesVisible(false); renderer1a[s].setBaseShapesVisible(true); renderer1a[s].setBaseShapesFilled(true); renderer1a[s].setBaseShape(series_shapes[s]); // different shape of dots for each pair of input subplot1.setDataset(s, dataset[s]); // System.out.println("setting renderer for " +s); subplot1.setRenderer(s, renderer1a[s]); } } if (kernels != null) { /* System.out.println("draw kernels"); System.out.println("num_series="+num_series); System.out.println("nkSelected="+nkSelected);*/ XYLineAndShapeRenderer[][] renderer2 = new XYLineAndShapeRenderer[num_series][nkSelected]; for (int s = 0; s < num_series; s++) { Paint[] color_kernels = mEMexperiment[s].getColorOfKernels(); // System.out.println("SOCR_EM creatChart adding kernels:" +num_kernels+ "ws[0]="+ws[0]); for (int i = 0; i < nkSelected; i++) { // System.out.println("drawing kernel for series: "+s +" kernel:" +i); // System.out.println("SOCR_EM creatChart: kernel "+i+" pt0="+kernels[i].getXValue(0, 0)+","+kernels[i].getYValue(0, 0)); renderer2[s][i] = new XYLineAndShapeRenderer(); renderer2[s][i].setBaseStroke(series_strokes[s]); renderer2[s][i].setBaseLinesVisible(true); renderer2[s][i].setBaseShapesVisible(false); // renderer2[s][i].setShapesFilled(false); renderer2[s][i].setBasePaint(color_kernels[i]); subplot1.setDataset(num_series - 1 + s * num_series + i + 1, kernels[s][i]); // System.out.println("setting renderer for " +(num_series-1+s*num_series+i+1)); subplot1.setRenderer(num_series - 1 + s * num_series + i + 1, renderer2[s][i]); //subplot1.setRangeAxis(i+1, rangeAxis); } //i } //s } segment_flag = false; return chart; }