Files
galaxy/tools/maf/maf_split_by_species.py
T

47 lines
1.3 KiB
Python

#!/usr/bin/env python
"""
Read a maf and split blocks by unique species combinations
"""
from __future__ import print_function
import sys
from bx.align import maf
from galaxy.tools.util import maf_utilities
from galaxy.util import string_as_bool
def __main__():
try:
maf_reader = maf.Reader(open(sys.argv[1]))
except Exception as e:
maf_utilities.tool_fail("Error opening MAF: %s" % e)
try:
out = maf.Writer(open(sys.argv[2], "w"))
except Exception as e:
maf_utilities.tool_fail("Error opening file for output: %s" % e)
try:
collapse_columns = string_as_bool(sys.argv[3])
except Exception as e:
maf_utilities.tool_fail("Error determining collapse columns value: %s" % e)
start_count = 0
end_count = 0
for start_count, start_block in enumerate(maf_reader): # noqa: B007
for block in maf_utilities.iter_blocks_split_by_species(start_block):
if collapse_columns:
block.remove_all_gap_columns()
out.write(block)
end_count += 1
out.close()
if end_count:
print("%i alignment blocks created from %i original blocks." % (end_count, start_count + 1))
else:
print("No alignment blocks were created.")
if __name__ == "__main__":
__main__()