#!/bin/bash

if [[ "$1" == *help ]]; then
  echo "Example usage: ${0} [-duration 10] [-sound plughw:1,0]"
  echo "optional -duration is in seconds"
  echo "optional -sound values: [none, anullsrc, plughw:1,0]"
  echo "default -sound is none i.e. no sound"
  exit 1
fi

#VIDEO_RES_HEIGHT="1080"
#VIDEO_RES_WIDTH="1920"
#VIDEO_FRAMERATE="30"

#VIDEO_RES_HEIGHT="972"
#VIDEO_RES_WIDTH="1296"
#VIDEO_FRAMERATE="43"

VIDEO_RES_HEIGHT="480"
VIDEO_RES_WIDTH="640"
VIDEO_FRAMERATE="10"
VIDEO_BITRATE="700000"

VIDEO_DIR="video-storage"
VIDEO_FILE="${VIDEO_DIR}/camera_$(date +%Y-%m-%d_%H-%M-%S)"
VIDEO_FILE_SEGMENTED="${VIDEO_DIR}/camera_%Y-%m-%d_%H-%M-%S"
VIDEO_FILE_EXT="mp4"
VIDEO_DURATION="1000000000"
SOUND_SOURCE="none"

LIVE_KEY="LIVE-STREAM-KEY"
PEERTUBE_LIVE="rtmp://peertube.virtonline.eu:1935/live/"

mkdir -p $VIDEO_DIR

while [[ $# -gt 0 ]]; do
  case $1 in
    -sound)
      if [ -n "$2" ]; then
        SOUND_SOURCE="$2"
        shift
      fi
      ;;
    -duration)
     if [ -n "$2" ]; then
        VIDEO_DURATION="$2"
        shift
     fi
      ;;
    *)
     echo "Unknown parameter $1"
     exit 1
  esac
  shift
done

# Set up a trap to check for SIGINT
trap "pkill rpicam-vid; exit 130" SIGINT

# Start the pipeline
rpicam-vid --inline \
		-n \
                -t 0 \
		--width ${VIDEO_RES_WIDTH} \
		--height ${VIDEO_RES_HEIGHT} \
		--nopreview \
		--exposure long \
		--sharpness 1.2 \
		--contrast 1.4 \
		--brightness 0.2 \
		--saturation 1.0 \
		--awb auto \
		--denoise auto \
		--rotation 0 \
		--autofocus-mode auto \
		--codec h264 \
		--framerate ${VIDEO_FRAMERATE} \
		-b ${VIDEO_BITRATE} \
		-o - | \

if [ "${SOUND_SOURCE}" = "none" ]; then
    # ffmpeg params without audio
    ffmpeg -hide_banner \
                -thread_queue_size 1024 \
		-t ${VIDEO_DURATION} \
                -i pipe:0 \
                -c:v copy \
		-f tee -map 0:0 -flags +global_header -flvflags no_duration_filesize \
		"[f=segment \
			:segment_time=3600 \
			:reset_timestamps=1 \
			:strftime=1 \
			:segment_format=mp4 \
			:onfail=ignore \
		]${VIDEO_FILE_SEGMENTED}.${VIDEO_FILE_EXT} \
		| \
		[f=fifo \
			:fifo_format=flv \
			:drop_pkts_on_overflow=1 \
			:attempt_recovery=1 \
			:recovery_wait_time=1 \
			:use_wallclock_as_timestamps=1 \
		]${PEERTUBE_LIVE}${LIVE_KEY}" \
		&
elif [ "${SOUND_SOURCE}" = "anullsrc" ]; then
    # ffmpeg params with silent audio
    ffmpeg -hide_banner \
                -f lavfi \
                -i anullsrc=channel_layout=stereo:sample_rate=44100 \
                -thread_queue_size 1024 \
                -use_wallclock_as_timestamps 1 \
                -i pipe:0 \
                -c:v copy \
                -c:a aac \
                -preset fast \
                -strict experimental \
                -f mp4 \
                -t ${VIDEO_DURATION} \
                ${VIDEO_FILE}.${VIDEO_FILE_EXT} &
else
    # ffmpeg params with audio
    ffmpeg -hide_banner \
		-f alsa \
                -thread_queue_size 1024 \
                -ac 2 \
                -i ${SOUND_SOURCE} \
                -thread_queue_size 1024 \
                -use_wallclock_as_timestamps 1 \
                -i pipe:0 \
                -c:v copy \
                -c:a aac \
		-b:a 32k \
                -ar 44100 \
                -f mp4 \
                -t ${VIDEO_DURATION} \
                ${VIDEO_FILE}.${VIDEO_FILE_EXT} &
fi

# Get process ID
pid=$!

# Wait for process to finish or get terminated signal
wait || pkill rpicam-vid; exit 130