#!/usr/bin/env python

# Convert a number for human consumption

# License: LGPLv2
# Author:
#    P@draigBrady.com
# Changes:
#    V1.0      09 Jan 2006     Initial release

#TODO: support ranges for --column option
#TODO: support converting from back from "human" numbers to "standard" numbers
#TODO: support aligning output like `column -t`
#TODO: support --col-delimiters option

# Divisor can be 1, 1000, 1024
#
# A divisor of 1 => the thousands separator
# appropriate to ones locale is inserted.
# So the locale must be set before this
# functionality is used (see below).
#
# With other divisors the output is aligned
# in a 7 or 8 character column respectively,
# which one can strip() if the display is not
# using a fixed width font.
def human_num(num, divisor=1, power=""):
    num=float(num)
    if divisor == 1:
        return locale.format("%ld",int(num),1)
    elif divisor == 1000:
        powers=[" ","K","M","G","T","P"]
    elif divisor == 1024:
        powers=["  ","Ki","Mi","Gi","Ti","Pi"]
    else:
        raise ValueError, "Invalid divisor"
    if not power: power=powers[0]
    while num >= 1000: #4 digits
        num /= divisor
        power=powers[powers.index(power)+1]
        human_num(num,divisor,power)
    if power.strip():
        return "%6.1f%s" % (num,power)
    else:
        return "%4ld  %s" % (num,power)

import locale
locale.setlocale(locale.LC_ALL,'')

import os
import sys
import getopt
def Usage():
    print "Usage: %s [OPTION] [PATH]" % os.path.split(sys.argv[0])[1]
    print "    --divisor=value   The default value is 1 which means insert thousands separator."
    print "                      Other possible values are 1000 and 1024."
    print "    --columns=1,2,3"
    print "    --help            display help"

try:
    lOpts, lArgs = getopt.getopt(sys.argv[1:], "", ["help","divisor=","columns="])

    if len(lArgs) == 0:
        infile = sys.stdin
    elif len(lArgs) == 1:
        infile = file(lArgs[0])
    else:
        Usage()
        sys.exit(None)

    if ("--help","") in lOpts:
        Usage()
        sys.exit(None)

    divisor=1
    columns=[]
    for opt in lOpts:
        if opt[0] == "--divisor":
            divisor=opt[1]
            if divisor == "1":
                divisor = 1
            elif divisor=="1000" or divisor=="1024":
                divisor = float(divisor)
            else:
                raise getopt.error, "Invalid divisor"
        if opt[0] == "--columns":
            columns=[int(col) for col in opt[1].split(",")]

except getopt.error, msg:
    print msg
    print
    Usage()
    sys.exit(2)

for line in infile:
    line = line.split()
    column=0
    for str in line:
        column+=1
        if not len(columns) or column in columns:
            try:
                str = human_num(str,divisor)
            except:
                pass
        print str,
    print