annotate gene_family_phylogeny_builder.py @ 24:61825c063bf3 draft

Uploaded
author greg
date Fri, 12 May 2017 09:39:06 -0400
parents 451ad91530b8
children 2de4563e70cf
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
f4361d941aa3 Uploaded
greg
parents:
diff changeset
1 #!/usr/bin/env python
f4361d941aa3 Uploaded
greg
parents:
diff changeset
2 import argparse
19
f33369f4392a Uploaded
greg
parents: 14
diff changeset
3 import os
0
f4361d941aa3 Uploaded
greg
parents:
diff changeset
4
f4361d941aa3 Uploaded
greg
parents:
diff changeset
5 import utils
f4361d941aa3 Uploaded
greg
parents:
diff changeset
6
19
f33369f4392a Uploaded
greg
parents: 14
diff changeset
7 OUTPUT_DIR = 'geneFamilyPhylogenies_dir'
0
f4361d941aa3 Uploaded
greg
parents:
diff changeset
8
f4361d941aa3 Uploaded
greg
parents:
diff changeset
9 parser = argparse.ArgumentParser()
f4361d941aa3 Uploaded
greg
parents:
diff changeset
10
5
f73151ec679b Uploaded
greg
parents: 0
diff changeset
11 parser.add_argument('--alignment_type', dest='alignment_type', help='Input alignments type produced by the GeneFamilyAligner')
0
f4361d941aa3 Uploaded
greg
parents:
diff changeset
12 parser.add_argument('--bootstrap_replicates', dest='bootstrap_replicates', type=int, default=None, help='Number of replicates for rapid bootstrap analysis')
f4361d941aa3 Uploaded
greg
parents:
diff changeset
13 parser.add_argument('--config_dir', dest='config_dir', help='Directory containing default configuration files')
f4361d941aa3 Uploaded
greg
parents:
diff changeset
14 parser.add_argument('--max_orthogroup_size', dest='max_orthogroup_size', type=int, help='Maximum number of sequences in orthogroup alignments')
f4361d941aa3 Uploaded
greg
parents:
diff changeset
15 parser.add_argument('--method', dest='method', help='Protein clustering method')
f4361d941aa3 Uploaded
greg
parents:
diff changeset
16 parser.add_argument('--min_orthogroup_size', dest='min_orthogroup_size', type=int, help='Minimum number of sequences in orthogroup alignments')
f4361d941aa3 Uploaded
greg
parents:
diff changeset
17 parser.add_argument('--num_threads', dest='num_threads', type=int, help='Number of threads to use for execution')
14
2f1de64cc3ae Uploaded
greg
parents: 5
diff changeset
18 parser.add_argument('--orthogroup_aln', dest='orthogroup_aln', help='Input dataset files_path')
24
61825c063bf3 Uploaded
greg
parents: 21
diff changeset
19 parser.add_argument('--output_phylip', dest='output_phylip', default=None, help='Output for orthogroup phylip multiple sequence alignments')
61825c063bf3 Uploaded
greg
parents: 21
diff changeset
20 parser.add_argument('--output_phylip_dir', dest='output_phylip_dir', default=None, help='output_phylip.files_path')
21
451ad91530b8 Uploaded
greg
parents: 19
diff changeset
21 #parser.add_argument('--output_tree', dest='output_tree', help='Output for phylogenetic trees')
451ad91530b8 Uploaded
greg
parents: 19
diff changeset
22 #parser.add_argument('--output_tree_dir', dest='output_tree_dir', help='output_tree.files_path')
0
f4361d941aa3 Uploaded
greg
parents:
diff changeset
23 parser.add_argument('--rooting_order', dest='rooting_order', default=None, help='Rooting order configuration for rooting trees')
f4361d941aa3 Uploaded
greg
parents:
diff changeset
24 parser.add_argument('--scaffold', dest='scaffold', help='Orthogroups or gene families proteins scaffold')
14
2f1de64cc3ae Uploaded
greg
parents: 5
diff changeset
25 parser.add_argument('--sequence_type', dest='sequence_type', help='Sequence type used in the phylogenetic inference')
0
f4361d941aa3 Uploaded
greg
parents:
diff changeset
26 parser.add_argument('--tree_inference', dest='tree_inference', help='Phylogenetic trees inference method')
f4361d941aa3 Uploaded
greg
parents:
diff changeset
27
f4361d941aa3 Uploaded
greg
parents:
diff changeset
28 args = parser.parse_args()
f4361d941aa3 Uploaded
greg
parents:
diff changeset
29
f4361d941aa3 Uploaded
greg
parents:
diff changeset
30 # Build the command line.
f4361d941aa3 Uploaded
greg
parents:
diff changeset
31 cmd = 'GeneFamilyPhylogenyBuilder'
5
f73151ec679b Uploaded
greg
parents: 0
diff changeset
32 cmd += ' --alignment_type %s' % args.alignment_type
0
f4361d941aa3 Uploaded
greg
parents:
diff changeset
33 if args.bootstrap_replicates is not None:
f4361d941aa3 Uploaded
greg
parents:
diff changeset
34 cmd += ' --bootstrap_replicates %d' % args.bootstrap_replicates
f4361d941aa3 Uploaded
greg
parents:
diff changeset
35 cmd += ' --config_dir %s' % args.config_dir
f4361d941aa3 Uploaded
greg
parents:
diff changeset
36 cmd += ' --max_orthogroup_size %d' % args.max_orthogroup_size
f4361d941aa3 Uploaded
greg
parents:
diff changeset
37 cmd += ' --method %s' % args.method
f4361d941aa3 Uploaded
greg
parents:
diff changeset
38 cmd += ' --min_orthogroup_size %d' % args.min_orthogroup_size
f4361d941aa3 Uploaded
greg
parents:
diff changeset
39 cmd += ' --num_threads %d' % args.num_threads
f4361d941aa3 Uploaded
greg
parents:
diff changeset
40 cmd += ' --orthogroup_aln %s' % args.orthogroup_aln
f4361d941aa3 Uploaded
greg
parents:
diff changeset
41 if args.rooting_order is not None:
f4361d941aa3 Uploaded
greg
parents:
diff changeset
42 cmd += ' --rooting_order %s' % args.rooting_order
f4361d941aa3 Uploaded
greg
parents:
diff changeset
43 cmd += ' --scaffold %s' % args.scaffold
f4361d941aa3 Uploaded
greg
parents:
diff changeset
44 cmd += ' --sequence_type %s' % args.sequence_type
f4361d941aa3 Uploaded
greg
parents:
diff changeset
45 cmd += ' --tree_inference %s' % args.tree_inference
14
2f1de64cc3ae Uploaded
greg
parents: 5
diff changeset
46
0
f4361d941aa3 Uploaded
greg
parents:
diff changeset
47 # Run the command.
14
2f1de64cc3ae Uploaded
greg
parents: 5
diff changeset
48 utils.run_command(cmd)
2f1de64cc3ae Uploaded
greg
parents: 5
diff changeset
49
2f1de64cc3ae Uploaded
greg
parents: 5
diff changeset
50 # Handle outputs.
19
f33369f4392a Uploaded
greg
parents: 14
diff changeset
51 if args.output_phylip is not None and args.output_phylip_dir is not None:
f33369f4392a Uploaded
greg
parents: 14
diff changeset
52 src_output_dir = os.path.join(OUTPUT_DIR, 'phylip_aln')
f33369f4392a Uploaded
greg
parents: 14
diff changeset
53 utils.move_directory_files(src_output_dir, args.output_phylip_dir)
f33369f4392a Uploaded
greg
parents: 14
diff changeset
54 utils.write_html_output(args.output_phylip, 'Orthogroup phylip multiple sequence alignments', args.output_phylip_dir)
24
61825c063bf3 Uploaded
greg
parents: 21
diff changeset
55 #src_output_dir = os.path.join(OUTPUT_DIR, 'orthogroups_tree')
61825c063bf3 Uploaded
greg
parents: 21
diff changeset
56 #utils.move_directory_files(src_output_dir, args.output_tree_dir)
61825c063bf3 Uploaded
greg
parents: 21
diff changeset
57 #utils.write_html_output(args.output_tree, 'Phylogenetic trees', args.output_tree_dir)