-
Notifications
You must be signed in to change notification settings - Fork 0
/
register.go
58 lines (50 loc) · 1.31 KB
/
register.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
package victronDevice
import (
"github.com/koestler/go-iotdevice/v3/dataflow"
"github.com/koestler/go-victron/veconst"
"github.com/koestler/go-victron/veregister"
)
// Register is used as a wrapper for veregister.Register to implement dataflow.Register
type Register struct {
veregister.Register
}
func (r Register) RegisterType() dataflow.RegisterType {
switch r.Type() {
case veregister.Number:
return dataflow.NumberRegister
case veregister.Text:
return dataflow.TextRegister
case veregister.Enum:
return dataflow.EnumRegister
case veregister.FieldList:
return dataflow.TextRegister
default:
return dataflow.UndefinedRegister
}
}
type enumFactory interface {
Factory() veconst.EnumFactory
}
func (r Register) Enum() map[int]string {
if ef, ok := r.Register.(enumFactory); ok {
return ef.Factory().IntToStringMap()
}
return nil
}
type uniter interface {
Unit() string
}
func (r Register) Unit() string {
if nr, ok := r.Register.(uniter); ok {
return nr.Unit()
}
return ""
}
func addToRegisterDb(rdb *dataflow.RegisterDb, rl veregister.RegisterList) {
registers := rl.GetRegisters()
dataflowRegisters := make([]dataflow.RegisterStruct, len(registers))
for i, r := range registers {
dataflowRegisters[i] = dataflow.NewRegisterStructByInterface(Register{r})
}
rdb.AddStruct(dataflowRegisters...)
}