Example usage for org.jfree.data.statistics DefaultMultiValueCategoryDataset add

List of usage examples for org.jfree.data.statistics DefaultMultiValueCategoryDataset add

Introduction

In this page you can find the example usage for org.jfree.data.statistics DefaultMultiValueCategoryDataset add.

Prototype

public void add(List values, Comparable rowKey, Comparable columnKey) 

Source Link

Document

Adds a list of values to the dataset (null and Double.NaN items are automatically removed) and sends a DatasetChangeEvent to all registered listeners.

Usage

From source file:org.jfree.chart.demo.ScatterRendererDemo1.java

private static MultiValueCategoryDataset createDataset() {
    DefaultMultiValueCategoryDataset defaultmultivaluecategorydataset = new DefaultMultiValueCategoryDataset();
    defaultmultivaluecategorydataset.add(listOfValues(new double[] { 1.0D, 2D, 3D }), "Series 1", "C1");
    defaultmultivaluecategorydataset.add(
            listOfValues(new double[] { 1.2D, 2.2000000000000002D, 3.2000000000000002D }), "Series 1", "C2");
    defaultmultivaluecategorydataset.add(
            listOfValues(new double[] { 1.3999999999999999D, 2.3999999999999999D, 3.3999999999999999D }),
            "Series 1", "C3");
    defaultmultivaluecategorydataset.add(listOfValues(new double[] { 1.0D, 3D }), "Series 2", "C1");
    defaultmultivaluecategorydataset.add(listOfValues(new double[] { 1.2D, 3.2000000000000002D }), "Series 2",
            "C2");
    defaultmultivaluecategorydataset/* www . j a  v  a2 s.  c  om*/
            .add(listOfValues(new double[] { 1.3999999999999999D, 3.6000000000000001D }), "Series 2", "C3");
    return defaultmultivaluecategorydataset;
}

From source file:org.jfree.expdemo.SelectionDemo7ScatterRenderer.java

/**
 * Creates a sample dataset./*  w ww  .  j ava  2s  .  c  om*/
 * 
 * @return A dataset.
 */
private static MultiValueCategoryDataset createDataset() {
    DefaultMultiValueCategoryDataset dataset = new DefaultMultiValueCategoryDataset();
    dataset.add(listOfValues(new double[] { 1.0, 2.0, 3.0 }), "Series 1", "C1");
    dataset.add(listOfValues(new double[] { 1.2, 2.2, 3.2 }), "Series 1", "C2");
    dataset.add(listOfValues(new double[] { 1.4, 2.4, 3.4 }), "Series 1", "C3");
    dataset.add(listOfValues(new double[] { 1.0, 2.1, 3.2 }), "Series 1", "C1");
    dataset.add(listOfValues(new double[] { 1.2, 2.15, 3.5 }), "Series 1", "C2");
    dataset.add(listOfValues(new double[] { 1.4, 2.5, 3.2 }), "Series 1", "C3");

    dataset.add(listOfValues(new double[] { 1.4, 3.0, 3.2 }), "Series 1", "C3");
    dataset.add(listOfValues(new double[] { 1.4, 3.0 }), "Series 2", "C1");

    dataset.add(listOfValues(new double[] { 1.0, 3.0 }), "Series 2", "C1");
    dataset.add(listOfValues(new double[] { 1.2, 3.2 }), "Series 2", "C2");
    dataset.add(listOfValues(new double[] { 1.4, 3.6 }), "Series 2", "C3");
    dataset.add(listOfValues(new double[] { 1.2, 3.1 }), "Series 2", "C1");
    dataset.add(listOfValues(new double[] { 1.4, 3.4 }), "Series 2", "C2");
    dataset.add(listOfValues(new double[] { 1.5, 3.6 }), "Series 2", "C3");

    return dataset;
}

From source file:org.jfree.chart.demo.selection.SelectionDemo7ScatterRenderer.java

/**
 * Creates a sample dataset./*from   w w  w  . j ava2s .  c om*/
 * 
 * @return A dataset.
 */
private static MultiValueCategoryDataset createDataset() {
    DefaultMultiValueCategoryDataset dataset = new DefaultMultiValueCategoryDataset();
    dataset.add(listOfValues(new double[] { 1.0, 2.0, 3.0 }), "Series 1", "C1");
    dataset.add(listOfValues(new double[] { 1.2, 2.2, 3.2 }), "Series 1", "C2");
    dataset.add(listOfValues(new double[] { 1.4, 2.4, 3.4 }), "Series 1", "C3");
    dataset.add(listOfValues(new double[] { 1.0, 2.1, 3.2 }), "Series 1", "C1");
    dataset.add(listOfValues(new double[] { 1.2, 2.15, 3.5 }), "Series 1", "C2");
    dataset.add(listOfValues(new double[] { 1.4, 2.5, 3.2 }), "Series 1", "C3");
    dataset.add(listOfValues(new double[] { 1.4, 3.0, 3.2 }), "Series 1", "C3");
    dataset.add(listOfValues(new double[] { 1.4, 3.0 }), "Series 2", "C1");

    dataset.add(listOfValues(new double[] { 1.0, 3.0 }), "Series 2", "C1");
    dataset.add(listOfValues(new double[] { 1.2, 3.2 }), "Series 2", "C2");
    dataset.add(listOfValues(new double[] { 1.4, 3.6 }), "Series 2", "C3");
    dataset.add(listOfValues(new double[] { 1.2, 3.1 }), "Series 2", "C1");
    dataset.add(listOfValues(new double[] { 1.4, 3.4 }), "Series 2", "C2");
    dataset.add(listOfValues(new double[] { 1.5, 3.6 }), "Series 2", "C3");

    return dataset;
}

From source file:org.jfree.data.statistics.DefaultMultiValueCategoryDatasetTest.java

/**
 * Some checks for the add() method.//from   w w w.  j  ava  2s.  c  om
 */
@Test
public void testAddValue() {
    DefaultMultiValueCategoryDataset d1 = new DefaultMultiValueCategoryDataset();

    boolean pass = false;
    try {
        d1.add(null, "R1", "C1");
    } catch (IllegalArgumentException e) {
        pass = true;
    }
    assertTrue(pass);

    List values = new ArrayList();
    d1.add(values, "R2", "C1");
    assertEquals(values, d1.getValues("R2", "C1"));

    pass = false;
    try {
        d1.add(values, null, "C2");
    } catch (IllegalArgumentException e) {
        pass = true;
    }
    assertTrue(pass);
}

From source file:org.jfree.data.statistics.DefaultMultiValueCategoryDatasetTest.java

/**
 * Some tests for the getRowCount() method.
 *///from w ww. j a v  a 2 s .  c o  m
@Test
public void testGetRowCount() {
    DefaultMultiValueCategoryDataset d = new DefaultMultiValueCategoryDataset();
    assertTrue(d.getRowCount() == 0);
    List values = new ArrayList();
    d.add(values, "R1", "C1");
    assertTrue(d.getRowCount() == 1);

    d.add(values, "R2", "C1");
    assertTrue(d.getRowCount() == 2);

    d.add(values, "R2", "C1");
    assertTrue(d.getRowCount() == 2);
}

From source file:org.jfree.data.statistics.DefaultMultiValueCategoryDatasetTest.java

/**
 * Some tests for the getColumnCount() method.
 */// w  w w  .j  a v  a 2s  .  co m
@Test
public void testGetColumnCount() {
    DefaultMultiValueCategoryDataset d = new DefaultMultiValueCategoryDataset();
    assertTrue(d.getColumnCount() == 0);

    List values = new ArrayList();
    d.add(values, "R1", "C1");
    assertTrue(d.getColumnCount() == 1);

    d.add(values, "R1", "C2");
    assertTrue(d.getColumnCount() == 2);

    d.add(values, "R1", "C2");
    assertTrue(d.getColumnCount() == 2);

}

From source file:org.jfree.data.statistics.DefaultMultiValueCategoryDatasetTest.java

/**
 * Some checks for the getValue() method.
 *//*ww  w.  j  a  v a 2 s.co m*/
@Test
public void testGetValue() {
    DefaultMultiValueCategoryDataset d = new DefaultMultiValueCategoryDataset();
    List values = new ArrayList();
    values.add(new Integer(1));
    values.add(new Integer(2));
    d.add(values, "R1", "C1");
    assertEquals(new Double(1.5), d.getValue("R1", "C1"));
    boolean pass = false;
    try {
        d.getValue("XX", "C1");
    } catch (UnknownKeyException e) {
        pass = true;
    }
    assertTrue(pass);

    pass = false;
    try {
        d.getValue("R1", "XX");
    } catch (UnknownKeyException e) {
        pass = true;
    }
    assertTrue(pass);
}

From source file:org.jfree.data.statistics.DefaultMultiValueCategoryDatasetTest.java

/**
 * Confirm that the equals method can distinguish all the required fields.
 *///from   www . j a  va 2s  .  c  om
@Test
public void testEquals() {
    DefaultMultiValueCategoryDataset d1 = new DefaultMultiValueCategoryDataset();
    DefaultMultiValueCategoryDataset d2 = new DefaultMultiValueCategoryDataset();
    assertTrue(d1.equals(d2));
    assertTrue(d2.equals(d1));

    List values = new ArrayList();
    d1.add(values, "R1", "C1");
    assertFalse(d1.equals(d2));
    d2.add(values, "R1", "C1");
    assertTrue(d1.equals(d2));

    values.add(new Integer(99));
    d1.add(values, "R1", "C1");
    assertFalse(d1.equals(d2));
    d2.add(values, "R1", "C1");
    assertTrue(d1.equals(d2));

    values.add(new Integer(99));
    d1.add(values, "R1", "C2");
    assertFalse(d1.equals(d2));
    d2.add(values, "R1", "C2");
    assertTrue(d1.equals(d2));
}

From source file:org.jfree.data.statistics.DefaultMultiValueCategoryDatasetTest.java

/**
 * Confirm that cloning works.//from   www.j  ava2  s .c  o m
 */
@Test
public void testCloning() throws CloneNotSupportedException {
    DefaultMultiValueCategoryDataset d1 = new DefaultMultiValueCategoryDataset();
    DefaultMultiValueCategoryDataset d2 = (DefaultMultiValueCategoryDataset) d1.clone();
    assertTrue(d1 != d2);
    assertTrue(d1.getClass() == d2.getClass());
    assertTrue(d1.equals(d2));

    // try a dataset with some content...
    List values = new ArrayList();
    values.add(new Integer(99));
    d1.add(values, "R1", "C1");
    d2 = (DefaultMultiValueCategoryDataset) d1.clone();
    assertTrue(d1 != d2);
    assertTrue(d1.getClass() == d2.getClass());
    assertTrue(d1.equals(d2));

    // check that the clone doesn't share the same underlying arrays.
    List values2 = new ArrayList();
    values2.add(new Integer(111));
    d1.add(values2, "R2", "C2");
    assertFalse(d1.equals(d2));
    d2.add(values2, "R2", "C2");
    assertTrue(d1.equals(d2));
}

From source file:ubic.gemma.web.controller.expression.experiment.ExpressionExperimentQCController.java

private boolean writeDetailedFactorAnalysis(ExpressionExperiment ee, OutputStream os) throws Exception {
    SVDValueObject svdo = svdService.getSvdFactorAnalysis(ee.getId());
    if (svdo == null)
        return false;

    if (svdo.getFactors().isEmpty() && svdo.getDates().isEmpty()) {
        return false;
    }//from w w  w  .  ja va  2s  . c o  m
    Map<Integer, Map<Long, Double>> factorCorrelations = svdo.getFactorCorrelations();
    // Map<Integer, Map<Long, Double>> factorPvalues = svdo.getFactorPvalues();
    Map<Integer, Double> dateCorrelations = svdo.getDateCorrelations();

    assert ee.getId().equals(svdo.getId());

    ee = expressionExperimentService.thawLite(ee); // need the experimental design
    int maxWidth = 30;
    Map<Long, String> efs = this.getFactorNames(ee, maxWidth);
    Map<Long, ExperimentalFactor> efIdMap = EntityUtils
            .getIdMap(ee.getExperimentalDesign().getExperimentalFactors());
    Collection<Long> continuousFactors = new HashSet<>();
    for (ExperimentalFactor ef : ee.getExperimentalDesign().getExperimentalFactors()) {
        boolean isContinous = ExperimentalDesignUtils.isContinuous(ef);
        if (isContinous) {
            continuousFactors.add(ef.getId());
        }
    }

    /*
     * Make plots of the dates vs. PCs, factors vs. PCs.
     */
    int MAX_COMP = 3;

    Map<Long, List<JFreeChart>> charts = new LinkedHashMap<>();
    ChartFactory.setChartTheme(StandardChartTheme.createLegacyTheme());
    /*
     * FACTORS
     */
    String componentShorthand = "PC";
    for (Integer component : factorCorrelations.keySet()) {

        if (component >= MAX_COMP)
            break;
        String xaxisLabel = componentShorthand + (component + 1);

        for (Long efId : factorCorrelations.get(component).keySet()) {

            /*
             * Should not happen.
             */
            if (!efs.containsKey(efId)) {
                log.warn("No experimental factor with id " + efId);
                continue;
            }

            if (!svdo.getFactors().containsKey(efId)) {
                // this should not happen.
                continue;
            }

            boolean isCategorical = !continuousFactors.contains(efId);

            Map<Long, String> categories = new HashMap<>();

            if (isCategorical) {
                this.getCategories(efIdMap, efId, categories);
            }

            if (!charts.containsKey(efId)) {
                charts.put(efId, new ArrayList<JFreeChart>());
            }

            Double a = factorCorrelations.get(component).get(efId);
            String plotname = (efs.get(efId) == null ? "?" : efs.get(efId)) + " " + xaxisLabel; // unique?

            if (a != null && !Double.isNaN(a)) {
                String title = plotname + " " + String.format("%.2f", a);
                List<Double> values = svdo.getFactors().get(efId);
                Double[] eigenGene = this.getEigenGene(svdo, component);
                assert values.size() == eigenGene.length;

                /*
                 * Plot eigengene vs values, add correlation to the plot
                 */
                JFreeChart chart;
                if (isCategorical) {

                    /*
                     * Categorical factor
                     */

                    // use the absolute value of the correlation, since direction is arbitrary.
                    title = plotname + " " + String.format("r=%.2f", Math.abs(a));

                    DefaultMultiValueCategoryDataset dataset = new DefaultMultiValueCategoryDataset();

                    /*
                     * What this code does is organize the factor values by the groups.
                     */
                    Map<String, List<Double>> groupedValues = new TreeMap<>();
                    for (int i = 0; i < values.size(); i++) {
                        Long fvId = values.get(i).longValue();
                        String fvValue = categories.get(fvId);
                        if (fvValue == null) {
                            /*
                             * Problem ...eg gill2006fateinocean id=1748 -- missing values. We just don't plot
                             * anything for this sample.
                             */
                            continue; // is this all we need to do?
                        }
                        if (!groupedValues.containsKey(fvValue)) {
                            groupedValues.put(fvValue, new ArrayList<Double>());
                        }

                        groupedValues.get(fvValue).add(eigenGene[i]);

                        if (log.isDebugEnabled())
                            log.debug(fvValue + " " + values.get(i));
                    }

                    for (String key : groupedValues.keySet()) {
                        dataset.add(groupedValues.get(key), plotname, key);
                    }

                    // don't show the name of the X axis: it's redundant with the title.
                    NumberAxis rangeAxis = new NumberAxis(xaxisLabel);
                    rangeAxis.setAutoRangeIncludesZero(false);
                    // rangeAxis.setAutoRange( false );
                    rangeAxis.setAutoRangeMinimumSize(4.0);
                    // rangeAxis.setRange( new Range( -2, 2 ) );

                    CategoryPlot plot = new CategoryPlot(dataset, new CategoryAxis(null), rangeAxis,
                            new ScatterRenderer());
                    plot.setRangeGridlinesVisible(false);
                    plot.setDomainGridlinesVisible(false);

                    chart = new JFreeChart(title, new Font("SansSerif", Font.BOLD, 12), plot, false);

                    ScatterRenderer renderer = (ScatterRenderer) plot.getRenderer();
                    float saturationDrop = (float) Math.min(1.0, component * 0.8f / MAX_COMP);
                    renderer.setSeriesFillPaint(0, Color.getHSBColor(0.0f, 1.0f - saturationDrop, 0.7f));
                    renderer.setSeriesShape(0, new Ellipse2D.Double(0, 0, 3, 3));
                    renderer.setUseOutlinePaint(false);
                    renderer.setUseFillPaint(true);
                    renderer.setBaseFillPaint(Color.white);
                    CategoryAxis domainAxis = plot.getDomainAxis();
                    domainAxis.setCategoryLabelPositions(CategoryLabelPositions.UP_45);
                } else {

                    /*
                     * Continuous value factor
                     */

                    DefaultXYDataset series = new DefaultXYDataset();
                    series.addSeries(plotname,
                            new double[][] { ArrayUtils.toPrimitive(values.toArray(new Double[] {})),
                                    ArrayUtils.toPrimitive(eigenGene) });

                    // don't show x-axis label, which would otherwise be efs.get( efId )
                    chart = ChartFactory.createScatterPlot(title, null, xaxisLabel, series,
                            PlotOrientation.VERTICAL, false, false, false);
                    XYPlot plot = chart.getXYPlot();
                    plot.setRangeGridlinesVisible(false);
                    plot.setDomainGridlinesVisible(false);

                    XYItemRenderer renderer = plot.getRenderer();
                    renderer.setBasePaint(Color.white);
                    renderer.setSeriesShape(0, new Ellipse2D.Double(0, 0, 3, 3));
                    float saturationDrop = (float) Math.min(1.0, component * 0.8f / MAX_COMP);
                    renderer.setSeriesPaint(0, Color.getHSBColor(0.0f, 1.0f - saturationDrop, 0.7f));
                    plot.setRenderer(renderer);
                }

                chart.getTitle().setFont(new Font("SansSerif", Font.BOLD, 12));

                charts.get(efId).add(chart);
            }
        }
    }

    /*
     * DATES
     */
    charts.put(-1L, new ArrayList<JFreeChart>());
    for (Integer component : dateCorrelations.keySet()) {
        String xaxisLabel = componentShorthand + (component + 1);

        List<Date> dates = svdo.getDates();
        if (dates.isEmpty())
            break;

        long secspan = ubic.basecode.util.DateUtil.numberOfSecondsBetweenDates(dates);

        if (component >= MAX_COMP)
            break;
        Double a = dateCorrelations.get(component);

        if (a != null && !Double.isNaN(a)) {
            Double[] eigenGene = svdo.getvMatrix().getColObj(component);

            /*
             * Plot eigengene vs values, add correlation to the plot
             */
            TimeSeries series = new TimeSeries("Dates vs. eigen" + (component + 1));
            int i = 0;
            for (Date d : dates) {
                // if span is less than an hour, retain the minute.
                if (secspan < 60 * 60) {
                    series.addOrUpdate(new Minute(d), eigenGene[i++]);
                } else {
                    series.addOrUpdate(new Hour(d), eigenGene[i++]);
                }

            }
            TimeSeriesCollection dataset = new TimeSeriesCollection();
            dataset.addSeries(series);

            JFreeChart chart = ChartFactory.createTimeSeriesChart(
                    "Dates: " + xaxisLabel + " " + String.format("r=%.2f", a), null, xaxisLabel, dataset, false,
                    false, false);

            XYPlot xyPlot = chart.getXYPlot();

            chart.getTitle().setFont(new Font("SansSerif", Font.BOLD, 12));

            // standard renderer makes lines.
            XYDotRenderer renderer = new XYDotRenderer();
            renderer.setBaseFillPaint(Color.white);
            renderer.setDotHeight(3);
            renderer.setDotWidth(3);
            renderer.setSeriesShape(0, new Ellipse2D.Double(0, 0, 3, 3)); // has no effect, need dotheight.
            float saturationDrop = (float) Math.min(1.0, component * 0.8f / MAX_COMP);
            renderer.setSeriesPaint(0, Color.getHSBColor(0.0f, 1.0f - saturationDrop, 0.7f));
            ValueAxis domainAxis = xyPlot.getDomainAxis();
            domainAxis.setVerticalTickLabels(true);
            xyPlot.setRenderer(renderer);
            xyPlot.setRangeGridlinesVisible(false);
            xyPlot.setDomainGridlinesVisible(false);
            charts.get(-1L).add(chart);

        }
    }

    /*
     * Plot in a grid, with each factor as a column. FIXME What if we have too many factors to fit on the screen?
     */
    int columns = (int) Math.ceil(charts.size());
    int perChartSize = ExpressionExperimentQCController.DEFAULT_QC_IMAGE_SIZE_PX;
    BufferedImage image = new BufferedImage(columns * perChartSize, MAX_COMP * perChartSize,
            BufferedImage.TYPE_INT_ARGB);
    Graphics2D g2 = image.createGraphics();
    int currentX = 0;
    int currentY = 0;
    for (Long id : charts.keySet()) {
        for (JFreeChart chart : charts.get(id)) {
            this.addChartToGraphics(chart, g2, currentX, currentY, perChartSize, perChartSize);
            if (currentY + perChartSize < MAX_COMP * perChartSize) {
                currentY += perChartSize;
            } else {
                currentY = 0;
                currentX += perChartSize;
            }
        }
    }

    os.write(ChartUtilities.encodeAsPNG(image));
    return true;
}