feat(main): add a ton of features
- progress bar - frame res measurements - verbosity flag
This commit is contained in:
parent
801abef51b
commit
e8742aba21
SSH key fingerprint:
SHA256:WqtcVnDMrv1lnUlNah5k31iywFUI/DV+5yHzCTO4Vds
3 changed files with 132 additions and 8 deletions
9
go.mod
9
go.mod
|
|
@ -4,7 +4,16 @@ go 1.24.3
|
|||
|
||||
require (
|
||||
github.com/AlexEidt/Vidio v1.5.1 // indirect
|
||||
github.com/VividCortex/ewma v1.2.0 // indirect
|
||||
github.com/cheggaaa/pb v1.0.29 // indirect
|
||||
github.com/cheggaaa/pb/v3 v3.1.7 // indirect
|
||||
github.com/fatih/color v1.18.0 // indirect
|
||||
github.com/mattn/go-colorable v0.1.14 // indirect
|
||||
github.com/mattn/go-isatty v0.0.20 // indirect
|
||||
github.com/mattn/go-runewidth v0.0.16 // indirect
|
||||
github.com/rivo/uniseg v0.4.7 // indirect
|
||||
github.com/urfave/cli/v3 v3.3.3 // indirect
|
||||
github.com/zmb3/gogetdoc v0.0.0-20190228002656-b37376c5da6a // indirect
|
||||
golang.org/x/sys v0.30.0 // indirect
|
||||
golang.org/x/tools v0.0.0-20181207195948-8634b1ecd393 // indirect
|
||||
)
|
||||
|
|
|
|||
27
go.sum
27
go.sum
|
|
@ -1,8 +1,35 @@
|
|||
github.com/AlexEidt/Vidio v1.5.1 h1:tovwvtgQagUz1vifiL9OeWkg1fP/XUzFazFKh7tFtaE=
|
||||
github.com/AlexEidt/Vidio v1.5.1/go.mod h1:djhIMnWMqPrC3X6nB6ymGX6uWWlgw+VayYGKE1bNwmI=
|
||||
github.com/VividCortex/ewma v1.2.0 h1:f58SaIzcDXrSy3kWaHNvuJgJ3Nmz59Zji6XoJR/q1ow=
|
||||
github.com/VividCortex/ewma v1.2.0/go.mod h1:nz4BbCtbLyFDeC9SUHbtcT5644juEuWfUAUnGx7j5l4=
|
||||
github.com/cheggaaa/pb v1.0.29 h1:FckUN5ngEk2LpvuG0fw1GEFx6LtyY2pWI/Z2QgCnEYo=
|
||||
github.com/cheggaaa/pb v1.0.29/go.mod h1:W40334L7FMC5JKWldsTWbdGjLo0RxUKK73K+TuPxX30=
|
||||
github.com/cheggaaa/pb/v3 v3.1.7 h1:2FsIW307kt7A/rz/ZI2lvPO+v3wKazzE4K/0LtTWsOI=
|
||||
github.com/cheggaaa/pb/v3 v3.1.7/go.mod h1:/Ji89zfVPeC/u5j8ukD0MBPHt2bzTYp74lQ7KlgFWTQ=
|
||||
github.com/fatih/color v1.9.0/go.mod h1:eQcE1qtQxscV5RaZvpXrrb8Drkc3/DdQ+uUYCNjL+zU=
|
||||
github.com/fatih/color v1.18.0 h1:S8gINlzdQ840/4pfAwic/ZE0djQEH3wM94VfqLTZcOM=
|
||||
github.com/fatih/color v1.18.0/go.mod h1:4FelSpRwEGDpQ12mAdzqdOukCy4u8WUtOY6lkT/6HfU=
|
||||
github.com/mattn/go-colorable v0.1.4/go.mod h1:U0ppj6V5qS13XJ6of8GYAs25YV2eR4EVcfRqFIhoBtE=
|
||||
github.com/mattn/go-colorable v0.1.14 h1:9A9LHSqF/7dyVVX6g0U9cwm9pG3kP9gSzcuIPHPsaIE=
|
||||
github.com/mattn/go-colorable v0.1.14/go.mod h1:6LmQG8QLFO4G5z1gPvYEzlUgJ2wF+stgPZH1UqBm1s8=
|
||||
github.com/mattn/go-isatty v0.0.8/go.mod h1:Iq45c/XA43vh69/j3iqttzPXn0bhXyGjM0Hdxcsrc5s=
|
||||
github.com/mattn/go-isatty v0.0.11/go.mod h1:PhnuNfih5lzO57/f3n+odYbM4JtupLOxQOAqxQCu2WE=
|
||||
github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY=
|
||||
github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
|
||||
github.com/mattn/go-runewidth v0.0.4/go.mod h1:LwmH8dsx7+W8Uxz3IHJYH5QSwggIsqBzpuz5H//U1FU=
|
||||
github.com/mattn/go-runewidth v0.0.16 h1:E5ScNMtiwvlvB5paMFdw9p4kSQzbXFikJ5SQO6TULQc=
|
||||
github.com/mattn/go-runewidth v0.0.16/go.mod h1:Jdepj2loyihRzMpdS35Xk/zdY8IAYHsh153qUoGf23w=
|
||||
github.com/rivo/uniseg v0.2.0/go.mod h1:J6wj4VEh+S6ZtnVlnTBMWIodfgj8LQOQFoIToxlJtxc=
|
||||
github.com/rivo/uniseg v0.4.7 h1:WUdvkW8uEhrYfLC4ZzdpI2ztxP1I582+49Oc5Mq64VQ=
|
||||
github.com/rivo/uniseg v0.4.7/go.mod h1:FN3SvrM+Zdj16jyLfmOkMNblXMcoc8DfTHruCPUcx88=
|
||||
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/sys v0.0.0-20190222072716-a9d3bda3a223/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
||||
golang.org/x/sys v0.0.0-20191026070338-33540a1f6037/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||
golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.30.0 h1:QjkSwP/36a20jFYWkSue1YwXzLmsV5Gfq7Eiy72C1uc=
|
||||
golang.org/x/sys v0.30.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
|
||||
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=
|
||||
|
|
|
|||
104
main.go
104
main.go
|
|
@ -2,16 +2,21 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"cmp"
|
||||
"context"
|
||||
"encoding/csv"
|
||||
"fmt"
|
||||
"image"
|
||||
"image/draw"
|
||||
"image/png"
|
||||
"log"
|
||||
"os"
|
||||
"os/exec"
|
||||
"strconv"
|
||||
"strings"
|
||||
|
||||
vidio "github.com/AlexEidt/Vidio"
|
||||
"github.com/cheggaaa/pb"
|
||||
"github.com/urfave/cli/v3"
|
||||
)
|
||||
|
||||
|
|
@ -105,11 +110,22 @@ func main() {
|
|||
Usage: "Path to CSV file for frame data output",
|
||||
Value: "",
|
||||
},
|
||||
&cli.BoolFlag{
|
||||
Name: "resdet",
|
||||
Usage: "use the resdet cli to measure each frame's resoltion\nWARNING: This will slow the process down by a LOT",
|
||||
Value: false,
|
||||
},
|
||||
|
||||
&cli.BoolFlag{
|
||||
Name: "verbose",
|
||||
Usage: "print out total unique frames for every second of measurements",
|
||||
Value: false,
|
||||
},
|
||||
},
|
||||
Action: func(ctx context.Context, cmd *cli.Command) error {
|
||||
tolerance := uint64(cmd.Float64("tolerance"))
|
||||
csvOutput := cmd.String("csv-output")
|
||||
return analyzeFramePersistence(cmd.StringArg("video"), tolerance, csvOutput)
|
||||
return analyzeFramePersistence(cmd.StringArg("video"), tolerance, csvOutput, cmd.Bool("resdet"), cmd.Bool("verbose"))
|
||||
},
|
||||
},
|
||||
},
|
||||
|
|
@ -251,7 +267,7 @@ func getImageFromFilePath(filePath string) (image.Image, error) {
|
|||
return image, err
|
||||
}
|
||||
|
||||
func analyzeFramePersistence(videoPath string, tolerance uint64, csvOutput string) error {
|
||||
func analyzeFramePersistence(videoPath string, tolerance uint64, csvOutput string, toggleResdet bool, verbose bool) error {
|
||||
video, err := vidio.NewVideo(videoPath)
|
||||
if err != nil {
|
||||
return err
|
||||
|
|
@ -288,6 +304,8 @@ func analyzeFramePersistence(videoPath string, tolerance uint64, csvOutput strin
|
|||
effectiveFPS float64
|
||||
currentFrameTime float64
|
||||
realFrameTime float64
|
||||
frameWidth int
|
||||
frameHeight int
|
||||
}
|
||||
|
||||
var frameAnalysisData []FrameData
|
||||
|
|
@ -301,6 +319,8 @@ func analyzeFramePersistence(videoPath string, tolerance uint64, csvOutput strin
|
|||
var frameNumber int
|
||||
var uniqueFramesPerSecond []int
|
||||
var framePersistenceDurations []float64
|
||||
var frameWidthMeasurements []int
|
||||
var frameHeightMeasurements []int
|
||||
|
||||
currentSecond := 0
|
||||
uniqueFramesInCurrentSecond := 0
|
||||
|
|
@ -310,9 +330,44 @@ func analyzeFramePersistence(videoPath string, tolerance uint64, csvOutput strin
|
|||
|
||||
hasFirstFrame := false
|
||||
|
||||
bar := pb.StartNew(video.Frames())
|
||||
|
||||
for video.Read() {
|
||||
frameNumber++
|
||||
|
||||
// frame colum will be full of 0s normally, not the worst compromise
|
||||
frameWidth := 0
|
||||
frameHeight := 0
|
||||
|
||||
// mesure resoltion
|
||||
if toggleResdet {
|
||||
frameFile, err0 := os.Create("/tmp/frame.png")
|
||||
|
||||
err1 := png.Encode(frameFile, currentFrame)
|
||||
|
||||
out, err2 := exec.Command("resdet", "-v", "1", frameFile.Name()).Output()
|
||||
|
||||
err3 := frameFile.Close()
|
||||
|
||||
err4 := os.Remove(frameFile.Name())
|
||||
|
||||
formattedOutput := strings.Split(string(out), " ")
|
||||
|
||||
frameWidthOut, err5 := strconv.Atoi(formattedOutput[0])
|
||||
|
||||
frameHeightOut, err6 := strconv.Atoi(strings.TrimSuffix(formattedOutput[1], "\n"))
|
||||
|
||||
if err := cmp.Or(err0, err1, err2, err3, err4, err5, err6); err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
|
||||
frameWidth = frameWidthOut
|
||||
frameHeight = frameHeightOut
|
||||
}
|
||||
|
||||
frameWidthMeasurements = append(frameWidthMeasurements, frameWidth)
|
||||
frameHeightMeasurements = append(frameHeightMeasurements, frameHeight)
|
||||
|
||||
if !hasFirstFrame {
|
||||
copy(previousFrame.Pix, currentFrame.Pix)
|
||||
hasFirstFrame = true
|
||||
|
|
@ -330,6 +385,8 @@ func analyzeFramePersistence(videoPath string, tolerance uint64, csvOutput strin
|
|||
effectiveFPS: effectiveFPS,
|
||||
currentFrameTime: actualFrameTimeMs,
|
||||
realFrameTime: 0, // Will be calculated in second pass
|
||||
frameWidth: frameWidth,
|
||||
frameHeight: frameHeight,
|
||||
})
|
||||
continue
|
||||
}
|
||||
|
|
@ -385,17 +442,25 @@ func analyzeFramePersistence(videoPath string, tolerance uint64, csvOutput strin
|
|||
effectiveFPS: effectiveFPS,
|
||||
currentFrameTime: actualFrameTimeMs,
|
||||
realFrameTime: 0, // Will be calculated in second pass
|
||||
frameWidth: frameWidth,
|
||||
frameHeight: frameHeight,
|
||||
})
|
||||
|
||||
newSecond := int(float64(frameNumber-1) / fps)
|
||||
if newSecond > currentSecond {
|
||||
uniqueFramesPerSecond = append(uniqueFramesPerSecond, uniqueFramesInCurrentSecond)
|
||||
log.Default().Printf("Second %d: %d unique frames", currentSecond+1, uniqueFramesInCurrentSecond)
|
||||
currentSecond = newSecond
|
||||
uniqueFramesInCurrentSecond = 0
|
||||
if verbose {
|
||||
newSecond := int(float64(frameNumber-1) / fps)
|
||||
if newSecond > currentSecond {
|
||||
uniqueFramesPerSecond = append(uniqueFramesPerSecond, uniqueFramesInCurrentSecond)
|
||||
log.Default().Printf("Second %d: %d unique frames", currentSecond+1, uniqueFramesInCurrentSecond)
|
||||
currentSecond = newSecond
|
||||
uniqueFramesInCurrentSecond = 0
|
||||
}
|
||||
}
|
||||
|
||||
bar.Increment()
|
||||
}
|
||||
|
||||
bar.Finish()
|
||||
|
||||
// Record the final unique frame duration
|
||||
if totalUniqueFrames > 0 {
|
||||
if len(uniqueFrameDurations) < totalUniqueFrames {
|
||||
|
|
@ -415,6 +480,8 @@ func analyzeFramePersistence(videoPath string, tolerance uint64, csvOutput strin
|
|||
fmt.Sprintf("%.2f", frameData.currentFrameTime),
|
||||
strconv.Itoa(frameData.uniqueFrameCount),
|
||||
fmt.Sprintf("%.2f", realFrameTimeMs),
|
||||
strconv.Itoa(frameData.frameWidth),
|
||||
strconv.Itoa(frameData.frameHeight),
|
||||
})
|
||||
if err != nil {
|
||||
log.Default().Printf("Warning: failed to write CSV row %d: %v", i+1, err)
|
||||
|
|
@ -461,5 +528,26 @@ func analyzeFramePersistence(videoPath string, tolerance uint64, csvOutput strin
|
|||
log.Default().Printf("No frame persistence detected (all frames are unique)")
|
||||
}
|
||||
|
||||
if len(frameWidthMeasurements) > 0 && len(frameHeightMeasurements) > 0 {
|
||||
sumWidth := 0
|
||||
sumHeight := 0
|
||||
|
||||
for _, width := range frameWidthMeasurements {
|
||||
sumWidth += width
|
||||
}
|
||||
|
||||
if sumWidth != 0 {
|
||||
|
||||
for _, height := range frameHeightMeasurements {
|
||||
sumHeight += height
|
||||
}
|
||||
|
||||
avgWidth := float64(sumWidth) / float64(len(frameWidthMeasurements))
|
||||
avgHeight := float64(sumHeight) / float64(len(frameHeightMeasurements))
|
||||
log.Default().Printf("Average Width: %.2f", avgWidth)
|
||||
log.Default().Printf("Average Height: %.2f", avgHeight)
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue