File: grid.hpp

package info (click to toggle)
quantlib 0.2.1.cvs20020322-1
  • links: PTS
  • area: main
  • in suites: woody
  • size: 4,716 kB
  • ctags: 4,614
  • sloc: cpp: 19,601; sh: 7,389; makefile: 796; ansic: 22
file content (108 lines) | stat: -rw-r--r-- 3,363 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
/*
 Copyright (C) 2001, 2002 Sadruddin Rejeb

 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.
*/
/*! \file grid.hpp
    \brief Grid classes with useful constructors for trees and finite diffs

    \fullpath
    ql/%grid.hpp
*/

// $Id: grid.hpp,v 1.1 2002/03/05 01:10:36 sadrejeb Exp $

#ifndef quantlib_time_grid_h
#define quantlib_time_grid_h

#include <ql/array.hpp>
#include <ql/errors.hpp>
#include <ql/qldefines.hpp>
#include <ql/types.hpp>

#include <list>
#include <numeric>
#include <vector>

namespace QuantLib {

    //! spatial grid class
    class Grid : public Array {
      public:
        Grid(double center, double dx, Size steps)
        : Array(steps) {
            for (Size i=0; i<steps; i++)
                (*this)[i] = center + (i - steps/2.0)*dx;
        }
    };

    //! time grid class
    #if defined(QL_PATCH_MICROSOFT_BUGS)
    class TimeGrid : public std::vector<Time> {
      public:
    #else
    class TimeGrid : private std::vector<Time> {
      public:
        using std::vector<Time>::operator[];
        using std::vector<Time>::size;
        using std::vector<Time>::begin;
        using std::vector<Time>::end;
        using std::vector<Time>::const_iterator;
    #endif
        TimeGrid() {}
        TimeGrid(const std::list<Time>& times, Size steps)
        : std::vector<Time>(0) {
            Time last = times.back();
            Time dtMax;
            if (steps == 0) {
                std::vector<Time> diff;
                std::back_insert_iterator<std::vector<Time> > ii(diff);
                std::adjacent_difference(times.begin(), times.end(), ii);
                if (diff.front()==0.0)
                    diff.erase(diff.begin());
                dtMax = *(std::min_element(diff.begin(), diff.end()));
            } else {
                dtMax = last/steps;
            }

//            std::cout << "Building tree: dt set to " << dtMax << std::endl;

            Time begin = 0.0;
            std::list<Time>::const_iterator t;
            for (t = times.begin(); t != times.end(); t++) {
                Time end = *t;
                if (begin == end) continue;
                Size nSteps = (Size)((end - begin)/dtMax + 1.0);
                double dt = (end - begin)/nSteps;
                for (Size n=0; n<nSteps; n++)
                    push_back(begin + n*dt);
                begin = end;
            }
            push_back(begin);
        }

        Size findIndex(Time t) const {
            const_iterator result = std::find(begin(), end(), t);
            QL_REQUIRE(result!=end(), "Using inadequate tree");
            return result - begin();
        }

        Time dt(Size i) const {
            return (*this)[i+1]  - (*this)[i];
        }

    };

}

#endif