aboutsummaryrefslogtreecommitdiffstats
path: root/layout/split.go
blob: fb6d36ee5a264bba4a833c2c8a4e8853578ad664 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
package layout

import "fmt"

// SplitFunc represents a way to split a space among a number of elements.
// The length of the returned slice must be equal to the number of elements.
// The sum of all elements of the returned slice must be eqal to the space.
type SplitFunc func(elements int, space int) []int

// EvenSplit is a SplitFunc used to split a space (almost) evenly among the elements.
// It is almost evenly because width may not be divisible by elements.
func EvenSplit(elements int, width int) []int {
	if elements <= 0 {
		panic(fmt.Errorf("EvenSplit: elements must be greater than 0"))
	}
	ret := make([]int, 0, elements)
	for elements > 0 {
		v := width / elements
		width -= v
		elements -= 1
		ret = append(ret, v)
	}
	return ret
}