package org.activityinfo.legacy.shared.reports.model;

/*
 * #%L
 * ActivityInfo Server
 * %%
 * Copyright (C) 2009 - 2013 UNICEF
 * %%
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU 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 General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public 
 * License along with this program.  If not, see
 * <http://www.gnu.org/licenses/gpl-3.0.html>.
 * #L%
 */

import com.google.common.collect.Lists;
import org.activityinfo.legacy.shared.command.DimensionType;
import org.activityinfo.legacy.shared.reports.content.PivotChartContent;

import javax.xml.bind.annotation.XmlElement;
import javax.xml.bind.annotation.XmlElementWrapper;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

public class PivotChartReportElement extends PivotReportElement<PivotChartContent> {

    public enum Type {
        @Deprecated Bar,
        StackedBar,
        Line,
        ClusteredBar,
        Pie
    }

    private Type type = Type.ClusteredBar;
    private List<Dimension> categoryDimensions = new ArrayList<Dimension>();
    private List<Dimension> seriesDimensions = new ArrayList<Dimension>();
    private String categoryAxisTitle;
    private String valueAxisTitle;

    public PivotChartReportElement() {

    }

    public PivotChartReportElement(Type type) {
        this.type = type;
    }

    @Override
    public Set<Dimension> allDimensions() {
        Set<Dimension> set = new HashSet<Dimension>();
        set.addAll(categoryDimensions);
        set.addAll(seriesDimensions);
        return set;
    }

    @XmlElement(required = true)
    public Type getType() {
        return type;
    }

    public void setType(Type type) {
        this.type = type;
    }

    public void setIndicator(int indicatorId) {
        getFilter().clearRestrictions(DimensionType.Indicator);
        addIndicator(indicatorId);
    }

    public void addIndicator(int indicatorId) {
        getFilter().addRestriction(DimensionType.Indicator, indicatorId);
    }

    public void addIndicator(Integer indicatorId) {
        getFilter().addRestriction(DimensionType.Indicator, indicatorId);
    }

    @XmlElement(name = "dimension") @XmlElementWrapper(name = "series")
    public List<Dimension> getSeriesDimensions() {
        return seriesDimensions;
    }

    public void setSeriesDimensions(List<Dimension> seriesDimensions) {
        this.seriesDimensions = seriesDimensions;
    }

    public void addSeriesDimension(Dimension dimension) {
        this.seriesDimensions.add(dimension);
    }

    public void setSeriesDimension(Dimension dimension) {
        this.seriesDimensions = Lists.newArrayList(dimension);
    }

    @XmlElement(name = "dimension") @XmlElementWrapper(name = "categories")
    public List<Dimension> getCategoryDimensions() {
        return categoryDimensions;
    }

    public void setCategoryDimensions(List<Dimension> categoryDimensions) {
        this.categoryDimensions = categoryDimensions;
    }

    public void setCategoryDimension(Dimension dimension) {
        this.categoryDimensions = Lists.newArrayList(dimension);
    }

    public void addCategoryDimension(Dimension dimension) {
        this.categoryDimensions.add(dimension);
    }

    @XmlElement
    public String getCategoryAxisTitle() {
        return categoryAxisTitle;
    }

    public void setCategoryAxisTitle(String categoryAxisTitle) {
        this.categoryAxisTitle = categoryAxisTitle;
    }

    @XmlElement
    public String getValueAxisTitle() {
        return valueAxisTitle;
    }

    public void setValueAxisTitle(String valueAxisTitle) {
        this.valueAxisTitle = valueAxisTitle;
    }

}