package ecoparasite.nettoyage;

import ecoparasite.poisson.Poisson;

import java.util.HashSet;

public class Nettoyage {

    /**
     * 
     * @param tablePoisson
     * @return
     */
    public static HashSet<Poisson> nettoiePoissonMean(HashSet<Poisson> tablePoisson){

        Double mean = mean(tablePoisson); //Moyenne
        Double ecart = ecartType(tablePoisson); // Ecart Type

        Double z = 0.0;

        for (Poisson p : tablePoisson) {
            z = ( p.getInfestation() - mean ) / ecart;

            if ( z >= 2.5 ) {
                p.setInfestation(mean);
            }
        }

        return tablePoisson;
    }

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

        Double mean = 0.0;

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

        return mean / tablePoisson.size();
    }

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

        Double vari = 0.0;
        Double mean = mean(tablePoisson);

        for (Poisson p : tablePoisson) {
            vari += Math.pow( (p.getInfestation() - mean), 2);
        }

        return vari / tablePoisson.size();
    }

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

        Double vari = variance(tablePoisson);

        return Math.sqrt(vari);
    }


}