-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmain.go
106 lines (84 loc) · 2.15 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
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
package main
import (
"bytes"
"fmt"
"image"
"image/color"
"image/jpeg"
"os/exec"
"time"
"github.com/faiface/pixel"
"github.com/faiface/pixel/pixelgl"
"golang.org/x/image/colornames"
)
var blockSize = 8
func CalculateDominantColor(img image.Image, startX, startY, blockSize int) color.Color {
colorCount := make(map[color.Color]int)
for y := startY; y < startY+blockSize && y < img.Bounds().Dy(); y++ {
for x := startX; x < startX+blockSize && x < img.Bounds().Dx(); x++ {
c := img.At(x, y)
colorCount[c]++
}
}
var dominantColor color.Color
maxCount := 0
for c, count := range colorCount {
if count > maxCount {
dominantColor = c
maxCount = count
}
}
return dominantColor
}
func GetColorMap(img image.Image) *image.RGBA {
colorMap := image.NewRGBA(img.Bounds())
for y := 0; y < img.Bounds().Dy(); y += blockSize {
for x := 0; x < img.Bounds().Dx(); x += blockSize {
dominantColor := CalculateDominantColor(img, x, y, blockSize)
for by := y; by < y+blockSize && by < img.Bounds().Dy(); by++ {
for bx := x; bx < x+blockSize && bx < img.Bounds().Dx(); bx++ {
colorMap.Set(bx, by, dominantColor)
}
}
}
}
return colorMap
}
func run() {
windowCfg := pixelgl.WindowConfig{
Title: "Webcam",
Bounds: pixel.R(0, 0, 640, 480),
VSync: true,
}
win, err := pixelgl.NewWindow(windowCfg)
if err != nil {
panic(err)
}
for !win.Closed() {
cmd := exec.Command("ffmpeg", "-y", "-f", "video4linux2", "-i", "/dev/video0", "-vframes", "1", "-f", "image2pipe", "-")
var out bytes.Buffer
cmd.Stdout = &out
err := cmd.Run()
if err != nil {
fmt.Printf("Erro ao capturar a imagem: %v\n", err)
return
}
imageData := out.Bytes()
img, err := jpeg.Decode(bytes.NewReader(imageData))
if err != nil {
fmt.Printf("Erro ao decodificar a imagem: %v\n", err)
return
}
colorMap := GetColorMap(img)
pic := pixel.PictureDataFromImage(colorMap)
sprite := pixel.NewSprite(pic, pic.Bounds())
win.Clear(colornames.Black)
sprite.Draw(win, pixel.IM.Moved(win.Bounds().Center()))
win.Update()
// Esperar 1/10 de segundo
time.Sleep(1 * time.Millisecond)
}
}
func main() {
pixelgl.Run(run)
}