summaryrefslogtreecommitdiffstats
path: root/screencap/screencap.sh
blob: 53284b3e46ea422e3c1f394c428fe1b0465227b8 (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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
#!/bin/sh
#
# Copyright (C) 2017 Wolfgang Wiedmeyer <wolfgit@wiedmeyer.de>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

set -e

FFMEG="ffmpeg"
ADB="adb"
FB_PATH="/dev/graphics/fb0"

TMP_FB="fb_raw"
TMP_FB_RS="fb_raw_rs"
RECOVERY=false

print_usage () {
    echo
    echo "Usage: $0 [OPTIONS] -f OUTPUT.PNG"
    echo "       $0 [OPTIONS] --filename OUTPUT.PNG"
    echo
    echo "Options:"
    echo "-r, --recovery        Device is booted in recovery mode"
    echo "-s, --screen-size     Specify screen size (mandatory in recovery mode)"
    echo
}

print_recovery_advice () {
    echo "For screenshots in recovery mode, the screen size needs to be specified,"
    echo "e.g. with \"-s 720x1280\""
}

get_screen_size () {
    if [ -f $( $ADB shell command -v wm ) ]; then
	echo "Screen size can't be determined. Are you in recovery mode?"
	print_recovery_advice
	exit 1
    fi

    SCREEN_SIZE=$( $ADB shell wm size )
    SCREEN_SIZE=$( echo ${SCREEN_SIZE#*: } | tr -d '\r' )

    echo "Screen size is $SCREEN_SIZE"
}

take_screenshot () {
    SCREEN_SIZE_X=${SCREEN_SIZE%x*}
    SCREEN_SIZE_Y=${SCREEN_SIZE#*x}

    if [ $RECOVERY = true  ]; then
	BS=$(( $SCREEN_SIZE_X * 4 ))
	PIX_FMT="bgr0"
    else
	BS=$(( $SCREEN_SIZE_X * 2 ))
	PIX_FMT="rgb565"
    fi

    $ADB pull $FB_PATH $TMP_FB
    dd bs=$BS count=$SCREEN_SIZE_Y if=$TMP_FB of=$TMP_FB_RS

    $FFMEG -vcodec rawvideo -f rawvideo -pix_fmt $PIX_FMT -s $SCREEN_SIZE \
	   -i $TMP_FB_RS -f image2 -vcodec png $OUTFILE

    # cleanup
    rm -f $TMP_FB $TMP_FB_RS
}

while true
do
    case "$1" in
	-r|--recovery)
	    RECOVERY=true
	    shift
	    ;;
	-f|--filename)
	    OUTFILE="$2"
	    shift 2
	    ;;
	-s|--screen-size)
	    SCREEN_SIZE="$2"
	    shift 2
	    ;;
	"")
	    break
	    ;;
	*)
	    echo "Unknown option"
	    print_usage
	    exit 1
	    ;;
    esac
done

if [ "$OUTFILE" = "" ]; then
    echo "No output file specified"
    print_usage
    exit 1
fi

if [ $RECOVERY = true ] && [ -f $SCREEN_SIZE ]; then
    print_recovery_advice
    exit 1
fi

if [ $RECOVERY = false ] && [ -f $SCREEN_SIZE ]; then
    get_screen_size
fi

take_screenshot

echo "Finished successfully"