summaryrefslogtreecommitdiff
path: root/saturating/saturating.go
blob: f2656b07a59e7b61c60ec7d46b5b72b7a91423be (plain)
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
/*
 * This file is part of Go Responsiveness.
 *
 * Go Responsiveness 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.
 * Go Responsiveness 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 Foobar. If not, see <https://www.gnu.org/licenses/>.
 */

package saturating

type SaturatingInt struct {
	max       int
	value     int
	saturated bool
}

func NewSaturatingInt(max int) *SaturatingInt {
	return &SaturatingInt{max: max, value: 0, saturated: false}
}

func (s *SaturatingInt) Value() int {
	if s.saturated {
		return s.max
	}
	return s.value
}

func (s *SaturatingInt) Add(operand int) {
	if !s.saturated {
		s.value += operand
		if s.value >= s.max {
			s.saturated = true
		}
	}
}