home/bin/ps2dot

   1 #! /bin/sh
   2
   3 usage() {
   4     echo "EXAMPLE (whole tree)   : $0 | neato -T png > ps.png && open ps.png"
   5     echo "EXAMPLE (user clusters): $0 | sdp   -T png > ps.png && open ps.png"
   6 }
   7
   8
   9 compile() {
  10     awk -v kernel="$(uname -v)" -v whoami="$(whoami)" \
  11     '
  12         function num_scale(src_cur, src_max, dst_min, dst_max) {
  13             return dst_min + ((src_cur * (dst_max - dst_min)) / src_max)
  14         }
  15
  16         function vert_print(v,    _color, _fontcolor, _shape, _state, _size, _height, _label, _label_base, _label_ext) {
  17             _state = child2state[v]
  18             _style = "filled,solid"
  19
  20             # -----------------------------------------------------------------
  21             # Sleeping/idling
  22             # -----------------------------------------------------------------
  23             # D uninterruptible sleep (usually IO)
  24             if (_state == "D") {
  25                 _shape = "circle"
  26             # I Idle kernel thread
  27             } else if (_state == "I") {
  28                 _shape = "circle"
  29             # S interruptible sleep (waiting for an event to complete)
  30             } else if (_state == "S") {
  31                 _shape = "circle"
  32             # -----------------------------------------------------------------
  33             # Running
  34             # -----------------------------------------------------------------
  35             # R running or runnable (on run queue)
  36             } else if (_state == "R") {
  37                 _shape = "rarrow"
  38             # -----------------------------------------------------------------
  39             # Stopped
  40             # -----------------------------------------------------------------
  41             # T stopped by job control signal
  42             } else if (_state == "T") {
  43                 _shape = "square"
  44             # t stopped by debugger during the tracing
  45             } else if (_state == "t") {
  46                 _shape = "square"
  47             # -----------------------------------------------------------------
  48             # Dead
  49             # -----------------------------------------------------------------
  50             # Z defunct ("zombie") process, terminated but not reaped by its parent
  51             } else if (_state == "Z") {
  52                 _shape = "Msquare"
  53                 _style = "solid"
  54             # -----------------------------------------------------------------
  55             # UNKNOWN STATE
  56             # -----------------------------------------------------------------
  57             } else {
  58                 _shape = "doublecircle"
  59             }
  60
  61             _color =\
  62                 num_scale(\
  63                     child2cpu[v],
  64                     max_cpu,
  65                     VERT_COLORSCHEME_MAX,
  66                     VERT_COLORSCHEME_MIN\
  67                 )
  68             _size =\
  69                 num_scale(\
  70                     child2mem[v],
  71                     max_mem,
  72                     1,
  73                     4\
  74                 ) / 4
  75             _height = _size
  76             _width  = _size
  77             _fontcolor = \
  78                 _color == VERT_COLORSCHEME_MAX || _color == VERT_COLORSCHEME_MIN \
  79                 ? sprintf("/%s/%d", VERT_COLORSCHEME, VERT_COLORSCHEME_MID) \
  80                 : sprintf("/%s/%d", "greys9", 9)
  81             _fontcolor = \
  82                 _size < 0.5 \
  83                 ? sprintf("/%s/%d", "greys9", 9) \
  84                 : _fontcolor
  85             _label_base = \
  86                 sprintf("%s\n%d", child2comm[v], v)
  87             _label_ext = \
  88                 _size >= 0.5 \
  89                 ? sprintf("\ncpu: %.1f%%\nmem: %.1f%%", child2cpu[v], child2mem[v]) \
  90                 : ""
  91             _label = _label_base _label_ext
  92             printf(\
  93                 "\"%d\"\
  94                 [ fontsize=8 \
  95                 , fixedsize=true \
  96                 , height=%f \
  97                 , width=%f \
  98                 , border=1 \
  99                 , style=\"%s\" \
 100                 , fontname=Helvetica \
 101                 , label=\"%s\" \
 102                 , shape=\"%s\" \
 103                 , fillcolor=\"/%s/%d\" \
 104                 , fontcolor=\"%s\" \
 105                 ];",
 106                 v,
 107                 _height,
 108                 _width,
 109                 _style,
 110                 _label,
 111                 _shape,
 112                 VERT_COLORSCHEME,
 113                 _color,
 114                 _fontcolor\
 115             )
 116         }
 117
 118         function edge_print(child,    _parent) {
 119             _parent = child2parent[child]
 120             printf(\
 121                 "\"%s\" -> \"%s\"\
 122                 [ fontsize=8 \
 123                 , fontname=Helvetica \
 124                 , len=2.0 \
 125                 , color=\"%s\" \
 126                 ];\n",
 127                 _parent,
 128                 child,
 129                 EDGE_COLOR\
 130             )
 131         }
 132
 133         BEGIN {
 134             # Hot->Cold gradual colorschemes:
 135             # - rdbu11
 136             # - rdbu9
 137             # - rdbu8
 138             # - rdylgn10  # 3 - 11
 139
 140             # Light->Dark gradual colorschemes:
 141             # - reds9
 142             # - blues9
 143             # - orrd9
 144             # - oranges9
 145             # - bupu9
 146             # - greys9
 147
 148             VERT_COLORSCHEME_MIN = 1
 149             VERT_COLORSCHEME_MID = 4
 150             VERT_COLORSCHEME_MAX = 8
 151             VERT_COLORSCHEME = "rdylgn10"
 152
 153             EDGE_COLOR = "/ylorbr9/3"
 154
 155             child2comm[0] = "swapper/sched"
 156         }
 157
 158         NR > 1 {
 159             parent2child_count[$2]++
 160             max_children = \
 161                 parent2child_count[$2] > max_children\
 162                 ? parent2child_count[$2]\
 163                 : max_children
 164             child2parent[$1]    = $2
 165             child2user_id[$1]   = $3
 166             child2user_name[$1] = $4
 167             child2nice[$1]      = $5
 168             child2state[$1]     = $6
 169             child2cpu[$1]       = $7
 170             child2mem[$1]       = $8
 171             child2comm[$1]      = $9
 172             user_names[$4]      = 1
 173             max_cpu = $7 > max_cpu ? $7 : max_cpu
 174             max_mem = $8 > max_mem ? $8 : max_mem
 175         }
 176
 177         END {
 178             print "strict digraph G {";
 179
 180             print "start=0;";
 181             print "fontsize=8;";
 182             print "fontname=Helvetica;";
 183             print "label=\"" kernel "\";";
 184             print "fontcolor=\"/greys9/9\";"
 185
 186             ##### Vertices (clustered by user)
 187             for (user_name in user_names) {
 188                 printf "subgraph \"cluster_%s\" {\n", user_name
 189                 printf "label=\"%s\"\n", user_name
 190                 for (c in child2parent)
 191                     if (child2user_name[c] == user_name)
 192                         vert_print(c)
 193                 print "}"
 194             }
 195
 196             ##### Vertices (without a user)
 197             for (c in child2comm)
 198                 if (!child2user_name[c])
 199                     vert_print(c)
 200
 201             ##### Edges (across clusters)
 202             for (c in child2parent)
 203                 edge_print(c)
 204
 205             print "}";
 206         }
 207     '
 208 }
 209
 210
 211 procs() {
 212     if [ "$(uname)" = 'Linux' ]; then
 213         ps -eo  'pid,ppid,euid,euser,nice,s,%cpu,%mem,comm'
 214     else
 215         ps -eco 'pid,ppid,euid,euser,nice,s,%cpu,%mem,comm'
 216     fi
 217 }
 218
 219
 220 main() {
 221     case "$1" in
 222         '--help') usage
 223         ;;
 224         *) procs | grep "$1" | compile
 225         ;;
 226     esac
 227 }
 228
 229
 230 main "$1"