mirror of https://github.com/certusone/dc4bc.git
69 lines
1.4 KiB
Go
69 lines
1.4 KiB
Go
package qr
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"time"
|
|
|
|
"github.com/makiuchi-d/gozxing"
|
|
"github.com/makiuchi-d/gozxing/qrcode"
|
|
"gocv.io/x/gocv"
|
|
)
|
|
|
|
const timeToScan = time.Second * 5
|
|
|
|
func ReadQRFromCamera() ([]byte, error) {
|
|
webcam, err := gocv.OpenVideoCapture(0)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to OpenVideoCapture: %w", err)
|
|
}
|
|
window := gocv.NewWindow("Hello")
|
|
|
|
defer func() {
|
|
if err := webcam.Close(); err != nil {
|
|
log.Fatalf("failed to close camera: %v", err)
|
|
}
|
|
}()
|
|
defer func() {
|
|
if err := window.Close(); err != nil {
|
|
log.Fatalf("failed to close camera window: %v", err)
|
|
}
|
|
}()
|
|
|
|
img := gocv.NewMat()
|
|
tk := time.NewTimer(timeToScan)
|
|
|
|
// This loop reads an image from the webcam every millisecond
|
|
// for 5 seconds. The last image taken will be used as the final
|
|
//one.
|
|
loop:
|
|
for {
|
|
select {
|
|
case <-tk.C:
|
|
break loop
|
|
default:
|
|
webcam.Read(&img)
|
|
window.IMShow(img)
|
|
window.WaitKey(1)
|
|
}
|
|
}
|
|
|
|
imgObject, err := img.ToImage()
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to get image object: %w", err)
|
|
}
|
|
|
|
bmp, err := gozxing.NewBinaryBitmapFromImage(imgObject)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to get NewBinaryBitmapFromImage: %w", err)
|
|
}
|
|
|
|
qrReader := qrcode.NewQRCodeReader()
|
|
result, err := qrReader.Decode(bmp, nil)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to decode the QR-code contents: %w", err)
|
|
}
|
|
|
|
return result.GetRawBytes(), err
|
|
}
|