package ecoparasite.completion;

import ecoparasite.poisson.Poisson;

import java.util.HashSet;

public class Completion {

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

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

        double z = 0;

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

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

        return tablePoisson;
    }

    private double mean(HashSet<Poisson> tablePoisson){

        double mean = 0.0;

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

        return mean / tablePoisson.size();
    }

    private 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  double ecartType(HashSet<Poisson> tablePoisson){

        double vari = variance(tablePoisson);

        return Math.sqrt(vari);
    }


}