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
42
43
44
45
46
|
#!/usr/bin/python3
"""Cut up fasta file in non-overlapping or overlapping parts of equal length.
"""
import argparse
import sys
from Bio import SeqIO
def cut_up_fasta(fastfiles, chunk_size, overlap, min_length, merge_last):
for ff in fastfiles:
for record in SeqIO.parse(ff, "fasta"):
if (not merge_last and len(record.seq) > chunk_size) or (merge_last and len(record.seq) >= 2 * chunk_size):
i = 0
for split_seq in chunks(record.seq, chunk_size, overlap, merge_last):
start = i*chunk_size
end = start + len(split_seq)
print(">{}_({}_{})\n{}".format(record.id, start, end, split_seq))
i = i + 1
elif len(record.seq) >= min_length:
suffix = "" if "length" in record.id else "_length_" + str(len(record))
print(">{}{}\n{}".format(record.id, suffix, record.seq))
def chunks(l, n, o, merge_last):
""" Yield successive n-sized chunks from l with given overlap o between the
chunks.
"""
assert(n > o)
if not merge_last:
for i in xrange(0, len(l), n - o):
yield l[i:i + n]
else:
for i in xrange(0, len(l) - n + 1, n - o):
yield l[i:i + n] if i + n + n - o <= len(l) else l[i:]
if __name__ == "__main__":
parser = argparse.ArgumentParser(description=__doc__,
formatter_class=argparse.RawDescriptionHelpFormatter)
parser.add_argument("contigs", nargs="+", help="Fasta files with contigs\n")
parser.add_argument("-c", "--chunk_size", default=sys.maxsize, type=int, help="Chunk size\n")
parser.add_argument("-o", "--overlap_size", default=0, type=int, help="Overlap size\n")
parser.add_argument("-l", "--min_length", default=2000, type=int, help="Minimum split/contig length")
parser.add_argument("-m", "--merge_last", default=False, action="store_true", help="Concatenate final part to last contig\n")
args = parser.parse_args()
cut_up_fasta(args.contigs, args.chunk_size, args.overlap_size, args.min_length, args.merge_last)
|
