File: Interpolation.i

package info (click to toggle)
quantlib-python 0.2.1.cvs20020322-1
  • links: PTS
  • area: main
  • in suites: woody
  • size: 2,040 kB
  • ctags: 3,245
  • sloc: cpp: 32,997; python: 7,156; makefile: 70
file content (128 lines) | stat: -rw-r--r-- 4,361 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128

/*
 Copyright (C) 2000, 2001, 2002 RiskMap srl

 This file is part of QuantLib, a free-software/open-source library
 for financial quantitative analysts and developers - http://quantlib.org/

 QuantLib is free software: you can redistribute it and/or modify it under the
 terms of the QuantLib license.  You should have received a copy of the
 license along with this program; if not, please email ferdinando@ametrano.net
 The license is also available online at http://quantlib.org/html/license.html

 This program 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 license for more details.
*/

// $Id: Interpolation.i,v 1.11 2002/03/12 11:25:38 nando Exp $

#ifndef quantlib_interpolation_i
#define quantlib_interpolation_i

%include QLArray.i
%include Matrix.i

%{
using QuantLib::Array;
using QuantLib::Math::Matrix;
using QuantLib::Handle;
using QuantLib::Math::Interpolation;
using QuantLib::Math::LinearInterpolation;
using QuantLib::Math::CubicSpline;
using QuantLib::Math::Interpolation2D;
using QuantLib::Math::BilinearInterpolation;

// safe versions which copy their arguments
// Python users are not supposed to worry about lifetimes
class PyInterpolation {
  public:
    PyInterpolation(const Array& x, const Array& y, bool allowExtrapolation)
    : x_(x), y_(y) {}
    double __call__(double x) { return (*f_)(x); }
  protected:
    Array x_, y_;
    Handle<Interpolation<Array::const_iterator,
                         Array::const_iterator> > f_;
};

class PyInterpolation2D {
  public:
    PyInterpolation2D(const Array& x, const Array& y, const Matrix& m, bool allowExtrapolation)
    : x_(x), y_(y), m_(m) {}
    double __call__(double x, double y) { return (*f_)(x,y); }
  protected:
    Array x_, y_;
    Matrix m_;
    Handle<Interpolation2D<Array::const_iterator,
                           Array::const_iterator,
                           Matrix> > f_;
};


class PyLinearInterpolation : public PyInterpolation {
  public:
    PyLinearInterpolation(const Array& x, const Array& y, bool allowExtrapolation)
    : PyInterpolation(x,y,allowExtrapolation) {
        f_ = Handle<Interpolation<Array::const_iterator,
                                  Array::const_iterator> >(
                new LinearInterpolation<Array::const_iterator,
                                        Array::const_iterator>(
                    x_.begin(),x_.end(),y_.begin(), allowExtrapolation));
    }
};

class PyCubicSpline : public PyInterpolation {
  public:
    PyCubicSpline(const Array& x, const Array& y, bool allowExtrapolation)
    : PyInterpolation(x,y,allowExtrapolation) {
        f_ = Handle<Interpolation<Array::const_iterator,
                                  Array::const_iterator> >(
                new CubicSpline<Array::const_iterator,
                                Array::const_iterator>(
                    x_.begin(),x_.end(),y_.begin(), allowExtrapolation));
    }
};


class PyBilinearInterpolation : public PyInterpolation2D {
  public:
    PyBilinearInterpolation(const Array& x, const Array& y, const Matrix& m, bool allowExtrapolation)
    : PyInterpolation2D(x,y,m,allowExtrapolation) {
        f_ = Handle<Interpolation2D<Array::const_iterator,
                                    Array::const_iterator,
                                    Matrix> >(
                new BilinearInterpolation<Array::const_iterator,
                                          Array::const_iterator,
                                          Matrix>(
                    x_.begin(),x_.end(),y_.begin(),y_.end(),m_, allowExtrapolation));
    }
};
%}

%name(LinearInterpolation)
class PyLinearInterpolation {
  public:
    PyLinearInterpolation(const Array& x, const Array& y, bool allowExtrapolation);
    ~PyLinearInterpolation();
    double __call__(double x);
};

%name(CubicSpline)
class PyCubicSpline {
  public:
    PyCubicSpline(const Array& x, const Array& y, bool allowExtrapolation);
    ~PyCubicSpline();
    double __call__(double x);
};
    
%name(BilinearInterpolation)
class PyBilinearInterpolation {
  public:
    PyBilinearInterpolation(const Array& x, const Array& y, const Matrix& m, bool allowExtrapolation);
    ~PyBilinearInterpolation();
    double __call__(double x, double y);
};


#endif