1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41
|
#!/usr/bin/python3
"""
Pass through blocks from a maf file until a certain number of columns
have been passed.
usage: %prog -c cols < maf > maf
"""
import sys
from optparse import OptionParser
from bx.align import maf
def __main__():
parser = OptionParser()
parser.add_option("-c", "--cols", action="store")
(options, args) = parser.parse_args()
maf_reader = maf.Reader(sys.stdin, parse_e_rows=True)
maf_writer = maf.Writer(sys.stdout)
if not options.cols:
raise Exception("Cols argument is required")
cols = int(options.cols)
count = 0
for m in maf_reader:
maf_writer.write(m)
count += m.text_size
if count >= cols:
return
if __name__ == "__main__":
__main__()
|