summaryrefslogtreecommitdiff
path: root/ar/.local/bin/noisereduce
blob: 709b2575a61161961260f2feadd4e3230aa1c48c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
#!/bin/sh

usage() {
  printf "Usage : noisereduce <input video file> <output video file>\n"
  exit 1
}

# Tests for requirements
ifinstalled ffmpeg || {
  echo >&2 "We require 'ffmpeg' but it's not installed."
  exit 1
}
ifinstalled sox || {
  echo >&2 "We require 'ffmpeg' but it's not installed."
  exit 1
}

if [ "$#" -ne 2 ]; then
  usage
fi

if [ ! -e "$1" ]; then
  printf "File not found: %s\n" "$1"
  exit
fi

if [ -e "$2" ]; then
  printf "File %s already exists, overwrite? [y/N]\n: " "$2"
  read -r yn
  case $yn in
  [Yy]*) ;;
  *) exit ;;
  esac
fi

inBasename=$(basename "$1")
inExt="${inBasename##*.}"

isVideoStr=$(ffprobe -v warning -show_streams "$1" | grep codec_type=video)
if [ -n "$isVideoStr" ]; then
  isVideo=1
  printf "Detected %s as a video file\n" "$inBasename"
else
  isVideo=0
  printf "Detected %s as an audio file\n" "$inBasename"
fi

printf "Sample noise start time [00:00:00]: "
read -r sampleStart
if [ -z "$sampleStart" ]; then sampleStart="00:00:00"; fi
printf "Sample noise end time [00:00:00.900]: "
read -r sampleEnd
if [ -z "$sampleEnd" ]; then sampleEnd="00:00:00.900"; fi
printf "Noise reduction amount [0.21]: "
read -r sensitivity
if [ -z "$sensitivity" ]; then sensitivity="0.21"; fi

tmpVidFile="/tmp/noiseclean_tmpvid.$inExt"
tmpAudFile="/tmp/noiseclean_tmpaud.wav"
noiseAudFile="/tmp/noiseclean_noiseaud.wav"
noiseProfFile="/tmp/noiseclean_noise.prof"
tmpAudCleanFile="/tmp/noiseclean_tmpaud-clean.wav"

printf "Cleaning noise on %s...\n" "$1"

if [ $isVideo -eq "1" ]; then
  ffmpeg -v warning -y -i "$1" -qscale:v 0 -vcodec copy -an "$tmpVidFile"
  ffmpeg -v warning -y -i "$1" -qscale:a 0 "$tmpAudFile"
else
  cp "$1" "$tmpAudFile"
fi
ffmpeg -v warning -y -i "$1" -vn -ss "$sampleStart" -t "$sampleEnd" "$noiseAudFile"
sox "$noiseAudFile" -n noiseprof "$noiseProfFile"
sox "$tmpAudFile" "$tmpAudCleanFile" noisered "$noiseProfFile" "$sensitivity"
if [ $isVideo -eq "1" ]; then
  ffmpeg -v warning -y -i "$tmpAudCleanFile" -i "$tmpVidFile" -vcodec copy -qscale:v 0 -qscale:a 0 "$2"
else
  cp "$tmpAudCleanFile" "$2"
fi

printf "Done"