bp_bin/format_genome

   1 #!/usr/bin/env perl
   2
   3 # Copyright (C) 2007-2009 Martin A. Hansen.
   4
   5 # This program is free software; you can redistribute it and/or
   6 # modify it under the terms of the GNU General Public License
   7 # as published by the Free Software Foundation; either version 2
   8 # of the License, or (at your option) any later version.
   9
  10 # This program is distributed in the hope that it will be useful,
  11 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 # GNU General Public License for more details.
  14
  15 # You should have received a copy of the GNU General Public License
  16 # along with this program; if not, write to the Free Software
  17 # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  18
  19 # http://www.gnu.org/copyleft/gpl.html
  20
  21
  22 # >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>> DESCRIPTION <<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
  23
  24 # Format a genome creating specified indexes.
  25
  26 # >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>><<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
  27
  28
  29 use warnings;
  30 use strict;
  31 use Maasha::Fasta;
  32 use Maasha::Biopieces;
  33 use Maasha::Bowtie;
  34 use Maasha::NCBI;
  35 use Maasha::Match;
  36 use Maasha::UCSC;
  37
  38
  39 # >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>><<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
  40
  41
  42 my ( $default, $formats, $options, $in, $out, $record, $data_out, $entry,
  43      $genome, $dir, $fasta_dir, $phastcons_dir, $fh_out, $vals, $format, $tmp_dir );
  44
  45 $tmp_dir = Maasha::Biopieces::get_tmpdir();
  46 $default = $ENV{ 'BP_DATA' };
  47 $formats = 'fasta,blast,vmatch,bowtie,phastcons';
  48
  49 $options = Maasha::Biopieces::parse_options(
  50     [
  51         { long => 'no_stream', short => 'x', type => 'flag',   mandatory => 'no',  default => undef,    allowed => undef,    disallowed => undef },
  52         { long => 'dir',       short => 'd', type => 'dir!',   mandatory => 'no',  default => $default, allowed => undef,    disallowed => undef },
  53         { long => 'genome',    short => 'g', type => 'string', mandatory => 'yes', default => undef,    allowed => undef,    disallowed => undef },
  54         { long => 'formats',   short => 'f', type => 'list',   mandatory => 'yes', default => undef,    allowed => $formats, disallowed => '0' },
  55     ]
  56 );
  57
  58 $in  = Maasha::Biopieces::read_stream( $options->{ "stream_in" } );
  59 $out = Maasha::Biopieces::write_stream( $options->{ "stream_out" } );
  60
  61 $dir    = $options->{ 'dir' };
  62 $genome = $options->{ 'genome' };
  63
  64 Maasha::Filesys::dir_create_if_not_exists( "$dir/genomes" );
  65 Maasha::Filesys::dir_create_if_not_exists( "$dir/genomes/$genome" );
  66
  67 if ( grep { $_ =~ /fasta|blast|vmatch|bowtie/i } @{ $options->{ "formats" } } )
  68 {
  69     if ( -f "$dir/genomes/$genome/fasta/$genome.fna" )
  70     {
  71         $fasta_dir = "$dir/genomes/$genome/fasta";
  72     }
  73     else
  74     {
  75         Maasha::Filesys::dir_create_if_not_exists( "$dir/genomes/$genome/fasta" );
  76
  77         $fasta_dir = "$dir/genomes/$genome/fasta";
  78
  79         $fh_out = Maasha::Filesys::file_write_open( "$fasta_dir/$genome.fna" );
  80     }
  81 }
  82 elsif ( grep { $_ =~ /phastcons/i } @{ $options->{ "formats" } } )
  83 {
  84     Maasha::Filesys::dir_create_if_not_exists( "$dir/genomes/$genome/phastcons" );
  85
  86     $phastcons_dir = "$dir/genomes/$genome/phastcons";
  87
  88     $fh_out = Maasha::Filesys::file_write_open( "$phastcons_dir/$genome.pp" );
  89 }
  90
  91 while ( $record = Maasha::Biopieces::get_record( $in ) )
  92 {
  93     if ( $fh_out and $entry = Maasha::Fasta::biopiece2fasta( $record ) )
  94     {
  95         Maasha::Fasta::put_entry( $entry, $fh_out );
  96     }
  97     elsif ( $fh_out and $record->{ "CHR" } and $record->{ "CHR_BEG" } and $record->{ "STEP" } and $record->{ "VALS" } )
  98     {
  99         print $fh_out "fixedStep chrom=$record->{ 'CHR' } start=$record->{ 'CHR_BEG' } step=$record->{ 'STEP' }\n";
 100
 101         $vals = $record->{ 'VALS' };
 102
 103         $vals =~ tr/,/\n/;
 104
 105         print $fh_out "$vals\n";
 106     }
 107
 108     Maasha::Biopieces::put_record( $record, $out ) if not $options->{ "no_stream" };
 109 }
 110
 111 foreach $format ( @{ $options->{ 'formats' } } )
 112 {
 113     print STDERR qq(Creating format: $format for $genome ... ) if $options->{ 'verbose' };
 114
 115     if    ( $format =~ /^fasta$/i )     { Maasha::Fasta::fasta_index( "$fasta_dir/$genome.fna", "$dir/genomes/$genome/fasta/$genome.index" ) }
 116     elsif ( $format =~ /^blast$/i )     { Maasha::NCBI::blast_index( "$genome.fna", $fasta_dir, "$dir/genomes/$genome/blast", "dna", $genome ) }
 117     elsif ( $format =~ /^blat$/i )      { warn "BLAT FORMAT NOT IMPLEMENTED" }
 118     elsif ( $format =~ /^vmatch$/i )    { Maasha::Match::vmatch_index( "$genome.fna", $fasta_dir, "$dir/genomes/$genome/vmatch", $tmp_dir ) }
 119     elsif ( $format =~ /^bowtie$/i )    { Maasha::Bowtie::bowtie_index( "$fasta_dir/$genome.fna", "$dir/genomes/$genome/bowtie", $genome, $options->{ 'verbose' } ) }
 120     elsif ( $format =~ /^phastcons$/i ) { Maasha::UCSC::phastcons_index( "$genome.pp", $phastcons_dir ) }
 121
 122     print STDERR qq(done.\n) if $options->{ 'verbose' };
 123 }
 124
 125 close $fh_out if $fh_out;
 126
 127 Maasha::Biopieces::close_stream( $in );
 128 Maasha::Biopieces::close_stream( $out );
 129
 130
 131 # >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>><<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
 132
 133
 134 BEGIN
 135 {
 136     Maasha::Biopieces::status_set();
 137 }
 138
 139
 140 END
 141 {
 142     Maasha::Biopieces::status_log();
 143 }
 144
 145
 146 # >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>><<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
 147
 148
 149 __END__