[Windows/Linux] GC553 的 Switch 完美采集之路

1
#!/bin/bash
2

3
PROGNAME="$(basename $0)"
4

5
# mpv settings
6
MPV_FULLSCREEN=no
7
MPV_VIDEO_SIZE=1920x1080
8
MPV_TITLE="Live Gamer Ultra-Video"
9
MPV_SCREENSHOT_DIR="$(pwd)"
10

11
GETOPT_ARGS=$(getopt -o hc:w:g:de:nm: -l "help","fullscreen","fs" -n "$PROGNAME" -- "$@")
12
[[ $? -ne 0 ]] && exit 1
13
eval set -- "$GETOPT_ARGS"
14

15
while :; do
16
    case "$1" in
17
        -h|--help)
18
            usage
19
            exit 0
20
            ;;
21
        --video-size)
22
            shift
23
            VIDEO_SIZE="$1"
24
            shift
25
            ;;
26
        --video-device)
27
            shift
28
            MPV_VIDEO_SIZE="$1"
29
            shift
30
            ;;
31
        --fullscreen|--fs)
32
            shift
33
            MPV_FULLSCREEN=yes
34
            ;;
35
        --screenshot-directory)
36
            shift
37
            MPV_SCREENSHOT_DIR="$1"
38
            shift
39
            ;;
40
        --title)
41
            shift
42
            MPV_TITLE="$1"
43
            SHIFT
44
            ;;
45
        --)
46
            shift
47
            break
48
            ;;
49
    esac
50
done
51

52
function requirement_check() {
53
  which lsusb    > /dev/null 2>&1 &&
54
  which pactl    > /dev/null 2>&1 &&
55
  which v4l2-ctl > /dev/null 2>&1 &&
56
  which mpv      > /dev/null 2>&1 ||
57
  (echo "Please make sure lsusb, pactl, v4l2-ctl and mpv exist in your system!"; exit 1)
58
}
59

60
function device_check() {
61
  lsusb | grep -q AVerMedia
62
  [[ $? -ne 0 ]] && echo "No capture card found!" && exit 1
63
}
64

65
function get_source() {
66
  PACTL_SOURCE=$(LC_ALL=C pactl list sources | grep AVerMedia -B 3 | grep Source | cut -d '#' -f 2)
67
}
68

69
function audio_on() {
70
  # get audio source id
71
  get_source
72
  # retry once
73
  # may fail if audio source has not been initialized
74
  [[ -z "$PACTL_SOURCE" ]] && sleep 1 && get_source
75
  [[ -z "$PACTL_SOURCE" ]] && echo "Failed to get audio source!" && exit 1
76

77
  # enable audio loopback
78
  PACTL_MODULE_ID=$(pactl load-module module-loopback source="$PACTL_SOURCE")
79
  [[ $? -ne 0 ]] && echo "Failed to initialize audio playback!" && exit 1
80
}
81

82
function audio_off() {
83
  if [ ! -z "$PACTL_MODULE_ID" ]; then
84
    pactl unload-module $PACTL_MODULE_ID
85
  fi
86
}
87

88
function video_config() {
89
  # create tmp config dir
90
  CONFIG_DIR=$(mktemp -d -t ccplay-XXXXXXXX)
91

92
  # create tmp config file
93
  CONFIG_FILE=$(cat << EOF
94
osc=no
95
input-default-bindings=no
96
osd-font-size=24
97

98
profile=low-latency
99
untimed=yes
100

101
# screenshot
102
screenshot-format=png
103
screenshot-template=screenshot-%tY-%tm-%td-%tH:%tM:%tS
104
screenshot-directory=$MPV_SCREENSHOT_DIR
105

106
title=$MPV_TITLE
107
fullscreen=$MPV_FULLSCREEN
108
demuxer-lavf-o=video_size=$MPV_VIDEO_SIZE
109
EOF
110
)
111
  echo "$CONFIG_FILE" > "$CONFIG_DIR/mpv.conf"
112

113
  # create tmp input config
114
  CONFIG_FILE=$(cat << EOF
115
MBTN_LEFT_DBL  cycle fullscreen
116
ESC            set fullscreen no
117
ctrl+c         quit 0
118
F10            screenshot
119
EOF
120
)
121
  echo "$CONFIG_FILE" > "$CONFIG_DIR/input.conf"
122
}
123

124
function video_on() {
125
  # default video device
126
  if [ -z "$VIDEO_DEVICE" ]; then
127
    VIDEO_DEVICE=$(v4l2-ctl --list-devices | grep 'Live Gamer Ultra-Video' -A 5 | grep '/dev' | head -n 1 | xargs)
128
  fi
129

130
  # launch mpv
131
  mpv "av://v4l2:$VIDEO_DEVICE" --config-dir="$CONFIG_DIR" &>/dev/null &
132
}
133

134
function start() {
135
  requirement_check
136
  device_check
137
  audio_on
138
  video_config
139
  video_on
140
  wait $!
141

142
  # https://github.com/mpv-player/mpv/blob/master/DOCS/man/mpv.rst#exit-codes
143
  # file could not be played, retry once
144
  if [ $? -eq 2 ]; then
145
    video_on
146
    wait
147
  fi
148
}
149

150
function clean() {
151
  audio_off
152
  yes | rm "$CONFIG_DIR"/*.conf &>/dev/null
153
  yes | rmdir "$CONFIG_DIR" &>/dev/null
154
}
155

156
trap clean EXIT
157
start