TMDlib_InterpolationKS.h

Go to the documentation of this file.

//------------------------------------------------------------------------------
//
// Copyright (c) 2012, Sebastian Sapeta
//
//  This is part of KSgluon package.
//
//  KSgluon is free software; you can redistribute it and/or modify
//  it under the terms of the GNU General Public License as published by
//  the Free Software Foundation; either version 2 of the License, or
//  (at your option) any later version.
//
//  KSgluon 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.
//
//  You should have received a copy of the GNU General Public License
//  along with FastJet; if not, write to the Free Software
//  Foundation, Inc.:
//      59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
//
//------------------------------------------------------------------------------
#ifndef __TMDlib_INTERPOLATIONKS_H__
#define __TMDlib_INTERPOLATIONKS_H__
#include <cstdlib>
#include <cstdio>
#include <cmath>
#include <iostream>
#include <memory>
#include <vector>
#include <boost/multi_array/base.hpp>
#include <boost/shared_ptr.hpp>
#include <boost/multi_array.hpp>
#include <gsl/gsl_errno.h>
#include <gsl/gsl_interp.h>
#include <gsl/gsl_spline.h>


#include <gsl/gsl_interp2d.h>
#include <gsl/gsl_spline2d.h>
using namespace std;

//------------------------------------------------------------------------------
//------------------------------------------------------------------------------
class BaseInterpolation {
  public:
  BaseInterpolation() {}
  virtual ~BaseInterpolation() {}

  virtual vector<pair <double, double> > grid_limits() const { 
    return vector<pair <double, double> >(); }

  virtual double interp (const vector<double> &x) { return -1.0; }
  
  unsigned int eval_status() {return _eval_status;}

  protected:
  unsigned int _eval_status; 

};


//------------------------------------------------------------------------------
class Interpolation2D : public BaseInterpolation {

  public:
  Interpolation2D(const vector<double> &x1vv, const vector<double> &x2vv,
                 boost::multi_array<double, 2> ym, 
                 gsl_interp_type int_type = *gsl_interp_linear); 

  ~Interpolation2D();
  
  vector<pair <double, double> > grid_limits() const;

  double interp (const vector<double> &x);

  private:
   gsl_interp_type _int_type; //this must stay - we leave old constructor
   gsl_interp2d_type _int2d_type; //possib: gsl_interp2d_bilinear or bicubic
   unsigned int n1, n2;
   boost::multi_array<double, 2> y;
   gsl_interp_accel* x1acc; 
   gsl_interp_accel* x2acc; 
   gsl_spline2d* spline2d;
   vector<double> x1v, x2v;

};
//------------------------------------------------------------------------------
//
//------------------------------------------------------------------------------
class CubicSpline3D : public BaseInterpolation {

  public:
  CubicSpline3D(const vector<double> &x1vv, const vector<double> &x2vv, 
                const vector<double> &x3vv, boost::multi_array<double, 3> ym);
                
  ~CubicSpline3D();

  //double interp (double x3) {
  //      int i = 0, j = 0;
  //      return gsl_spline_eval (spline[i][j], x3, acc[i][j]);
  //}
  

  vector<pair <double, double> > grid_limits() const;
  double interp (const vector<double> &x);
  

  private:
  unsigned int n1, n2, n3; 
  boost::multi_array<double, 3> y;
  vector<gsl_interp_accel*> x1acc;
  vector<gsl_interp_accel*> x2acc;
  vector<gsl_spline2d*>  spline2d;
  vector<double> x1v, x2v, x3v;

};



#endif // __TMDlib_INTERPOLATIONKS_H__