| #!/bin/bash |
| set -xe |
| |
| while [[ "$#" -gt 0 ]]; do |
| case $1 in |
| -b|--avm_build_dir) avm_build_dir="$2"; shift ;; |
| -s|--stream) stream="$2"; shift ;; |
| -o|--output) output="$2"; shift ;; |
| -y|--yuv) yuv="$2"; shift ;; |
| -n|--limit) limit="$2"; shift ;; |
| *) echo "Unknown arg: $1"; exit 1 ;; |
| esac |
| shift |
| done |
| |
| if [[ -z ${avm_build_dir} || -z ${stream} || -z ${output} ]]; then |
| echo "Usage: ./convert_stream.sh --avm_build_dir <AVM_BUILD_PATH> --stream <STREAM_PATH> --output <ZIP_OUTPUT_PATH> [--yuv <PATH_TO_ORIG_YUV>] [--limit <NUM_FRAMES>]" |
| exit 1 |
| fi |
| |
| orig_yuv_arg="" |
| if [[ -n ${yuv} ]]; then |
| orig_yuv_arg="--orig_yuv=${yuv}" |
| fi |
| limit_arg="" |
| if [[ -n ${limit} ]]; then |
| limit_arg="--limit=${limit}" |
| fi |
| tmpdir=$(mktemp -d) |
| ${avm_build_dir}/extract_proto --stream ${stream} --output_folder ${tmpdir} ${orig_yuv_arg} ${limit_arg} |
| zip --filesync --recurse-paths --junk-paths ${output} ${tmpdir}/*.pb |
| rm ${tmpdir}/*.pb |
