[AddVideoFormat.sh] Frameinfos in alte Aufnahmen einfügen

#!/usr/bin/env bash

#
# Add Video frame informations to vdr recordings made with vdr <= 2.6.4
#
# VERSION=240123

# Variables
VIDEO='/video'  # Home of vdr's videos

f_find_frameinfo() {
  local f_tag='^F '  # The F at the beginning of the line
  IFS=' ' read -r -a f_line < <(grep "$f_tag" "$INFO" 2>/dev/null)
  [[ "$?" -ne 0 ]] && echo "->'Info' not found! Propably a recording from old vdr"
  # echo "> Found in 'Info': f${f_line[1]} w${f_line[2]} h${f_line[3]} s${f_line[4]} a${f_line[5]}"
  [[ -z "${f_line[1]}" ]] && return 1  # No Framerate. Propably a old pes recording
  [[ -z "${f_line[2]}" ]] && return 0  # Width not set; TS recording, but no extra infos
  [[ -n "${f_line[1]}" && -n "${f_line[5]}" ]] && return 2  # Frameinfos already set (VDR >= 2.6.5)
}

f_check_video() {
  local ts_file="${dir}/00001.ts"
                        # Read only 42 packets after seeking to position 01:23: 01:23%+#42
  mapfile -t result < <(ffprobe -v fatal -select_streams v:0 -read_intervals 9:23%+#100 \
                        -show_entries stream=width,height,display_aspect_ratio,r_frame_rate \
                        -of default=nw=1:nk=1 "$ts_file")
  [[ -n "${result[0]}" && -n "${result[1]}" && -n "${result[2]}" && -n "${result[3]}" ]] &&
    return 0  # Everything found…
  return 1  # Error detecting values
}

f_insert_framedata() {
  local fps fps_before="${result[3]%/*}" fps_after="${result[3]#*/}"
  if [[ -n "$fps_after" ]] ; then
    printf -v fps '%.0f' $((fps_before / fps_after))  # No decimals!
  else
    fps="$fps_before"
  fi
  # Order of parameters: Framerate, Width, Height, Scantype, Aspectratio
  sed -i -e "s|^F .*|F ${f_line[1]:-$fps} ${result[0]} ${result[1]} - ${result[2]:='-'}|" "$INFO"
}

f_check_backup() {
  if [[ -e "${dir}/info.bak" ]] ; then
    if [[ "${FORCE:=false}" == 'true' ]] ; then
      cp -f "${dir}/info.bak" "$INFO"
    else
      echo "> Backup of 'info' found! Use '-f' to rescan. Skipping…"
      return 1
    fi
  else
    cp "$INFO" "${dir}/info.bak"
  fi
}
### Start

[[ "$EUID" -ne 0 ]] && { echo 'Please run as root!' ; exit 1 ;}
[[ "$1" == '-f' ]] && FORCE='true'

while IFS= read -r dir; do
  INFO="${dir}/info"
  ((cnt+=1))
  echo -e "\n=> (${cnt}) Checking ${dir}…"
  if [[ ! -e "$INFO" ]] ; then
    echo "> 'info' file not found! Skipping…"
    continue
  fi

  f_check_backup || continue
  f_find_frameinfo || continue

  if f_check_video ; then
    # echo "ffprobe got width ${result[0]}, height ${result[1]}, ar ${result[2]:-}, fps ${result[3]}"
    f_insert_framedata
  else
    echo "!> ffprobe error!"
  fi

done < <(find "${VIDEO}/" -type d -name '*.rec')

echo "Setting owner vdr…"
chown -hR -c vdr:vdr "$VIDEO"

# End

#!/usr/bin/env bash

#
# Add Video frame informations to vdr recordings made with vdr <= 2.6.4
#
# VERSION=240125

# Variables
VIDEO='/video'  # Home of vdr's videos

f_find_frameinfo() {
  local f_tag='^F '  # The F at the beginning of the line
  IFS=' ' read -r -a f_line < <(grep "$f_tag" "$INFO" 2>/dev/null)
  echo "> Found in 'info': FPS: ${f_line[1]}, Width: ${f_line[2]}, Height: ${f_line[3]} ScanType: ${f_line[4]}, Aspect: ${f_line[5]}"
  [[ -z "${f_line[1]}" ]] && return 1  # No Framerate. Propably a old pes recording
  [[ -z "${f_line[2]}" ]] && return 0  # Width not set; TS recording, but no extra infos
  [[ -n "${f_line[1]}" && -n "${f_line[5]}" ]] && return 2  # Frameinfos already set (VDR >= 2.6.5)
}

f_check_video() {
  local ts_file="${dir}/00001.ts"
                        # Read only 42 packets after seeking to position 01:23: 01:23%+#42
  mapfile -t result < <(ffprobe -v fatal -select_streams v:0 -read_intervals 9:23%+#100 \
                        -show_entries stream=width,height,display_aspect_ratio,r_frame_rate \
                        -of default=nw=1:nk=1 "$ts_file")
  [[ -n "${result[0]}" && -n "${result[1]}" && -n "${result[2]}" && -n "${result[3]}" ]] &&
    return 0  # Everything found…
  return 1  # Error detecting values
}

f_insert_framedata() {
  local fps fps_before="${result[3]%/*}" fps_after="${result[3]#*/}"
  if [[ -n "$fps_after" ]] ; then
    printf -v fps '%.0f' $((fps_before / fps_after))  # No decimals!
  else
    fps="$fps_before"
  fi
  echo "> Inserting: F ${f_line[1]:-$fps} ${result[0]} ${result[1]} - ${result[2]:-'-'}"
  # Order of parameters: Framerate, Width, Height, Scantype, Aspectratio
  sed -i -e "s|^F .*|F ${f_line[1]:-$fps} ${result[0]} ${result[1]} - ${result[2]:-'-'}|" "$INFO"
}

f_check_backup() {
  if [[ -e "${dir}/info.bak" ]] ; then
    if [[ "${FORCE:=false}" == 'true' ]] ; then
      cp -f "${dir}/info.bak" "$INFO"
    else
      echo "> Backup of 'info' found! Use '-f' to rescan. Skipping…"
      return 1
    fi
  else
    cp "$INFO" "${dir}/info.bak"
  fi
}

f_restore() {  # Restore original 'info' and delete backup
  if [[ -e "${dir}/info.bak" ]] ; then
    echo "> Restoring original 'info'"
    if cp -f "${dir}/info.bak" "$INFO" ; then
      rm "${dir}/info.bak"
    fi
  else
    echo "!> No backup found! Skipping."
  fi
}

### Start

[[ "$EUID" -ne 0 ]] && { echo 'Please run as root!' ; exit 1 ;}
[[ "$1" == '-f' ]] && FORCE='true'
[[ "$1" == '-r' ]] && RESTORE='true'

while IFS= read -r dir; do
  INFO="${dir}/info"
  ((cnt+=1))
  echo "=> (${cnt}) Checking ${dir}…"
  if [[ "${RESTORE:=false}" == 'true' ]] ; then
    f_restore
  else
    if [[ ! -e "$INFO" ]] ; then
      echo "!> 'info' file not found! Skipping…"
      continue
    fi

    f_check_backup || continue
    f_find_frameinfo || continue

    if f_check_video ; then
      # echo "> ffprobe got width ${result[0]}, height ${result[1]}, ar ${result[2]:-}, fps ${result[3]}"
      f_insert_framedata
    else
      echo "!> ffprobe error!"
    fi
  fi  # if RESTORE
done < <(find "${VIDEO}/" -type d -name '*.rec')

echo "Setting owner of $VIDEO to vdr…"
chown -hR -c vdr:vdr "$VIDEO"

# End

#!/usr/bin/env bash

#
# Add Video frame information's to vdr recordings made with vdr <= 2.6.4
#
# VERSION=240127

### Functions
f_find_frameinfo() {
  f_line=($(grep '^F ' "$INFO" 2>/dev/null))
  #echo "  > Found in 'info': ${f_line[*]}"
  [[ "${#f_line[*]}" -lt 2 ]] && return 1   # No Framerate. Propably an old PES recording
  [[ "${#f_line[*]}" -eq 2 ]] && fps="${f_line[1]}" && return 0   # TS recording, but only framerate set
  [[ "${#f_line[*]}" -eq 6 ]] && return 2   # Frameinfos already set (VDR >= 2.6.5)
}

f_check_video() {
  width='' ; height='' ; aspect_ratio='' ; scantype='' ; frame_rate=''
  local key='' value=''
  local ts_file="${dir}/00001.ts"  # TODO: What if first file is too small?
            # Read only 100 packets after seeking to position 09:23: 09:23%+#100
  result=($(ffprobe -v fatal -select_streams v:0 -read_intervals 9:23%+#100 \
            -show_entries stream=width,height,display_aspect_ratio,field_order,r_frame_rate \
            -of default=nw=1:nk=0 "$ts_file"))
  #echo "  > ffprobe result: ${result[*]}"
  for line in "${result[@]}" ; do
    key="${line%=*}" ; value="${line#*=}"
    case "$key" in
      'width')                width="$value" ;;
      'height')               height="$value" ;;
      'display_aspect_ratio') aspect_ratio="$value" ;;
      'field_order')          scantype=$(echo "$value" | sed -r 's/[bt][bt]/i/g; s/progressive/p/; /^[ip]$/!s/.*/-/') ;;
      'r_frame_rate')         frame_rate="$value" ;;
    esac
  done
  [[ -n "$width" && -n "$height" && -n "$aspect_ratio" && -n "$scantype" && -n "$frame_rate" ]] &&
    return 0  # Everything found…
  return 1  # Error detecting values
}

f_insert_framedata() {
  # Order of parameters: Framerate, Width, Height, Scantype, Aspectratio
  echo "  > Inserting: F $fps $width $height $scantype $aspect_ratio"
  sed -i -e "s|^F .*|F $fps $width $height $scantype ${aspect_ratio}|" "$INFO"
}

f_check_backup() {
  if [[ -e "${INFO}.bak" ]] ; then
    if [[ "${FORCE:=false}" == 'true' ]] ; then
      cp -pf "${INFO}.bak" "$INFO"
    else
      echo "  > Backup of 'info' found! Use '-f' to rescan. Skipping"
      return 1
    fi
  else
    cp -p "$INFO" "${INFO}.bak"
  fi
}

f_restore() {  # Restore original 'info' and delete backup
  if [[ -e "${INFO}.bak" ]] ; then
    echo "  > Restoring original 'info'"
    if cp -pf "${INFO}.bak" "$INFO" ; then
      rm "${INFO}.bak"
    fi
  else
    echo "  !> No backup found! Skipping"
  fi
}

f_help() {
  echo "$0 Usage:"
  echo -e "\t$0 -f  Force processing of already processed 'info'. Reuse of original 'info'!"
  echo -e "\t$0 -r  Restore backed up 'info' files"
  echo -e "\t$0 -v VIDEO_DIR    Path to VDR's video directory (default: /video)"
  exit 0
}

### Start

while getopts ':frv:' opt ; do
  case "$opt" in
    f) FORCE=''true ;;
    r) RESTORE='true' ;;
    v) VIDEO="$OPTARG";;
    *) f_help ;;
  esac
done

if [[ ! -r "${VIDEO:=/video}" ]]; then
  echo "Video directory '${VIDEO}' not found or not readable - exiting"
  exit 1
fi

while IFS= read -r dir; do
  INFO="${dir}/info"
  ((cnt+=1))
  echo "=> (${cnt}) Checking ${dir}"
  if [[ "${RESTORE:=false}" == 'true' ]] ; then
    f_restore
  else
    if [[ -r "${INFO}.vdr" ]] ; then
      echo "  !> Skipping PES recording"
      continue
    fi
    if [[ ! -w "$INFO" ]] ; then
      echo "  !> 'info' file not found or is not writable! Skipping"
      continue
    fi

    f_check_backup || continue
    f_find_frameinfo || continue

    if f_check_video ; then
      echo "  > ffprobe got ${width}x${height} @ ${fps} scantype: $scantype  ar: ${aspect_ratio}"
      f_insert_framedata
    else
      echo "  !> ffprobe error!"
    fi
  fi  # if RESTORE
done < <(find "${VIDEO}/" -type d -name '*.rec')

# End