custom-code/runmean.R at master · indykpol/custom-code · GitHub

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
runmean <- function (x, k,endrule = c("mean", "NA", "trim", "keep", "constant", "func"), align = c("center","left", "right"))
{
	EndRule <- function (x, y, k, dimx, endrule = c("NA", "trim", "keep", "constant", "func"), align = c("center", "left", "right"), Func, ...) {
	align = match.arg(align)
	k = as.integer(k)
	k2 = k%/%2
	if (k2 < 1)
		k2 = 1
	yIsVec = is.null(dimx)
	if (yIsVec)
		dimx = c(length(y), 1)
	dim(x) <- dimx
	dim(y) <- dimx
	n = nrow(x)
	m = ncol(x)
	if (k > n)
		k2 = (n - 1)%/%2
	k1 = k - k2 - 1
	if (align == "center" && k == 2)
	align = "right"
	if (endrule == "trim") {
		y = y[(k1 + 1):(n - k2), ]
	}
	else if (align == "center") {
		idx1 = 1:k1
		idx2 = (n - k2 + 1):n
		if (endrule == "NA") {
			y[idx1, ] = NA
			y[idx2, ] = NA
		}
		else if (endrule == "keep") {
			y[idx1, ] = x[idx1, ]
			y[idx2, ] = x[idx2, ]
		}
		else if (endrule == "constant") {
			y[idx1, ] = y[k1 + 1 + integer(m), ]
			y[idx2, ] = y[n - k2 + integer(m), ]
		}
		else if (endrule == "func" || !yIsVec) {
			for (j in 1:m) {
				for (i in idx1) y[i, j] = Func(x[1:(i + k2),j], ...)
				for (i in idx2) y[i, j] = Func(x[(i - k1):n,j], ...)
			}
		}
	}
	else if (align == "left") {
		y[1:(n - k1), ] = y[(k1 + 1):n, ]
		idx = (n - k + 2):n
		if (endrule == "NA") {
			y[idx, ] = NA
		}
		else if (endrule == "keep") {
			y[idx, ] = x[idx, ]
		}
		else if (endrule == "constant") {
			y[idx, ] = y[n - k + integer(m) + 1, ]
		}
		else {
			for (j in 1:m) for (i in idx) y[i, j] = Func(x[i:n,j], ...)
		}
	}
	else if (align == "right") {
		y[(k2 + 1):n, ] = y[1:(n - k2), ]
		idx = 1:(k - 1)
		if (endrule == "NA") {
			y[idx, ] = NA
		}
		else if (endrule == "keep") {
			y[idx, ] = x[idx, ]
		}
		else if (endrule == "constant") {
			y[idx, ] = y[k + integer(m), ]
		}
		else {
			for (j in 1:m) for (i in idx) y[i, j] = Func(x[1:i,j], ...)
		}
	}
	if (yIsVec)
		y = as.vector(y)
	return(y)
}

	endrule = match.arg(endrule)
	align = match.arg(align)
	dimx = dim(x)
	x = as.vector(x)
	n = length(x)
	if (k <= 1)
		return(x)
	if (k > n)
		k = n
	k2 = k%/%2
	y = double(n)
	k1 = k - k2 - 1
	y = c(sum(x[1:k]), diff(x, k))
	y = cumsum(y)/k
	y = c(rep(0, k1), y, rep(0, k2))
	if (endrule == "mean")
		endrule = "func"
	y = EndRule(x, y, k, dimx, endrule, align, mean, na.rm = TRUE)
	return(y)
}