| #!/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 |