EDIT

/**

* Copyright (C) 2018 EDIT

* European Distributed Institute of Taxonomy

* http://www.e-taxonomy.eu

*

* The contents of this file are subject to the Mozilla Public License Version 1.1

* See LICENSE.TXT at the top of this package for the full license terms.

*/

package eu.etaxonomy.taxeditor.editor.descriptiveDataSet.matrix;

import java.util.HashMap;

import java.util.Map;

import eu.etaxonomy.cdm.model.description.Feature;

import eu.etaxonomy.cdm.model.description.State;

/**

 * @author pplitzner

 * @since Jul 12, 2018

 *

 */

public class CategoricalDataHistogram {

    private Feature feature;

    private Map<State, Integer> stateCountMap;

    public CategoricalDataHistogram(Feature feature) {

        this.feature = feature;

        this.stateCountMap = new HashMap<>();

        feature.getSupportedCategoricalEnumerations()

        .forEach(voc->voc.getTerms()

                .forEach(state->stateCountMap.put(state, 0)));

    }

    public void addState(State state){

        Integer integer = stateCountMap.get(state);

        stateCountMap.put(state, integer+1);

    }

    public Map<State, Integer> getStateCountMap() {

        return stateCountMap;

    }

    public Feature getFeature() {

        return feature;

    }

}