Commit b89b6c9ce4c35decc74d18628a5def8299d6c41f - golang-github-vbauerster-mpb

add count to medianEwma Vladimir Bauer 7 years ago

1 changed file(s) with 8 addition(s) and 4 deletion(s). Raw diff Collapse all Expand all

-4

decor/moving-average.go less more

50	50	}
51	51
52	52	type medianEwma struct {
	53	count uint
	54	median MovingAverage
53	55	MovingAverage
54		median MovingAverage
55	56	}
56	57
57		func (s medianEwma) Add(v float64) {
	58	func (s *medianEwma) Add(v float64) {
58	59	s.median.Add(v)
59		s.MovingAverage.Add(s.median.Value())
	60	s.count++
	61	if s.count >= 3 {
	62	s.MovingAverage.Add(s.median.Value())
	63	}
60	64	}
61	65
62	66	// NewMedianEwma is ewma based MovingAverage, which gets its values from median MovingAverage.
63	67	func NewMedianEwma(age ...float64) MovingAverage {
64		return medianEwma{
	68	return &medianEwma{
65	69	MovingAverage: ewma.NewMovingAverage(age...),
66	70	median: NewMedian(),
67	71	}