/
oneregion.go
78 lines (62 loc) · 1.87 KB
/
oneregion.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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
package engine
import (
"fmt"
"github.com/mumax/3/cuda"
"github.com/mumax/3/data"
"github.com/mumax/3/util"
)
func sInRegion(q Quantity, r int) ScalarField {
return AsScalarField(inRegion(q, r))
}
func vInRegion(q Quantity, r int) VectorField {
return AsVectorField(inRegion(q, r))
}
func sOneRegion(q Quantity, r int) *sOneReg {
util.Argument(q.NComp() == 1)
return &sOneReg{oneReg{q, r}}
}
func vOneRegion(q Quantity, r int) *vOneReg {
util.Argument(q.NComp() == 3)
return &vOneReg{oneReg{q, r}}
}
type sOneReg struct{ oneReg }
func (q *sOneReg) Average() float64 { return q.average()[0] }
type vOneReg struct{ oneReg }
func (q *vOneReg) Average() data.Vector { return unslice(q.average()) }
// represents a new quantity equal to q in the given region, 0 outside.
type oneReg struct {
parent Quantity
region int
}
func inRegion(q Quantity, region int) Quantity {
return &oneReg{q, region}
}
func (q *oneReg) NComp() int { return q.parent.NComp() }
func (q *oneReg) Name() string { return fmt.Sprint(NameOf(q.parent), ".region", q.region) }
func (q *oneReg) Unit() string { return UnitOf(q.parent) }
func (q *oneReg) Mesh() *data.Mesh { return MeshOf(q.parent) }
func (q *oneReg) EvalTo(dst *data.Slice) { EvalTo(q, dst) }
// returns a new slice equal to q in the given region, 0 outside.
func (q *oneReg) Slice() (*data.Slice, bool) {
src := ValueOf(q.parent)
defer cuda.Recycle(src)
out := cuda.Buffer(q.NComp(), q.Mesh().Size())
cuda.RegionSelect(out, src, regions.Gpu(), byte(q.region))
return out, true
}
func (q *oneReg) average() []float64 {
slice, r := q.Slice()
if r {
defer cuda.Recycle(slice)
}
avg := sAverageUniverse(slice)
sDiv(avg, regions.volume(q.region))
return avg
}
func (q *oneReg) Average() []float64 { return q.average() }
// slice division
func sDiv(v []float64, x float64) {
for i := range v {
v[i] /= x
}
}