/*
 *  See header file for a description of this class.
 *
 *  \author G. Cerminara - INFN Torino
 */
#include "CalibMuon/DTCalibration/plugins/DTTTrigCalibration.h"
#include "CalibMuon/DTCalibration/interface/DTTimeBoxFitter.h"
#include "CalibMuon/DTDigiSync/interface/DTTTrigSyncFactory.h"
#include "CalibMuon/DTDigiSync/interface/DTTTrigBaseSync.h"

#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/ESHandle.h"

#include "DataFormats/MuonDetId/interface/DTWireId.h"

#include "CondFormats/DTObjects/interface/DTTtrig.h"

#include "DataFormats/GeometryVector/interface/GlobalPoint.h"

#include "CalibMuon/DTCalibration/interface/DTCalibDBUtils.h"

#include "CondFormats/DataRecord/interface/DTStatusFlagRcd.h"
#include "CondFormats/DTObjects/interface/DTStatusFlag.h"

#include "TFile.h"
#include "TH1F.h"
#include "TGraph.h"

class DTLayer;

using namespace std;
using namespace edm;
// using namespace cond;

// Constructor
DTTTrigCalibration::DTTTrigCalibration(const edm::ParameterSet& pset) {
  // Get the debug parameter for verbose output
  debug = pset.getUntrackedParameter<bool>("debug");

  // Get the label to retrieve digis from the event
  std::string digiLabel = pset.getUntrackedParameter<string>("digiLabel");
  digiToken = consumes<DTDigiCollection>(edm::InputTag(digiLabel));

  // Switch on/off the DB writing
  findTMeanAndSigma = pset.getUntrackedParameter<bool>("fitAndWrite", false);

  // The TDC time-window (ns)
  maxTDCCounts = 5000 * pset.getUntrackedParameter<int>("tdcRescale", 1);
  //The maximum number of digis per layer
  maxDigiPerLayer = pset.getUntrackedParameter<int>("maxDigiPerLayer", 10);

  // The root file which will contain the histos
  string rootFileName = pset.getUntrackedParameter<string>("rootFileName");
  theFile = new TFile(rootFileName.c_str(), "RECREATE");
  theFile->cd();
  theFitter = std::make_unique<DTTimeBoxFitter>();
  if (debug)
    theFitter->setVerbosity(1);

  double sigmaFit = pset.getUntrackedParameter<double>("sigmaTTrigFit", 10.);
  theFitter->setFitSigma(sigmaFit);

  doSubtractT0 = pset.getUntrackedParameter<bool>("doSubtractT0", "false");
  // Get the synchronizer
  if (doSubtractT0) {
    theSync = DTTTrigSyncFactory::get()->create(pset.getUntrackedParameter<string>("tTrigMode"),
                                                pset.getUntrackedParameter<ParameterSet>("tTrigModeConfig"),
                                                consumesCollector());
  }

  checkNoisyChannels = pset.getUntrackedParameter<bool>("checkNoisyChannels", "false");

  // the kfactor to be uploaded in the ttrig DB
  kFactor = pset.getUntrackedParameter<double>("kFactor", -0.7);

  if (debug)
    cout << "[DTTTrigCalibration]Constructor called!" << endl;

  if (checkNoisyChannels) {
    theStatusMapToken = esConsumes();
  }
}

// Destructor
DTTTrigCalibration::~DTTTrigCalibration() {
  if (debug)
    cout << "[DTTTrigCalibration]Destructor called!" << endl;

  //   // Delete all histos
  //   for(map<DTSuperLayerId, TH1F*>::const_iterator slHisto = theHistoMap.begin();
  //       slHisto != theHistoMap.end();
  //       slHisto++) {
  //     delete (*slHisto).second;
  //   }

  theFile->Close();
}

/// Perform the real analysis
void DTTTrigCalibration::analyze(const edm::Event& event, const edm::EventSetup& eventSetup) {
  if (debug)
    cout << "[DTTTrigCalibration] #Event: " << event.id().event() << endl;

  // Get the digis from the event
  const edm::Handle<DTDigiCollection>& digis = event.getHandle(digiToken);

  ESHandle<DTStatusFlag> statusMap;
  if (checkNoisyChannels) {
    // Get the map of noisy channels
    statusMap = eventSetup.getHandle(theStatusMapToken);
  }

  if (doSubtractT0)
    theSync->setES(eventSetup);

  //The chambers too noisy in this event
  vector<DTChamberId> badChambers;

  // Iterate through all digi collections ordered by LayerId
  DTDigiCollection::DigiRangeIterator dtLayerIt;
  for (dtLayerIt = digis->begin(); dtLayerIt != digis->end(); ++dtLayerIt) {
    // Get the iterators over the digis associated with this LayerId
    const DTDigiCollection::Range& digiRange = (*dtLayerIt).second;

    const DTLayerId layerId = (*dtLayerIt).first;
    const DTSuperLayerId slId = layerId.superlayerId();
    const DTChamberId chId = slId.chamberId();
    bool badChamber = false;

    if (debug)
      cout << "----------- Layer " << layerId << " -------------" << endl;

    //Check if the layer is inside a noisy chamber
    for (vector<DTChamberId>::const_iterator chamber = badChambers.begin(); chamber != badChambers.end(); ++chamber) {
      if ((*chamber) == chId) {
        badChamber = true;
        break;
      }
    }
    if (badChamber)
      continue;

    //Check if the layer has too many digis
    if ((digiRange.second - digiRange.first) > maxDigiPerLayer) {
      if (debug)
        cout << "Layer " << layerId << "has too many digis (" << (digiRange.second - digiRange.first) << ")" << endl;
      badChambers.push_back(chId);
      continue;
    }

    // Get the histo from the map
    TH1F* hTBox = theHistoMap[slId];
    if (hTBox == nullptr) {
      // Book the histogram
      theFile->cd();
      hTBox =
          new TH1F(getTBoxName(slId).c_str(), "Time box (ns)", int(0.25 * 32.0 * maxTDCCounts / 25.0), 0, maxTDCCounts);
      if (debug)
        cout << "  New Time Box: " << hTBox->GetName() << endl;
      theHistoMap[slId] = hTBox;
    }
    TH1F* hO = theOccupancyMap[layerId];
    if (hO == nullptr) {
      // Book the histogram
      theFile->cd();
      hO = new TH1F(getOccupancyName(layerId).c_str(), "Occupancy", 100, 0, 100);
      if (debug)
        cout << "  New Time Box: " << hO->GetName() << endl;
      theOccupancyMap[layerId] = hO;
    }

    // Loop over all digis in the given range
    for (DTDigiCollection::const_iterator digi = digiRange.first; digi != digiRange.second; ++digi) {
      const DTWireId wireId(layerId, (*digi).wire());

      // Check for noisy channels and skip them
      if (checkNoisyChannels) {
        bool isNoisy = false;
        bool isFEMasked = false;
        bool isTDCMasked = false;
        bool isTrigMask = false;
        bool isDead = false;
        bool isNohv = false;
        statusMap->cellStatus(wireId, isNoisy, isFEMasked, isTDCMasked, isTrigMask, isDead, isNohv);
        if (isNoisy) {
          if (debug)
            cout << "Wire: " << wireId << " is noisy, skipping!" << endl;
          continue;
        }
      }
      theFile->cd();
      double offset = 0;
      if (doSubtractT0) {
        const DTLayer* layer = nullptr;  //fake
        const GlobalPoint glPt;          //fake
        offset = theSync->offset(layer, wireId, glPt);
      }
      hTBox->Fill((*digi).time() - offset);
      if (debug) {
        cout << "   Filling Time Box:   " << hTBox->GetName() << endl;
        cout << "           offset (ns): " << offset << endl;
        cout << "           time(ns):   " << (*digi).time() - offset << endl;
      }
      hO->Fill((*digi).wire());
    }
  }
}

void DTTTrigCalibration::endJob() {
  if (debug)
    cout << "[DTTTrigCalibration]Writing histos to file!" << endl;

  // Write all time boxes to file
  theFile->cd();
  for (map<DTSuperLayerId, TH1F*>::const_iterator slHisto = theHistoMap.begin(); slHisto != theHistoMap.end();
       ++slHisto) {
    (*slHisto).second->Write();
  }
  for (map<DTLayerId, TH1F*>::const_iterator slHisto = theOccupancyMap.begin(); slHisto != theOccupancyMap.end();
       ++slHisto) {
    (*slHisto).second->Write();
  }

  if (findTMeanAndSigma) {
    // Create the object to be written to DB
    DTTtrig* tTrig = new DTTtrig();

    // Loop over the map, fit the histos and write the resulting values to the DB
    for (map<DTSuperLayerId, TH1F*>::const_iterator slHisto = theHistoMap.begin(); slHisto != theHistoMap.end();
         ++slHisto) {
      pair<double, double> meanAndSigma = theFitter->fitTimeBox((*slHisto).second);
      tTrig->set((*slHisto).first, meanAndSigma.first, meanAndSigma.second, kFactor, DTTimeUnits::ns);

      if (debug) {
        cout << " SL: " << (*slHisto).first << " mean = " << meanAndSigma.first << " sigma = " << meanAndSigma.second
             << endl;
      }
    }

    // Print the ttrig map
    dumpTTrigMap(tTrig);

    // Plot the tTrig
    plotTTrig(tTrig);

    if (debug)
      cout << "[DTTTrigCalibration]Writing ttrig object to DB!" << endl;

    // FIXME: to be read from cfg?
    string tTrigRecord = "DTTtrigRcd";

    // Write the object to DB
    DTCalibDBUtils::writeToDB(tTrigRecord, tTrig);
  }
}

string DTTTrigCalibration::getTBoxName(const DTSuperLayerId& slId) const {
  string histoName;
  stringstream theStream;
  theStream << "Ch_" << slId.wheel() << "_" << slId.station() << "_" << slId.sector() << "_SL" << slId.superlayer()
            << "_hTimeBox";
  theStream >> histoName;
  return histoName;
}

string DTTTrigCalibration::getOccupancyName(const DTLayerId& slId) const {
  string histoName;
  stringstream theStream;
  theStream << "Ch_" << slId.wheel() << "_" << slId.station() << "_" << slId.sector() << "_SL" << slId.superlayer()
            << "_L" << slId.layer() << "_Occupancy";
  theStream >> histoName;
  return histoName;
}

void DTTTrigCalibration::dumpTTrigMap(const DTTtrig* tTrig) const {
  static const double convToNs = 25. / 32.;
  for (DTTtrig::const_iterator ttrig = tTrig->begin(); ttrig != tTrig->end(); ++ttrig) {
    cout << "Wh: " << (*ttrig).first.wheelId << " St: " << (*ttrig).first.stationId
         << " Sc: " << (*ttrig).first.sectorId << " Sl: " << (*ttrig).first.slId
         << " TTrig mean (ns): " << (*ttrig).second.tTrig * convToNs
         << " TTrig sigma (ns): " << (*ttrig).second.tTrms * convToNs << endl;
  }
}

void DTTTrigCalibration::plotTTrig(const DTTtrig* tTrig) const {
  TH1F* tTrig_YB1_Se10 = new TH1F("tTrig_YB1_Se10", "tTrig YB1_Se10", 15, 1, 16);
  TH1F* tTrig_YB2_Se10 = new TH1F("tTrig_YB2_Se10", "tTrig YB2_Se10", 15, 1, 16);
  TH1F* tTrig_YB2_Se11 = new TH1F("tTrig_YB2_Se11", "tTrig YB2_Se11", 12, 1, 13);

  static const double convToNs = 25. / 32.;
  for (DTTtrig::const_iterator ttrig = tTrig->begin(); ttrig != tTrig->end(); ++ttrig) {
    // avoid to have wired numbers in the plot
    float tTrigValue = 0;
    float tTrmsValue = 0;
    if ((*ttrig).second.tTrig * convToNs > 0 && (*ttrig).second.tTrig * convToNs < 32000) {
      tTrigValue = (*ttrig).second.tTrig * convToNs;
      tTrmsValue = (*ttrig).second.tTrms * convToNs;
    }

    int binx;
    string binLabel;
    stringstream binLabelStream;
    if ((*ttrig).first.sectorId != 14) {
      binx = ((*ttrig).first.stationId - 1) * 3 + (*ttrig).first.slId;
      binLabelStream << "MB" << (*ttrig).first.stationId << "_SL" << (*ttrig).first.slId;
    } else {
      binx = 12 + (*ttrig).first.slId;
      binLabelStream << "MB14_SL" << (*ttrig).first.slId;
    }
    binLabelStream >> binLabel;

    if ((*ttrig).first.wheelId == 2) {
      if ((*ttrig).first.sectorId == 10 || (*ttrig).first.sectorId == 14) {
        tTrig_YB2_Se10->Fill(binx, tTrigValue);
        tTrig_YB2_Se10->SetBinError(binx, tTrmsValue);
        tTrig_YB2_Se10->GetXaxis()->SetBinLabel(binx, binLabel.c_str());
        tTrig_YB2_Se10->GetYaxis()->SetTitle("ns");
      } else {
        tTrig_YB2_Se11->Fill(binx, tTrigValue);
        tTrig_YB2_Se11->SetBinError(binx, tTrmsValue);
        tTrig_YB2_Se11->GetXaxis()->SetBinLabel(binx, binLabel.c_str());
        tTrig_YB2_Se11->GetYaxis()->SetTitle("ns");
      }
    } else {
      tTrig_YB1_Se10->Fill(binx, tTrigValue);
      tTrig_YB1_Se10->SetBinError(binx, tTrmsValue);
      tTrig_YB1_Se10->GetXaxis()->SetBinLabel(binx, binLabel.c_str());
      tTrig_YB1_Se10->GetYaxis()->SetTitle("ns");
    }
  }

  tTrig_YB1_Se10->Write();
  tTrig_YB2_Se10->Write();
  tTrig_YB2_Se11->Write();
}