package ecoparasite.completion;

import ecoparasite.poisson.Poisson;

import java.util.HashSet;


/**
 * Class incluant des méthodes Statiques de Completion des données
 */
public class Completion {


    public static HashSet<Poisson> completePoissonMean(HashSet<Poisson> tablePoisson){

        Double mean = mean(tablePoisson);

        for (Poisson p : tablePoisson) {
            if ( p.getInfestation() == null ) {
                p.setInfestation(mean);
            }
        }

        return tablePoisson;
    }

    private static Double mean(HashSet<Poisson> tablePoisson){

        Double mean = 0.0;

        for (Poisson p : tablePoisson){
            if (p.getInfestation() != null) {
                mean += p.getInfestation();
            }
        }

        return mean / tablePoisson.size();
    }


}