/**
 * DataCleaner (community edition)
 * Copyright (C) 2014 Neopost - Customer Information Management
 *
 * This copyrighted material is made available to anyone wishing to use, modify,
 * copy, or redistribute it subject to the terms and conditions of the GNU
 * Lesser General Public License, as published by the Free Software Foundation.
 *
 * 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 Lesser General Public License
 * for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this distribution; if not, write to:
 * Free Software Foundation, Inc.
 * 51 Franklin Street, Fifth Floor
 * Boston, MA  02110-1301  USA
 */
package org.datacleaner.visualization;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.apache.metamodel.util.ObjectComparator;
import org.datacleaner.api.InputColumn;
import org.datacleaner.util.ReflectionUtils;

public class JavaStackedAreaAnalyzerResult implements IStackedAreaAnalyzerResult {
    private static final long serialVersionUID = 1L;

    private final InputColumn<?> _categoryColumn;
    private final InputColumn<Number>[] _measureColumns;
    private final Map<Object, List<Number>> _measureMap = new HashMap<>();

    public JavaStackedAreaAnalyzerResult(final InputColumn<?> categoryColumn,
            final InputColumn<Number>[] measureColumns) {
        _categoryColumn = categoryColumn;
        _measureColumns = measureColumns;
    }

    @Override
    public boolean isTimeCategory() {
        return ReflectionUtils.isDate(_categoryColumn.getDataType());
    }

    @Override
    public boolean isNumberCategory() {
        return ReflectionUtils.isNumber(_categoryColumn.getDataType());
    }

    @Override
    public InputColumn<?> getCategoryColumn() {
        return _categoryColumn;
    }

    @Override
    public int getCategoryCount() {
        return _measureMap.size();
    }

    @Override
    public List<?> getCategories() {
        final List<Object> categories = new ArrayList<>(_measureMap.keySet());
        categories.sort(ObjectComparator.getComparator());
        return categories;
    }

    @Override
    public InputColumn<Number>[] getMeasureColumns() {
        return _measureColumns;
    }

    @Override
    public Number[] getMeasures(final Object category) {
        final List<Number> list = _measureMap.get(category);
        if (list != null) {
            return list.toArray(new Number[list.size()]);
        } else {
            return null;
        }
    }

    @Override
    public void addMeasures(final Object category, final Number[] measures) {
        final List<Number> oldMeasures = _measureMap.get(category);
        if (oldMeasures != null) {
            for (int i = 0; i < oldMeasures.size(); i++) {
                final Number oldValue = oldMeasures.get(i);
                final Number addition = measures[i];
                oldMeasures.remove(i);
                oldMeasures.add(i, sum(oldValue, addition));
            }
        } else {
            _measureMap.put(category, new ArrayList<>(Arrays.asList(measures)));
        }
    }

    @Override
    public Number sum(final Number x, final Number y) {
        if (x == null) {
            return y;
        }
        if (y == null) {
            return x;
        }
        return x.doubleValue() + y.doubleValue();
    }
}