feat: first wave of tests
I can: count how many frames in a video, compare 2 frames for differences and compare 2 videos to see if their frames match up
This commit is contained in:
commit
6dcf46c597
SSH key fingerprint:
SHA256:WqtcVnDMrv1lnUlNah5k31iywFUI/DV+5yHzCTO4Vds
4 changed files with 245 additions and 0 deletions
7
.cz.toml
Normal file
7
.cz.toml
Normal file
|
|
@ -0,0 +1,7 @@
|
||||||
|
[tool.commitizen]
|
||||||
|
name = "cz_conventional_commits"
|
||||||
|
tag_format = "$version"
|
||||||
|
version_scheme = "semver2"
|
||||||
|
version = "0.0.1"
|
||||||
|
update_changelog_on_bump = true
|
||||||
|
major_version_zero = true
|
||||||
10
go.mod
Normal file
10
go.mod
Normal file
|
|
@ -0,0 +1,10 @@
|
||||||
|
module fps-go-brr
|
||||||
|
|
||||||
|
go 1.24.3
|
||||||
|
|
||||||
|
require (
|
||||||
|
github.com/AlexEidt/Vidio v1.5.1 // indirect
|
||||||
|
github.com/urfave/cli/v3 v3.3.3 // indirect
|
||||||
|
github.com/zmb3/gogetdoc v0.0.0-20190228002656-b37376c5da6a // indirect
|
||||||
|
golang.org/x/tools v0.0.0-20181207195948-8634b1ecd393 // indirect
|
||||||
|
)
|
||||||
8
go.sum
Normal file
8
go.sum
Normal file
|
|
@ -0,0 +1,8 @@
|
||||||
|
github.com/AlexEidt/Vidio v1.5.1 h1:tovwvtgQagUz1vifiL9OeWkg1fP/XUzFazFKh7tFtaE=
|
||||||
|
github.com/AlexEidt/Vidio v1.5.1/go.mod h1:djhIMnWMqPrC3X6nB6ymGX6uWWlgw+VayYGKE1bNwmI=
|
||||||
|
github.com/urfave/cli/v3 v3.3.3 h1:byCBaVdIXuLPIDm5CYZRVG6NvT7tv1ECqdU4YzlEa3I=
|
||||||
|
github.com/urfave/cli/v3 v3.3.3/go.mod h1:FJSKtM/9AiiTOJL4fJ6TbMUkxBXn7GO9guZqoZtpYpo=
|
||||||
|
github.com/zmb3/gogetdoc v0.0.0-20190228002656-b37376c5da6a h1:00UFliGZl2UciXe8o/2iuEsRQ9u7z0rzDTVzuj6EYY0=
|
||||||
|
github.com/zmb3/gogetdoc v0.0.0-20190228002656-b37376c5da6a/go.mod h1:ofmGw6LrMypycsiWcyug6516EXpIxSbZ+uI9ppGypfY=
|
||||||
|
golang.org/x/tools v0.0.0-20181207195948-8634b1ecd393 h1:0P8IF6+RwCumULxvjp9EtJryUs46MgLIgeHbCt7NU4Q=
|
||||||
|
golang.org/x/tools v0.0.0-20181207195948-8634b1ecd393/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
|
||||||
220
main.go
Normal file
220
main.go
Normal file
|
|
@ -0,0 +1,220 @@
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"image"
|
||||||
|
"image/draw"
|
||||||
|
"log"
|
||||||
|
"os"
|
||||||
|
"strconv"
|
||||||
|
|
||||||
|
vidio "github.com/AlexEidt/Vidio"
|
||||||
|
"github.com/urfave/cli/v3"
|
||||||
|
)
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
cmd := &cli.Command{
|
||||||
|
Commands: []*cli.Command{
|
||||||
|
{
|
||||||
|
Name: "count-frames",
|
||||||
|
Usage: "Count frames",
|
||||||
|
Action: func(ctx context.Context, cmd *cli.Command) error {
|
||||||
|
// fmt.Println("added task: ", cmd.Args().First())
|
||||||
|
return count_video_frames(cmd.Args().First())
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Name: "compare-frames",
|
||||||
|
Usage: "Are two frames different?",
|
||||||
|
Arguments: []cli.Argument{
|
||||||
|
&cli.StringArg{
|
||||||
|
Name: "frame1",
|
||||||
|
},
|
||||||
|
&cli.StringArg{
|
||||||
|
Name: "frame2",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
|
||||||
|
Action: func(ctx context.Context, cmd *cli.Command) error {
|
||||||
|
// fmt.Println("added task: ", cmd.Args().First())
|
||||||
|
first_frame, _ := getImageFromFilePath(cmd.StringArg("frame1"))
|
||||||
|
second_frame, _ := getImageFromFilePath(cmd.StringArg("frame2"))
|
||||||
|
|
||||||
|
first_rgba := imageToRGBA(first_frame)
|
||||||
|
second_rgba := imageToRGBA(second_frame)
|
||||||
|
return compare_frames(first_rgba, second_rgba)
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Name: "count-frames-differing-pixels",
|
||||||
|
Usage: "Are two frames different?",
|
||||||
|
Arguments: []cli.Argument{
|
||||||
|
&cli.StringArg{
|
||||||
|
Name: "frame1",
|
||||||
|
},
|
||||||
|
&cli.StringArg{
|
||||||
|
Name: "frame2",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
|
||||||
|
Action: func(ctx context.Context, cmd *cli.Command) error {
|
||||||
|
// fmt.Println("added task: ", cmd.Args().First())
|
||||||
|
first_frame, _ := getImageFromFilePath(cmd.StringArg("frame1"))
|
||||||
|
second_frame, _ := getImageFromFilePath(cmd.StringArg("frame2"))
|
||||||
|
|
||||||
|
first_rgba := imageToRGBA(first_frame)
|
||||||
|
second_rgba := imageToRGBA(second_frame)
|
||||||
|
return compare_frames_alt(first_rgba, second_rgba)
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Name: "count-unique-video-frames",
|
||||||
|
Usage: "Are two frames different?",
|
||||||
|
Arguments: []cli.Argument{
|
||||||
|
&cli.StringArg{
|
||||||
|
Name: "video1",
|
||||||
|
},
|
||||||
|
&cli.StringArg{
|
||||||
|
Name: "video2",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
|
||||||
|
Action: func(ctx context.Context, cmd *cli.Command) error {
|
||||||
|
return countUniqueVideoFrames(cmd.StringArg("video1"), cmd.StringArg("video2"), 1)
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := cmd.Run(context.Background(), os.Args); err != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// count_video_frames
|
||||||
|
// Prints out the total ammount of frames within `video`
|
||||||
|
//
|
||||||
|
// Parameters:
|
||||||
|
// - video string - path to video file
|
||||||
|
//
|
||||||
|
// Returns:
|
||||||
|
// - error
|
||||||
|
func count_video_frames(video string) error {
|
||||||
|
log.Default().Print("Trying to open video at: " + video)
|
||||||
|
video_file, _ := vidio.NewVideo(video)
|
||||||
|
count := 0
|
||||||
|
for video_file.Read() {
|
||||||
|
count++
|
||||||
|
}
|
||||||
|
log.Default().Println("Video total frames: " + strconv.Itoa(count))
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func compare_frames(frame1 *image.RGBA, frame2 *image.RGBA) error {
|
||||||
|
// diff_frame := image.NewRGBA(frame1.Rect)
|
||||||
|
|
||||||
|
accumError := int64(0)
|
||||||
|
|
||||||
|
for i := 0; i < len(frame1.Pix); i++ {
|
||||||
|
accumError += int64(sqDiffUInt8(frame1.Pix[i], frame2.Pix[i]))
|
||||||
|
}
|
||||||
|
|
||||||
|
log.Default().Println("Total image error: " + strconv.FormatInt(accumError, 10))
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func compare_frames_alt(frame1 *image.RGBA, frame2 *image.RGBA) error {
|
||||||
|
// diff_frame := image.NewRGBA(frame1.Rect)
|
||||||
|
|
||||||
|
accumError := int64(0)
|
||||||
|
|
||||||
|
for i := 0; i < len(frame1.Pix); i++ {
|
||||||
|
if isDiffUInt8(frame1.Pix[i], frame2.Pix[i]) {
|
||||||
|
accumError++
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
log.Default().Println("Total differing pixels: " + strconv.FormatInt(accumError, 10))
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func sqDiffUInt8(x, y uint8) uint64 {
|
||||||
|
d := uint64(x) - uint64(y)
|
||||||
|
return d * d
|
||||||
|
}
|
||||||
|
|
||||||
|
func isDiffUInt8(x, y uint8) bool {
|
||||||
|
d := uint64(x) - uint64(y)
|
||||||
|
sq := d * d
|
||||||
|
|
||||||
|
if sq > 0 {
|
||||||
|
return true
|
||||||
|
} else {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func imageToRGBA(src image.Image) *image.RGBA {
|
||||||
|
|
||||||
|
// No conversion needed if image is an *image.RGBA.
|
||||||
|
if dst, ok := src.(*image.RGBA); ok {
|
||||||
|
return dst
|
||||||
|
}
|
||||||
|
|
||||||
|
// Use the image/draw package to convert to *image.RGBA.
|
||||||
|
b := src.Bounds()
|
||||||
|
dst := image.NewRGBA(image.Rect(0, 0, b.Dx(), b.Dy()))
|
||||||
|
draw.Draw(dst, dst.Bounds(), src, b.Min, draw.Src)
|
||||||
|
return dst
|
||||||
|
}
|
||||||
|
|
||||||
|
func getImageFromFilePath(filePath string) (image.Image, error) {
|
||||||
|
f, err := os.Open(filePath)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
defer f.Close()
|
||||||
|
image, _, err := image.Decode(f)
|
||||||
|
return image, err
|
||||||
|
}
|
||||||
|
|
||||||
|
func countUniqueVideoFrames(video_path1 string, video_path2 string, min_diff int64) error {
|
||||||
|
video1, _ := vidio.NewVideo(video_path1)
|
||||||
|
video2, _ := vidio.NewVideo(video_path2)
|
||||||
|
|
||||||
|
video1_frame := image.NewRGBA(image.Rect(0, 0, video1.Width(), video1.Height()))
|
||||||
|
video2_frame := image.NewRGBA(image.Rect(0, 0, video2.Width(), video2.Height()))
|
||||||
|
video1.SetFrameBuffer(video1_frame.Pix)
|
||||||
|
video2.SetFrameBuffer(video2_frame.Pix)
|
||||||
|
|
||||||
|
total_frames := 0
|
||||||
|
unique_frames := 0
|
||||||
|
for video1.Read() {
|
||||||
|
total_frames++
|
||||||
|
video2.Read()
|
||||||
|
|
||||||
|
accumError := int64(0)
|
||||||
|
|
||||||
|
for i := 0; i < len(video1_frame.Pix); i++ {
|
||||||
|
if isDiffUInt8(video1_frame.Pix[i], video2_frame.Pix[i]) {
|
||||||
|
accumError++
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if min_diff <= accumError {
|
||||||
|
unique_frames++
|
||||||
|
log.Default().Println("[" + strconv.Itoa(total_frames) + "]Unique frame")
|
||||||
|
} else {
|
||||||
|
log.Default().Println("[" + strconv.Itoa(total_frames) + "]Non-unique frame")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
video1.Close()
|
||||||
|
video2.Close()
|
||||||
|
|
||||||
|
log.Default().Println(strconv.Itoa(unique_frames) + "/" + strconv.Itoa(total_frames) + " are unique!")
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
Loading…
Add table
Add a link
Reference in a new issue