Replicator.java

package umontreal.ssj.mcqmctools.anova;
 
import umontreal.ssj.stat.Tally;
import umontreal.ssj.stat.list.ListOfTallies;
import umontreal.ssj.rng.*;
// import umontreal.ssj.hups.*;
import umontreal.ssj.mcqmctools.*;

public class Replicator implements RandomIntegrator {
 
   protected int nReplicates;
   protected RandomIntegrator integrator;
 
   // internal tallies, to be allocated only once, and only if needed
   protected Tally statValue = null;
   protected ListOfTallies<Tally> statValueList = null;

   public Replicator(int nReplicates, RandomIntegrator integrator) {
      this.nReplicates = nReplicates;
      this.integrator = integrator;
   }

   public RandomIntegrator getIntegrator() {
      return integrator;
   }

   public int getNumReplicates() {
      return nReplicates;
   }

   public int getNumPoints() {
      return integrator.getTotalSimulations();
   }

   public int getTotalSimulations() {
      return getNumReplicates() * getNumPoints();
   }

   public void setStream(RandomStream stream) {
      integrator.setStream(stream);
   }

   public RandomStream getStream() {
      return integrator.getStream();
   }

   public void integrate(MonteCarloModelDouble model, Tally statValue) {
      statValue.init();
      for (int i = 0; i < nReplicates; i++)
         statValue.add(integrator.integrate(model));
   }

   public double integrate(MonteCarloModelDouble model) {
      if (statValue == null)
         this.statValue = new Tally();
      else
         statValue.init();
      integrate(model, statValue);
      return statValue.average();
   }

   public void integrate(MonteCarloModel<double[]> model, ListOfTallies<? extends Tally> stat) {
      double[] val = new double[stat.size()];
      ListOfTallies<Tally> innerStat = ListOfTallies.createWithTally(stat.size());
 
      for (int i = 0; i < nReplicates; i++) {
         innerStat.init();
         integrator.integrate(model, innerStat);
         innerStat.average(val);
         stat.add(val);
      }
   }

   public void integrate(MonteCarloModel<double[]> model, double[] values) {
      if (statValueList == null || statValueList.size() != values.length)
         this.statValueList = ListOfTallies.createWithTally(values.length);
      else
         statValueList.init();
      integrate(model, statValueList);
      statValueList.average(values);
   }
 
   @Override
   public String toString() {
      return "Replicator" + " [replicates=" + nReplicates + "]" + " [integrator=" + integrator + "]";
   }
}