#!/usr/bin/env python2.3

"""
Read a MAF from stdin and break into a set of mafs containing 
no more than a certain number of columns
"""

usage = "usage: %prog"

import sys, string
import bx.align.maf
from optparse import OptionParser

import psyco_full

INF="inf"

def __main__():

    # Parse command line arguments

    parser = OptionParser( usage=usage )
    parser.add_option( "-o", "--outprefix", action="store", default="" )
    ( options, args ) = parser.parse_args()

    out_prefix = options.outprefix

    maf_reader = bx.align.maf.Reader( sys.stdin )

    writers = {}

    for m in maf_reader:
        
        writer_key = m.components[0].src

        if not writers.has_key( writer_key ):
            writer = bx.align.maf.Writer( file( "%s%s.maf" % ( out_prefix, writer_key ), "w" ) )
            writers[ writer_key ] = writer
        else:
            writer = writers[ writer_key ] 

        writer.write( m )
        sys.stdout.write( "." )
        sys.stdout.flush()

    for key in writers:
        writers[ key ].close()

if __name__ == "__main__": __main__()