#include "CbmRichMCbmToTShifter.h"

#include "TH1.h"
#include "TH1D.h"
#include "TCanvas.h"
#include "TClonesArray.h"
#include "TF1.h"
#include "TStyle.h"
#include "TEllipse.h"
#include "TLine.h"
#include "TMarker.h"
#include "TGeoNode.h"
#include "TGeoManager.h"
#include "TGeoBBox.h"
#include "TMath.h"
#include <TFile.h>


#include "CbmRichDigi.h"
#include "TLatex.h"
#include "CbmDrawHist.h"
#include "CbmRichGeoManager.h"
#include "CbmRichPoint.h"
#include "CbmRichConverter.h"

#include "CbmUtils.h"
#include "CbmHistManager.h"
#include "CbmDigiManager.h"

#include <iostream>
#include <string>
#include <boost/assign/list_of.hpp>
#include <sstream>
#include <cmath>

using boost::assign::list_of;

CbmRichMCbmToTShifter::CbmRichMCbmToTShifter() :
                        FairTask("CbmRichMCbmToTShifter"),
                        fEventNum(0),
                        fOutputDir("result_ToTOffset"),
                        fhTotMap(),
                        fGeneratePDFs(false),
                        fShowTdcId(false)
{
    std::cout << "CbmRichMCbmToTShifter::Constructor.." << std::endl;
}

InitStatus CbmRichMCbmToTShifter::Init()
{
    std::cout << "CbmRichMCbmToTShifter::Init" << std::endl;

     FairRootManager* ioman = FairRootManager::Instance();
     if (nullptr == ioman) { Fatal("CbmRichMCbmToTShifter::Init","RootManager not instantised!"); }
 
     fDigiMan = CbmDigiManager::Instance();
     fDigiMan->Init();
     if (! fDigiMan->IsPresent(kRich) ) {
       Fatal("CbmRichMCbmToTShifter::Init", "No Rich Digis!");
     }

    return kSUCCESS;
}



void CbmRichMCbmToTShifter::Exec(Option_t* /*option*/)
{
    fEventNum++;
    int nofDigis = fDigiMan->GetNofDigis(kRich);

    for (int i=0; i < nofDigis; ++i){
        const CbmRichDigi* digi = fDigiMan->Get<CbmRichDigi>(i);
        TH1* h = GetTotH1(getDirichAddress(digi->GetAddress()), getDirichChannel(digi->GetAddress()));
	if (h != nullptr) h->Fill(digi->GetToT());
    }
}


void CbmRichMCbmToTShifter::Finish()
{   
    
     std::cout<<"Drawing Hists..."<<std::endl;
     std::stringstream s;
     Int_t mean_cnt = 0;
     Double_t mean = 0.;
     for (auto const &outer : fhTotMap) {
        for (auto const &inner : outer.second){
           mean += static_cast<Double_t>(inner.second->GetEntries());
           mean_cnt++;
        }
     }
     
     if (mean_cnt != 0) mean /= mean_cnt;
     
     auto it = std::begin(fhTotMap);
     for (auto const &outer : fhTotMap) {
        int tdc = outer.first;
        TCanvas* c = new TCanvas(Form("fhToT_%x",outer.first),Form("fhToT_%x",outer.first), 2000 , 2000);
        c->Divide(6,6);
        if (fShowTdcId) s << "TDC 0x"<<std::hex<<outer.first<<std::dec<<" !";
        s<<"  0.00";
        for (int i=0;i<32;++i){
          c->cd(1+i);   
          TH1* h = GetTotH1(tdc, i+1);
          if (h != nullptr) {  
            h->Draw();
            if (GetMaxH1(h)<20 || h->GetEntries() < mean*0.1) {
              s<< "  0.00";
            } else {
              s<< "  "<< GetMaxH1(h)-25.00;
            }
          }
        }

         
        if (it == fhTotMap.begin()) {
            if (fGeneratePDFs) c->Print("ToTs/Tdc_all.pdf(","pdf");
            s<< " \\"<< std::endl;

        } else if (std::next(it) == fhTotMap.end()) {
            if (fGeneratePDFs) c->Print("ToTs/Tdc_all.pdf)","pdf");
            s<<std::endl;
        } else {
            if (fGeneratePDFs) c->Print("ToTs/Tdc_all.pdf","pdf");
            s<< " \\"<< std::endl;
        }
        
        ++it;
     }
     
     std::cout<<s.str()<<std::endl;
}


TH1* CbmRichMCbmToTShifter::GetTotH1(Int_t tdc, Int_t channel)
{
	TH1* h = fhTotMap[tdc][channel];
	if (h == nullptr) {
		TString name, title, subFolder;
		name.Form("ToT_tdc_0x%x_ch%u", tdc, channel);
		title.Form("%s;ToT [ns];Entries", name.Data());
		h = new TH1D(name, title, 500, -1., 49.);
		fhTotMap[tdc][channel] = h;
	}
	return h;
}

Double_t CbmRichMCbmToTShifter::GetMaxH1(TH1 * h)
{
    Double_t max = 0.0;
    Int_t b = 0;
	for (Int_t i=1; i< h->GetNbinsX();++i){
           Double_t val = h->GetBinContent(i);
           if (val > max) {
               max = val;
               b = i;
           }
        }
	return h->GetBinCenter(b);
}

ClassImp(CbmRichMCbmToTShifter)