org.vaadin.addons.chart.js.ChartJS.java Source code

Introduction

Here is the source code for org.vaadin.addons.chart.js.ChartJS.java

Source

/*
 * Copyright 2014 fatalix.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.vaadin.addons.chart.js;

import com.vaadin.annotations.JavaScript;
import com.vaadin.annotations.StyleSheet;
import com.vaadin.ui.AbstractJavaScriptComponent;
import java.util.List;
import org.vaadin.addons.chart.js.data.AbstractSeriesDataSet;
import org.vaadin.addons.chart.js.data.BarSeriesDataContainer;
import org.vaadin.addons.chart.js.data.BarSeriesDataSet;
import org.vaadin.addons.chart.js.data.DoughnutDataContainer;
import org.vaadin.addons.chart.js.data.DoughnutSeriesDataSet;
import org.vaadin.addons.chart.js.data.LineSeriesDataContainer;
import org.vaadin.addons.chart.js.data.LineSeriesDataSet;
import org.vaadin.addons.chart.js.data.PieDataContainer;
import org.vaadin.addons.chart.js.data.PieSeriesDataSet;
import org.vaadin.addons.chart.js.data.RadarSeriesDataContainer;
import org.vaadin.addons.chart.js.data.RadarSeriesDataSet;

/**
 *
 * @author felix.husse
 * @param <T>
 */
@JavaScript({ "vaadin://chartjs/Chart.min.js", "chartjs-connector.js" })
@StyleSheet({ "vaadin://chartjs/chartjs-styles.css" })
public class ChartJS<T extends AbstractSeriesDataSet> extends AbstractJavaScriptComponent {

    private static int componentCount = 0;

    private final int componentId;

    public ChartJS(ChartConfiguration chartConfiguration, List<String> labels, List<T> dataSets) {
        super();
        componentId = componentCount;
        setId("chartjs-wrapper-" + componentId);
        componentCount++;
        ChartData chartData = new ChartData();
        chartData.componentID = componentId;
        chartData.chartConfiguration = chartConfiguration;

        if (dataSets.get(0) instanceof RadarSeriesDataSet) {
            RadarSeriesDataContainer dataContainer = new RadarSeriesDataContainer();
            dataContainer.setLabels(labels);
            dataContainer.setDatasets((List<RadarSeriesDataSet>) dataSets);
            chartData.chartType = ChartType.RADAR;
            chartData.radarSeriesDataContainer = dataContainer;
        } else if (dataSets.get(0) instanceof LineSeriesDataSet) {
            LineSeriesDataContainer dataContainer = new LineSeriesDataContainer();
            dataContainer.setLabels(labels);
            dataContainer.setDatasets((List<LineSeriesDataSet>) dataSets);
            chartData.chartType = ChartType.LINE;
            chartData.lineSeriesDataContainer = dataContainer;
        } else if (dataSets.get(0) instanceof BarSeriesDataSet) {
            BarSeriesDataContainer dataContainer = new BarSeriesDataContainer();
            dataContainer.setLabels(labels);
            dataContainer.setDatasets((List<BarSeriesDataSet>) dataSets);
            chartData.chartType = ChartType.BAR;
            chartData.barSeriesDataContainer = dataContainer;
        } else if (dataSets.get(0) instanceof PieSeriesDataSet) {
            PieDataContainer[] dcs = new PieDataContainer[dataSets.size()];
            int counter = 0;
            for (PieSeriesDataSet dataSet : (List<PieSeriesDataSet>) dataSets) {
                PieDataContainer dataContainer = new PieDataContainer();
                dataContainer.setColor(dataSet.getColor());
                dataContainer.setHighlight(dataSet.getHighlight());
                dataContainer.setLabel(dataSet.label);
                dataContainer.setValue(dataSet.getValue());
                dcs[counter] = dataContainer;
                counter++;
            }
            chartData.chartType = ChartType.PIE;
            chartData.pieSeriesDataContainer = dcs;
        } else if (dataSets.get(0) instanceof DoughnutSeriesDataSet) {

            DoughnutDataContainer[] dcs = new DoughnutDataContainer[dataSets.size()];
            int counter = 0;
            for (DoughnutSeriesDataSet dataSet : (List<DoughnutSeriesDataSet>) dataSets) {
                DoughnutDataContainer dataContainer = new DoughnutDataContainer();
                dataContainer.setColor(dataSet.getColor());
                dataContainer.setHighlight(dataSet.getHighlight());
                dataContainer.setLabel(dataSet.label);
                dataContainer.setValue(dataSet.getValue());
                dcs[counter] = dataContainer;
                counter++;
            }
            chartData.chartType = ChartType.DOUGHNUT;
            chartData.doughnutDataContainer = dcs;
        }

        getState().chartData = chartData;
    }

    @Override
    protected ChartJSState getState() {
        return (ChartJSState) super.getState();
    }

}