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