package ecoparasite.completion;

import ecoparasite.poisson.Poisson;

import java.util.HashSet;

public class Completion {

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

        double mean = mean(tablePoisson);

        for (Poisson p : tablePoisson) {

            if ( p.getInfestation().isNaN() ) {
                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();
    }


}