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
|
/*Copyright (C) 2015 Olivier Delaneau, Halit Ongen, Emmanouil T. Dermitzakis
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
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
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.*/
#ifndef _BASIC_STATS_H
#define _BASIC_STATS_H
#include <vector>
class basic_stats {
protected:
uint32_t m_n;
double m_oldM;
double m_newM;
double m_oldS;
double m_newS;
public:
basic_stats() {
m_n = 0;
m_oldM = 0;
m_newM = 0;
m_oldS = 0;
m_newS = 0;
}
template <class T>
basic_stats(vector < T > & X) {
m_n = 0;
m_oldM = 0;
m_newM = 0;
m_oldS = 0;
m_newS = 0;
for (uint32_t e = 0 ; e < X.size() ; e ++) push(X[e]);
}
void clear() {
m_n = 0;
m_oldM = 0;
m_newM = 0;
m_oldS = 0;
m_newS = 0;
}
template <class T>
void push(T x) {
m_n++;
if (m_n == 1) {
m_oldM = m_newM = x;
m_oldS = 0.0;
} else {
m_newM = m_oldM + (x - m_oldM)/m_n;
m_newS = m_oldS + (x - m_oldM)*(x - m_newM);
m_oldM = m_newM;
m_oldS = m_newS;
}
}
int size() const {
return m_n;
}
double mean() const {
return (m_n > 0) ? m_newM : 0.0;
}
double variance() const {
return ( (m_n > 1) ? m_newS/(m_n - 1) : 0.0 );
}
double sd() const {
return sqrt( variance() );
}
};
#endif
|