WSL/SLF GitLab Repository

FilterMedianAvg.cc 3.43 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
/***********************************************************************************/
/*  Copyright 2009 WSL Institute for Snow and Avalanche Research    SLF-DAVOS      */
/***********************************************************************************/
/* This file is part of MeteoIO.
    MeteoIO is free software: you can redistribute it and/or modify
    it under the terms of the GNU Lesser General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    MeteoIO 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 Lesser General Public License for more details.

    You should have received a copy of the GNU Lesser General Public License
    along with MeteoIO.  If not, see <http://www.gnu.org/licenses/>.
*/
18
#include <meteoio/meteofilters/FilterMedianAvg.h>
19
#include <cmath>
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35

using namespace std;

namespace mio {

FilterMedianAvg::FilterMedianAvg(const std::vector<std::string>& vec_args) : WindowedFilter("MEAN_AVG")
{
	parse_args(vec_args);

	//This is safe, but maybe too imprecise:
	properties.time_before = min_time_span;
	properties.time_after  = min_time_span;
	properties.points_before = min_data_points;
	properties.points_after = min_data_points;
}

36
void FilterMedianAvg::process(const unsigned int& param, const std::vector<MeteoData>& ivec,
37
                              std::vector<MeteoData>& ovec)
38
{
39
40
	ovec = ivec;
	for (size_t ii=0; ii<ovec.size(); ii++){ //for every element in ivec, get a window
41
		double& value = ovec[ii](param);
42

43
		size_t start, end;
44
45
		if( get_window_specs(ii, ivec, start, end) ) {
			value = calc_median(ivec, param, start, end);
46
		} else if(!is_soft) value = IOUtils::nodata;
47
48
49
	}
}

50
double FilterMedianAvg::calc_median(const std::vector<MeteoData>& ivec, const unsigned int& param, const size_t& start, const size_t& end)
51
52
{
	vector<double> vecTemp;
53
	vecTemp.reserve( end-start+1 );
54
55
	for(size_t ii=start; ii<=end; ii++){ //get rid of nodata elements
		const double& value = ivec[ii](param);
56
57
58
59
		if (value != IOUtils::nodata)
			vecTemp.push_back(value);
	}

60
61
	const size_t vecSize = vecTemp.size();
	if (vecSize == 0)
62
		return IOUtils::nodata;
63

64
65
66
	if ((vecSize % 2) == 1){ //uneven
		const int middle = (int)(vecSize/2);
		nth_element(vecTemp.begin(), vecTemp.begin()+middle, vecTemp.end());
67
68
		return *(vecTemp.begin()+middle);
	} else { //use arithmetic mean of element n/2 and n/2-1
69
70
71
72
73
74
		const int middle = (int)(vecSize/2);
		nth_element(vecTemp.begin(), vecTemp.begin()+middle-1, vecTemp.end());
		const double m1 = *(vecTemp.begin()+middle-1);
		nth_element(vecTemp.begin(), vecTemp.begin()+middle, vecTemp.end());
		const double m2 = *(vecTemp.begin()+middle);
		return Interpol1D::weightedMean( m1, m2, 0.5);
75
	}
76
}
77
78
79
80
81
82

void FilterMedianAvg::parse_args(std::vector<std::string> vec_args)
{
	vector<double> filter_args;

	if (vec_args.size() > 2){
83
		is_soft = ProcessingBlock::is_soft(vec_args);
84
85
86
87
	}

	if (vec_args.size() > 2)
		centering = (WindowedFilter::Centering)WindowedFilter::get_centering(vec_args);
88

89
	convert_args(2, 2, vec_args, filter_args);
90
91

	if ((filter_args[0] < 1) || (filter_args[1] < 0)){
92
		throw InvalidArgumentException("Invalid window size configuration for filter " + getName(), AT);
93
94
	}

95
	min_data_points = (unsigned int)floor(filter_args[0]);
96
	min_time_span = Duration(filter_args[1] / 86400.0, 0.);
97
}
98

99
} //namespace