ExplicitFDM.cpp

/**
* ==================================================================================================================================
* Name: 		David Turner
* Description: Solves Black Scholes equation using Explicit finite difference method
*
* */

# include <cmath>

using namespace std;

/**
 * Solves Black Scholes equation using Explicit finite difference method
 * Inputs: double S (spot price)
 *         double K (strike price)
 *         double r (risk free rate)
 *         double q (dividend rate)
 *         double sigma (volatility)
 *         double expiry (time to expiry)
 *         double dx (step size in space)
 *         double dtau (step size in time)
 *         int call_or_put (+1 for call, -1 for put)
 *         int amer_or_eur (0 for European option, 1 for American)
 * Output: double value (value of option)
 */
double ExplicitFDM(double S, double K, double r, double q, double sigma, double expiry, double dx, double dtau, int call_or_put, int amer_or_eur) {
  double *b;
  double w;
  double *ymat;
  int i, j;
  double *t, *x;
  double t_max, t_min, x_max, x_min;
  int N, M;

  double rp = 2*r/(sigma*sigma);
  double qp = 2*(r-q)/(sigma*sigma);
  double alpha = -0.5*(qp-1);
  double beta = -0.25*(qp-1)*(qp-1) + rp;

  // x = log(S/K) ranges from -2.5 to 2.5
  // proxy for 0 to "infinity"
  x_min = -2.5;
  x_max = 2.5;
  M = 1 + ((x_max - x_min)/dx);

  x = new double[M];

  // setup x vector with M elements at steps of dx
  for(i=0; i<M; i++) {
    x[i] = x_min + i*dx;
  }

  // tau vector ranges from 0 to 0.5*sigma^2*expiry
  t_min = 0.0;
  t_max = 0.5*(sigma*sigma)*expiry;
  N = 1 + ((t_max - t_min)/dtau);

  t = new double[N];

  // setup t vector with N elements at steps of dtau
  for(j=0; j<N; j++) {
    t[j] = t_min + j*dtau;
  }

  // FDM matrix
  ymat = new double[M*N];

  for(i=0; i<M; i++) {
    // Initial condition (at tau=0)
    ymat[i] = fmax(call_or_put*(exp(0.5*x[i]*(qp+1))-exp(0.5*x[i]*(qp-1))),0.0);
  }

  w = dtau/(dx*dx); // for explicit FDM, w <= 0.5 for stability
  b = new double[M]; // this contains current column

  for(j=1; j<N; j++) {
    // Boundary condition at x=-2.5
    ymat[0+j*M]=(call_or_put>0)?0.0:w*exp(0.5*(qp-1)*x[0]+0.25*(qp-1)*(qp-1)*t[j]);
    for(i=1; i<M-1; i++) {
      // Update interior points
      b[i] = ymat[i+(j-1)*M] + w*(
	ymat[i-1+(j-1)*M]-2.0*ymat[i+(j-1)*M]+ymat[i+1+(j-1)*M]);
      if(amer_or_eur==1) {
	ymat[i+j*M] = fmax(b[i],w*call_or_put*(exp(0.5*x[i]*(qp+1))-exp(0.5*x[i]*(qp-1))));  // check for early exercise
      } else {
	ymat[i+j*M] = b[i];
      }
    }
    // Boundary condition at x=2.5
    ymat[M-1+j*M]= (call_or_put>0)? w*exp(0.5*(qp+1)*x[M-1]+0.25*(qp+1)*(qp+1)*t[j]):0.0;
  }
  i = M/2; // value of option at the money (S == K)
  j = N-1; // value at tau (t=0)

  double value = ymat[i+j*M]*K*exp(alpha*x[i]+beta*t[j]);

  delete [] ymat;
  delete [] t;
  delete [] x;

  return value;
}