kkonganti@1: #!/usr/bin/env perl
kkonganti@1: 
kkonganti@1: # Kranti Konganti
kkonganti@1: # Takes in a gzipped multi-fasta file
kkonganti@1: # and joins contigs by 10 N's
kkonganti@1: 
kkonganti@1: use strict;
kkonganti@1: use warnings;
kkonganti@1: use Cwd;
kkonganti@1: use Bio::SeqIO;
kkonganti@1: use Getopt::Long;
kkonganti@1: use File::Find;
kkonganti@1: use File::Basename;
kkonganti@1: use File::Spec::Functions;
kkonganti@1: 
kkonganti@1: my ( $in_dir, $out_dir, $suffix, @uncatted_genomes );
kkonganti@1: 
kkonganti@1: GetOptions(
kkonganti@1:     'in_dir=s'  => \$in_dir,
kkonganti@1:     'out_dir=s' => \$out_dir,
kkonganti@1:     'suffix=s'  => \$suffix
kkonganti@1: ) or die usage();
kkonganti@1: 
kkonganti@1: $in_dir  = getcwd            if ( !defined $in_dir );
kkonganti@1: $out_dir = getcwd            if ( !defined $out_dir );
kkonganti@1: $suffix  = '_genomic.fna.gz' if ( !defined $suffix );
kkonganti@1: 
kkonganti@1: find(
kkonganti@1:     {
kkonganti@1:         wanted => sub {
kkonganti@1:             push @uncatted_genomes, $File::Find::name if ( $_ =~ m/$suffix$/ );
kkonganti@1:         }
kkonganti@1:     },
kkonganti@1:     $in_dir
kkonganti@1: );
kkonganti@1: 
kkonganti@1: if ( $out_dir ne getcwd && !-d $out_dir ) {
kkonganti@1:     mkdir $out_dir || die "\nCannot create directory $out_dir: $!\n\n";
kkonganti@1: }
kkonganti@1: 
kkonganti@1: open( my $geno_path, '>genome_paths.txt' )
kkonganti@1:   || die "\nCannot open file genome_paths.txt: $!\n\n";
kkonganti@1: 
kkonganti@1: foreach my $uncatted_genome_path (@uncatted_genomes) {
kkonganti@1:     my $catted_genome_header = '>' . basename( $uncatted_genome_path, $suffix );
kkonganti@1:     $catted_genome_header =~ s/(GC[AF]\_\d+\.\d+)\_*.*/$1/;
kkonganti@1: 
kkonganti@1:     my $catted_genome =
kkonganti@1:       catfile( $out_dir, $catted_genome_header . '_scaffolded' . $suffix );
kkonganti@1: 
kkonganti@1:     $catted_genome =~ s/\/\>(GC[AF])/\/$1/;
kkonganti@1: 
kkonganti@1:     print $geno_path "$catted_genome\n";
kkonganti@1: 
kkonganti@1:     open( my $fh, "gunzip -c $uncatted_genome_path |" )
kkonganti@1:       || die "\nCannot create pipe for $uncatted_genome_path: $!\n\n";
kkonganti@1: 
kkonganti@1:     open( my $fho, '|-', "gzip -c > $catted_genome" )
kkonganti@1:       || die "\nCannot pipe to gzip: $!\n\n";
kkonganti@1: 
kkonganti@1:     my $seq_obj = Bio::SeqIO->new(
kkonganti@1:         -fh     => $fh,
kkonganti@1:         -format => 'Fasta'
kkonganti@1:     );
kkonganti@1: 
kkonganti@1:     my $joined_seq = '';
kkonganti@1:     while ( my $seq = $seq_obj->next_seq ) {
kkonganti@1:         $joined_seq = $joined_seq . 'NNNNNNNNNN' . $seq->seq;
kkonganti@1:     }
kkonganti@1: 
kkonganti@1:     $joined_seq =~ s/NNNNNNNNNN$//;
kkonganti@1:     $joined_seq =~ s/^NNNNNNNNNN//;
kkonganti@1: 
kkonganti@1:     # $joined_seq =~ s/.{80}\K/\n/g;
kkonganti@1:     # $joined_seq =~ s/\n$//;
kkonganti@1:     print $fho $catted_genome_header, "\n", $joined_seq, "\n";
kkonganti@1: 
kkonganti@1:     $seq_obj->close();
kkonganti@1:     close $fh;
kkonganti@1:     close $fho;
kkonganti@1: }
kkonganti@1: 
kkonganti@1: sub usage {
kkonganti@1:     print
kkonganti@1: "\nUsage: $0 [-in IN_DIR] [-ou OUT_DIR] [-su Filename Suffix for Header]\n\n";
kkonganti@1:     exit;
kkonganti@1: }
kkonganti@1: