#include "TAwagsSisMapProc.h"

#include <math.h>
#include <vector>

#include "TH1.h"
#include "TH2.h"
#include "TROOT.h"

#include "TGo4WinCond.h"
#include "TGo4Analysis.h"

#include "TAwagsSisMapEvent.h"
#include "TAwagsSisBasicEvent.h"

#include "TAwagsSisParam.h"
//-----------------------------------------------------------
TAwagsSisMapProc::TAwagsSisMapProc() :
    TGo4EventProcessor()
{
}

//-----------------------------------------------------------
TAwagsSisMapProc::TAwagsSisMapProc(const char* name) :
    TGo4EventProcessor(name), fParam(0)
{
  TGo4Log::Info("TAwagsSisMapProc: Create");
  //// init user analysis objects:
  SetMakeWithAutosave(kTRUE);
  // here already exisitng parameter will be returned
  // one not need to specify macro here - it is already executed in first step
  fParam = dynamic_cast<TAwagsSisMapParam*> (MakeParameter("AwagsSisMapParam", "TAwagsSisMapParam", "set_AwagsSisMapParam.C"));
  TString obname;
  TString obtitle;
  //TString foldername;



  Int_t spillsamples = 200;
#ifdef AWAGS_STORE_TRACES
  Int_t tracelength=CSA_TRACE_SIZE;    // TODO: dynamically get trace size from firhSampleChargeProfilest step?
#endif
  TAwagsSisParam* firstparam = dynamic_cast<TAwagsSisParam*>(GetParameter("AwagsSisParam"));
  if (firstparam)
  {
    spillsamples = firstparam->fNumEventsPerSpill;
    printf("QQQQ took number of events per spill %d from parameter %s", spillsamples, firstparam->GetName());
  }

 Int_t spilllen=CSA_TRACE_SIZE;
 if(fParam && fParam->fSpillLength>0)
      spilllen=fParam->fSpillLength;


  Int_t numchambers = GetNumChambers();
  for (Int_t dev = 0; dev < numchambers; ++dev)
  {
    Int_t numwires = GetNumWires(dev);
    for (Int_t wire = 0; wire < numwires; ++wire)
    {
#ifdef AWAGS_STORE_TRACES
      obname.Form("Mapped/Chamber%d/Traces/SpillTrace_%d_%d", dev, dev, wire);
      obtitle.Form("Chamber %d Wire %d Raw Traces during spill", dev, wire);
      hWireTraces[dev][wire] = MakeTH1('I', obname.Data(), obtitle.Data(), tracelength, 0, tracelength, "time (bins)",
          "counts");
#endif
      obname.Form("Mapped/Chamber%d/Charge/ChargeTrend_%d_%d", dev, dev, wire);
      obtitle.Form("Chamber %d Wire %d Charge trend during spill", dev, wire);
      TH1* testhis = GetHistogram(obname.Data());
      if (testhis && (testhis->GetNbinsX() != spillsamples))
      {
        //printf("QQQ MBS event range for Spill changed in parameter to %d, recreated spill charge histograms for chamber %d\n",
        //    spillsamples,dev);
        SetMakeWithAutosave(kFALSE);
      }
      if(spillsamples>1){
      hWireSpillCharge[dev][wire] = MakeTH1('D', obname.Data(), obtitle.Data(), spillsamples, 0, spillsamples,
          "MBS Event sample index", "#Delta Q");
      }
      else{
        hWireSpillCharge[dev][wire] = MakeTH1('D', obname.Data(), obtitle.Data(), spilllen, 0, spilllen,
                 "Spill sample", "#Delta Q");

      }


      SetMakeWithAutosave(kTRUE);

    }    // wires
#ifdef AWAGS_STORE_TRACES
    Int_t trbinsize= tracelength/10;
    obname.Form("Mapped/Chamber%d/OverviewSpillTrace_%d", dev, dev);
    obtitle.Form("Chamber %d Trace wire map", dev);
    hMapTrace[dev]=MakeTH2('I', obname.Data(), obtitle.Data(), trbinsize, 0, tracelength, numwires, 0, numwires,"time (bins)", "wire number", "counts");

    obname.Form("Mapped/Chamber%d/SumOverviewSpillTrace_%d", dev, dev);
    obtitle.Form("Chamber %d Trace wire map accumulated", dev);
    hMapTrace_Sum[dev]=MakeTH2('I', obname.Data(), obtitle.Data(), trbinsize, 0, tracelength, numwires, 0, numwires,"time (bins)", "wire number", "counts");

    obname.Form("Mapped/Chamber%d/WireProfile_%d", dev, dev);
    obtitle.Form("Chamber %d Wire Profile", dev);
    hWireProfile[dev] = MakeTH1('I', obname.Data(), obtitle.Data(), numwires, 0, numwires, "wire number", "counts");

    obname.Form("Mapped/Chamber%d/SumWireProfile_%d", dev, dev);
    obtitle.Form("Chamber %d Wire Profile accumulated", dev);
    hWireProfile_Sum[dev] = MakeTH1('I', obname.Data(), obtitle.Data(), numwires, 0, numwires, "wire number", "counts");
#endif
    obname.Form("Mapped/Chamber%d/OverviewSpillCharge_%d", dev, dev);
    obtitle.Form("Chamber %d Charge wire map", dev);
    // check for dynamic recreation here:
    TH1* testhis = GetHistogram(obname.Data());
    if (testhis && (testhis->GetNbinsX() != spillsamples))
    {
      printf(
          "QQQ MBS event range for Spill changed in parameter to %d, recreated spill charge histograms for chamber %d\n",
          spillsamples, dev);
      SetMakeWithAutosave(kFALSE);
    }
    if (testhis && (testhis->GetNbinsY() != numwires))
    {
      printf("QQQ Number  of wires changed in parameter to %d, recreated spill charge histograms for chamber %d\n",
          numwires, dev);
      SetMakeWithAutosave(kFALSE);
    }


    if(spillsamples>1)
    {
      // multiple events for each spill
      hMapSpillCharge[dev] = MakeTH2('I', obname.Data(), obtitle.Data(), spillsamples, 0, spillsamples, numwires, 0,
          numwires, "MBS Event sample", "wire number", "#Delta Q");

      obname.Form("Mapped/Chamber%d/SumOverviewSpillCharge_%d", dev, dev);
      obtitle.Form("Chamber %d Charge wire map accumulated", dev);
      hMapSpillCharge_Sum[dev] = MakeTH2('I', obname.Data(), obtitle.Data(), spillsamples, 0, spillsamples, numwires, 0,
          numwires, "MBS Event sample", "wire number", "#Delta Q");

    }
    else
    {
       // each event traces contain full spill
        hMapSpillCharge[dev] = MakeTH2('I', obname.Data(), obtitle.Data(), spilllen, 0, spilllen, numwires, 0,
               numwires, "Spill sample", "wire number", "#Delta Q");

        obname.Form("Mapped/Chamber%d/SumOverviewSpillCharge_%d", dev, dev);
        obtitle.Form("Chamber %d Charge wire map accumulated", dev);
        hMapSpillCharge_Sum[dev] = MakeTH2('I', obname.Data(), obtitle.Data(), spilllen, 0, spilllen, numwires, 0,
            numwires, "Spill sample", "wire number", "#Delta Q");
    }

    obname.Form("Mapped/Chamber%d/WireChargeProfile_%d", dev, dev);
    obtitle.Form("Chamber %d Wire ChargeProfile", dev);
    hWireChargeProfile[dev] = MakeTH1('D', obname.Data(), obtitle.Data(), numwires, 0, numwires, "wire number",
        "#Delta Q");

    obname.Form("Mapped/Chamber%d/SumWireChargeProfile_%d", dev, dev);
    obtitle.Form("Chamber %d Wire ChargeProfile accumulated", dev);
    hWireChargeProfile_Sum[dev] = MakeTH1('D', obname.Data(), obtitle.Data(), numwires, 0, numwires, "wire number",
        "#Delta Q");

    if(spillsamples>1)
      {
        // multiple events for each spill
    obname.Form("Mapped/Chamber%d/SampleChargeProfile_%d", dev, dev);
    obtitle.Form("Chamber %d Sample ChargeProfile", dev);
    hSampleChargeProfile[dev] = MakeTH1('D', obname.Data(), obtitle.Data(), spillsamples, 0, spillsamples,
        "MBS Event sample", "#Delta Q");

    obname.Form("Mapped/Chamber%d/SumSampleChargeProfile_%d", dev, dev);
    obtitle.Form("Chamber %d Sample ChargeProfile accumulated", dev);
    hSampleChargeProfile_Sum[dev] = MakeTH1('D', obname.Data(), obtitle.Data(), spillsamples, 0, spillsamples,
        "MBS Event sample", "#Delta Q");

      }
    else
       {
          // each event traces contain full spill
      obname.Form("Mapped/Chamber%d/SampleChargeProfile_%d", dev, dev);
         obtitle.Form("Chamber %d Sample ChargeProfile", dev);
         hSampleChargeProfile[dev] = MakeTH1('D', obname.Data(), obtitle.Data(), spilllen, 0, spilllen,
             "Spill sample", "#Delta Q");

         obname.Form("Mapped/Chamber%d/SumSampleChargeProfile_%d", dev, dev);
         obtitle.Form("Chamber %d Sample ChargeProfile accumulated", dev);
         hSampleChargeProfile_Sum[dev] = MakeTH1('D', obname.Data(), obtitle.Data(), spilllen, 0, spilllen,
             "Spill sample", "#Delta Q");

       }

    SetMakeWithAutosave(kTRUE);

  }    // devs

  if(spillsamples>1)
       {
        // signal traces only useful for old evaluation of spill with several MBS traces, do not use for new mode 13-05-25 JAM
        obname.Form("Mapped/AllSigTrace");
        obtitle.Form("Stitched trace of signal region during last spill");

         TH1* testhis = GetHistogram(obname.Data());
            if (testhis && (testhis->GetNbinsX() != spillsamples*CSA_SIGNAL_SIZE))
            {
              printf(
                  "QQQ MBS event range for Spill changed in parameter to %d, recreate signal trace histograms\n",
                  spillsamples);
              SetMakeWithAutosave(kFALSE);
            }
         hSignalTrace = MakeTH1('I', obname.Data(), obtitle.Data(), CSA_SIGNAL_SIZE * spillsamples, 0,
                    CSA_SIGNAL_SIZE * spillsamples,"time [febex samples]");

         obname.Form("Mapped/AllSigTraceSum");
         obtitle.Form("Stitched trace of signal region accumulated");
         hSignalTraceSum = MakeTH1('I', obname.Data(), obtitle.Data(), CSA_SIGNAL_SIZE * spillsamples, 0,
             CSA_SIGNAL_SIZE * spillsamples,"time [febex samples]");
       }
  else
      {
        hSignalTrace=nullptr;
        hSignalTraceSum=nullptr;
      }

  /////////////////////// JAM 06-06-25: additional filter regions for baseline correction and beam spot selection:


  fxInSpillRegion=MakeWinCond("InSpillRegion",4,25);
  fxOffSpillRegion=MakeWinCond("OffSpillRegion", 40,100);



   SetMakeWithAutosave(kTRUE);
}
//-----------------------------------------------------------
TAwagsSisMapProc::~TAwagsSisMapProc()
{
  TGo4Log::Info("TAwagsSisMapProc: Delete");

}
//-----------------------------------------------------------
Bool_t TAwagsSisMapProc::BuildEvent(TGo4EventElement* dest)
{
  TAwagsSisBasicEvent* inp_evt = (TAwagsSisBasicEvent*) GetInputEvent();
  TAwagsSisMapEvent* out_evt = (TAwagsSisMapEvent*) dest;
  out_evt->SetValid(kFALSE);
  if (inp_evt->IsValid())
  {
    ResetTraces();
    Int_t numchambers = GetNumChambers();
    for (Int_t dev = 0; dev < numchambers; ++dev)
    {
      Int_t numwires = GetNumWires(dev);
      for (Int_t wire = 0; wire < numwires; ++wire)
      {
        Int_t sfp = fParam->fSFP[dev][wire];
        Int_t slave = fParam->fSlave[dev][wire];
        Int_t chan = fParam->fChannel[dev][wire];
        if (sfp < 0 || slave < 0 || chan < 0)
          continue;
        if (sfp > MAX_SFP || slave > MAX_SLAVE || chan > N_CHA)
          continue;

#ifdef AWAGS_STORE_TRACES
        std::vector<Double_t> & theTrace = inp_evt->fSpillTrace[sfp][slave][chan];
        for (size_t bin = 0; bin < theTrace.size(); bin++)
        {
          if ((Int_t) bin >= hWireTraces[dev][wire]->GetNbinsX())
          break;    // avoid crash by accidental overflow
//          printf("TAwagsSisMapProc::BuildEvent for sfp %d slave %d chan %d access spilltrace bin %d , size is %ld\n",
//              sfp, slave, chan, bin, theTrace.size());
          if ((Int_t) (1 + bin) < hWireTraces[dev][wire]->GetNbinsX())
          {


            hWireTraces[dev][wire]->SetBinContent(1 + bin, theTrace[bin]);
          }
          hWireProfile[dev]->Fill(wire, theTrace[bin]);
          hWireProfile_Sum[dev]->Fill(wire, theTrace[bin]);
          hMapTrace[dev]->Fill(bin, wire, theTrace[bin]);
          hMapTrace_Sum[dev]->Fill(bin, wire, theTrace[bin]);
          //out_evt->fWireTrace[dev][wire].push_back(theTrace[bin]);

        }
#endif
        std::vector<Double_t> & theCharge = inp_evt->fChargeTrend[sfp][slave][chan];
        // JAM 06-06-25: add off spill subtraction and in spill selection filtering

        // TODO: first scan off spill region and evaluate mean value
        Double_t baseline=0;
        Bool_t inspill=kTRUE;
        Double_t value=0;
        size_t firstbin=0;
        if(fParam->fOffSpillBackground) {
          Double_t count=0;
          for (size_t i = 0; i < theCharge.size(); i++)
               {
                if(fxOffSpillRegion->Test(i))
                  {
                    baseline+=theCharge[i];
                    count++;
                  }
               } // for size i
          if(count>0) baseline /=count;
          firstbin=1; // in background correction mode, we also suppress the first bin of spilltrace here
        }
        // transfer (corrected) spill trace to display histograms:
//        printf("TAwagsSisMapProc::BuildEvent has vector length %ld and histogram size %d for dev:%d, wire %d\n",theCharge.size(), hWireSpillCharge[dev][wire]->GetNbinsX(),dev, wire);
//        std::cout<<std::endl;
        for (size_t bin = firstbin; bin < theCharge.size(); bin++)
        {
          if ((Int_t) bin >= hWireSpillCharge[dev][wire]->GetNbinsX())
            continue; //break;    // avoid crash by accidental overflow
//          Double_t value=theCharge[bin];
//          printf("TAwagsSisMapProc::BuildEvent for sfp %d slave %d chan %d access charge bin %d , size is %ld, val=%e\n",
//                      sfp, slave, chan, bin, theCharge.size(), value);
//          continue;

          value = theCharge[bin]  - baseline;
          value *=fParam->fProfileScaleFactor; // JAM 09-07-25: possibility to use factor -1 or any other
          inspill = fParam->fInSpillProfile ? fxInSpillRegion->Test(bin) : kTRUE;

          if ((Int_t) (1 + bin) < hWireSpillCharge[dev][wire]->GetNbinsX())
          {
            hWireSpillCharge[dev][wire]->SetBinContent(1 + bin, value);
          }

          if(inspill)
          {
            hWireChargeProfile[dev]->Fill(wire, value);
            hWireChargeProfile_Sum[dev]->Fill(wire, value);
          }

          hSampleChargeProfile[dev]->Fill(bin, value);
          hSampleChargeProfile_Sum[dev]->Fill(bin, value);



          hMapSpillCharge[dev]->Fill(bin, wire, value);
          hMapSpillCharge_Sum[dev]->Fill(bin, wire, value);
        }
      }
    }

    if(hSignalTrace && hSignalTraceSum)
    {
      // only for old mode with signal regions selected in traces
        std::vector<Double_t> & theSignals = inp_evt->fSignalTrace;
            for (size_t bin = 0; bin < theSignals.size(); bin++)
            {
//              if ((Int_t) bin >= hSignalTrace->GetNbinsX())
//                break;    // avoid crash by accidental overflow

              hSignalTrace->Fill(bin, theSignals[bin]);
              hSignalTraceSum->Fill(bin, theSignals[bin]);
            }
    }
    out_evt->SetValid(kTRUE);


    if (fParam->fStopAtEachSpill)
     {
       GO4_STOP_ANALYSIS_MESSAGE(
           "Stopped after Spill %d was complete. Click green arrow for next event. please.", inp_evt->fuSpillCount);
     }


  }    // if(inp_evt->IsValid())

  if (fParam->fSlowMotion)
  {
    Int_t evnum = -1;
    TGo4MbsEvent* mbsevent = dynamic_cast<TGo4MbsEvent*>(GetInputEvent("Unpacker"));
    if (mbsevent)
      evnum = mbsevent->GetCount();
    GO4_STOP_ANALYSIS_MESSAGE(
        "Stopped for slow motion mode after MBS event count %d. Click green arrow for next event. please.", evnum);
  }

  return kTRUE;
}

void TAwagsSisMapProc::ResetTraces()
{
  if(hSignalTrace) hSignalTrace->Reset("");
  for (Int_t dev = 0; dev < GetNumChambers(); ++dev)
  {
    for (Int_t wire = 0; wire < GetNumWires(dev); ++wire)
    {
#ifdef AWAGS_STORE_TRACES
      hWireTraces[dev][wire]->Reset("");
#endif
      hWireSpillCharge[dev][wire]->Reset("");
    }
#ifdef AWAGS_STORE_TRACES
    hMapTrace[dev]->Reset("");
    hWireProfile[dev]->Reset("");
#endif
    hMapSpillCharge[dev]->Reset("");
    hWireChargeProfile[dev]->Reset("");

    hSampleChargeProfile[dev]->Reset("");
  }
}