From 5d6bf8bdbab113e079f5557ecaa0ee1e6a2f3cb5 Mon Sep 17 00:00:00 2001
From: Markus Lehtonen <markus.lehtonen@linux.intel.com>
Date: Mon, 29 Aug 2016 11:48:34 +0300
Subject: build-perf-bisect: multiple possible test methods

Implement -m command line argument for specifying the test method to
use. Default is the existing 'buildtime' method. The patch also adds a
new 'tmpsize' method that measures the size of the tmp directory.

Signed-off-by: Markus Lehtonen <markus.lehtonen@linux.intel.com>
---
 build-perf-bisect.sh | 93 +++++++++++++++++++++++++++++++++++++++-------------
 1 file changed, 71 insertions(+), 22 deletions(-)

diff --git a/build-perf-bisect.sh b/build-perf-bisect.sh
index c98ebe728b..1bd496da36 100755
--- a/build-perf-bisect.sh
+++ b/build-perf-bisect.sh
@@ -21,25 +21,30 @@
 #
 script=`basename $0`
 workdir=`realpath build-perf-bisect`
+test_method="buildtime"
 
 usage () {
 cat << EOF
-Usage: $script [-h] [-d DL_DIR] [-w WORKDIR] BUILD_TARGET THRESHOLD
+Usage: $script [-h] [-d DL_DIR] [-m TEST_METHOD] [-w WORKDIR] BUILD_TARGET THRESHOLD
 
 Optional arguments:
   -h                show this help and exit.
   -d                DL_DIR to use
+  -m                test method, available options are: buildtime, tmpsize
+                        (default: $test_method)
   -w                work directory to use
 EOF
 }
 
-while getopts "hd:w:" opt; do
+while getopts "hd:m:w:" opt; do
     case $opt in
         h)  usage
             exit 0
             ;;
         d)  downloaddir=`realpath "$OPTARG"`
             ;;
+        m)  test_method="$OPTARG"
+            ;;
         w)  workdir=`realpath "$OPTARG"`
             ;;
         *)  usage
@@ -103,6 +108,10 @@ s_to_hms () {
     printf "%d:%02d:%02d" $(($1 / 3600)) $((($1 % 3600) / 60)) $(($1 % 60))
 }
 
+kib_to_gib () {
+    echo `echo -e "scale=2\n$1 / 1024^2" | bc -l`G
+}
+
 time_cmd () {
     log "timing $*"
     /usr/bin/time -o time.log -f '%e' $@ &>> "$log_file"
@@ -117,16 +126,70 @@ run_cmd () {
 }
 
 
+#
+# TEST METHODS
+#
+buildtime () {
+    log "cleaning up build directory"
+    run_cmd rm -rf bitbake.lock conf/sanity_info cache tmp sstate-cache
+
+    log "syncing and dropping caches"
+    run_cmd sync
+    echo 3 | sudo -n -k /usr/bin/tee /proc/sys/vm/drop_caches > /dev/null || exit 255
+    sleep 2
+
+    result=`time_cmd bitbake $1` || exit 125
+    result_h=`s_to_hms $result`
+
+    log "removing build directory"
+    cd $workdir
+    run_cmd rm -rf $builddir
+}
+
+tmpsize () {
+    log "cleaning up build directory"
+    run_cmd rm -rf bitbake.lock conf/sanity_info cache tmp sstate-cache
+
+    log "syncing and dropping caches"
+    run_cmd sync
+    echo 3 | sudo -n -k /usr/bin/tee /proc/sys/vm/drop_caches > /dev/null || exit 255
+    sleep 2
+
+    _time=`time_cmd bitbake $1` || exit 125
+
+    result=`du -s tmp* | cut -f1`
+    result_h=`kib_to_gib $result`
+
+    log "removing build directory"
+    cd $workdir
+    run_cmd rm -rf $builddir
+}
+
+
 #
 # MAIN SCRIPT
 #
 build_target=$1
-threshold=`hms_to_s $2`
+
+builddir="$workdir/build-$git_rev-$timestamp"
+case "$test_method" in
+    buildtime)
+        threshold=`hms_to_s $2`
+        threshold_h=`s_to_hms $threshold`
+        ;;
+    tmpsize)
+        threshold=$2
+        threshold_h=`kib_to_gib $2`
+        ;;
+    *)
+        echo "Invalid test method $test_method"
+        exit 255
+esac
+
 
 #Initialize build environment
 mkdir -p $workdir
-. ./oe-init-build-env "$workdir/build-$git_rev-$timestamp" > /dev/null || exit 255
-builddir=`pwd`
+. ./oe-init-build-env "$builddir" > /dev/null || exit 255
 
 echo DL_DIR = \"$downloaddir\" >> conf/local.conf
 echo CONNECTIVITY_CHECK_URIS = \"\" >> conf/local.conf
@@ -136,26 +199,12 @@ log "TESTING REVISION $git_rev (#$git_rev_cnt)"
 log "fetching sources"
 run_cmd bitbake $build_target -c fetchall
 
-log "cleaning up build directory"
-run_cmd rm -rf bitbake.lock conf/sanity_info cache tmp sstate-cache
-
-log "syncing and dropping caches"
-run_cmd sync
-echo 3 | sudo -n -k /usr/bin/tee /proc/sys/vm/drop_caches > /dev/null || exit 255
-sleep 2
-
-result=`time_cmd bitbake $build_target` || exit 125
-
-log "removing build directory"
-cd $workdir
-run_cmd rm -rf $builddir
+$test_method $build_target
 
-result_hms=`s_to_hms $result`
-threshold_hms=`s_to_hms $threshold`
 if [ $result -lt $threshold ]; then
-    log "OK ($git_rev): $result ($result_hms) < $threshold ($threshold_hms)"
+    log "OK ($git_rev): $result ($result_h) < $threshold ($threshold_h)"
     exit 0
 else
-    log "FAIL ($git_rev): $result ($result_hms) >= $threshold ($threshold_hms)"
+    log "FAIL ($git_rev): $result ($result_h) >= $threshold ($threshold_h)"
     exit 1
 fi
-- 
cgit 1.2.3-korg