Gisle Aune
3 years ago
7 changed files with 447 additions and 2 deletions
-
7app/api/util.go
-
36app/services/scene/manager.go
-
87app/services/scene/scene.go
-
53models/device.go
-
7models/errors.go
-
204models/scene.go
-
55models/shared.go
@ -0,0 +1,36 @@ |
|||||
|
package scene |
||||
|
|
||||
|
import ( |
||||
|
"git.aiterp.net/lucifer/new-server/models" |
||||
|
"sync" |
||||
|
) |
||||
|
|
||||
|
type Manager struct { |
||||
|
mu sync.Mutex |
||||
|
assignments map[int]*Scene |
||||
|
scenes []*Scene |
||||
|
} |
||||
|
|
||||
|
func (mgr *Manager) UpdateDevice(device *models.Device) { |
||||
|
|
||||
|
} |
||||
|
|
||||
|
func (mgr *Manager) GetUnassigned(devices []models.Device) []models.Device { |
||||
|
mgr.mu.Lock() |
||||
|
defer mgr.mu.Unlock() |
||||
|
|
||||
|
res := make([]models.Device, 0, len(devices)) |
||||
|
for _, device := range devices { |
||||
|
if _, ok := mgr.assignments[device.ID]; !ok { |
||||
|
res = append(res, device) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
return res |
||||
|
} |
||||
|
|
||||
|
var globalManager = &Manager{} |
||||
|
|
||||
|
func Get() *Manager { |
||||
|
return globalManager |
||||
|
} |
@ -0,0 +1,87 @@ |
|||||
|
package scene |
||||
|
|
||||
|
import ( |
||||
|
"git.aiterp.net/lucifer/new-server/models" |
||||
|
"time" |
||||
|
) |
||||
|
|
||||
|
type Scene struct { |
||||
|
startTime time.Time |
||||
|
duration time.Duration |
||||
|
tag string |
||||
|
data models.Scene |
||||
|
roleList [][]models.Device |
||||
|
deviceMap map[int]*models.Device |
||||
|
roleMap map[int]int |
||||
|
changes map[int]models.NewDeviceState |
||||
|
} |
||||
|
|
||||
|
func (s *Scene) UpdateScene(scene models.Scene) { |
||||
|
s.data = scene |
||||
|
s.roleMap = make(map[int]int) |
||||
|
|
||||
|
for _, device := range s.deviceMap { |
||||
|
s.moveDevice(*device) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
func (s *Scene) UpsertDevice(device models.Device) { |
||||
|
s.deviceMap[device.ID] = &device |
||||
|
s.moveDevice(device) |
||||
|
} |
||||
|
|
||||
|
func (s *Scene) RemoveDevice(device models.Device) { |
||||
|
delete(s.deviceMap, device.ID) |
||||
|
delete(s.roleMap, device.ID) |
||||
|
} |
||||
|
|
||||
|
func (s *Scene) runRole(index int) { |
||||
|
|
||||
|
} |
||||
|
|
||||
|
func (s *Scene) moveDevice(device models.Device, run bool) { |
||||
|
oldRole, hasOldRole := s.roleMap[device.ID] |
||||
|
newRole := s.data.RoleIndex(&device) |
||||
|
|
||||
|
if hasOldRole && oldRole == newRole { |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
if hasOldRole { |
||||
|
for i, device2 := range s.roleList[oldRole] { |
||||
|
if device2.ID == device.ID { |
||||
|
s.roleList[oldRole] = append(s.roleList[oldRole][:i], s.roleList[oldRole][i+1:]...) |
||||
|
break |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
s.runRole(oldRole) |
||||
|
} |
||||
|
|
||||
|
s.roleMap[device.ID] = newRole |
||||
|
s.runRole(newRole) |
||||
|
} |
||||
|
|
||||
|
func (s *Scene) intervalNumber() int64 { |
||||
|
intervalDuration := time.Duration(s.data.IntervalMS) * time.Millisecond |
||||
|
|
||||
|
return int64(time.Since(s.startTime) / intervalDuration) |
||||
|
} |
||||
|
|
||||
|
func (s *Scene) intervalFac() float64 { |
||||
|
return clamp(float64(time.Since(s.startTime)) / float64(s.duration)) |
||||
|
} |
||||
|
|
||||
|
func (s *Scene) positionFac(i, len int) float64 { |
||||
|
return clamp(float64(i) / float64(len)) |
||||
|
} |
||||
|
|
||||
|
func clamp(fac float64) float64 { |
||||
|
if fac > 1.00 { |
||||
|
return 1.00 |
||||
|
} else if fac < 0.00 { |
||||
|
return 0.00 |
||||
|
} else { |
||||
|
return fac |
||||
|
} |
||||
|
} |
@ -0,0 +1,204 @@ |
|||||
|
package models |
||||
|
|
||||
|
import ( |
||||
|
"math" |
||||
|
"math/rand" |
||||
|
"sort" |
||||
|
"strings" |
||||
|
) |
||||
|
|
||||
|
type Scene struct { |
||||
|
ID int `json:"id"` |
||||
|
Name string `json:"name"` |
||||
|
IntervalMS int64 `json:"interval"` |
||||
|
Roles []SceneRole `json:"roles"` |
||||
|
} |
||||
|
|
||||
|
func (s *Scene) Validate() error { |
||||
|
if s.IntervalMS < 0 { |
||||
|
return ErrSceneInvalidInterval |
||||
|
} |
||||
|
if len(s.Roles) == 0 { |
||||
|
return ErrSceneNoRoles |
||||
|
} |
||||
|
|
||||
|
for _, role := range s.Roles { |
||||
|
err := role.Validate() |
||||
|
if err != nil { |
||||
|
return err |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
return nil |
||||
|
} |
||||
|
|
||||
|
func (s *Scene) Role(device *Device) *SceneRole { |
||||
|
index := s.RoleIndex(device) |
||||
|
if index == -1 { |
||||
|
return nil |
||||
|
} |
||||
|
|
||||
|
return &s.Roles[index] |
||||
|
} |
||||
|
|
||||
|
func (s *Scene) RoleIndex(device *Device) int { |
||||
|
for i, role := range s.Roles { |
||||
|
if role.TargetKind.Matches(device, role.TargetValue) { |
||||
|
return i |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
return -1 |
||||
|
} |
||||
|
|
||||
|
type SceneEffect string |
||||
|
|
||||
|
const ( |
||||
|
SEStatic SceneEffect = "Plain" |
||||
|
SERandom SceneEffect = "Random" |
||||
|
SEGradient SceneEffect = "Gradient" |
||||
|
SETransition SceneEffect = "Transition" |
||||
|
) |
||||
|
|
||||
|
type SceneRole struct { |
||||
|
Effect SceneEffect `json:"effect"` |
||||
|
PowerMode ScenePowerMode `json:"overridePower"` |
||||
|
TargetKind ReferenceKind `json:"targetKind"` |
||||
|
TargetValue string `json:"targetValue"` |
||||
|
Interpolate bool `json:"interpolate"` |
||||
|
Order string `json:"order"` |
||||
|
States []NewDeviceState `json:"states"` |
||||
|
} |
||||
|
|
||||
|
func (r *SceneRole) Validate() error { |
||||
|
if len(r.States) == 0 { |
||||
|
return ErrSceneRoleNoStates |
||||
|
} |
||||
|
|
||||
|
switch r.TargetKind { |
||||
|
case RKTag, RKBridgeID, RKDeviceID, RKName, RKAll: |
||||
|
default: |
||||
|
return ErrBadInput |
||||
|
} |
||||
|
|
||||
|
switch r.PowerMode { |
||||
|
case SPScene, SPDevice, SPOverride: |
||||
|
default: |
||||
|
return ErrSceneRoleUnknownPowerMode |
||||
|
} |
||||
|
|
||||
|
switch r.Effect { |
||||
|
case SEStatic, SERandom, SEGradient, SETransition: |
||||
|
default: |
||||
|
return ErrSceneRoleUnknownEffect |
||||
|
} |
||||
|
|
||||
|
switch r.Order { |
||||
|
case "", "-name", "name", "+name", "-id", "id", "+id": |
||||
|
default: |
||||
|
return ErrSceneRoleUnsupportedOrdering |
||||
|
} |
||||
|
|
||||
|
return nil |
||||
|
} |
||||
|
|
||||
|
func (r *SceneRole) ApplyOrder(devices []Device) { |
||||
|
if r.Order == "" { |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
desc := strings.HasPrefix(r.Order, "-") |
||||
|
|
||||
|
orderKey := r.Order |
||||
|
if desc || strings.HasPrefix(r.Order, "+") { |
||||
|
orderKey = orderKey[1:] |
||||
|
} |
||||
|
|
||||
|
switch orderKey { |
||||
|
case "id": |
||||
|
sort.Slice(devices, func(i, j int) bool { |
||||
|
if desc { |
||||
|
return devices[i].ID > devices[j].ID |
||||
|
} |
||||
|
|
||||
|
return devices[i].ID < devices[j].ID |
||||
|
}) |
||||
|
case "name": |
||||
|
sort.Slice(devices, func(i, j int) bool { |
||||
|
if desc { |
||||
|
return strings.Compare(devices[i].Name, devices[j].Name) < 0 |
||||
|
} |
||||
|
|
||||
|
return strings.Compare(devices[i].Name, devices[j].Name) > 0 |
||||
|
}) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
func (r *SceneRole) ApplyEffect(device *Device, intervalFac float64, positionFac float64) (newState NewDeviceState, deviceChanged bool) { |
||||
|
switch r.Effect { |
||||
|
case SEStatic: |
||||
|
newState = r.States[0] |
||||
|
case SERandom: |
||||
|
newState = r.State(rand.Float64()) |
||||
|
case SEGradient: |
||||
|
newState = r.State(positionFac) |
||||
|
case SETransition: |
||||
|
newState = r.State(intervalFac) |
||||
|
} |
||||
|
|
||||
|
switch r.PowerMode { |
||||
|
case SPDevice: |
||||
|
newState.Power = nil |
||||
|
case SPScene: |
||||
|
// Do nothing
|
||||
|
case SPOverride: |
||||
|
if newState.Power != nil && device.State.Power != *newState.Power { |
||||
|
device.State.Power = *newState.Power |
||||
|
deviceChanged = true |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
return |
||||
|
} |
||||
|
|
||||
|
func (r *SceneRole) State(fac float64) NewDeviceState { |
||||
|
if len(r.States) == 0 { |
||||
|
return NewDeviceState{} |
||||
|
} else if len(r.States) == 1 { |
||||
|
return r.States[0] |
||||
|
} |
||||
|
|
||||
|
if r.Interpolate { |
||||
|
if len(r.States) == 2 { |
||||
|
return r.States[0].Interpolate(r.States[1], fac) |
||||
|
} else { |
||||
|
pos := fac * float64(len(r.States)-1) |
||||
|
start := math.Floor(pos) |
||||
|
localFac := pos - start |
||||
|
|
||||
|
return r.States[int(start)].Interpolate(r.States[int(start)+1], localFac) |
||||
|
} |
||||
|
} else { |
||||
|
index := int(fac * float64(len(r.States))) |
||||
|
if index == len(r.States) { |
||||
|
index = len(r.States) - 1 |
||||
|
} |
||||
|
|
||||
|
return r.States[index] |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
type ScenePowerMode string |
||||
|
|
||||
|
const ( |
||||
|
SPDevice ScenePowerMode = "Device" // Device state decides power. Scene state may only power off.
|
||||
|
SPScene ScenePowerMode = "Scene" // Scene state decide power.
|
||||
|
SPOverride ScenePowerMode = "Override" // Same as above, but Scene state set Device state' power. This is good for "wake up" scenes.
|
||||
|
) |
||||
|
|
||||
|
type SceneAssignment struct { |
||||
|
Scene *Scene |
||||
|
Device *Device |
||||
|
Tag string |
||||
|
} |
||||
|
|
Write
Preview
Loading…
Cancel
Save
Reference in new issue