package subgroups;

use strict;
use warnings;
use Exporter;
our @ISA = qw( Exporter );
our @EXPORT_OK = qw( &subgroups );

use POSIX;
use File::Copy;
use FindBin;
use lib $FindBin::Bin;
use align qw ( get_hash_alignment );

sub subgroups
{
	my ($fin, $dir, $mis, $mis_TE, $proc, $tRNAs, $rRNAs, $snRNAs, $miRNAs, $transcripts, $TE, $min_si, $max_si, $min_pi, $max_pi, $report ) = @_;
	my (@files, $sum, $pie, $repar, %ismapped, %isjunk, %repartition, @junk_ref, @all_ref );

	srand();
	print $report "----------------------------\n";
	print $report "Create subgroups:\nfastq_in: $fin\ndirectory_out: $dir\nmismatches: $mis\nmismatches TE: $mis_TE\n";

	mkdir $dir;
	$dir = $dir.'/' unless $dir =~ /(.*)\/$/;

	my $accept_miRNas = $dir.'miRNAs.fastq'; 
	my $reject_miRNAs = $dir.'miRNAs_rejected.fastq';
	my $sam_miRNAs = $dir.'miRNAs.sam'; 
	my @tmp = get_hash_alignment($miRNAs, $mis, 1, 1, $accept_miRNas, $reject_miRNAs, $fin, $proc, 'miRNAs',$sam_miRNAs, $report);
	my $mi = $tmp[0]; my $sam = '';
	$repartition{'miRNAs'} = $mi;


	my $reject_rRNAs = $dir.'rRNAs_rejected.fastq';
	if ( $rRNAs eq 'None')
	{
		move($reject_miRNAs,$reject_rRNAs);
	}
	else
	{
		$sam = new String::Random;
		$sam = $sam->randpattern("CCcccccc");
		@tmp = get_hash_alignment($rRNAs, $mis, 0, 1, 'NA', $reject_rRNAs, $reject_miRNAs, $proc, 'rRNAs', $sam, $report);
		$repartition{'rRNAs'} = $tmp[0];
		unlink $sam, $sam.'_aln.err', $sam.'_samse.err';
	}

	my $reject_tRNAs = $dir.'rRNAs_rejected.fastq';
	if ( $rRNAs eq 'None')
	{
		move($reject_rRNAs,$reject_tRNAs);
	}
	else
	{
		$sam = new String::Random;
		$sam = $sam->randpattern("CCcccccc");
		@tmp = get_hash_alignment($tRNAs, $mis, 0, 1, 'NA', $reject_tRNAs, $reject_rRNAs, $proc, 'tRNAs', $sam, $report);
		$repartition{'tRNAs'} = $tmp[0];
		unlink $sam, $sam.'_aln.err', $sam.'_samse.err';
	}


	my $bonafide = $dir.'bonafide_reads.fastq';
	if ( $rRNAs eq 'None')
	{
		move($reject_tRNAs,$bonafide);
	}
	else
	{
		$sam = new String::Random;
		$sam = $sam->randpattern("CCcccccc");
		@tmp = get_hash_alignment($snRNAs, $mis, 0, 1, 'NA', $bonafide, $reject_tRNAs, $proc, 'snRNAs', $sam, $report);
		$repartition{'snRNAs'} = $tmp[0];

		unlink $sam, $sam.'_aln.err', $sam.'_samse.err';
	}
	my $bo = $tmp[1];

	my $sam_transcripts = $dir.'transcripts.sam'; 
	my $reject_transcripts = $dir.'rejected_transcripts.fastq';
	@tmp = get_hash_alignment($transcripts, $mis, 0, 1, 'NA', $reject_transcripts, $bonafide, $proc, 'transcripts', $sam_transcripts, $report, $dir.'transcripts.fai');
	$repartition{'transcripts'} = $tmp[0];

	
	my $sam_TEs = $dir.'TEs.sam';
	my $reject_TEs = $dir.'rejected.fastq';
	@tmp = get_hash_alignment($TE, $mis_TE, 0, 1, 'NA', $reject_TEs, $reject_transcripts, $proc, 'TEs', $sam_TEs, $report, $dir.'TEs.fai' );
	$repartition{'TEs'} = $tmp[0] ; $repartition{'others'} = $tmp[1];
	unlink $sam, $sam.'_aln.err', $sam.'_samse.err';
	unlink $reject_transcripts;
	unlink $reject_rRNAs;
	unlink $reject_miRNAs;
	unlink $reject_tRNAs;

	#create repartition
	my $pi = fastqSubgroups($bonafide, $dir, $min_si, $max_si, $min_pi, $max_pi );

	open (my $re, '>'.$dir.'repartition.txt') || die "cannot open $dir repartition.txt $!\n";
	print $re "type\tnumber\tpercentage\n";
	$sum += $_ foreach values %repartition;
	foreach my $k  ( sort keys  %repartition ) 
	{
		my $prct = 0;
		$prct = $repartition{$k} / $sum * 100 if $sum != 0;
		print $re "$k\t$repartition{$k}\t"; printf $re "%.2f\n",$prct;
	}
	return ( $bo, $mi, $pi);
}

sub fastqSubgroups
{
  my ( $fastq, $output_directory, $min_si, $max_si, $min_pi, $max_pi ) = @_;
  my $fastq_siRNA = $output_directory."siRNAs.fastq";
  my $fastq_piRNA = $output_directory."piRNAs.fastq";

  open my $fic, '<', $fastq || die "cannot open input file $! \n";
  open my $si, '>', $fastq_siRNA || die "cannot open siRNA.fastq $! \n";
  open my $pi, '>', $fastq_piRNA || die "cannot open piRNA.fastq $! \n";
  
  my ($length, $cmp, $type, $siRNA_number, $miRNA_h_number, $piRNA_number, $not_pi_number) = (0,0,0,0,0,0,0);
  my (@fastq) =(); my $seq_name;
  my $out;
  while(<$fic>)
  {
    chomp $_;
    $cmp++;
    if ($cmp == 1)
    {
      die "file do not contain a @ at line $cmp\n" unless ($_ =~ /^\@/ );
      $type = 0; @fastq = ();
      if ($_ =~ /^\@(.*)\s.*/) { $seq_name = $1;}
      elsif ($_ =~ /^\@(.*)/) {$seq_name = $1;}
      push(@fastq,$_);
    }
    elsif ($cmp == 2)
    {
      #die "unrecognized symbol at line $cmp\n" unless $_ =~ /[atcgATCGnN]+/;
      push(@fastq,$_);
      $length = length($_);
      $type = 0;
      if ( $length >= $min_si  && $length <= $max_si )
      {
         $type = 2;
         $siRNA_number++;
      }
      if ($length >= $min_pi  && $length <= $max_pi )
      {
        $type += 4;
        $piRNA_number++;
      }      
    }
    elsif ($cmp == 3 )
    {
      die "file do not contain a + at line $cmp\n" unless $_ =~ /^\+/;
      push(@fastq,$_);
    }
    elsif ($cmp == 4 )
    {
      push(@fastq,$_);
      $cmp = 0;
      if ($type != 0)
      {
        if ($type & 4 ) { foreach my $t (@fastq) { print $pi $t."\n";} }
        if ($type & 2 ) { foreach my $t (@fastq) { print $si $t."\n";} }        
      }
    }
  }
  close $fic;
  close $si; close $pi;
  return ($piRNA_number);
}

1;