#!/bin/bash
#
# Copyright (C) 2017 The Android Open Source Project
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#

WINSCOPE_URL='http://go/winscope/'

set -e

help=

for arg in "$@"; do
  case $arg in
    -h|--help) help=1;;
    --);;
    -*) echo "Unknown option: $arg"; help=1;;
    *) outfile="$arg";;
  esac
done

outfileTrans=${outfile}_transactiontrace.pb
outfileSurf=${outfile}_layerstrace.pb

outfileTrans_abs="$(cd "$(dirname "$outfileTrans")"; pwd)/$(basename "$outfileTrans")"
outfileSurf_abs="$(cd "$(dirname "$outfileSurf")"; pwd)/$(basename "$outfileSurf")"

if [ "$help" != "" ]; then
  echo "usage: $0 [-h | --help] [OUTFILE]"
  echo
  echo "Records Transaction traces (default transactiontrace.pb)."
  echo "Records Surface traces (default layerstrace.pb)."
  echo "To view the traces, use $WINSCOPE_URL."
  exit 1
fi

function log_error() {
  echo "FAILED"
}
trap log_error ERR

function start_tracing() {
  echo -n "Starting transaction and surface recording..."
  echo
  adb shell su root service call SurfaceFlinger 1020 i32 1 >/dev/null
  adb shell su root service call SurfaceFlinger 1025 i32 1 >/dev/null
  echo "DONE"
  trap stop_tracing EXIT
}

function stop_tracing() {
  echo -n "Stopping transaction and surface recording..."
  echo
  adb shell su root service call SurfaceFlinger 1020 i32 0 >/dev/null
  adb shell su root service call SurfaceFlinger 1025 i32 0 >/dev/null
  echo "DONE"
  trap - EXIT
}

which adb >/dev/null 2>/dev/null || { echo "ERROR: ADB not found."; exit 1; }
adb get-state 2>/dev/null | grep -q device || { echo "ERROR: No device connected or device is unauthorized."; exit 1; }

start_tracing
read -p "Press ENTER to stop recording" -s x
echo
stop_tracing
adb exec-out su root cat /data/misc/wmtrace/transaction_trace.pb >"$outfileTrans"
adb exec-out su root cat /data/misc/wmtrace/layers_trace.pb >"$outfileSurf"

echo
echo "To view the trace, go to $WINSCOPE_URL, and open"
echo "${outfileTrans_abs}"
echo "${outfileSurf_abs}"