# include "relphot.h"
void plot_setMcal (double *list, int Npts, StatType *stats, float clouds);

// the MeasureToImage, ImageToCatalog, and ImageToMeasure arrays are a substantial part of the memory footprint.
// total data volume is currently:
// a) Naverage*sizeof(AverageTiny)      [averages]       : 32
// b) Naverage*sizeof(Secfilt)*Nsecfilt [secfilt values] : 8 * 32
// b) Nmeasure*sizeof(Measure)          [measurements]   : 72
// c) Nmeasure*sizeof(IDX_T)*3          [image idx]      : 3 * 16
// d) Nmeasure*sizeof(IDX_T)*3          [mosaic idx]     : 3 * 16
// e) Nimage*sizeof(Image)              [image data]     : 360

// for 3pi analysis in 2012, we have Nmeasure ~ 20 x Naverage
// so, each measurement costs ~14.5B (Ave + Sec) + 72B (Meas) + 96B (idx)!
// we are using off_t (64bit) to avoid the 32bit limit of an int, but
// if we really had 2^31 measurements in a single analysis, we would be using > 350GB of ram...
// until we reach that point, it is sort of silly to use IDX_T = off_t here.

// in fact, we are safer than this, because the number of detections per table is rarely so large.

// with IDX_T = int, each measurement costs ~14.5B (Ave + Sec) + 72B (Meas) + 48B (idx)!

// elsewhere (not in relphot_images), we need to use off_t because a single catalog 

// we have an array of images: image[ImageIndex] (ImageIndex : 0 < Nimage)
static off_t        Nimage;   // number of available images
static Image        *image;   // array of available images  

// if we read only a subset of the rows from the Image FITS, LineNumber tells us to which row
// each image belongs
static off_t       *LineNumber; // match of subset to full image table

// to search by image ID, we sort (imageIDs, imageIdx) by imageIDs to get a sorted index
static off_t        *imageIDs; // list of all image IDs
static off_t        *imageIdx; // list of index for image IDs 

// elsewhere, we have loaded a set of catalogs with measures (catalog[cat].measure[meas])
// each image has N_onImage[ImageIndex] measurements
static off_t        *N_onImage;   // actual number of measurements on image	     
static off_t        *N_ONIMAGE;   // allocated number of measurements on image   

// relationships between the measure,catalog set and the images:
static IDX_T       **MeasureToImage = NULL; // image index from measure,catalog   : MeasureToImage[cat][meas] = ImageIndex
static IDX_T       **ImageToCatalog = NULL; // catalog for given measure on image : ImageCatalog[ImageIndex][i] = cat (i : 0 < NonImage[ImageIndex])
static IDX_T       **ImageToMeasure = NULL; // measure for given measure on image : ImageMeasure[ImageIndex][i] = cat (i : 0 < NonImage[ImageIndex])

// Projection Cell / SkyCell ID : for stack primary detection, we need to know the projection cell and skycell ID for each
// stack image.  for now, we generate these ID arrays based on the image names when we load in the image table (initImages).
// When we pass data to the remote clients via the ImageSubset, the projID/skycellID values are carried directly in the table.
int *tessID    = NULL;
int *projectID = NULL;
int *skycellID = NULL;

// MeasureToImage was 'bin'
// ImageToCatalog was 'clist'
// ImageToMeasure was 'mlist'

// N_onImage was 'Nlist'
// N_ONIMAGE was 'NLIST'

Image *getimages (off_t *N, off_t **line_number) {

  *N = Nimage;
  if (line_number) *line_number = LineNumber;
  return (image);
}

Image *getimage (off_t N) {
  return (&image[N]);
}

void initImages (Image *input, off_t *line_number, off_t N) {

  off_t i;

  image = input;
  LineNumber = line_number;
  Nimage = N;

  ALLOCATE (imageIDs, off_t, Nimage);
  ALLOCATE (imageIdx, off_t, Nimage);

  // for stack images, assign projection cell ID and skycell ID based on filenames
  ALLOCATE (tessID,      int, Nimage);
  ALLOCATE (projectID,   int, Nimage);
  ALLOCATE (skycellID,   int, Nimage);

  for (i = 0; i < Nimage; i++) {
    imageIdx[i] = i;
    imageIDs[i] = image[i].imageID;

    tessID[i]    = -1;
    projectID[i] = -1;
    skycellID[i] = -1;

    TessellationIDsByImageName (&tessID[i], &projectID[i], &skycellID[i], image[i].name);
  }

  // sort the image index by the IDs
  // XXX does this break the imageID <-> projectID, etc match?
  llsortpair (imageIDs, imageIdx, Nimage);
}

void initImagesSubset (ImageSubset *input, off_t *line_number, off_t N) {

  off_t i;

  // we have been given an ImageSubset array, containing a reduced set of image fields
  // create full a Image array and save the needed values
  ALLOCATE (image, Image, N);

  ALLOCATE (tessID,    int, N);
  ALLOCATE (projectID, int, N);
  ALLOCATE (skycellID, int, N);

  for (i = 0; i < N; i++) {
    image[i].imageID       = input[i].imageID      ;
    image[i].photom_map_id = input[i].photom_map_id;
    image[i].flags         = input[i].flags        ;
    image[i].Mcal          = input[i].Mcal         ;
    image[i].dMcal         = input[i].dMcal        ;
    image[i].tzero         = input[i].tzero        ;
    image[i].trate         = input[i].trate        ;
    image[i].ubercalDist   = input[i].ubercalDist  ;
    tessID[i]              = input[i].tessID       ;
    projectID[i]           = input[i].projID       ;
    skycellID[i]           = input[i].skycellID    ;
  }
  LineNumber = line_number;
  Nimage = N;

  ALLOCATE (imageIDs, off_t, Nimage);
  ALLOCATE (imageIdx, off_t, Nimage);

  for (i = 0; i < Nimage; i++) {
    imageIdx[i] = i;
    imageIDs[i] = image[i].imageID;
  }
  llsortpair (imageIDs, imageIdx, Nimage);
}

ImageSubset *getimages_subset (off_t *N) {

  *N = Nimage;

  off_t i;

  // we have been given an ImageSubset array, containing a reduced set of image fields
  // create full a Image array and save the needed values
  ImageSubset *subset = NULL;
  ALLOCATE (subset, ImageSubset, Nimage);
  for (i = 0; i < Nimage; i++) {
    subset[i].imageID       = image[i].imageID      ;
    subset[i].photom_map_id = image[i].photom_map_id;
    subset[i].flags         = image[i].flags        ;
    subset[i].Mcal          = image[i].Mcal         ;
    subset[i].dMcal         = image[i].dMcal        ;
    subset[i].tzero         = image[i].tzero        ;
    subset[i].trate         = image[i].trate        ;
    subset[i].ubercalDist   = image[i].ubercalDist  ;
    subset[i].tessID        = tessID[i];
    subset[i].projID        = projectID[i];
    subset[i].skycellID     = skycellID[i];
  }
  return subset;
}

off_t getImageByID (off_t ID) {

  // we have a pair of vectors (imageIDs, imageIdx) sorted by imageIDs
  // use bisection to find the specified image ID

  off_t Nlo, Nhi, N;

  Nlo = 0; Nhi = Nimage;
  while (Nhi - Nlo > 10) {
    N = 0.5*(Nlo + Nhi);
    if (imageIDs[N] < ID) {
      Nlo = MAX(N, 0);
    } else {
      Nhi = MIN(N + 1, Nimage);
    }
  }

  for (N = Nlo; N < Nhi; N++) {
    if (imageIDs[N] == ID)
      return (imageIdx[N]);
  }

  return (-1);
}

void initImageBins (Catalog *catalog, int Ncatalog, int doImageList) {

  IDX_T i, j;

  ALLOCATE (MeasureToImage, IDX_T *, Ncatalog);
  for (i = 0; i < Ncatalog; i++) {
    ALLOCATE (MeasureToImage[i], IDX_T, MAX (catalog[i].Nmeasure, 1));
    for (j = 0; j < catalog[i].Nmeasure; j++) MeasureToImage[i][j] = -1;
  }

  if (doImageList) {
    ALLOCATE (N_onImage, off_t, Nimage);
    ALLOCATE (N_ONIMAGE, off_t, Nimage);
    ALLOCATE (ImageToCatalog, IDX_T *, Nimage);
    ALLOCATE (ImageToMeasure, IDX_T *, Nimage);

    for (i = 0; i < Nimage; i++) {
      N_onImage[i] = 0;
      N_ONIMAGE[i] = 30;
      ALLOCATE (ImageToCatalog[i], IDX_T, N_ONIMAGE[i]);
      ALLOCATE (ImageToMeasure[i], IDX_T, N_ONIMAGE[i]);
    }
  }
}

void freeImageBins (int Ncatalog, int doImageList) {

  off_t i;

  for (i = 0; i < Ncatalog; i++) {
    free (MeasureToImage[i]);
  }
  free (MeasureToImage);

  if (doImageList) {
    for (i = 0; i < Nimage; i++) {
      free (ImageToCatalog[i]);
      free (ImageToMeasure[i]);
    }
    free (ImageToCatalog);
    free (ImageToMeasure);
    free (N_onImage);
    free (N_ONIMAGE);
  }
}

/* select all images equivalent to the active photcode set */
void findImages (Catalog *catalog, int Ncatalog, int doImageList) {

  off_t j;
  int i, ecode, Ns, found;
  
  int Nmatch = 0;
  for (i = 0; i < Ncatalog; i++) {
    for (j = 0; j < catalog[i].Nmeasure; j++) {
      catalog[i].measureT[j].myDet = FALSE; // a detetion is not mine until proven otherwise
      ecode = GetPhotcodeEquivCodebyCode (catalog[i].measureT[j].photcode);

      // skip measurements which do not match one of the requested photcodes (
      // (do we not already exclude in bcatalog -- maybe needed for reload_objects?
      found = FALSE;
      for (Ns = 0; !found && (Ns < Nphotcodes); Ns++) {
	if (ecode == photcodes[Ns][0].code) found = TRUE;
      }
      if (!found) continue; 

      // if we match one of our images, myDet gets set to TRUE
      matchImage (catalog, j, i, doImageList);
      Nmatch ++;
    }
  }
 // fprintf (stderr, "matched %d detections to images\n", Nmatch);
}

int findCCD (off_t idx, off_t meas, int cat, MeasureTiny *measure) {

  int ccdnum, found, Ns, ecode;
  double X, Y;
  char *pname, *filter, *p, base[256];

  /* identify the ccd on the basis of the photcode name */
  pname = GetPhotcodeNamebyCode (image[idx].photcode);

  // XXX this seems quite terrible...
  ecode = GetPhotcodeEquivCodebyCode (measure[0].photcode);
  found = FALSE;
  for (Ns = 0; !found && (Ns < Nphotcodes); Ns++) {
    if (ecode == photcodes[Ns][0].code) found = TRUE;
  }
  if (!found) return (FALSE); 

  filter = photcodes[Ns][0].name;
  sprintf (base, "%s.%s.", MOSAICNAME, filter);
  if (strncmp (pname, base, strlen (base))) return (FALSE);
  p = pname + strlen(base);

  /* p is pointing at the DETECTOR ID part of the photcode: CAMERA.FILTER.DETECTOR 
     for now, we will let this be of the form SSSSnn where SSS is an arbitrary string
     and nn is an integer -> ccdnum */
      
  /// XXX we have an inconsistency here wrt dvo.layout and dvo.photcode

  while (!isdigit(*p) && *p) p++;
  if (*p == 0) return (FALSE);
  ccdnum = atoi (p);

  /* ccdnum is an integer, but not necessarily a sequence number.  when the camera layout is
     constructed, there will be null values for undefined ccdnums */

  // old code to add this measurement to the grid cell for this chip 
  // ave = measureT[0].averef;
  // ra  = catalog[cat].averageT[ave].R - measureT[0].dR / 3600.0;
  // dec = catalog[cat].averageT[ave].D - measureT[0].dD / 3600.0;
  // RD_to_XY (&X, &Y, ra, dec, &image[i].coords);

  // XXX we can now use these values (but need to be careful about old formats)
  X = measure[0].Xccd;
  Y = measure[0].Yccd;
  setGridMeasure (meas, cat, X, Y, ccdnum);

  return (TRUE);
}

void matchImage (Catalog *catalog, off_t meas, int cat, int doImageList) {

  off_t idx, ID;
  int status;
  MeasureTiny *measure;
  
  measure = &catalog[cat].measureT[meas];

  ID = measure[0].imageID;
  idx = getImageByID (ID);
  if (idx == -1) {
    if (VERBOSE2) fprintf (stderr, "can't match detection to image?\n");
    return;
  }
  catalog[cat].measureT[meas].myDet = TRUE;

  if (USE_GRID) {
    status = findCCD (idx, meas, cat, measure);
    if (!status) {
      if (VERBOSE2) fprintf (stderr, "failed to determine CCD for "OFF_T_FMT", %d\n",  meas, cat);
      return;
    }
  }

  // index for (catalog, measure) -> image
  MeasureToImage[cat][meas] = idx;

  if (doImageList) {
    // index for image, Nentry -> catalog
    ImageToCatalog[idx][N_onImage[idx]] = cat;

    // index for image, Nentry -> measure
    ImageToMeasure[idx][N_onImage[idx]] = meas;
    N_onImage[idx] ++;

    if (N_onImage[idx] == N_ONIMAGE[idx]) {
      N_ONIMAGE[idx] += 30;
      REALLOCATE (ImageToCatalog[idx], IDX_T, N_ONIMAGE[idx]);
      REALLOCATE (ImageToMeasure[idx], IDX_T, N_ONIMAGE[idx]);
    }	
  }

  return;
}

off_t getImageEntry (off_t meas, int cat) {

  off_t i;

  if (!MeasureToImage) return -1;

  i = MeasureToImage[cat][meas];
  return (i);
}

// returns image.Mcal - ff(x,y)
float getMcal (off_t meas, int cat, FlatCorrectionTable *flatcorr, Catalog *catalog) {

  off_t i;
  float value, offset;

  i = MeasureToImage[cat][meas];
  if (i == -1) return (NAN);

  if (image[i].flags & IMAGE_BAD) return (NAN);  
  value = image[i].Mcal;
  offset = 0.0;

  // to do this, I need to pass in the catalog and flatcorr pointers
  int flat_id = image[i].photom_map_id;
  if (flat_id) {
    offset = FlatCorrectionOffset (flatcorr, flat_id, catalog[cat].measureT[meas].Xccd, catalog[cat].measureT[meas].Yccd);
  }
  value -= offset;

  return (value);
}

// returns image.Mcal - ff(x,y)
float getMcal_alt (off_t meas, int cat, FlatCorrectionTable *flatcorr, float Xccd, float Yccd) {

  off_t i;
  float value, offset;

  i = MeasureToImage[cat][meas];
  if (i == -1) return (NAN);

  if (image[i].flags & IMAGE_BAD) return (NAN);  
  value = image[i].Mcal;
  offset = 0.0;

  // to do this, I need to pass in the catalog and flatcorr pointers
  int flat_id = image[i].photom_map_id;
  if (flat_id) {
    offset = FlatCorrectionOffset (flatcorr, flat_id, Xccd, Yccd);
  }
  value -= offset;

  return (value);
}

// returns image.Mcal - ff(x,y)
short getUbercalDist (off_t meas, int cat) {

  off_t i;
  short distance;

  if (!MeasureToImage) return -1;

  i = MeasureToImage[cat][meas];
  if (i == -1) return (1000);

  if (image[i].flags & IMAGE_BAD) return (1000);  
  distance = image[i].ubercalDist; // was dummy3 in structure
  return (distance);
}

// returns image.Mcal - ff(x,y)
float getCenterOffset (off_t meas, int cat, Measure *measure, unsigned int *myID) {

  off_t i;
  float distance;

  if (!MeasureToImage) return -1;

  i = MeasureToImage[cat][meas];
  if (i == -1) return (1000);

  float Xcenter = 0.5*image[i].NX;
  float Ycenter = 0.5*image[i].NY;

  *myID = image[i].imageID;

  distance = hypot (measure[0].Xccd - Xcenter, measure[0].Yccd - Ycenter);
  return (distance);
}

// returns image.Mcal - ff(x,y)
int MatchImageName (off_t meas, int cat, char *name) {

  off_t i;

  if (!name) return FALSE;
  if (!name[0]) return FALSE;

  if (!MeasureToImage) return FALSE;

  i = MeasureToImage[cat][meas];
  if (i == -1) return FALSE;

  // this is a bit crude: stack image names are of the form:
  // RINGS.V3.skycell.1495.027.sky.191211.stk.988232.cmf.

  // the primaryCell has a name of the form RINGS.V3.skycell.1495 or RINGS.V3.skycell.1495.027
  // (if we use projection or skycell as the primary)

  if (!strncmp(image[i].name, name, strlen(name))) return TRUE;
  return FALSE;
}

// returns image.Mcal - ff(x,y)
int MatchImageSkycellID (off_t meas, int cat, int myTessID, int myProjectionID, int mySkycellID) {

  off_t i;

  if (!MeasureToImage) return FALSE;

  i = MeasureToImage[cat][meas];
  if (i == -1) return FALSE;

  if (tessID[i]    == -1) return FALSE;
  if (projectID[i] == -1) return FALSE;
  if (skycellID[i] == -1) return FALSE;

  if (tessID[i]    != myTessID) return FALSE;
  if (projectID[i] != myProjectionID) return FALSE;
  if (skycellID[i] != mySkycellID) return FALSE;
  
  return TRUE;
}

float getMflat (off_t meas, int cat, FlatCorrectionTable *flatcorr, Catalog *catalog) {

  off_t i = MeasureToImage[cat][meas];
  if (i == -1) return (NAN);

  float offset = 0.0;

  // to do this, I need to pass in the catalog and flatcorr pointers
  int flat_id = image[i].photom_map_id;
  if (flat_id) {
    offset = FlatCorrectionOffset (flatcorr, flat_id, catalog[cat].measureT[meas].Xccd, catalog[cat].measureT[meas].Yccd);
  }

  return (offset);
}

Coords *getCoords (off_t meas, int cat) {

  off_t i;

  i = MeasureToImage[cat][meas];
  if (i == -1) return (NULL);
  return (&image[i].coords);
}

/* determine Mcal values for all images */
void setMcal (Catalog *catalog, int PoorImages, FlatCorrectionTable *flatcorr) {

  off_t i, j, m, c, n, N, Nmax;
  int mark, bad, Nfew, Nbad, Nmos, Nrel, Ngrid, Nsys, Nbright;
  float Msys, Mrel, Mmos, Mgrid, Mflat;
  double *list, *dlist, *Mlist, *dMlist;

  StatType stats;
  liststats_setmode (&stats, STATMODE);

  if (FREEZE_IMAGES) return;

  fprintf (stderr, "limiting negative clouds to %f\n", CLOUD_TOLERANCE);

  int Nsecfilt = GetPhotcodeNsecfilt ();

  if (PoorImages) {
    // XXX use bad stars and measurements for PoorImages? or not?
    // IMAGE_BAD = STAR_BAD = MEAS_BAD = 0;
    IMAGE_BAD = 0;
  }

  Nmax = 0;
  for (i = 0; i < Nimage; i++) {
    Nmax = MAX (Nmax, N_onImage[i]);
  }
  ALLOCATE (list, double, Nmax);
  ALLOCATE (dlist, double, Nmax);
  ALLOCATE (Mlist, double, Nmax);
  ALLOCATE (dMlist, double, Nmax);

  Nfew = Nbad = Nmos = Ngrid = Nrel = Nsys = 0;

  for (i = 0; i < Nimage; i++) {
    
    /* on PoorImages run, skip good images */
    if (PoorImages) {
      bad = image[i].flags & (ID_IMAGE_PHOTOM_FEW | ID_IMAGE_PHOTOM_POOR | ID_IMAGE_PHOTOM_SKIP);
      if (!bad) continue;
    }      

    // UBERCAL image: if this is an ubercal image, set minUbercalDist to 0:
    // we optionally do not recalibrate images with UBERCAL zero points 
    if (image[i].flags & ID_IMAGE_PHOTOM_UBERCAL) {
      image[i].ubercalDist = 0; // was dummy3
      if (KEEP_UBERCAL) continue;
    }

    int minUbercalDist = 1000;

    // number of stars to measure the bright-end scatter
    Nbright = 0;

    N = 0;
    for (j = 0; j < N_onImage[i]; j++) {
      
      m = ImageToMeasure[i][j];
      c = ImageToCatalog[i][j];
      
      if (catalog[c].measureT[m].dbFlags & MEAS_BAD) {
	  Nbad++;
	  continue;
      }
      Mmos  = getMmos  (m, c);
      if (isnan(Mmos)) {
	  Nmos ++;
	  continue;
      }
      Mgrid = getMgrid (m, c);
      if (isnan(Mgrid)) {
	  Ngrid++;
	  continue;
      }
      Mrel  = getMrel  (catalog, m, c);
      if (isnan(Mrel)) {
	  Nrel ++;
	  continue;
      }
      
      // image.Mcal is not supposed to include the flat-field correction, so we need to
      // apply that offset as well here for this image (in other words, each detection is
      // being compared to the model, excluding the zero point, Mcal.  The model includes
      // the flat-correction.  NOTE the sign of Mflat (Image.Mcal = Measure.Mcal - Mflat)

      Mflat = getMflat (m, c, flatcorr, catalog);

      n = catalog[c].measureT[m].averef;
      Msys = PhotSysTiny (&catalog[c].measureT[m], &catalog[c].averageT[n], &catalog[c].secfilt[n*Nsecfilt]);
      if (isnan(Msys)) {
	Nsys++;
	continue;
      }

      PhotCode *code = GetPhotcodebyCode (catalog[c].measureT[m].photcode);
      if (!code) goto skip;
      if (code->equiv < 1) goto skip;
      int Nsec = GetPhotcodeNsec (code->equiv);
      if (Nsec == -1) goto skip;
      minUbercalDist = MIN (catalog[c].secfilt[n*Nsecfilt + Nsec].ubercalDist, minUbercalDist);

      // fprintf (stderr, "%1d, %3d : %3d, %3d : %10.6f %10.6f : %6.3f  %6.3f  %6.3f  %6.3f  %6.3f\n", (int) i, (int) j, (int) c, (int) m, catalog[c].averageT[n].R, catalog[c].averageT[n].D, Msys, Mrel, Mmos, Mgrid, Mflat);

    skip:
      list[N] = Msys - Mrel - Mmos - Mgrid + Mflat;
      dlist[N] = MAX (catalog[c].measureT[m].dM, MIN_ERROR);
      if (catalog[c].measureT[m].dM < IMFIT_SYS_SIGMA_LIM) {
	Mlist[Nbright] = list[N];
	dMlist[Nbright] = dlist[N];
	Nbright ++;
      }
      N++;
    }
    /* N_onImage[i] is all measurements, N is good measurements */

    // if (VERBOSE2) fprintf (stderr, "meas skipped: (Nbad: %d, Nmos: %d, Ngrid: %d, Nrel: %d, Nsys: %d)\n", Nbad, Nmos, Ngrid, Nrel, Nsys);

    /* too few good measurements or too many bad measurements */
    if (!PoorImages) {
      mark = (N < IMAGE_TOOFEW) || (N < IMAGE_GOOD_FRACTION*N_onImage[i]);
      if (mark) {
	image[i].flags |= ID_IMAGE_PHOTOM_FEW;
	Nfew ++;
      } else {
	image[i].flags &= ~ID_IMAGE_PHOTOM_FEW;
      }      
    }

    // no additional weight modification (we treat all stars on an image equally -- note an image is either ubercal-tied or not)
    liststats (list, dlist, NULL, N, &stats);
    image[i].Mcal  = stats.mean;
    image[i].dMcal = stats.error;
    image[i].nFitPhotom = N;
    image[i].Xm    = 100.0*log10(stats.chisq);

    if (PLOTSTUFF) {
      fprintf (stderr, "Mcal for : %s\n", image[i].name);
      plot_setMcal (list, N, &stats, CLOUD_TOLERANCE);
    }

    // bright end scatter
    liststats (Mlist, dMlist, NULL, Nbright, &stats);
    image[i].dMagSys = stats.sigma;

    if (image[i].Mcal < -CLOUD_TOLERANCE) {
      image[i].Mcal = 0.0;
    }

    // minUbercalDist calculated here is the min value for any star owned by this image
    // since this particular image is tied to that star, bump its distance by 1
    image[i].ubercalDist = minUbercalDist + 1;
  }
  free (list);
  free (dlist);
  free (Mlist);
  free (dMlist);

  fprintf (stderr, "%d images marked having too few measurements (Nbad: %d, Nmos: %d, Ngrid: %d, Nrel: %d, Nsys: %d)\n", Nfew, Nbad, Nmos, Ngrid, Nrel, Nsys);

  if (PoorImages) {
    IMAGE_BAD = ID_IMAGE_PHOTOM_POOR | ID_IMAGE_PHOTOM_FEW | ID_IMAGE_PHOTOM_SKIP;
    STAR_BAD  = ID_STAR_POOR | ID_STAR_FEW;
    MEAS_BAD  = ID_MEAS_NOCAL | ID_MEAS_POOR_PHOTOM | ID_MEAS_SKIP_PHOTOM | ID_MEAS_AREA;
  }
  return;
}

/* mark image if: abs(Mcal) too large, dMcal too large */
void clean_images () {

  int mark, Nmark;
  off_t i, N;
  double *mlist, *slist, *dlist;
  double MaxOffset, MaxScatter, MedOffset;

  if (FREEZE_IMAGES) return;

  if (VERBOSE) fprintf (stderr, "marking poor images\n");

  ALLOCATE (mlist, double, Nimage);
  ALLOCATE (slist, double, Nimage);
  ALLOCATE (dlist, double, Nimage);

  // measure stats for Mcal and dMcal
  for (i = N = 0; i < Nimage; i++) {
    if (image[i].flags & IMAGE_BAD) continue;
    mlist[N] = image[i].Mcal;
    slist[N] = image[i].dMcal;
    dlist[N] = 1;
    N++;
  }

  // use a straight mean to find the global image statistics (no weighting)
  StatType stats;
  liststats_setmode (&stats, "MEAN");

  liststats (mlist, dlist, NULL, N, &stats);
  MaxOffset = MAX (IMAGE_OFFSET, 3*stats.sigma);
  MedOffset = stats.median;

  liststats (slist, dlist, NULL, N, &stats);
  MaxScatter = MAX (IMAGE_SCATTER, 2*stats.median);
  fprintf (stderr, "Mrel: %f, dMrel: %f, Max Scatter: %f, Max Offset: %f\n", MedOffset, stats.median, MaxScatter, MaxOffset);
  
  Nmark = 0;
  for (i = 0; i < Nimage; i++) {
    // if we are keeping ubercal sacrosanct, then we should not be allowed to break them...
    if (KEEP_UBERCAL && (image[i].flags & ID_IMAGE_PHOTOM_UBERCAL)) continue;

    mark = FALSE;
    image[i].flags &= ~ID_IMAGE_PHOTOM_POOR;
    mark = (image[i].dMcal > MaxScatter) || (fabs(image[i].Mcal - MedOffset) > MaxOffset);
    if (mark) { 
      Nmark ++;
      image[i].flags |= ID_IMAGE_PHOTOM_POOR;
    } else {
      image[i].flags &= ~ID_IMAGE_PHOTOM_POOR;
    }
  }

  fprintf (stderr, "%d images marked poor\n", Nmark);
  free (mlist);
  free (slist);
  free (dlist);
}

static int setMcal_init_done = FALSE;
void plot_setMcal (double *list, int Npts, StatType *stats, float clouds) {

  off_t i;
  double *xlist;
  Graphdata graphdata;

  if (!PLOTSTUFF) return;

  if (Npts == 0) {
    // fprintf (stderr, "cannot set image Mcal values yet (nan Mave)\n");
    return;
  }

  plot_defaults (&graphdata);
  graphdata.xmin = -0.01;
  graphdata.xmax = +1.01;
  graphdata.ymin = -0.21;
  graphdata.ymax = +0.21;

  if (!setMcal_init_done) {
    int kapa = get_graph(0);
    if (kapa < 1) {
      kapa = open_graph(0);
      if (!kapa) return;
    }

    KapaClearSections (kapa);
    KapaClearPlots (kapa);
    KapaSetLimits (kapa, &graphdata);
    KapaSetFont (kapa, "helvetica", 14);
    KapaBox (kapa, &graphdata);
    setMcal_init_done = TRUE;
  }

  ALLOCATE (xlist, double, Npts);
  for (i = 0; i < Npts; i++) {
    xlist[i] = i / (float) Npts;
  }

  plot_list_add (&graphdata, xlist, list, Npts);

  free (xlist);
}

void plot_images () {

  off_t i, bin;
  double *xlist, *Mlist, *dlist;
  Graphdata graphdata;

  if (FREEZE_IMAGES) return;

  ALLOCATE (xlist, double, Nimage);
  ALLOCATE (dlist, double, Nimage);
  ALLOCATE (Mlist, double, Nimage);

  /**** dMcal vs airmass ****/
  float minAirmass = 1000.0;
  float maxAirmass =    0.0;
  float minMcal    = +100.0;
  float maxMcal    = -100.0;
  float mindMcal    = +100.0;
  float maxdMcal    = -100.0;
  for (i = 0; i < Nimage; i++) {
    Mlist[i] = image[i].Mcal;
    dlist[i] = image[i].dMcal;
    xlist[i] = image[i].secz;
    minAirmass = MIN (image[i].secz, minAirmass);
    maxAirmass = MAX (image[i].secz, maxAirmass);
    minMcal = MIN (image[i].Mcal, minMcal);
    maxMcal = MAX (image[i].Mcal, maxMcal);
    mindMcal = MIN (image[i].dMcal, mindMcal);
    maxdMcal = MAX (image[i].dMcal, maxdMcal);
  }

  float AirmassRange  = MAX(1.2*(maxAirmass - minAirmass), 0.25);
  float AirmassCenter = 0.5*(maxAirmass + minAirmass);

  float McalRange  = MAX(1.2*(maxMcal - minMcal), 0.21);
  float McalCenter = 0.5*(maxMcal + minMcal);

  float dMcalRange  = MAX(1.2*(maxdMcal - mindMcal), 0.21);
  float dMcalCenter = 0.5*(maxdMcal + mindMcal);

  plot_defaults (&graphdata);
  graphdata.xmin = AirmassCenter - 0.5*AirmassRange;
  graphdata.xmax = AirmassCenter + 0.5*AirmassRange;
  graphdata.ymin = McalCenter - 0.5*McalRange;
  graphdata.ymax = McalCenter + 0.5*McalRange;
  plot_list (&graphdata, xlist, Mlist, Nimage, "airmass vs Mcal", "%s.airmass.png", OUTROOT);

  plot_defaults (&graphdata);
  graphdata.xmin = McalCenter - 0.5*McalRange;
  graphdata.xmax = McalCenter + 0.5*McalRange;
  graphdata.ymin = dMcalCenter - 0.5*dMcalRange;
  graphdata.ymax = dMcalCenter + 0.5*dMcalRange;
  plot_list (&graphdata, Mlist, dlist, Nimage, "Mcal vs dMcal", "%s.Mcal.dMcal.png", OUTROOT);

# define NBIN 200
  REALLOCATE (xlist, double, NBIN);
  REALLOCATE (Mlist, double, NBIN);

  /**** dMcal histgram ****/
  for (i = 0; i < NBIN; i++) xlist[i] = 0.00025*i;
  bzero (Mlist, NBIN*sizeof(double));
  for (i = 0; i < Nimage; i++) {
    bin = image[i].dMcal / 0.00025;
    bin = MAX (0, MIN (NBIN - 1, bin));
    Mlist[bin] += 1.0;
  }

  plot_defaults (&graphdata);
  graphdata.style = 1;
  graphdata.xmin = dMcalCenter - 0.5*dMcalRange;
  graphdata.xmax = dMcalCenter + 0.5*dMcalRange;
  plot_list (&graphdata, xlist, Mlist, NBIN, "dMcal hist", "%s.dMcalhist.png", OUTROOT);

  free (dlist);
  free (xlist);
  free (Mlist);
}

StatType statsImageN (Catalog *catalog, FlatCorrectionTable *flatcorr) {

  off_t i, j, m, c, n, N;
  double *list, *dlist;
  float Mcal, Mmos, Mgrid;

  StatType stats;
  bzero (&stats, sizeof (StatType));
  if (FREEZE_IMAGES) return (stats);

  ALLOCATE (list, double, Nimage);
  ALLOCATE (dlist, double, Nimage);

  n = 0;
  for (i = 0; i < Nimage; i++) {
    if (image[i].flags & IMAGE_BAD)  continue;

    N = 0;
    for (j = 0; j < N_onImage[i]; j++) {

      m = ImageToMeasure[i][j];
      c = ImageToCatalog[i][j];

      Mcal  = getMcal  (m, c, flatcorr, catalog);
      if (isnan(Mcal)) continue;
      Mmos  = getMmos  (m, c);
      if (isnan(Mmos)) continue;
      Mgrid = getMgrid (m, c);
      if (isnan(Mgrid)) continue;
      N++;
    }
    list[n] = N;
    dlist[n] = 1;
    n++;
  }

  liststats_setmode (&stats, "MEAN");

  liststats (list, dlist, NULL, n, &stats);
  free (list);
  free (dlist);
  return (stats);
}

StatType statsImageX (Catalog *catalog) {

  off_t i, n;
  double *list, *dlist;
  StatType stats;

  bzero (&stats, sizeof (StatType));
  if (FREEZE_IMAGES) return (stats);

  ALLOCATE (list, double, Nimage);
  ALLOCATE (dlist, double, Nimage);

  n = 0;
  for (i = 0; i < Nimage; i++) {

    if (image[i].flags & IMAGE_BAD)  continue;

    list[n] = pow (10.0, 0.01*image[i].Xm);
    dlist[n] = 1;
    n++;
  }

  liststats_setmode (&stats, "MEAN");

  liststats (list, dlist, NULL, n, &stats);
  free (list);
  free (dlist);
  return (stats);
}

StatType statsImageM (Catalog *catalog) {

  off_t i, n;
  double *list, *dlist;
  StatType stats;

  bzero (&stats, sizeof (StatType));
  if (FREEZE_IMAGES) return (stats);

  ALLOCATE (list, double, Nimage);
  ALLOCATE (dlist, double, Nimage);

  n = 0;
  for (i = 0; i < Nimage; i++) {

    if (image[i].flags & IMAGE_BAD)  continue;

    list[n] = image[i].Mcal;
    dlist[n] = 1;
    n++;
  }

  liststats_setmode (&stats, "MEAN");

  liststats (list, dlist, NULL, n, &stats);
  free (list);
  free (dlist);
  return (stats);
}

StatType statsImagedM (Catalog *catalog) {

  off_t i, n;
  double *list, *dlist;
  StatType stats;

  bzero (&stats, sizeof (StatType));
  if (FREEZE_IMAGES) return (stats);

  ALLOCATE (list, double, Nimage);
  ALLOCATE (dlist, double, Nimage);

  n = 0;
  for (i = 0; i < Nimage; i++) {

    if (image[i].flags & IMAGE_BAD)  continue;

    list[n] = image[i].dMcal;
    dlist[n] = 1;
    n++;
  }

  liststats_setmode (&stats, "MEAN");

  liststats (list, dlist, NULL, n, &stats);
  free (list);
  free (dlist);
  return (stats);
}
