/* <tapsetdescription>
 * The indent tapset provides functions to generate indented lines for
 * nested kinds of trace messages.  Each line contains a relative
 * timestamp, and the process name / pid.
 * </tapsetdescription>
 */

global _indent_counters, _indent_timestamps

function _generic_indent (idx, desc, delta) 
{
  ts = __indent_timestamp ()
  if (! _indent_counters[idx]) _indent_timestamps[idx] = ts

  # pre-increment for positive delta and post-decrement for negative delta
  x = _indent_counters[idx] + (delta > 0 ? delta : 0)
  _indent_counters[idx] += delta

  return sprintf("%6d %s:%-*s", (ts - _indent_timestamps[idx]), desc, (x>0 ? x-1 : 0), "")
}

/**
 * sfunction thread_indent - returns an amount of space with the current task information
 *
 * @delta: the amount of space added/removed for each call
 *
 * Description: This function returns a string with appropriate
 * indentation for a thread. Call it with a small positive or
 * matching negative delta. If this is the real outermost,
 * initial level of indentation, then the function resets the 
 * relative timestamp base to zero. An example is shown at the 
 * end of this file.
 */
function thread_indent:string (delta:long)
{
  return _generic_indent (tid(), sprintf("%s(%d)", execname(), tid()), delta)
}

/**
 * sfunction indent - returns an amount of space to indent
 *
 * @delta: the amount of space added/removed for each call
 *
 * Description: This function returns a string with appropriate
 * indentation. Call it with a small positive or matching negative
 * delta. Unlike the thread_indent function, the indent does not
 * track individual indent values on a per thread basis.
 */
function indent:string (delta:long){
  return _generic_indent(-1, "",  delta)
}

/* The following example uses thread_indent() to trace the functions
 * called in the drivers/usb/core kernel source. It prints a relative
 * timestamp and the name and ID of the current process, followed by 
 * the appropriate indent and the function name. Note that 
 * 'char' swapper(0) indicates the kernel is running in interrupt
 * context and there is no valid current process.
 *
 * probe kernel.function("*@drivers/usb/core/*") {
 * 	printf("%s -> %s\n, thread_indent(1), probefunc())
 * }
 * probe kernel.function("*@drivers/usb/core/*").return {
 * 	printf("%s <- %s\n", thread_indent(-1), probefunc())
 * }
 *
 * //This prints:
 *  0 swapper(0): -> usb_hcd_irq
 *  8 swapper(0): <- usb_hcd_irq
 *  0 swapper(0): -> usb_hcd_irq
 * 10 swapper(0):  -> usb_hcd_giveback_urb
 * 16 swapper(0):   -> urb_unlink
 * 22 swapper(0):   <- usb_unlink
 * 29 swapper(0):   -> usb_free_urb
 * 35 swapper(0):   <- usb_hcd_urb
 * 39 swapper(0):  <- usb_hcd_giveback_urb
 * 45 swapper(0): <- usb_hcd_irq
 *  0 usb-storage(1338): -> usb_submit_urb
 *  6 usb-storage(1338):  -> usb_hcd_submit_urb
 * 12 usb-storage(1338):   -> usb_get_urb
 * 18 usb-storage(1338):   <- usb_get_urb
 * 25 usb-storage(1338):  <- usb_hcd_submit_urb
 * 29 usb-storage(1338): -> usb_submit_urb
 *  0 swapper(0): -> usb_hcd_irq
 *  7 swapper(0): <- usb_hcd_irq
 */