CDPP / AMDA_Kernel

Blame view

src/ExternLib/StatisticFunctions/VarFunc.hh 2.34 KB
Edit Raw Normal View History
74a11471   Benjamin Renard   Add min, max, var... 
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
 
/*
 * VarFunc.hh
 *
 *  Created on: Jun 22, 2018
 *      Author: benjamin
 */

#ifndef VARFUNC_HH_
#define VARFUNC_HH_

#include "AbstractFunc.hh"
#include "Toolbox.hh"

namespace AMDA {
namespace Parameters {
namespace StatisticFunctions {

template <typename InputElemType, typename OutputElemType>
OutputElemType computeVar(std::list<std::pair<double,InputElemType>>& mem, OutputElemType& nanVal) {
	OutputElemType result = nanVal;
	if (mem.empty()) {
		return result;
	}
	result << ElemNull();
	OutputElemType sum = result;
	int n = 0;
	for (auto val : mem) {
		if (isNAN(val.second))
			continue;
		sum = sum+val.second;
		result = result+StatisticFunctions::square(val.second);
		++n;
	}
	if (n == 0) {
		result << NotANumber();
		return result;
	}
	result = 1./n*result;
	OutputElemType mean = sum/n;
	result = result - StatisticFunctions::square(mean);
	return result;
}

/**
 * @class VarFunc
 * @brief 
 * @details This class implements AbstractFunc.
 */
template <typename InputElemType, typename OutputElemType> 
class VarFunc : public ClassicAbstractFunc<InputElemType,OutputElemType> {
public:
	/**
	 * @brief Constructor.
	 */
	VarFunc(Process& pProcess, TimeIntervalListSPtr pTimeIntervalList, ParamDataSpec<InputElemType>& paramInput, double windowtime)
	: ClassicAbstractFunc<InputElemType,OutputElemType>(pProcess, pTimeIntervalList, paramInput, windowtime) {
	}

	virtual ~VarFunc() {
	}

	OutputElemType compute() {
		return computeVar(ClassicAbstractFunc<InputElemType,OutputElemType>::_mem, ClassicAbstractFunc<InputElemType,OutputElemType>::_nanVal);
	}
};  

/**
 * @class VarFuncSm
 * @brief
 * @details This class implements AbstractFunc.
 */
template <typename InputElemType, typename OutputElemType>
class VarSmFunc : public SmAbstractFunc<InputElemType,OutputElemType> {
public:
	/**
	 * @brief Constructor.
	 */
	VarSmFunc(Process& pProcess, TimeIntervalListSPtr pTimeIntervalList, ParamDataSpec<InputElemType>& paramInput, double windowtime)
	: SmAbstractFunc<InputElemType,OutputElemType>(pProcess, pTimeIntervalList, paramInput, windowtime) {
        }

	virtual ~VarSmFunc() {
	}

	OutputElemType compute() {
		return computeVar(SmAbstractFunc<InputElemType,OutputElemType>::_mem, SmAbstractFunc<InputElemType,OutputElemType>::_nanVal);
        }
};


} /* namespace StatisticFunctions */
} /* namespace Parameters */
} /* namespace AMDA */
#endif /* MINFUNC_HH_ */