#! /usr/bin/perl 
$version ="2.00";
$date  ="2020-10-15";
$author = "kaa and km";

# define the binning breaks and xissimarfgen arguments

$chan1 = 700;
$chan2 = 2696;
$energy1 = 1201;
$energy2 = 5501;

$fast_num_photons = 50000;
$fast_accuracy = 0.05;
$fast_estepfile = "sparse";

$medium_num_photons = 100000;
$medium_accuracy = 0.01;
$medium_estepfile = "medium";

$slow_num_photons = 200000;
$slow_accuracy = 0.005;
$slow_estepfile = "dense";




# This script creates a Suzaku XIS response.

if(@ARGV < 1 || @ARGV > 7)
{
    print "\n usage : xisresp filename <fast|medium|slow> regionfile extend? echo?\n\n";
    print "Runs xisrmfgen to make an rmf, xissimarfgen to make an arf, combines\n";
    print "them using marfrmf then optionally rebins the response using rbnrmf\n\n";
    print "The input filename should be the spectrum file. For the medium or fast\n";
    print "options this will be rebinned to match the channel binning of the response\n\n";
    print "The three speed options are as follows :\n\n";
    print "     fast : channel binning  2x then 4x from $chan1 and 8x from $chan2\n";
    print "            energy binning   2x then 4x from $energy1 and 8x from $energy2\n";
    print "            xissimarfgen run with num_photons=$fast_num_photons\n";
    print "                                  accuracy=$fast_accuracy\n";
    print "                                  estepfile=$fast_estepfile\n\n";
    print "   medium : channel binning  1x then 2x from $chan1 and 4x from $chan2\n";
    print "            energy binning   1x then 2x from $energy1 and 4x from $energy2\n";
    print "            xissimarfgen run with num_photons=$medium_num_photons\n";
    print "                                  accuracy=$medium_accuracy\n";
    print "                                  estepfile=$medium_estepfile\n\n";
    print "     slow : no response binning\n";
    print "            xissimarfgen run with num_photons=$slow_num_photons\n";
    print "                                  accuracy=$slow_accuracy\n";
    print "                                  estepfile=$slow_estepfile\n\n";
    print "For the extend=no option the arf is calculated assuming a point source at the\n";
    print "center of the selected region. The extend=yes option uses the detector image\n";
    print "in the WMAP (the primary extension of the spectrum file) as the source from which\n";
    print "to estimate the arf.\n\n";
    exit(0);
}

$filename = $ARGV[0];
$speed = $ARGV[1];
$regionfile = $ARGV[2];
$extend = "no";
if ( @ARGV >= 4 && $ARGV[3] eq "yes" ) { $extend = "yes"; }
$echo = "no";
if ( @ARGV >= 5 && $ARGV[4] eq "yes" ) { $echo = "yes"; }
$debug = "no";
if ( @ARGV == 6 && $ARGV[5] eq "yes" ) { $debug = "yes"; }

print "filename = $filename, speed = $speed, regionfile = $regionfile, extend = $extend, echo = $echo, debug = $debug\n";


# construct the rmf and arf filenames

$dot = index($filename,".");
$rootname = substr($filename, 0, $dot);

$rmffile = $rootname . ".rmf";
$arffile = $rootname . ".arf";
$respfile = $rootname . ".rsp";

# save an original version of the input spectrum

$origfile = $filename . "-orig";
$command = "cp $filename $origfile";
if ( $echo eq "yes" ) { print $command,"\n"; }
if ( $debug eq "no" ) { system($command); }

# run xisrmfgen to make the RMF.

$command = "xisrmfgen phafile=$filename outfile=$rmffile";
if ( $echo eq "yes" ) { print $command,"\n"; }
if ( $debug eq "no" ) { system($command); }

# run xissimarfgen to make the arf.- first need to extract some information from the spectrum
# file which the program ought to be able to do itself but doesn't yet

$command = "fkeypar fitsfile=$filename\[SPECTRUM\] keyword=INSTRUME";
system($command);
($instrume = `pget fkeypar value`) =~ tr/\n//d;

if ( $extend eq "no" ) {

# find the DETX/Y coordinates of the center of the WMAP - assumed to be the source center

    $command = "fkeypar fitsfile=$filename+0 keyword=NAXIS1";
    system($command);
    ($naxis1 = `pget fkeypar value`) =~ tr/\n//d;
    $command = "fkeypar fitsfile=$filename+0 keyword=NAXIS2";
    system($command);
    ($naxis2 = `pget fkeypar value`) =~ tr/\n//d;

    $command = "fkeypar fitsfile=$filename+0 keyword=CRPIX1P";
    system($command);
    ($crpix1p = `pget fkeypar value`) =~ tr/\n//d;
    $command = "fkeypar fitsfile=$filename+0 keyword=CRPIX2P";
    system($command);
    ($crpix2p = `pget fkeypar value`) =~ tr/\n//d;

    $command = "fkeypar fitsfile=$filename+0 keyword=CDELT1P";
    system($command);
    ($cdelt1p = `pget fkeypar value`) =~ tr/\n//d;
    $command = "fkeypar fitsfile=$filename+0 keyword=CDELT2P";
    system($command);
    ($cdelt2p = `pget fkeypar value`) =~ tr/\n//d;

    $command = "fkeypar fitsfile=$filename+0 keyword=CRVAL1P";
    system($command);
    ($crval1p = `pget fkeypar value`) =~ tr/\n//d;
    $command = "fkeypar fitsfile=$filename+0 keyword=CRVAL2P";
    system($command);
    ($crval2p = `pget fkeypar value`) =~ tr/\n//d;

    $sourcex = ($naxis1/2.0 - $crpix1p)*$cdelt1p + $crval1p;
    $sourcey = ($naxis2/2.0 - $crpix2p)*$cdelt2p + $crval2p;

}

# get the Euler angles to do the sky to detector conversion

$command = "fkeypar fitsfile=$filename+0 keyword=MEAN_EA1";
system($command);
($euler1 = `pget fkeypar value`) =~ tr/\n//d;
$command = "fkeypar fitsfile=$filename+0 keyword=MEAN_EA2";
system($command);
($euler2 = `pget fkeypar value`) =~ tr/\n//d;
$command = "fkeypar fitsfile=$filename+0 keyword=MEAN_EA3";
system($command);
($euler3 = `pget fkeypar value`) =~ tr/\n//d;

# set the speed options for xissimarfgen

if ( $speed eq "fast" ) {
    $num_photons = $fast_num_photons;
    $accuracy = $fast_accuracy;
    $estepfile = "$fast_estepfile";
} elsif ( $speed eq "medium" ) {
    $num_photons = $medium_num_photons;
    $accuracy = $medium_accuracy;
    $estepfile = "$medium_estepfile";
} elsif ( $speed eq "slow" ) {
    $num_photons = $slow_num_photons;
    $accuracy = $slow_accuracy;
    $estepfile = "$slow_estepfile";
}

# extended source case

if ( $extend eq "yes" ) {

    $command = "xissimarfgen instrume=$instrume source_mode=DETFITS source_image=$filename+0 " .
	       "num_region=1 region_mode=SKYREG regfile1=$regionfile limit_mode=MIXED " .
               "num_photon=$num_photons accuracy=$accuracy phafile=$filename detmask=none " .
               "gtifile=$filename attitude=none ea1=$euler1 ea2=$euler2 ea3=$euler3 rmffile=$rmffile " .
               "arffile1=$arffile estepfile=$estepfile clobber=yes";

} else {

# point source case

    $command = "xissimarfgen instrume=$instrume source_mode=DETXY source_x=$sourcex source_y=$sourcey " .
	       "num_region=1 region_mode=SKYREG regfile1=$regionfile limit_mode=MIXED " .
               "num_photon=$num_photons accuracy=$accuracy phafile=$filename detmask=none " .
               "gtifile=$filename attitude=none ea1=$euler1 ea2=$euler2 ea3=$euler3 rmffile=$rmffile " .
               "arffile1=$arffile estepfile=$estepfile clobber=yes";

}

if ( $echo eq "yes" ) { print $command,"\n"; }
if ( $debug eq "no" ) { system($command); }

# now merge the rmf and arf to make the response file

unlink($respfile);
$command = "marfrmf rmfil=$rmffile arfil=$arffile outfil=$respfile clobber=yes";
if ( $echo eq "yes" ) { print $command,"\n"; }
if ( $debug eq "no" ) { system($command); }

# set up the text files needed to do the rebinning

$chan1_1 = $chan1 - 1;
$chan2_1 = $chan2 - 1;
$energy1_1 = $energy1 - 1;
$energy2_1 = $energy2 - 1;

if ( $speed eq "fast" ) {
    open(TMPFILE,">chanfile.txt");
    print TMPFILE "0 $chan1_1 2\n";
    print TMPFILE "$chan1 $chan2_1 4\n";
    print TMPFILE "$chan2 4095 8\n";
    close(TMPFILE);
    open(TMPFILE,">energyfile.txt");
    print TMPFILE "1 $energy1_1 2\n";
    print TMPFILE "$energy1 $energy2_1 4\n";
    print TMPFILE "$energy2 7900 8\n";
    close(TMPFILE);
} elsif ( $speed eq "medium" ) {
    open(TMPFILE,">chanfile.txt");
    print TMPFILE "0 $chan1_1 1\n";
    print TMPFILE "$chan1 $chan2_1 2\n";
    print TMPFILE "$chan2 4095 4\n";
    close(TMPFILE);
    open(TMPFILE,">energyfile.txt");
    print TMPFILE "1 $energy1_1 1\n";
    print TMPFILE "$energy1 $energy2_1 2\n";
    print TMPFILE "$energy2 7900 4\n";
    close(TMPFILE);
}

# rebin the response file if the medium or fast options are given

if ( $speed eq "fast" || $speed eq "medium" ) {
    $tmpfile = $rootname . ".tmp";
    $command = "ftrbnrmf infile=$respfile cmpmode=binfile binfile=chanfile.txt ecmpmode=ebinfile ebinfile=energyfile.txt outfile=$tmpfile clobber=yes";
    if ( $echo eq "yes" ) { print $command,"\n"; }
    if ( $debug eq "no" ) { system($command); }
    if ( $debug eq "no" ) { rename($tmpfile, $respfile); }
}

# rebin the input spectrum file if the medium or fast options are given

if ( $speed eq "fast" || $speed eq "medium" ) {
    $tmpfile = $rootname . ".tmp";
    $command = "ftrbnpha infile=$filename binfile=chanfile.txt outfile=$tmpfile properr=yes clobbe=yes";
    if ( $echo eq "yes" ) { print $command,"\n"; }
    if ( $debug eq "no" ) { system($command); }
    if ( $debug eq "no" ) { rename($tmpfile, $filename); }
}

# set the RESPFILE and ANCRFILE keywords in the spectrum file

$command = "fparkey value=$respfile fitsfile=$filename\[SPECTRUM\] keyword=RESPFILE";
if ( $echo eq "yes" ) { print $command,"\n"; }
if ( $debug eq "no" ) { system($command); }

$command = "fparkey value=NONE fitsfile=$filename\[SPECTRUM\] keyword=ANCRFILE";
if ( $echo eq "yes" ) { print $command,"\n"; }
if ( $debug eq "no" ) { system($command); }

# tidy up temporary files

if ( $debug eq "no" ) { 
    unlink($tmpfile);
    unlink($arffile);
    unlink($rmffile);
    unlink("chanfile.txt");
    unlink("energyfile.txt");
}