-
Notifications
You must be signed in to change notification settings - Fork 0
/
excitation.go
133 lines (117 loc) · 3.94 KB
/
excitation.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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
package engine
import (
"github.com/mumax/3/cuda"
"github.com/mumax/3/data"
"github.com/mumax/3/script"
"github.com/mumax/3/util"
"math"
"reflect"
)
// An excitation, typically field or current,
// can be defined region-wise plus extra mask*multiplier terms.
type excitation struct {
name string
perRegion VectorParam // Region-based excitation
extraTerms []mulmask // add extra mask*multiplier terms
}
// space-dependent mask plus time dependent multiplier
type mulmask struct {
mul func() float64
mask *data.Slice
}
func (e *excitation) init(name, unit, desc string) {
e.name = name
e.perRegion.init("_"+name+"_perRegion", unit, "(internal)") // name starts with underscore: unexported
DeclLValue(name, e, cat(desc, unit))
}
func (e *excitation) AddTo(dst *data.Slice) {
if !e.perRegion.isZero() {
cuda.RegionAddV(dst, e.perRegion.gpuLUT(), regions.Gpu())
}
for _, t := range e.extraTerms {
var mul float32 = 1
if t.mul != nil {
mul = float32(t.mul())
}
cuda.Madd2(dst, dst, t.mask, 1, mul)
}
}
func (e *excitation) isZero() bool {
return e.perRegion.isZero() && len(e.extraTerms) == 0
}
func (e *excitation) Slice() (*data.Slice, bool) {
buf := cuda.Buffer(e.NComp(), e.Mesh().Size())
cuda.Zero(buf)
e.AddTo(buf)
return buf, true
}
// After resizing the mesh, the extra terms don't fit the grid anymore
// and there is no reasonable way to resize them. So remove them and have
// the user re-add them.
func (e *excitation) RemoveExtraTerms() {
if len(e.extraTerms) == 0 {
return
}
LogOut("REMOVING EXTRA TERMS FROM", e.Name())
for _, m := range e.extraTerms {
m.mask.Free()
}
e.extraTerms = nil
}
// Add an extra mask*multiplier term to the excitation.
func (e *excitation) Add(mask *data.Slice, f script.ScalarFunction) {
var mul func() float64
if f != nil {
if Const(f) {
val := f.Float()
mul = func() float64 {
return val
}
} else {
mul = func() float64 {
return f.Float()
}
}
}
e.AddGo(mask, mul)
}
// An Add(mask, f) equivalent for Go use
func (e *excitation) AddGo(mask *data.Slice, mul func() float64) {
if mask != nil {
checkNaN(mask, e.Name()+".add()") // TODO: in more places
mask = data.Resample(mask, e.Mesh().Size())
mask = assureGPU(mask)
}
e.extraTerms = append(e.extraTerms, mulmask{mul, mask})
}
func (e *excitation) SetRegion(region int, f script.VectorFunction) { e.perRegion.SetRegion(region, f) }
func (e *excitation) SetValue(v interface{}) { e.perRegion.SetValue(v) }
func (e *excitation) Set(v data.Vector) { e.perRegion.setRegions(0, NREGION, slice(v)) }
func (e *excitation) getRegion(region int) []float64 { return e.perRegion.getRegion(region) } // for gui
func (e *excitation) SetRegionFn(region int, f func() [3]float64) {
e.perRegion.setFunc(region, region+1, func() []float64 {
return slice(f())
})
}
func (e *excitation) average() []float64 { return qAverageUniverse(e) }
func (e *excitation) Average() data.Vector { return unslice(qAverageUniverse(e)) }
func (e *excitation) IsUniform() bool { return e.perRegion.IsUniform() }
func (e *excitation) Name() string { return e.name }
func (e *excitation) Unit() string { return e.perRegion.Unit() }
func (e *excitation) NComp() int { return e.perRegion.NComp() }
func (e *excitation) Mesh() *data.Mesh { return Mesh() }
func (e *excitation) Region(r int) *vOneReg { return vOneRegion(e, r) }
func (e *excitation) Comp(c int) *comp { return Comp(e, c) }
func (e *excitation) Eval() interface{} { return e }
func (e *excitation) Type() reflect.Type { return reflect.TypeOf(new(excitation)) }
func (e *excitation) InputType() reflect.Type { return script.VectorFunction_t }
func checkNaN(s *data.Slice, name string) {
h := s.Host()
for _, h := range h {
for _, v := range h {
if math.IsNaN(float64(v)) || math.IsInf(float64(v), 0) {
util.Fatal("NaN or Inf in", name)
}
}
}
}