97 lines
2.2 KiB
Go
97 lines
2.2 KiB
Go
package main
|
||
|
||
import (
|
||
"github.com/juju/gnuflag"
|
||
|
||
"range-gen/output"
|
||
|
||
"errors"
|
||
"os"
|
||
)
|
||
|
||
var (
|
||
ArgInput string
|
||
ArgInputIsSet bool = false
|
||
ArgOutput string
|
||
ArgOutputIsSet bool = false
|
||
ArgThreshold string
|
||
ArgThresholdIsSet bool = false
|
||
|
||
ArgJobs int = 0
|
||
ArgDefaultNoiseLevel int = 0
|
||
|
||
ArgVersion bool = false
|
||
ArgHelp bool = false
|
||
)
|
||
|
||
func printHelp() {
|
||
output.Println("Сreates a list of scene ranges based on a set of frames from the video")
|
||
output.Println("")
|
||
output.Println("Usage: range-gen [option]... <input_dir> <output_file> <threshold>")
|
||
output.Println("input_dir Directory with png images")
|
||
output.Println("output_file Range list file")
|
||
output.Println("threshold Image similarity threshold (0-1024)")
|
||
output.Println("")
|
||
output.Println("Options:")
|
||
output.Println(" -j, --jobs N Allow N jobs at once")
|
||
output.Println(" -n, --noise Default noise level for each range")
|
||
output.Println(" -h, --help Show this page")
|
||
output.Println(" -v, --version Show version")
|
||
}
|
||
|
||
func printVersion() {
|
||
output.Println("range-gen ", Version)
|
||
}
|
||
|
||
func init() {
|
||
gnuflag.IntVar(&ArgJobs, "jobs", 0, "")
|
||
gnuflag.IntVar(&ArgJobs, "j", 0, "")
|
||
gnuflag.IntVar(&ArgDefaultNoiseLevel, "noise", -1, "")
|
||
gnuflag.IntVar(&ArgDefaultNoiseLevel, "n", -1, "")
|
||
gnuflag.BoolVar(&ArgVersion, "version", false, "")
|
||
gnuflag.BoolVar(&ArgVersion, "v", false, "")
|
||
gnuflag.BoolVar(&ArgHelp, "help", false, "")
|
||
gnuflag.BoolVar(&ArgHelp, "h", false, "")
|
||
}
|
||
|
||
func parseArgs() error {
|
||
gnuflag.Parse(false)
|
||
|
||
switch {
|
||
case ArgHelp:
|
||
printHelp()
|
||
os.Exit(EXIT_SUCCESS)
|
||
case ArgVersion:
|
||
printVersion()
|
||
os.Exit(EXIT_SUCCESS)
|
||
}
|
||
|
||
for i := 0; i < 3 && i < gnuflag.NArg(); i++ {
|
||
switch i {
|
||
case 0:
|
||
ArgInput = gnuflag.Arg(0)
|
||
ArgInputIsSet = true
|
||
case 1:
|
||
ArgOutput = gnuflag.Arg(1)
|
||
ArgOutputIsSet = true
|
||
case 2:
|
||
ArgThreshold = gnuflag.Arg(2)
|
||
ArgThresholdIsSet = true
|
||
}
|
||
}
|
||
|
||
if !ArgInputIsSet {
|
||
return errors.New("Input directory not specified")
|
||
}
|
||
|
||
if !ArgOutputIsSet {
|
||
return errors.New("Output file not specified")
|
||
}
|
||
|
||
if !ArgThresholdIsSet {
|
||
return errors.New("Threshold not specified")
|
||
}
|
||
|
||
return nil
|
||
}
|