hc
2025-02-14 bbb9540dc49f70f6b703d1c8d1b85fa5f602d86e
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
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
#!/bin/bash
# SPDX-License-Identifier: GPL-2.0-only
 
# Sergey Senozhatsky, 2015
# sergey.senozhatsky.work@gmail.com
#
 
 
# This program is intended to plot a `slabinfo -X' stats, collected,
# for example, using the following command:
#   while [ 1 ]; do slabinfo -X >> stats; sleep 1; done
#
# Use `slabinfo-gnuplot.sh stats' to pre-process collected records
# and generate graphs (totals, slabs sorted by size, slabs sorted
# by size).
#
# Graphs can be [individually] regenerate with different ranges and
# size (-r %d,%d and -s %d,%d options).
#
# To visually compare N `totals' graphs, do
# slabinfo-gnuplot.sh -t FILE1-totals FILE2-totals ... FILEN-totals
#
 
min_slab_name_size=11
xmin=0
xmax=0
width=1500
height=700
mode=preprocess
 
usage()
{
   echo "Usage: [-s W,H] [-r MIN,MAX] [-t|-l] FILE1 [FILE2 ..]"
   echo "FILEs must contain 'slabinfo -X' samples"
   echo "-t             - plot totals for FILE(s)"
   echo "-l             - plot slabs stats for FILE(s)"
   echo "-s %d,%d        - set image width and height"
   echo "-r %d,%d        - use data samples from a given range"
}
 
check_file_exist()
{
   if [ ! -f "$1" ]; then
       echo "File '$1' does not exist"
       exit 1
   fi
}
 
do_slabs_plotting()
{
   local file=$1
   local out_file
   local range="every ::$xmin"
   local xtic=""
   local xtic_rotate="norotate"
   local lines=2000000
   local wc_lines
 
   check_file_exist "$file"
 
   out_file=`basename "$file"`
   if [ $xmax -ne 0 ]; then
       range="$range::$xmax"
       lines=$((xmax-xmin))
   fi
 
   wc_lines=`cat "$file" | wc -l`
   if [ $? -ne 0 ] || [ "$wc_lines" -eq 0 ] ; then
       wc_lines=$lines
   fi
 
   if [ "$wc_lines" -lt "$lines" ]; then
       lines=$wc_lines
   fi
 
   if [ $((width / lines)) -gt $min_slab_name_size ]; then
       xtic=":xtic(1)"
       xtic_rotate=90
   fi
 
gnuplot -p << EOF
#!/usr/bin/env gnuplot
 
set terminal png enhanced size $width,$height large
set output '$out_file.png'
set autoscale xy
set xlabel 'samples'
set ylabel 'bytes'
set style histogram columnstacked title textcolor lt -1
set style fill solid 0.15
set xtics rotate $xtic_rotate
set key left above Left title reverse
 
plot "$file" $range u 2$xtic title 'SIZE' with boxes,\
   '' $range u 3 title 'LOSS' with boxes
EOF
 
   if [ $? -eq 0 ]; then
       echo "$out_file.png"
   fi
}
 
do_totals_plotting()
{
   local gnuplot_cmd=""
   local range="every ::$xmin"
   local file=""
 
   if [ $xmax -ne 0 ]; then
       range="$range::$xmax"
   fi
 
   for i in "${t_files[@]}"; do
       check_file_exist "$i"
 
       file="$file"`basename "$i"`
       gnuplot_cmd="$gnuplot_cmd '$i' $range using 1 title\
           '$i Memory usage' with lines,"
       gnuplot_cmd="$gnuplot_cmd '' $range using 2 title \
           '$i Loss' with lines,"
   done
 
gnuplot -p << EOF
#!/usr/bin/env gnuplot
 
set terminal png enhanced size $width,$height large
set autoscale xy
set output '$file.png'
set xlabel 'samples'
set ylabel 'bytes'
set key left above Left title reverse
 
plot $gnuplot_cmd
EOF
 
   if [ $? -eq 0 ]; then
       echo "$file.png"
   fi
}
 
do_preprocess()
{
   local out
   local lines
   local in=$1
 
   check_file_exist "$in"
 
   # use only 'TOP' slab (biggest memory usage or loss)
   let lines=3
   out=`basename "$in"`"-slabs-by-loss"
   `cat "$in" | grep -A "$lines" 'Slabs sorted by loss' |\
       grep -E -iv '\-\-|Name|Slabs'\
       | awk '{print $1" "$4+$2*$3" "$4}' > "$out"`
   if [ $? -eq 0 ]; then
       do_slabs_plotting "$out"
   fi
 
   let lines=3
   out=`basename "$in"`"-slabs-by-size"
   `cat "$in" | grep -A "$lines" 'Slabs sorted by size' |\
       grep -E -iv '\-\-|Name|Slabs'\
       | awk '{print $1" "$4" "$4-$2*$3}' > "$out"`
   if [ $? -eq 0 ]; then
       do_slabs_plotting "$out"
   fi
 
   out=`basename "$in"`"-totals"
   `cat "$in" | grep "Memory used" |\
       awk '{print $3" "$7}' > "$out"`
   if [ $? -eq 0 ]; then
       t_files[0]=$out
       do_totals_plotting
   fi
}
 
parse_opts()
{
   local opt
 
   while getopts "tlr::s::h" opt; do
       case $opt in
           t)
               mode=totals
               ;;
           l)
               mode=slabs
               ;;
           s)
               array=(${OPTARG//,/ })
               width=${array[0]}
               height=${array[1]}
               ;;
           r)
               array=(${OPTARG//,/ })
               xmin=${array[0]}
               xmax=${array[1]}
               ;;
           h)
               usage
               exit 0
               ;;
           \?)
               echo "Invalid option: -$OPTARG" >&2
               exit 1
               ;;
           :)
               echo "-$OPTARG requires an argument." >&2
               exit 1
               ;;
       esac
   done
 
   return $OPTIND
}
 
parse_args()
{
   local idx=0
   local p
 
   for p in "$@"; do
       case $mode in
           preprocess)
               files[$idx]=$p
               idx=$idx+1
               ;;
           totals)
               t_files[$idx]=$p
               idx=$idx+1
               ;;
           slabs)
               files[$idx]=$p
               idx=$idx+1
               ;;
       esac
   done
}
 
parse_opts "$@"
argstart=$?
parse_args "${@:$argstart}"
 
if [ ${#files[@]} -eq 0 ] && [ ${#t_files[@]} -eq 0 ]; then
   usage
   exit 1
fi
 
case $mode in
   preprocess)
       for i in "${files[@]}"; do
           do_preprocess "$i"
       done
       ;;
   totals)
       do_totals_plotting
       ;;
   slabs)
       for i in "${files[@]}"; do
           do_slabs_plotting "$i"
       done
       ;;
   *)
       echo "Unknown mode $mode" >&2
       usage
       exit 1
   ;;
esac