root/galaxy-central/eggs/bx_python-0.5.0_dev_f74aec067563-py2.6-macosx-10.6-universal-ucs2.egg/EGG-INFO/scripts/tfloc_summary.py

#!/usr/bin/python2.6

"""
Read TFLOC output from stdin and write out a summary in which the nth line 
contains the number of sites found in the nth alignment of the input.

TODO: This is very special case, should it be here?
"""

import sys

counts = dict()

max_index = -1

for line in sys.stdin:
    if line[0].isdigit():
        current_index = int( line )
        max_index = max( current_index, max_index )        
    elif line[0] == "'":
        try: counts[ current_index ] += 1
        except: counts[ current_index ] = 1
    else:
        raise "Invalid input line " + line

for i in range( max_index + 1 ):
    print counts.get( i, 0 )