Chris@16
|
1 ///////////////////////////////////////////////////////////////////////////////
|
Chris@16
|
2 // weighted_sum_kahan.hpp
|
Chris@16
|
3 //
|
Chris@16
|
4 // Copyright 2011 Simon West. Distributed under the Boost
|
Chris@16
|
5 // Software License, Version 1.0. (See accompanying file
|
Chris@16
|
6 // LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
|
Chris@16
|
7
|
Chris@16
|
8 #ifndef BOOST_ACCUMULATORS_STATISTICS_WEIGHTED_SUM_KAHAN_HPP_EAN_11_05_2011
|
Chris@16
|
9 #define BOOST_ACCUMULATORS_STATISTICS_WEIGHTED_SUM_KAHAN_HPP_EAN_11_05_2011
|
Chris@16
|
10
|
Chris@16
|
11 #include <boost/mpl/placeholders.hpp>
|
Chris@16
|
12 #include <boost/accumulators/framework/accumulator_base.hpp>
|
Chris@16
|
13 #include <boost/accumulators/framework/extractor.hpp>
|
Chris@16
|
14 #include <boost/accumulators/numeric/functional.hpp>
|
Chris@16
|
15 #include <boost/accumulators/framework/parameters/sample.hpp>
|
Chris@16
|
16 #include <boost/accumulators/framework/parameters/weight.hpp>
|
Chris@16
|
17 #include <boost/accumulators/framework/accumulators/external_accumulator.hpp>
|
Chris@16
|
18 #include <boost/accumulators/framework/depends_on.hpp>
|
Chris@16
|
19 #include <boost/accumulators/statistics_fwd.hpp>
|
Chris@16
|
20 #include <boost/accumulators/statistics/weighted_sum.hpp>
|
Chris@16
|
21 #include <boost/numeric/conversion/cast.hpp>
|
Chris@16
|
22
|
Chris@16
|
23 namespace boost { namespace accumulators
|
Chris@16
|
24 {
|
Chris@16
|
25
|
Chris@16
|
26 namespace impl
|
Chris@16
|
27 {
|
Chris@16
|
28 #if _MSC_VER > 1400
|
Chris@16
|
29 # pragma float_control(push)
|
Chris@16
|
30 # pragma float_control(precise, on)
|
Chris@16
|
31 #endif
|
Chris@16
|
32
|
Chris@16
|
33 ///////////////////////////////////////////////////////////////////////////////
|
Chris@16
|
34 // weighted_sum_kahan_impl
|
Chris@16
|
35 template<typename Sample, typename Weight, typename Tag>
|
Chris@16
|
36 struct weighted_sum_kahan_impl
|
Chris@16
|
37 : accumulator_base
|
Chris@16
|
38 {
|
Chris@16
|
39 typedef typename numeric::functional::multiplies<Sample, Weight>::result_type weighted_sample;
|
Chris@16
|
40
|
Chris@16
|
41 // for boost::result_of
|
Chris@16
|
42 typedef weighted_sample result_type;
|
Chris@16
|
43
|
Chris@16
|
44 template<typename Args>
|
Chris@16
|
45 weighted_sum_kahan_impl(Args const &args)
|
Chris@16
|
46 : weighted_sum_(
|
Chris@16
|
47 args[parameter::keyword<Tag>::get() | Sample()] * numeric::one<Weight>::value),
|
Chris@16
|
48 compensation(boost::numeric_cast<weighted_sample>(0.0))
|
Chris@16
|
49 {
|
Chris@16
|
50 }
|
Chris@16
|
51
|
Chris@16
|
52 template<typename Args>
|
Chris@16
|
53 void
|
Chris@16
|
54 #if BOOST_ACCUMULATORS_GCC_VERSION > 40305
|
Chris@16
|
55 __attribute__((__optimize__("no-associative-math")))
|
Chris@16
|
56 #endif
|
Chris@16
|
57 operator ()(Args const &args)
|
Chris@16
|
58 {
|
Chris@16
|
59 const weighted_sample myTmp1 = args[parameter::keyword<Tag>::get()] * args[weight] - this->compensation;
|
Chris@16
|
60 const weighted_sample myTmp2 = this->weighted_sum_ + myTmp1;
|
Chris@16
|
61 this->compensation = (myTmp2 - this->weighted_sum_) - myTmp1;
|
Chris@16
|
62 this->weighted_sum_ = myTmp2;
|
Chris@16
|
63
|
Chris@16
|
64 }
|
Chris@16
|
65
|
Chris@16
|
66 result_type result(dont_care) const
|
Chris@16
|
67 {
|
Chris@16
|
68 return this->weighted_sum_;
|
Chris@16
|
69 }
|
Chris@16
|
70
|
Chris@16
|
71 private:
|
Chris@16
|
72 weighted_sample weighted_sum_;
|
Chris@16
|
73 weighted_sample compensation;
|
Chris@16
|
74 };
|
Chris@16
|
75
|
Chris@16
|
76 #if _MSC_VER > 1400
|
Chris@16
|
77 # pragma float_control(pop)
|
Chris@16
|
78 #endif
|
Chris@16
|
79
|
Chris@16
|
80 } // namespace impl
|
Chris@16
|
81
|
Chris@16
|
82 ///////////////////////////////////////////////////////////////////////////////
|
Chris@16
|
83 // tag::weighted_sum_kahan
|
Chris@16
|
84 // tag::weighted_sum_of_variates_kahan
|
Chris@16
|
85 //
|
Chris@16
|
86 namespace tag
|
Chris@16
|
87 {
|
Chris@16
|
88 struct weighted_sum_kahan
|
Chris@16
|
89 : depends_on<>
|
Chris@16
|
90 {
|
Chris@16
|
91 /// INTERNAL ONLY
|
Chris@16
|
92 ///
|
Chris@16
|
93 typedef accumulators::impl::weighted_sum_kahan_impl<mpl::_1, mpl::_2, tag::sample> impl;
|
Chris@16
|
94 };
|
Chris@16
|
95
|
Chris@16
|
96 template<typename VariateType, typename VariateTag>
|
Chris@16
|
97 struct weighted_sum_of_variates_kahan
|
Chris@16
|
98 : depends_on<>
|
Chris@16
|
99 {
|
Chris@16
|
100 /// INTERNAL ONLY
|
Chris@16
|
101 ///
|
Chris@16
|
102 typedef accumulators::impl::weighted_sum_kahan_impl<VariateType, mpl::_2, VariateTag> impl;
|
Chris@16
|
103 };
|
Chris@16
|
104
|
Chris@16
|
105 }
|
Chris@16
|
106
|
Chris@16
|
107 ///////////////////////////////////////////////////////////////////////////////
|
Chris@16
|
108 // extract::weighted_sum_kahan
|
Chris@16
|
109 // extract::weighted_sum_of_variates_kahan
|
Chris@16
|
110 //
|
Chris@16
|
111 namespace extract
|
Chris@16
|
112 {
|
Chris@16
|
113 extractor<tag::weighted_sum_kahan> const weighted_sum_kahan = {};
|
Chris@16
|
114 extractor<tag::abstract_weighted_sum_of_variates> const weighted_sum_of_variates_kahan = {};
|
Chris@16
|
115
|
Chris@16
|
116 BOOST_ACCUMULATORS_IGNORE_GLOBAL(weighted_sum_kahan)
|
Chris@16
|
117 BOOST_ACCUMULATORS_IGNORE_GLOBAL(weighted_sum_of_variates_kahan)
|
Chris@16
|
118 }
|
Chris@16
|
119
|
Chris@16
|
120 using extract::weighted_sum_kahan;
|
Chris@16
|
121 using extract::weighted_sum_of_variates_kahan;
|
Chris@16
|
122
|
Chris@16
|
123 // weighted_sum(kahan) -> weighted_sum_kahan
|
Chris@16
|
124 template<>
|
Chris@16
|
125 struct as_feature<tag::weighted_sum(kahan)>
|
Chris@16
|
126 {
|
Chris@16
|
127 typedef tag::weighted_sum_kahan type;
|
Chris@16
|
128 };
|
Chris@16
|
129
|
Chris@16
|
130 template<typename VariateType, typename VariateTag>
|
Chris@16
|
131 struct feature_of<tag::weighted_sum_of_variates_kahan<VariateType, VariateTag> >
|
Chris@16
|
132 : feature_of<tag::abstract_weighted_sum_of_variates>
|
Chris@16
|
133 {
|
Chris@16
|
134 };
|
Chris@16
|
135
|
Chris@16
|
136 }} // namespace boost::accumulators
|
Chris@16
|
137
|
Chris@16
|
138 #endif
|