StratifiedUnitCube.java

/*
 * Class:        StratifiedUnitCube
 * Description:  
 * Environment:  Java
 * Software:     SSJ 
 * Copyright (C) 2001  Pierre L'Ecuyer and Universite de Montreal
 * Organization: DIRO, Universite de Montreal
 * @author       
 * @since
 
 * SSJ is free software: you can redistribute it and/or modify it under
 * the terms of the GNU General Public License (GPL) as published by the
 * Free Software Foundation, either version 3 of the License, or
 * any later version.
 
 * SSJ is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 
 * A copy of the GNU General Public License is available at
   <a href="http://www.gnu.org/licenses">GPL licence site</a>.
 */
package umontreal.ssj.hups;
 
import umontreal.ssj.rng.RandomStream;

public class StratifiedUnitCube extends CachedPointSet {

   protected int[] numDiv; // Number of divisions in each dimension, k_j

   protected double[] delta;

   public StratifiedUnitCube(int[] k, int dim) {
      this.dim = dim;
      delta = new double[dim];
      numDiv = new int[dim];
      numPoints = 1;
      for (int j = 0; j < dim; j++) {
         numDiv[j] = k[j];
         numPoints *= k[j];
         delta[j] = 1.0 / (double) numDiv[j];
      }
      x = new double[numPoints][dim];
   }

   public StratifiedUnitCube(int k, int dim) {
      this.dim = dim;
      delta = new double[dim];
      numDiv = new int[dim];
      numPoints = 1;
      for (int j = 0; j < dim; j++) {
         numDiv[j] = k;
         numPoints *= k;
         delta[j] = 1.0 / (double) k;
      }
      x = new double[numPoints][dim];
   }

   public void randomize(RandomStream stream) {
      int[] current = new int[dim]; // current[j] = current division for
                                    // dim j when we enumerate the points
      for (int j = 0; j < dim; j++)
         current[j] = 0;
      for (int i = 0; i < numPoints; i++) {
         // Generate random point in current box; this is point i.
         for (int j = 0; j < dim; j++)
            x[i][j] = (current[j] + stream.nextDouble()) * delta[j];
         // Find the next box.
         for (int l = 0; l < dim; l++) {
            current[l]++;
            if (current[l] >= numDiv[l])
               current[l] = 0; // next, we will add the carry to current[l+1].
            else
               l = dim; // Exit loop.
         }
      }
   }

   public void addRandomShift(int fromDim, int toDim, RandomStream stream) {
      randomize(stream);
   }

   public void randomize(PointSetRandomization rand) {
      randomize(rand.getStream());
   }
 
   public String toString() {
      return "StratifiedUnitCube: stratified point set over the unit cube in " + dim + "dimensions.";
   }
}