#!/bin/bash

set -e

set -o pipefail

# Build WebKit, run benchmarks, and spit out compressed PGO profiles

DisplayHelp() {

echo "Usage: build-and-collect-pgo-profiles [ options ]"

echo " -h Show this help message."

echo " -o <directory> Directory in which output files are stored. Default: /Volumes/WebKit/BenchmarkProfiles/."

echo " -a <app> Path to Safari.app to generate profiles for. If not specified, the script will build WebKit."

echo " -d <directory> Path to build directory. Use seperately from -a."

echo " -y Use for automated collection. No user input."

}

BASE="/Volumes/WebKit/BenchmarkProfiles/"

while getopts "o:a:d:yh" flag; do

case "${flag}" in

o) BASE=${OPTARG};;

a) APP=${OPTARG};;

d) BUILD=${OPTARG};;

y) NINPUT=true ;;

h)

DisplayHelp

exit;;

esac

done

if [[ -n $APP ]] ; then

echo "app: $APP"

fi

if [[ -n $BUILD ]]; then

echo "build: $BUILD"

fi

echo "base: $BASE"

if [ ! -z $APP ] && [ ! -z $BUILD ] ; then

echo "These flags (-a and -d) cannot be used together."

DisplayHelp

exit

fi

while [[ ! "$NINPUT" ]] ; do

read -p "Have you read the source of this script, and do you understand that it is potentially destructive? [y/N]" yn

case $yn in

[Yy]* ) break;;

[Nn]* ) exit;;

* ) echo "Please answer yes or no.";;

esac

done

echo "Using output directory: $BASE"

if [[ ! -d "$BASE" ]] ; then

echo "$BASE is missing, creating directory."

mkdir -p "$BASE"

fi

if [[ -z $APP ]] ; then

cd Internal

echo "Building WebKit..."

rm -rf ../OpenSource/WebKitBuild

make release WK_LTO_MODE=thin ENABLE_LLVM_PROFILE_GENERATION=ON

cd ../

else

if [[ ! -e "$APP" ]] ; then

echo "$APP is missing, aborting."

DisplayHelp

exit

fi

echo "Using .app: $APP"

fi

benchmarks=(jetstream3 speedometer3 motionmark)

if [[ -n $APP ]] ; then

run_benchmark_args=(--browser-path "$APP")

deployment_target=$(vtool -show-build "$APP"/Contents/Frameworks/JavaScriptCore.framework/Versions/A/JavaScriptCore | grep -m1 minos | tr -d -c .0-9)

else

run_benchmark_args=(--build-directory $BUILD)

deployment_target=$(vtool -show-build "$BUILD"/JavaScriptCore.framework/Versions/A/JavaScriptCore | grep -m1 minos | tr -d -c .0-9)

fi

# The reported target triple is based on an arbitrary binary (JavaScriptCore)

# and this machine's architecture.

target_triple=$(machine)-apple-macos${deployment_target}

COMPRESSED_PROFILE_SUB_PATH="Internal/WebKit/WebKitAdditions/Profiling/$target_triple"

SPTH='OpenSource/Tools/Scripts'

$SPTH/collect-pgo-profiles --benchmarks ${benchmarks[@]} --output-directory $BASE --compressed-profile-sub-path $COMPRESSED_PROFILE_SUB_PATH ${run_benchmark_args[@]}