Home | History | Annotate | Download | only in scripts 
      1 #!/bin/bash
      2 
      3 ########################
      4 # Function definitions #
      5 ########################
      6 
      7 ## Print the logarithm base 2 of $1 on stdout.
      8 function log2 {
      9   local i
     10 
     11   for ((i=0;i<64;i++))
     12   do
     13     if [ $((2**i)) = "$1" ]; then
     14       echo $i
     15       return 0
     16     fi
     17   done
     18   echo error
     19   return 1
     20 }
     21 
     22 ## Print the size of the level 2 cache in bytes on stdout.
     23 function get_cache_size {
     24   local s
     25   if [ -e /sys/devices/system/cpu/cpu0/cache/index2/size ]; then
     26     s="$(</sys/devices/system/cpu/cpu0/cache/index2/size)"
     27   else
     28     s="$(cat /proc/cpuinfo|while read a b c d e ; do if [ "$a" = cache -a "$b" = size ]; then echo $d $e; break; fi; done)"
     29     s="${s%B}"
     30   fi
     31   if [ "${s%M}" != "$s" ]; then
     32     echo $((${s%M}*1024*1024))
     33   elif [ "${s%K}" != "$s" ]; then
     34     echo $((${s%K}*1024))
     35   else
     36     echo $s
     37   fi
     38 }
     39 
     40 ## Read zero or more lines from stdin, and print the average and standard
     41 #  deviation per column. n is the number of lines, m the number of columns.
     42 #  Each line must have the same number of columns.
     43 function avgstddev {
     44   awk '{n++;m=NF;for(i=1;i<=NF;i++){sum[i]+=$i;sumsq[i]+=$i*$i}}END{for(i=1;i<=m;i++){d=sumsq[i]/n-sum[i]*sum[i]/n/n;printf "%.2f %.2f ",sum[i]/n,(d>=0.0001?sqrt(d):0.01)}}'
     45 }
     46 
     47 ## Query the virtual memory size for the last invocation of command $1 from
     48 #  the information logged by the kernel (BSD process accounting).
     49 function query_cmd_vsz {
     50   local pacct
     51 
     52   if [ ! -e /usr/sbin/dump-acct ]; then
     53     echo "Error: userspace tools for BSD process accounting have not been" >&2
     54     echo "installed. Please install the acct package (Debian systems)."    >&2
     55     return 1
     56   fi
     57 
     58   if ! { dump-acct -h 2>&1 | grep -q -w format; }; then
     59     echo "Error: the installed version of dump-acct is not recent enough." >&2
     60     return 1
     61   fi
     62 
     63   if [ -e /var/log/account/pacct ]; then
     64     pacct=/var/log/account/pacct
     65   elif [ -e /var/account/pacct ]; then
     66     pacct=/var/account/pacct
     67   else
     68     echo "Where is the pacct file ?" >&2
     69     return 1
     70   fi
     71   /usr/sbin/dump-acct "${pacct}" | \
     72     grep -- "^$(basename "$1").*|v3|" | \
     73     cut -f8 -d'|' | \
     74     tail -n 1 | \
     75     { read vsz; echo ${vsz:-0}; }
     76 }
     77 
     78 ## Query the virtual memory size for the last invocation of command $1 from
     79 #  the information logged by the kernel (BSD process accounting).
     80 function query_vsz {
     81   local cmd tool
     82 
     83   cmd="$(basename "$1")"
     84   if [ "${cmd}" = "vg-in-place" ]; then
     85     tool="tool-not-found"
     86     for arg in "${@}"
     87     do
     88       if [ "${arg#--tool=}" != "${arg}" ]; then
     89         tool="${arg#--tool=}"
     90 	break
     91       fi
     92     done
     93     vsz_tool="$(query_cmd_vsz "${tool}")"
     94     awk "END{print $(query_cmd_vsz ${cmd}) + ${vsz_tool:-0}}" \
     95       </dev/null
     96   else
     97     query_cmd_vsz "${cmd}"
     98   fi
     99 }
    100 
    101 ## Echo all arguments on stderr, run the command passed in $1 .. ${$#} three
    102 #  times, pass the output of ${test_input} -p${p} to the command, write the
    103 #  command output to the file specified in ${test_output}, and print the
    104 #  runtime of the command on stdout.
    105 function measure_runtime {
    106   local i
    107 
    108   echo "$@" >&2
    109   if [ "${test_output}" != "" ]; then
    110     echo "$@" >"${test_output}"
    111   fi
    112   for ((i=0;i<3;i++))
    113   do
    114     echo -n "$("${test_input:-true}" $p | \
    115       /usr/bin/time --format="%e" "$@" 2>&1 | \
    116       tee -a "${test_output:-/dev/null}" | \
    117       tail -n 1) "
    118     query_vsz "$@"
    119   done
    120 }
    121 
    122 ## Print the average runtime of the command passed in $5 .. ${$#}, the ratio
    123 #  of the runtime to $1 +/- $2 and the ratio of the VSZ to $3 +/- $4.
    124 function print_runtime_ratio {
    125   local tmp avg1="$1" stddev1="$2" vsz1="$3" vszdev1="$4"
    126   local avg2 stddev2 vsz2 vszdev2
    127 
    128   if [ "${avg1}" = "" -o "${stddev1}" = "" -o "${vsz1}" = "" -o "${vszdev1}" = "" ]; then
    129     echo "Error: invalid arguments ($@)."
    130     exit 1
    131   fi
    132 
    133   shift
    134   shift
    135   shift
    136   shift
    137 
    138   tmp="/tmp/test-timing.$$"
    139   rm -f "${tmp}"
    140 
    141   measure_runtime "$@" | avgstddev > "$tmp"
    142   read avg2 stddev2 vsz2 vszdev2 < "$tmp"
    143   echo "Average time: ${avg2} +/- ${stddev2} seconds / VSZ ${vsz2} +/- ${vszdev2} KB"
    144   awk "END{printf "'"'"Ratio = %.2f +/- %.2f; VSZ ratio: %.2f +/- %.2f\n"'"'", ${avg2}/${avg1}, ${avg2}/${avg1}*(${stddev1}/${avg1}+${stddev2}/${avg2}), ${vsz2}/${vsz1}, ${vsz2}/${vsz1}*(${vszdev1}/${vsz1}+${vszdev2}/${vsz2})}" </dev/null
    145 }
    146 
    147