/* ACS2d -- basic 2-D image reduction */

# include <stdio.h>
# include <stdlib.h>		/* calloc */
# include <time.h>
# include <string.h>

int status = 0;			/* zero is OK */

# include <c_iraf.h>		/* for c_irafinit */
# include <ximio.h>
# include <hstio.h>

# include "../acs.h"
# include "../acsinfo.h"
# include "../acserr.h"
# include "../acscorr.h"		/* calibration switch names for cs1 */

static void FreeNames (char *, char *, char *, char *);

/* This is the main module for acs2d.  It gets the input and output
   file names, calibration switches, and flags, and then calls acs2d.

   Warren Hack, 1998 June 9:
   	Initial version for ACS.
*/

int main (int argc, char **argv) {

	char *inlist;		/* list of input file names */
	char *outlist;		/* list of output file names */
	int switch_on = 0;	/* was any switch specified? */
	int printtime = NO;	/* print time after each step? */
	int verbose = NO;	/* print additional info? */
	int quiet = NO;		/* suppress STDOUT messages? */
	int too_many = 0;	/* too many command-line arguments? */
	int i, j;			/* loop indexes */

	IRAFPointer i_imt, o_imt;	/* imt list pointers */
	char *input;		/* name of input science file */
	char *output;		/* optional name of output file */
	int n_in, n_out;	/* number of files in each list */
	int n;

	/* Input and output suffixes. */
	char *isuffix[] = {"_raw", "_blv_tmp", "_crj_tmp"};
	char *osuffix[] = {"_flt", "_flt",     "_crj"};

	int nsuffix = 3;

	/* A structure to pass the calibration switches to acs2d */
	CalSwitch acs2d_sw;

	/* reference file keywords and names */
	RefFileInfo refnames;
	void InitRefFile (RefFileInfo *);
	void FreeRefFile (RefFileInfo *);

	int ACS2d (char *, char *, CalSwitch *, RefFileInfo *, int, int);
	int DefSwitch (char *);
	int MkOutName (char *, char **, char **, int, char *, int);
 	void WhichError (int);
	int CompareNumbers (int, int, char *);
	void initSwitch (CalSwitch *);

	c_irafinit (argc, argv);
	
	/* Allocate space for file names. */
	inlist = calloc (ACS_LINE+1, sizeof (char));
	outlist = calloc (ACS_LINE+1, sizeof (char));
	input = calloc (ACS_LINE+1, sizeof (char));
	output = calloc (ACS_LINE+1, sizeof (char));
	if (inlist == NULL || outlist == NULL ||
		input == NULL || output == NULL) {
	    printf ("Can't even begin; out of memory.\n");
	    exit (ERROR_RETURN);
	}
        
	/* Initialize the lists of reference file keywords and names. */
	InitRefFile (&refnames);

	/* Initial values. */
	initSwitch (&acs2d_sw);

	for (i = 1;  i < argc;  i++) {

	    if (strcmp (argv[i], "-dqi") == 0) {	/* turn on */
		acs2d_sw.dqicorr = PERFORM;
		switch_on = 1;
	    } else if (strcmp (argv[i], "-glin") == 0) {
		acs2d_sw.glincorr = PERFORM;
		switch_on = 1;
	    } else if (strcmp (argv[i], "-lflg") == 0) {
		acs2d_sw.lflgcorr = PERFORM;
		switch_on = 1;
	    } else if (strcmp (argv[i], "-dark") == 0) {
		acs2d_sw.darkcorr = PERFORM;
		switch_on = 1;
	    } else if (strcmp (argv[i], "-flat") == 0) {
		acs2d_sw.flatcorr = PERFORM;
		switch_on = 1;
	    } else if (strcmp (argv[i], "-shad") == 0) {
		acs2d_sw.shadcorr = PERFORM;
		switch_on = 1;
	    } else if (strcmp (argv[i], "-phot") == 0) {
		acs2d_sw.photcorr = PERFORM;
		switch_on = 1;
	    } else if (strcmp (argv[i], "-stat") == 0) {
		acs2d_sw.statcorr = PERFORM;
		switch_on = 1;
	    } else if (argv[i][0] == '-') {
		for (j = 1;  argv[i][j] != '\0';  j++) {
		    if (argv[i][j] == 't') {
				printtime = YES;
		    } else if (argv[i][j] == 'v') {
				verbose = YES;
		    } else if (argv[i][j] == 'q') {
				quiet = YES;
		    } else {
				printf ("Unrecognized option %s\n", argv[i]);
	    		FreeNames (inlist, outlist, input, output);
				exit (1);
		    }
		}
	    } else if (inlist[0] == '\0') {
		strcpy (inlist, argv[i]);
	    } else if (outlist[0] == '\0') {
		strcpy (outlist, argv[i]);
	    } else {
		too_many = 1;
	    }
	}
	if (inlist[0] == '\0' || too_many) {
	    printf ("syntax:  acs2d [-t] [-v] [-q] input output\n");
	    printf ("  command-line switches:\n");
	    printf ("       -dqi  -atod\n");
	    printf ("       -glin -lflg\n");
	    printf ("       -dark -flat -shad -phot -stat\n");
	    FreeNames (inlist, outlist, input, output);
	    exit (ERROR_RETURN);
	}
	
	/* Initialize the structure for managing trailer file comments */
	InitTrlBuf ();
	
	/* Copy command-line value for QUIET to structure */
	SetTrlQuietMode(quiet);

	/* Was no calibration switch specified on command line? */
	if (!switch_on) {	/* default values (mostly PERFORM) */
	    acs2d_sw.dqicorr  = DefSwitch ("dqicorr");
	    acs2d_sw.glincorr = DefSwitch ("glincorr");
	    acs2d_sw.lflgcorr = DefSwitch ("lflgcorr");
	    acs2d_sw.darkcorr = DefSwitch ("darkcorr");
	    acs2d_sw.flatcorr = DefSwitch ("flatcorr");
	    acs2d_sw.shadcorr = DefSwitch ("shadcorr");
	    acs2d_sw.photcorr = DefSwitch ("photcorr");
	    acs2d_sw.statcorr = DefSwitch ("statcorr");
	}

	/* Expand the templates. */
	i_imt = c_imtopen (inlist);
	o_imt = c_imtopen (outlist);
	n_in = c_imtlen (i_imt);
	n_out = c_imtlen (o_imt);

	/* The number of input and output files must be the same. */
	if (CompareNumbers (n_in, n_out, "output"))
	    status = 1;
		
	if (status) {
		CloseTrlBuf ();
	    FreeNames (inlist, outlist, input, output);
	    exit (ERROR_RETURN);
	}

	/* Loop over the list of input files. */
	for (n = 0;  n < n_in;  n++) {

	    j = c_imtgetim (i_imt, input, ACS_LINE);
	    if (n_out > 0)
		j = c_imtgetim (o_imt, output, ACS_LINE);
	    else
		output[0] = '\0';

	    if (MkOutName (input, isuffix, osuffix, nsuffix, output, ACS_LINE)) {
			WhichError (status);
			sprintf (MsgText, "Skipping %s", input);
			trlmessage (MsgText);
			continue;
	    }
										
	    /* Calibrate the current input file. */
	    if (ACS2d (input, output, &acs2d_sw, &refnames, printtime, verbose)) {
			sprintf (MsgText, "Error processing %s.", input);
			trlerror (MsgText);
			WhichError (status);
	    }
	}

	/* Close lists of file names, and free name buffers 
		and trailer file buffer. 
	*/
	c_imtclose (i_imt);
	c_imtclose (o_imt);
	FreeRefFile (&refnames);
	FreeNames (inlist, outlist, input, output);
	CloseTrlBuf ();
	
	if (status)
	    exit (ERROR_RETURN);
	else
	    exit (ACS_OK);
}

static void FreeNames (char *inlist, char *outlist,
		char *input, char *output) {

	free (output);
	free (input);
	free (outlist);
	free (inlist);
}