#! /usr/bin/perl -w

use File::Copy;
use strict;
use warnings;

# Storing command line arguments in an array
print "The following arguments were passed to this script:\n";
print "@ARGV\n";

if (! @ARGV) {
    print "No arguments given.\n";
    print "Call adcirc.pl as:\n";
    print "./sl15_adcirc2.pl <forward speed> <year> <username> <system> <path> <directory name> <output directory> <email id> <processors> <walltime> <job name> <alloca
tion name> \n";    
    exit 1;
}

my $forward_speed=shift @ARGV;
my $year=shift @ARGV;
my $user=shift @ARGV;
my $system=shift @ARGV;
my $path=shift @ARGV;
my $dirname= shift @ARGV;
my $output=shift @ARGV;
my $email_id=shift @ARGV;
my $nprocs=shift @ARGV;
my $walltime=shift @ARGV;
my $job_name=shift @ARGV;
my $allocation_name=shift @ARGV;

#Creating "output" sub directory
chdir "$path/$dirname/" or die "Cannot move to the directory";
mkdir $output,0755 or die "Cannot create the directory";

#Location of source files
my $par_13="$path/$dirname/par/fort.13";
my $par_14="$path/$dirname/par/fort.14";
my $par_15="$path/$dirname/par/fort.15";
my $par_22="$path/$dirname/par/fort.22";
my $run_pbs="$path/$dirname/run/adcirc.pbs";
my $exe_adcirc="$path/$dirname/exe/adcirc";
my $exe_adcpost="$path/$dirname/exe/adcpost";
my $exe_adcprep="$path/$dirname/exe/adcprep";
my $exe_padcirc="$path/$dirname/exe/padcirc";
my $source_meta="$path/$dirname/metadata";

#Location of destination files
my $output_pbs="$path/$dirname/$output/adcirc.pbs";
my $output_13="$path/$dirname/$output/fort.13";
my $output_14="$path/$dirname/$output/fort.14";
my $output_15="$path/$dirname/$output/fort.15";
my $output_22="$path/$dirname/$output/fort.22";
my $output_adcirc="$path/$dirname/$output/adcirc";
my $output_adcpost="$path/$dirname/$output/adcpost";
my $output_adcprep="$path/$dirname/$output/adcprep";
my $output_padcirc="$path/$dirname/$output/padcirc";
my $output_meta="$path/$dirname/$output/metadata";

# Copying files from source to destination
copy($par_13,$output_13) or die "File cannot be copied";
copy($par_14,$output_14) or die "File cannot be copied";
copy($par_15,$output_15) or die "File cannot be copied";
copy($par_22,$output_22) or die "File cannot be copied";
copy($run_pbs,$output_pbs) or die "File cannot be copied";
copy($exe_adcirc,$output_adcirc) or die "File cannot be copied";
copy($exe_adcpost,$output_adcpost) or die "File cannot be copied";
copy($exe_adcprep,$output_adcprep) or die "File cannot be copied";
copy($exe_padcirc,$output_padcirc) or die "File cannot be copied";
copy($source_meta,$output_meta) or die "File cannot be copied";

# Change permissions for the copied files
chmod(0777,$output_adcprep) or die "Cannot change permissions";
chmod(0777,$output_padcirc) or die "Cannot change permissions";
chmod(0777,$output_adcpost) or die "Cannot change permissions";
chmod(0777,$output_adcirc) or die "Cannot change permissions";
chmod(0777,$output_13) or die "Cannot change permissions";
chmod(0777,$output_14) or die "Cannot change permissions";
chmod(0777,$output_15) or die "Cannot change permissions";
chmod(0777,$output_22) or die "Cannot change permissions";
chmod(0777,$output_pbs) or die "Cannot change permissions";
chmod(0777,$output_meta) or die "Cannot change permissions";

#Manipulating the date and time of the fort.22 file w.r.t the forward speed
if($forward_speed!=0)
{
system "python $path/dateparser.py $output_22 $forward_speed";
}

# Modifications to be done in metadata file
$year="Year=$year\n";
my $wallclock="Wall Clocl Time=$walltime\n";
$user="User=$user\n";
my $storm="Storm Name=$dirname\n";
my $procs="Number of Procs=$nprocs\n";
my $machine=' ';
if($system eq "ranger")
{
  $machine="Machine Name=ranger.tacc.teragrid.org\n";
}
if($system eq "queenbee")
{
  $machine="Machine Name=queenbee.loni.org\n";
}
if($system eq "tezpur")
{
  $machine="Machine Name=tezpur.hpc.lsu.edu\n";
}
my $allocation="Allocation=$allocation_name\n";

# Read metadata script
open FILEmeta, "<", $output_meta or die "Cannot open the metadata file: $!";
my @lines_meta = <FILEmeta>;
close FILEmeta;

# Modify lines in metadata file
$lines_meta[0] = $year;
$lines_meta[1] = $wallclock;
$lines_meta[2] = $user;
$lines_meta[4] = $storm;
$lines_meta[5] = $procs;
$lines_meta[6] = $machine;
$lines_meta[9] = $allocation;

# Write into metadata script
open FILEmeta, ">", $output_meta or die "Cannot open the PBS script: $!";
print FILEmeta join('',@lines_meta);
close FILEmeta;

# Modifications to be done in the PBS script
my $nodes= 0;
my $nprocs1=0;
my $work_dir=' ';
my $out_file=' ';
my $error_file=' ';
my $dollar=' ';
my $ppn = 4;
if($system ne "ranger")
{
  $dollar='PBS';
  $email_id="#$dollar -M $email_id\n";
  
  if($system eq "queenbee")
  {
   $ppn = 8;
  }
  
  $nodes=$nprocs/$ppn;
  $nprocs1="#$dollar -l nodes=$nodes:ppn=$ppn\n";

  $walltime="#$dollar -l walltime=$walltime\n";

  $job_name="#$dollar -N $job_name\n";

  $allocation_name="#$dollar -A $allocation_name\n";

  $work_dir="WORK_DIR=$path/$dirname/$output";
} else {
  $dollar='$';

  $email_id="#$dollar -M $email_id\n";
  
  $nprocs1="#$dollar -pe 16way $nprocs\n"; 
  
  $walltime="#$dollar -l h_rt=$walltime\n";

  $out_file="#$dollar -o $path/$dirname/$output/sl15stdout32.out\n";

  $error_file="#$dollar -e $path/$dirname/$output/sl15stderr32.err\n";

  $job_name="#$dollar -N $job_name\n";

  $allocation_name="#$dollar -A $allocation_name\n";

  $work_dir="WORK_DIR=$path/$dirname/$output"; 
 
  $nprocs= "NPROCS=$nprocs\n";
}
#Modifications done to the PBS script

# Read PBS script
open FILEpbs, "<", $run_pbs or die "Cannot open the PBS script: $!";
my @lines_pbs = <FILEpbs>;
close FILEpbs;

# Modify lines in PBS script
if($system eq "ranger")
{
  $lines_pbs[3] = $email_id;
  $lines_pbs[4] = $nprocs1;
  $lines_pbs[5] = $walltime;
  $lines_pbs[10] = $out_file;
  $lines_pbs[11] = $error_file;
  $lines_pbs[14] = $job_name;
  $lines_pbs[17] = $allocation_name;
  $lines_pbs[24] = $work_dir;
  $lines_pbs[26] = $nprocs;
} else {
  $lines_pbs[3] = $email_id;
  $lines_pbs[4] = $nprocs1;
  $lines_pbs[5] = $walltime;
  $lines_pbs[14] = $job_name;
  $lines_pbs[17] = $allocation_name;
  $lines_pbs[23] = $work_dir;
}

# Write into PBS script
open FILEpbs, ">", $output_pbs or die "Cannot open the PBS script: $!";
print FILEpbs join('',@lines_pbs);
close FILEpbs;

# Moving into the new directory created
chdir "$path/$dirname/$output/" or die "Cannot move to the output directory";

# Submittiing the PBS script
system "qsub adcirc.pbs";