/
main.go
77 lines (64 loc) · 1.5 KB
/
main.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
package main
import (
"image/color"
"log"
"time"
"dev.acmcsuf.com/christmas/lib/intmath"
"github.com/Jon-Bright/ledctl/pixarray"
)
const wormSpeed = 200 * time.Millisecond
var ledOrder = []int{
0, 1, 49, 2, 3, 4, 48, 46, 43, 5, 47, 44, 45, 6, 42, 7, 41, 8, 9, 40, 39,
10, 38, 11, 12, 37, 36, 35, 13, 14, 34, 15, 33, 16, 32, 31, 17, 18, 30, 29,
19, 20, 21, 28, 27, 22, 23, 26, 25, 24,
}
var maxLEDHeight = (func() int {
var max int
for _, i := range ledOrder {
max = intmath.Max(max, i)
}
return max
})()
func main() {
strip, err := pixarray.NewWS281x(
len(ledOrder), // 50 LEDs
3, // 3 bytes per pixel
pixarray.BGR, // BGR channel order
800000, // 800 KHz
10, // DMA 10
[]int{12}, // GPIO 12
)
if err != nil {
log.Fatalln("failed to create pixarray:", err)
}
ticker := time.NewTicker(wormSpeed)
defer ticker.Stop()
var i int
for range ticker.C {
strip.SetPixel(ledOrder[i], pixarray.Pixel{})
i = (i + 1) % len(ledOrder)
strip.SetPixel(ledOrder[i], rgbaToPixel(ledColor(i)))
if err := strip.Write(); err != nil {
log.Println("failed to write:", err)
}
}
}
var transColors = []color.RGBA{
{91, 206, 250, 255},
{245, 169, 184, 255},
{255, 255, 255, 255},
{245, 169, 184, 255},
{91, 206, 250, 255},
}
func ledColor(i int) color.RGBA {
y := ledOrder[i]
c := y * len(transColors) / maxLEDHeight
return transColors[c]
}
func rgbaToPixel(c color.RGBA) pixarray.Pixel {
return pixarray.Pixel{
R: int(c.R),
G: int(c.G),
B: int(c.B),
}
}