#include <TASImage.h>

#include <iostream>
#include <iomanip>
#include <sstream>
#include <fstream>
#include <string>

Double_t *sumArray;

TASImage *summer(char *prefix, Int_t seq, Int_t end, Int_t start=1)
{
   bool oldFliCam = true;

   std::stringstream sst;
   sst << prefix << "_"
       << std::setw(3) << std::setfill('0') << seq << "_" 
       << std::setw(3) << std::setfill('0') << 0 << ".tif";
   TASImage *img = new TASImage(sst.str().c_str());
 
   // the new Andor camera uses a different filename convention
   if (!img) {
      sst.str("");
      sst << prefix << "_scan" 
          << std::setfill('0') << std::setw(3) << seq << "_"
          << std::setfill('0') << std::setw(3) << start << ".tiff";
      img = new TASImage(sst.str().c_str());
      oldFliCam = false;
   }

   // read the dimensions of the topographs from the first image file
   Int_t height = img->GetHeight();
   Int_t width = img->GetWidth();
   UInt_t *reference = (UInt_t *)img->GetRgbaArray();
   sumArray = new Double_t[width*height];
   for (Int_t index=start; index<end; ++index) {
      if (oldFliCam) {
         sst.str("");
         sst << prefix << "_" 
             << std::setw(3) << std::setfill('0') << seq << "_" 
             << std::setw(3) << std::setfill('0') << index << ".tif";
      }
      else {
         sst.str("");
         sst << prefix << "_scan" 
             << std::setfill('0') << std::setw(3) << seq << "_"
             << std::setfill('0') << std::setw(3) << start << ".tiff";
      }
      TASImage *frame = new TASImage(sst.str().c_str());
      UInt_t *pixel = (UInt_t *)frame->GetRgbaArray();
      for (Int_t iv=0,k=0; iv<height; ++iv) {
         for (Int_t iu=0; iu<width; ++iu,k++) {
            Int_t ref_red = (reference[k]>>24) & 0xff;
            Int_t ref_green = (reference[k]>>16) & 0xff;
            Int_t ref_blue = (reference[k]>>8) & 0xff;
            //Int_t ref_alpha = (reference[k]>>0) & 0xff;
            Int_t red = (pixel[k]>>24) & 0xff;
            Int_t green = (pixel[k]>>16) & 0xff;
            Int_t blue = (pixel[k]>>8) & 0xff;
            //Int_t alpha = (pixel[k]>>0) & 0xff;
            Double_t intensity = red-ref_red +
                                 green-ref_green +
                                 blue-ref_blue;
            if (intensity > 0) {
               sumArray[k] += intensity;
            }
          }
       }
      delete [] pixel;
      delete frame;
   }
   img->SetImage(sumArray,width,height);
   delete [] sumArray;
   delete reference;
   return img;
}