diff options
Diffstat (limited to 'tools')
| -rwxr-xr-x | tools/video-util.sh | 333 |
1 files changed, 333 insertions, 0 deletions
diff --git a/tools/video-util.sh b/tools/video-util.sh new file mode 100755 index 0000000..5c4cfd1 --- /dev/null +++ b/tools/video-util.sh @@ -0,0 +1,333 @@ +#!/bin/bash + +set -e + +DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )" +PROGNAME="$0" +BOLD=$(tput bold) +RST=$(tput sgr0) +RED=$(tput setaf 1) +GREEN=$(tput setaf 2) +YELLOW=$(tput setaf 3) +CYAN=$(tput setaf 6) + +input= +command= +roi_file= +motion_threshold=1 +ffmpeg_args="-nostats -loglevel error" +dvr_scan_args="-q" +verbose= +config_dir=$HOME/.config/video-util +config_dir_set= + +debug() { + if [ -n "$verbose" ]; then + >&2 echo "$@" + fi +} + +echoinfo() { + echo "${CYAN}$@${RST}" +} + +echoerr() { + >&2 echo "${RED}${BOLD}error:${RST}${RED} $@${RST}" +} + +echowarn() { + >&2 echo "${YELLOW}${BOLD}warning:${RST}${YELLOW} $@${RST}" +} + +die() { + >&2 echo "error: $@" + exit 1 +} + +file_in_use() { + [ -n "$(lsof "$1")" ] +} + +get_mtime() { + stat -c %Y "$1" +} + +# converts date in format yyyy-mm-dd-hh.ii.ss to unixtime +date_to_unixtime() { + local date="$1" + date=${date//./-} + date=${date//-/ } + + local nums=($date) + + local y=${nums[0]} + local m=${nums[1]} + local d=${nums[2]} + local h=${nums[3]} + local i=${nums[4]} + local s=${nums[5]} + + date --date="$y-$m-$d $h:$i:$s" +"%s" +} + +filename_as_unixtime() { + local name="$1" + name="$(basename "$name")" + name="${name/record_/}" + name="${name/.mp4/}" + date_to_unixtime "$name" +} + +config_get_prev_mtime() { + local prefix="$1" + [ -z "$prefix" ] && die "config_get_prev_mtime: no prefix" + + local file="$config_dir/${prefix}_mtime" + if [ -f "$file" ]; then + debug "config_get_prev_mtime: $(cat "$file")" + cat "$file" + else + debug "config_get_prev_mtime: 0" + echo "0" + fi +} + +config_set_prev_mtime() { + local prefix="$1" + [ -z "$prefix" ] && die "config_set_prev_mtime: no prefix" + + local mtime="$2" + [ -z "$mtime" ] && die "config_set_prev_mtime: no mtime" + + local file="$config_dir/${prefix}_mtime" + + debug "config_set_prev_mtime: writing '$mtime' to '$file'" + echo "$mtime" > "$file" +} + +usage() { + cat <<-_EOF + usage: $PROGNAME OPTIONS command + + Options: + -i|--input input file/directory + -o|--output output file/directory + --roi-file path to file with ROI sets + -mt motion threshold + -v, -vv, vx be verbose + --config-dir config directory + + Commands: + fix, mass-fix fix video timestamps + mass-fix-mtime fix mtimes of recordings + motion detect motion + snapshot take video snapshot + + _EOF + exit 1 +} + +check_input_file() { + [ -z "$input" ] && die "input file not specified" + [ -f "$input" ] || die "input file '$input' doesn't exist" +} + +check_input_dir() { + [ -z "$input" ] && die "input directory not specified" + [ -d "$input" ] || die "input directory '$input' doesn't exist" +} + +fix_video_timestamps() { + local input="$1" + local dir=$(dirname "$input") + local temp="$dir/.temporary_fixing.mp4" + + local mtime=$(get_mtime "$input") + + # debug "fix_video_timestamps: ffmpeg $ffmpeg_args -i \"$input\" -y -c copy \"$temp\"" + ffmpeg $ffmpeg_args -i "$input" -y -c copy "$temp" </dev/null + rm "$input" + mv "$temp" "$input" + + local unixtime=$(filename_as_unixtime "$input") + touch --date=@$unixtime "$input" +} + +do_mass_fix() { + local mtime=$(config_get_prev_mtime "fix") + local tmpfile=$(mktemp) + debug "do_mass_fix: created temporary file $tmpfile" + + touch --date=@$mtime "$tmpfile" + debug "do_mass_fix: mtime of temp file: $(get_mtime $tmpfile)" + + [ -f "$input/.temporary_fixing.mp4" ] && { + echowarn "do_mass_fix: '$input/.temporary_fixing.mp4' exists, deleting" + rm "$input/.temporary_fixing.mp4" + } + + # find all files in $input directory, newer than $tmpfile's time, + # sort them in ascending order, and finally remove timestamps + # leaving only file names. Then loop through each line + find "$input" -type f -newer "$tmpfile" -printf "%T+ %p\n" | sort | awk '{print $2}' | while read file; do + if ! file_in_use "$file"; then + debug "do_mass_fix: calling fix_video_timestamps($file)" + + fix_video_timestamps "$file" + echoinfo "fixed $file" + + config_set_prev_mtime "fix" "$(filename_as_unixtime "$file")" + else + echowarn "file '$file' is in use" + fi + done + + rm "$tmpfile" +} + +do_mass_fix_mtime() { + local time + find "$input" -type f -name "*.mp4" | while read file; do + if [[ "$(basename "$file")" =~ ^record_.* ]]; then + time="$(filename_as_unixtime "$file")" + debug "$file: $time" + touch --date=@$time "$file" + else + echowarn "unrecognized file: $file" + fi + done +} + +do_motion() { + local timecodes=() + if [ -z "$roi_file" ]; then + timecodes+=($(dvr_scan "$input")) + else + [ -f "$roi_file" ] || die "specified ROI sets file does not exists" + echoinfo "using roi sets from file: ${BOLD}$roi_file" + while read line; do + if ! [[ "$line" =~ ^#.* ]]; then + timecodes+=("$(dvr_scan "$input" "$line")") + fi + done < <(cat "$roi_file") + fi + + timecodes="${timecodes[@]}" + timecodes=${timecodes// /,} + + echo "all timecodes: $timecodes" +} + +dvr_scan_fake() { + echo "00:05:06.930,00:05:24.063" +} + +dvr_scan() { + local input="$1" + local args= + if [ ! -z "$2" ]; then + args="-roi $2" + echoinfo "starting dvr-scan, roi=($2), mt=$motion_threshold" + else + echoinfo "starting dvr-scan, no roi, mt=$motion_threshold" + fi + dvr-scan $dvr_scan_args -i "$input" -so --min-event-length 3s -df 3 --frame-skip 2 -t $motion_threshold $args | tail -1 +} + +[[ $# -lt 1 ]] && usage + +while [[ $# -gt 0 ]]; do + case $1 in + -i|--input) + input="$2" + shift; shift + ;; + + --roi-file) + roi_file="$2" + shift; shift + ;; + + -mt) + motion_threshold="$2" + shift; shift + ;; + + fix|mass-fix|motion|snapshot|mass-fix-mtime) + command="$1" + shift + ;; + + -v) + verbose=1 + shift + ;; + + -vx) + verbose=1 + set -x + shift + ;; + + -vv) + verbose=1 + ffmpeg_args="-loglevel info" + dvr_scan_args="" + shift + ;; + + --config-dir) + config_dir="$2" + config_dir_set=1 + shift; shift + ;; + + *) + die "unrecognized option $1" + exit 1 + ;; + esac +done + +[ -z "$config_dir_set" ] && echowarn "no --config-dir specified, using default ($config_dir)" +if [ ! -d "$config_dir" ]; then + mkdir "$config_dir" || die "failed to create config directory ($config_dir)" +fi + +[ -z "$command" ] && die "command not specified" +case "$command" in + fix) + check_input_file + fix_video_timestamps "$input" + echo "done" + ;; + + mass-fix) + check_input_dir + do_mass_fix + ;; + + mass-fix-mtime) + check_input_dir + do_mass_fix_mtime + ;; + + motion) + check_input_file + do_motion + ;; + + snapshot) + check_input_file + [ -z "$output" ] && { + echowarn "--output not specified, using snapshot.jpg as default" + output="snapshot.jpg" + } + ffmpeg $ffmpeg_args -i "$input" -frames:v 1 -q:v 2 "$output" </dev/null + echoinfo "saved to $output" + ;; + + *) + echo "error: invalid command '$command'" + ;; +esac |