forked from cpmech/gosl
/
deepslices.go
56 lines (51 loc) · 1.32 KB
/
deepslices.go
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
// Copyright 2016 The Gosl Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package utl
// Deep3alloc allocates a slice of slice of slice
func Deep3alloc(n1, n2, n3 int) (a [][][]float64) {
a = make([][][]float64, n1)
for i := 0; i < n1; i++ {
a[i] = make([][]float64, n2)
for j := 0; j < n2; j++ {
a[i][j] = make([]float64, n3)
}
}
return
}
// Deep4alloc allocates a slice of slice of slice of slice
func Deep4alloc(n1, n2, n3, n4 int) (a [][][][]float64) {
a = make([][][][]float64, n1)
for i := 0; i < n1; i++ {
a[i] = make([][][]float64, n2)
for j := 0; j < n2; j++ {
a[i][j] = make([][]float64, n3)
for k := 0; k < n3; k++ {
a[i][j][k] = make([]float64, n4)
}
}
}
return
}
// Deep3set sets deep slice of slice of slice with v values
func Deep3set(a [][][]float64, v float64) {
for i := 0; i < len(a); i++ {
for j := 0; j < len(a[i]); j++ {
for k := 0; k < len(a[i][j]); k++ {
a[i][j][k] = v
}
}
}
}
// Deep4set sets deep slice of slice of slice of slice with v values
func Deep4set(a [][][][]float64, v float64) {
for i := 0; i < len(a); i++ {
for j := 0; j < len(a[i]); j++ {
for k := 0; k < len(a[i][j]); k++ {
for l := 0; l < len(a[i][j][k]); l++ {
a[i][j][k][l] = v
}
}
}
}
}