/**
 * Copyright (C) 2001-2017 by RapidMiner and the contributors
 * 
 * Complete list of developers available at our web site:
 * 
 * http://rapidminer.com
 * 
 * This program is free software: you can redistribute it and/or modify it under the terms of the
 * GNU Affero General Public License as published by the Free Software Foundation, either version 3
 * of the License, or (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without
 * even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * Affero General Public License for more details.
 * 
 * You should have received a copy of the GNU Affero General Public License along with this program.
 * If not, see http://www.gnu.org/licenses/.
*/
package com.rapidminer.gui.new_plotter.data;

import com.rapidminer.gui.new_plotter.configuration.DimensionConfig;
import com.rapidminer.gui.new_plotter.configuration.DimensionConfig.PlotDimension;
import com.rapidminer.gui.new_plotter.configuration.ValueSource;
import com.rapidminer.gui.new_plotter.configuration.ValueSource.SeriesUsageType;

import java.util.HashMap;
import java.util.Map;


/**
 * @author Marius Helf
 * 
 */
public class GroupCellData {

	Map<SeriesUsageType, Map<PlotDimension, double[]>> dimensionToDataMap = new HashMap<ValueSource.SeriesUsageType, Map<PlotDimension, double[]>>();

	public Map<PlotDimension, double[]> getDataForUsageType(SeriesUsageType usageType) {
		return dimensionToDataMap.get(usageType);
	}

	public void setDataForUsageType(SeriesUsageType usageType, Map<PlotDimension, double[]> data) {
		if (data != null) {
			dimensionToDataMap.put(usageType, data);
		} else {
			dimensionToDataMap.remove(usageType);
		}
	}

	/**
	 * Returns the number of data points in the DOMAIN dimension of the MAIN_SERIES series of this
	 * GroupCellData. All other usage types should have the same number of values.
	 */
	public int getSize() {
		Map<PlotDimension, double[]> mainData = getDataForUsageType(SeriesUsageType.MAIN_SERIES);
		if (mainData == null) {
			return 0;
		}

		double[] domainData = mainData.get(PlotDimension.DOMAIN);
		if (domainData == null) {
			return 0;
		}

		return domainData.length;
	}

	/**
	 * Initializes a map from Dimension to double[] for the given SeriesUsageType. Each double array
	 * is initialized with size valueCount.
	 * 
	 * The created map is added to this GroupCellData and then returned to the caller.
	 */
	public Map<PlotDimension, double[]> initDataForUsageType(SeriesUsageType usageType, Iterable<PlotDimension> dimensions,
			int valueCount) {
		Map<PlotDimension, double[]> data = new HashMap<DimensionConfig.PlotDimension, double[]>();
		for (PlotDimension dimension : dimensions) {
			data.put(dimension, new double[valueCount]);
		}
		setDataForUsageType(usageType, data);
		return data;
	}
}