File: multivariateaccumulator.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 (122 lines) | stat: -rw-r--r-- 4,082 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


/*
 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.

*/
/*! \file multivariateaccumulator.hpp
    \brief A simple accumulator for vector-type samples

    \fullpath
    ql/Math/%multivariateaccumulator.hpp
*/

// $Id: multivariateaccumulator.hpp,v 1.11 2002/01/17 12:09:35 aleppo Exp $

#ifndef quantlib_math_multivariate_accumulator_h
#define quantlib_math_multivariate_accumulator_h

#include <ql/null.hpp>
#include <ql/dataformatters.hpp>
#include <ql/Math/matrix.hpp>
#include <vector>

namespace QuantLib {

    namespace Math {

        //! A sample accumulator for multivariate analysis
        /*! MultivariateAccumulator can accumulate vector-type samples and
            return the average vector, both in Array form and
            std::vector<double> form, and the covariance matrix
        */
        class MultivariateAccumulator {
          public:
            MultivariateAccumulator();
            MultivariateAccumulator(Size size);
            //! \name Inspectors
            //@{
            //! size of each sample
            Size size() const;
            //! number of samples collected
            Size samples() const;
            //! sum of data weights
            double weightSum() const;
            //! returns the mean as an Array
            Array mean() const;
            //! returns the mean as a std::vector<double>
            std::vector<double> meanVector() const;
            //! returns the covariance Matrix
            Matrix covariance() const;
            //! returns the correlation Matrix
            Matrix correlation() const;

            //@}

            //! \name Modifiers
            //@{
            //! adds an Array to the collection, possibly with a weight
            void add(const Array &arr, double weight = 1.0);
            //! adds a vector<double> to the collection, possibly with a weight
            void add(const std::vector<double> &vec, double weight = 1.0);
            //! adds a sequence of data to the collection
            template <class DataIterator>
            void addSequence(DataIterator begin, DataIterator end) {
                for (;begin!=end;++begin)
                    add(*begin);
            }
            //! adds a sequence of data to the collection, each with its weight
            template <class DataIterator, class WeightIterator>
            void addSequence(DataIterator begin, DataIterator end,
                WeightIterator wbegin) {
                    for(;begin!=end;++begin,++wbegin)
                        add(*begin, *wbegin);
            }
            //! resets the data to a null set
            void reset();
            //@}
          private:
            Size size_;
            Size sampleNumber_;
            double sampleWeight_;
            Array sum_;
            Matrix quadraticSum_;
        };


        // inline definitions
        inline Size MultivariateAccumulator::size() const {
            return size_;
        }

        inline Size MultivariateAccumulator::samples() const {
            return sampleNumber_;
        }

        inline double MultivariateAccumulator::weightSum() const {
            return sampleWeight_;
        }

        inline Array MultivariateAccumulator::mean() const {
            QL_REQUIRE(sampleWeight_ > 0.0,
                "Stat::mean() : sampleWeight_=0, unsufficient");
            return sum_/sampleWeight_;
        }

    }

}

#endif