CIRProcessEuler.java

/*
 * Class:        CIRProcessEuler
 * Description:  
 * Environment:  Java
 * Software:     SSJ 
 * Copyright (C) 2001  Pierre L'Ecuyer and Universite de Montreal
 * Organization: DIRO, Universite de Montreal
 * @author       
 * @since
 *
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 */
package umontreal.ssj.stochprocess;
 
import umontreal.ssj.rng.*;
import umontreal.ssj.probdist.*;
import umontreal.ssj.randvar.*;

public class CIRProcessEuler extends StochasticProcess {
   protected NormalGen gen;
   protected double alpha, beta, sigma;
   // Precomputed values
   protected double[] alphadt, sigmasqrdt;

   public CIRProcessEuler(double x0, double alpha, double b, double sigma, RandomStream stream) {
      this(x0, alpha, b, sigma, new NormalGen(stream, new NormalDist()));
   }

   public CIRProcessEuler(double x0, double alpha, double b, double sigma, NormalGen gen) {
      this.alpha = alpha;
      this.beta = b;
      this.sigma = sigma;
      this.x0 = x0;
      this.gen = gen;
   }
 
   public double nextObservation() {
      double xOld = path[observationIndex];
      double x;
      x = xOld + (beta - xOld) * alphadt[observationIndex]
            + sigmasqrdt[observationIndex] * Math.sqrt(xOld) * gen.nextDouble();
      observationIndex++;
      if (x >= 0.0)
         path[observationIndex] = x;
      else
         path[observationIndex] = 0.;
      return x;
   }

   public double nextObservation(double nextTime) {
      double previousTime = t[observationIndex];
      double xOld = path[observationIndex];
      observationIndex++;
      t[observationIndex] = nextTime;
      double dt = nextTime - previousTime;
      double x = xOld + alpha * (beta - xOld) * dt + sigma * Math.sqrt(dt * xOld) * gen.nextDouble();
      if (x >= 0.0)
         path[observationIndex] = x;
      else
         path[observationIndex] = 0.;
      return x;
   }

   public double nextObservation(double x, double dt) {
      x = x + alpha * (beta - x) * dt + sigma * Math.sqrt(dt * x) * gen.nextDouble();
      if (x >= 0.0)
         return x;
      return 0.0;
   }
 
   public double[] generatePath() {
      double x;
      double xOld = x0;
      for (int j = 0; j < d; j++) {
         x = xOld + (beta - xOld) * alphadt[j] + sigmasqrdt[j] * Math.sqrt(xOld) * gen.nextDouble();
         if (x < 0.0)
            x = 0.0;
         path[j + 1] = x;
         xOld = x;
      }
      observationIndex = d;
      return path;
   }

   public double[] generatePath(RandomStream stream) {
      gen.setStream(stream);
      return generatePath();
   }

   public void setParams(double x0, double alpha, double b, double sigma) {
      this.alpha = alpha;
      this.beta = b;
      this.sigma = sigma;
      this.x0 = x0;
      if (observationTimesSet)
         init(); // Otherwise not needed.
   }

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

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

   public double getAlpha() {
      return alpha;
   }

   public double getB() {
      return beta;
   }

   public double getSigma() {
      return sigma;
   }

   public NormalGen getGen() {
      return gen;
   }
 
   // This is called by setObservationTimes to precompute constants
   // in order to speed up the path generation.
   protected void init() {
      super.init();
      alphadt = new double[d];
      sigmasqrdt = new double[d];
      double dt;
      for (int j = 0; j < d; j++) {
         dt = t[j + 1] - t[j];
         alphadt[j] = alpha * (dt);
         sigmasqrdt[j] = sigma * Math.sqrt(dt);
      }
   }
 
}