Skip to content

Commit b5d0e20

Browse files
committed
feat: add sequence placement functionality with command line interface
1 parent 7685774 commit b5d0e20

File tree

1 file changed

+26
-0
lines changed

1 file changed

+26
-0
lines changed

tests/data/seqplacer.py

+26
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,26 @@
1+
from polytomy.sequence_placer import SequencePlacer
2+
from polytomy.tree_parser import TreeParser
3+
import argparse
4+
5+
# parse command line arguments: --tree, --alignment, --output, --threads, --model, --prefix, --keep-files, --output-dir
6+
parser = argparse.ArgumentParser()
7+
parser.add_argument("--intree", "-i", required=True)
8+
parser.add_argument("--alignment", "-a", required=True)
9+
parser.add_argument("--output", "-o", required=True)
10+
parser.add_argument("--threads", "-t", type=int, default=6)
11+
parser.add_argument("--model", "-m", default="GTRGAMMA")
12+
parser.add_argument("--prefix", "-p", default="seq_placement")
13+
parser.add_argument("--keep-files", "-k", action="store_true")
14+
parser.add_argument("--dir", "-d", default=".")
15+
args = parser.parse_args()
16+
17+
parser = TreeParser()
18+
backbone_tree = parser.parse_from_file(args.intree)
19+
placer = SequencePlacer(backbone_tree, config={
20+
'threads': args.threads,
21+
'model': args.model,
22+
'prefix': args.prefix,
23+
'keep_files': args.keep_files,
24+
'output_dir': args.dir
25+
})
26+
result_tree = placer.place_sequences(args.alignment, prefilter=True)

0 commit comments

Comments
 (0)