/**************************************************************************
 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 *                                                                        *
 * Author: The ALICE Off-line Project.                                    *
 * Contributors are mentioned in the code where appropriate.              *
 *                                                                        *
 * Permission to use, copy, modify and distribute this software and its   *
 * documentation strictly for non-commercial purposes is hereby granted   *
 * without fee, provided that the above copyright notice appears in all   *
 * copies and that both the copyright notice and this permission notice   *
 * appear in the supporting documentation. The authors make no claims     *
 * about the suitability of this software for any purpose. It is          *
 * provided "as is" without express or implied warranty.                  *
 **************************************************************************/


/* $Id$ */

//---
//  Produces the data needed to calculate the quality assurance. 
//  T0 QA for Hits, Digits, RAW and RecPoints
//  Alla.Maevskaya@cern.ch
//  
//---

// --- ROOT system ---
#include <TClonesArray.h>
#include <TFile.h> 
#include <TH1F.h> 
#include <TDirectory.h>
// --- Standard library ---

// --- AliRoot header files ---
#include "AliESDEvent.h"
#include "AliLog.h"
#include "AliT0digit.h" 
#include "AliT0hit.h"
#include "AliT0RecPoint.h"
#include "AliT0QADataMaker.h"
#include "AliQAChecker.h"
#include "AliT0RawReader.h"

ClassImp(AliT0QADataMaker)
           
//____________________________________________________________________________ 
  AliT0QADataMaker::AliT0QADataMaker() : 
  AliQADataMaker(AliQAv1::GetDetName(AliQAv1::kT0), "T0 Quality Assurance Data Maker")

{
  // ctor
  /*
  for(Int_t i=0; i<24; i++) {
    fhHitsTime[i]=0x0;
   fhDigCFD[i]=0x0;
    fhDigLEDamp[i]=0x0;
    fhRecCFD[i]=0x0;
    fhRecLEDamp[i]=0x0;
    fhRecQTC[i]=0x0;
  }
  */
 //   fDetectorDir = fOutput->GetDirectory(GetName()) ;  
//   if (!fDetectorDir) 
//     fDetectorDir = fOutput->mkdir(GetName()) ;  
}

//____________________________________________________________________________ 
AliT0QADataMaker::AliT0QADataMaker(const AliT0QADataMaker& qadm) :
  AliQADataMaker() 
{
  //copy ctor 
  /*
  for(Int_t i=0; i<24; i++) {
    fhHitsTime[i]=0x0;
    fhDigCFD[i]=0x0;
    fhDigLEDamp[i]=0x0;
    fhRecCFD[i]=0x0;
    fhRecLEDamp[i]=0x0;
    fhRecQTC[i]=0x0;
  }
  */
  SetName((const char*)qadm.GetName()) ; 
  SetTitle((const char*)qadm.GetTitle()); 
}

//__________________________________________________________________
AliT0QADataMaker& AliT0QADataMaker::operator = (const AliT0QADataMaker& qadm )
{
  // Equal operator.
  this->~AliT0QADataMaker();
  new(this) AliT0QADataMaker(qadm);
  return *this;
}
//____________________________________________________________________________
void AliT0QADataMaker::EndOfDetectorCycle(AliQAv1::TASKINDEX task, TObjArray * list)
{
  //Detector specific actions at end of cycle
  // do the QA checking
  AliQAChecker::Instance()->Run(AliQAv1::kT0, task, list) ;
}

//____________________________________________________________________________
void AliT0QADataMaker::StartOfDetectorCycle()
{
  //Detector specific actions at start of cycle

}
 
//____________________________________________________________________________ 
void AliT0QADataMaker::InitHits()
{
  // create Hits histograms in Hits subdir
  TString timename;
  TH1F *    fhHitsTime[24];
  for (Int_t i=0; i<24; i++)
    {
      timename ="hHitTime";
      timename += i;
      if(i<12)  fhHitsTime[i] = new TH1F(timename.Data(),timename.Data(),100,2000,3000);
      else  
	fhHitsTime[i] = new TH1F(timename.Data(),timename.Data(),100,12000,13000);
	Add2HitsList( fhHitsTime[i],i);
    }
  /*
  TH2F *fhHitsEffA = new TH2F("hHitsEffA", "Hits Efficiency A side", 25,-0.5,24.5, 100,12,13 );
  Add2HitsList(fhHitsEffA,0);
  TH2F *fhHitsEffC = new TH2F("hHitsEffC", "Hits Efficiency C side", 25,-0.5,24.5, 100,2,3 );
  Add2HitsList(fhHitsEffC,1);
  */
  //
  ClonePerTrigClass(AliQAv1::kHITS); // this should be the last line
}

//____________________________________________________________________________ 
void AliT0QADataMaker::InitDigits()
{
  // create Digits histograms in Digits subdir

  /*
  TH2F * fhDigCFD = new TH2F("fhDigCFD", " CFD digits",25,-0.5,24.5,100,100,1000);
  Add2DigitsList( fhDigCFD,0);
  TH2F *fhDigLEDamp = new TH2F("fhDigLEDamp", " LED-CFD digits",25,-0.5,24.5,100,100,1000);
  Add2DigitsList( fhDigLEDamp,1);
  TH2F * fhDigQTC = new TH2F("fhDigQTC", " QTC digits",25,-0.5,24.5,100,100,1000);
  Add2DigitsList( fhDigQTC,2);
  TH1F * fhDigMean = new TH1F("hDigMean","online mean signal", 100,500,600);
  Add2DigitsList( fhDigMean,23);
  */
  
  TString timename, ampname, qtcname;

  TH1F *fhDigCFD[24]; TH1F * fhDigLEDamp[24]; TH1F *fhDigQTC[24];

  for (Int_t i=0; i<24; i++)
    {
      timename ="hDigCFD";
      ampname = "hDigLED";
      qtcname = "hDigQTC";
      timename += i;
      ampname += i;
      qtcname += i;
      fhDigCFD[i] = new TH1F(timename.Data(), timename.Data(),100,100,5000);
      Add2DigitsList( fhDigCFD[i],i);
      fhDigLEDamp[i] = new TH1F(ampname.Data(), ampname.Data(),100,120000,150000);
      Add2DigitsList( fhDigLEDamp[i],i+24);
      fhDigQTC[i] = new TH1F(qtcname.Data(), qtcname.Data(),100,100,500);
      Add2DigitsList( fhDigQTC[i],i+48);
     }
  
  TH1F* fhDigEff = new TH1F("hDigEff","digits efficiency", 25,-0.5,24.5);
  Add2DigitsList( fhDigEff,72);
  TH1F* fhDigMean = new TH1F("hDigMean","online mean signal", 100,500,600);
  Add2DigitsList( fhDigMean,73);
  //
  ClonePerTrigClass(AliQAv1::kDIGITS); // this should be the last line  
}

//____________________________________________________________________________ 
void AliT0QADataMaker::InitRaws()
{
  // create Raw histograms in Raw subdir
  printf("   AliT0QADataMaker::InitRaws() started\n");
  TString timename, ampname, qtcname;

  TH1F *fhRawCFD[24]; TH1F * fhRawLEDamp[24]; TH1F *fhRawQTC[24];

  for (Int_t i=0; i<24; i++)
    {
      timename ="hRawCFD";
      ampname = "hRawLED";
      qtcname = "hRawQTC";
      timename += i;
      ampname += i;
      qtcname += i;
      fhRawCFD[i] = new TH1F(timename.Data(), timename.Data(),100,100,5000);
      Add2RawsList( fhRawCFD[i],i);
      fhRawLEDamp[i] = new TH1F(ampname.Data(), ampname.Data(),100,120000,150000);
      Add2RawsList( fhRawLEDamp[i],i+24);
      fhRawQTC[i] = new TH1F(qtcname.Data(), qtcname.Data(),100,100,500);
      Add2RawsList( fhRawQTC[i],i+48);
     }
  
  TH1F* fhRawMean = new TH1F("hRawMean","online mean signal", 100,500,600);
  Add2RawsList( fhRawMean,72);
  //
  ClonePerTrigClass(AliQAv1::kRAWS); // this should be the last line  
}

//____________________________________________________________________________ 

void AliT0QADataMaker::InitRecPoints()
{
  // create cluster histograms in RecPoint subdir
  /* 
     TH2F * fhRecCFD = new TH2F("fhRecCFD", " CFD reconstructed",25,-0.5,24.5,100,12,13);
     Add2DigitsList( fhRecCFD,0);
     TH2F *fhRecLEDamp = new TH2F("fhRecLEDamp", " amplitude LED reconstructed",25,-0.5,24.5,100,1000,1000);
     Add2DigitsList( fhRecLEDamp,1);
     TH2F * fhRecQTC = new TH2F("fhRecQTC", " amplitude QTC reconstructed",25,-0.5,24.5,100,1000,1000);
     Add2DigitsList( fhRecQTC,2);
     TH1F * fhRecMean = new TH1F("hRecMean"," reconstructed mean signal",100,500,600);
     Add2DigitsList( fhRecMean,3);
  */ 
  
  TString timename,ampname, qtcname;
  TH1F *fhRecCFD[24]; TH1F *fhRecLEDAmp[24];  TH1F * fhRecQTC[24];
  for (Int_t i=0; i<24; i++)
    {
      timename ="hRecCFD";
      ampname = "hRecLED";
      qtcname = "hRecQTC";
      timename += i;
      ampname += i;
      qtcname += i;
      fhRecCFD[i] = new TH1F(timename.Data(), timename.Data(),100,0,1000);
     Add2RecPointsList ( fhRecCFD[i],i);
      fhRecLEDAmp[i] = new TH1F(ampname.Data(), ampname.Data(),100,0,200);
    Add2RecPointsList ( fhRecLEDAmp[i],i+24);
      fhRecQTC[i] = new TH1F(qtcname.Data(), qtcname.Data(),100,0,200);
    Add2RecPointsList ( fhRecQTC[i],i+48);
     }
  
  TH1F *fhRecEff = new TH1F("hRecEff","Efficiency rec.points",25,-0.5,24.5);
  Add2RecPointsList ( fhRecEff,72);
  TH1F * fhRecMean = new TH1F("hRecMean"," reconstructed mean signal",100,500,600);
  Add2RecPointsList( fhRecMean,73);
  //
  ClonePerTrigClass(AliQAv1::kRECPOINTS); // this should be the last line  
}
//____________________________________________________________________________
void AliT0QADataMaker::InitESDs()
{
  //create ESDs histograms in ESDs subdir
  TH1F *fhESDMean = new TH1F("hESDmean"," ESD mean",100,0,100);
  Add2ESDsList(fhESDMean, 0) ;
  TH1F * fhESDVertex = new TH1F("hESDvertex","EAD vertex",100,-50,50);
  Add2ESDsList(fhESDVertex, 1) ;
  //
  ClonePerTrigClass(AliQAv1::kESDS); // this should be the last line  
}
//____________________________________________________________________________

void AliT0QADataMaker::MakeHits(TTree *hitTree)
{
  //fills QA histos for Hits
  TClonesArray * hits = new TClonesArray("AliT0hit", 1000);
  
  TBranch * branch = hitTree->GetBranch("T0") ;
  if ( ! branch ) {
    AliWarning("T0 branch in Hit Tree not found") ;
  } else {

   if (branch) {
      branch->SetAddress(&hits);
    }else{
      AliError("Branch T0 hit not found");
      exit(111);
    } 
    Int_t ntracks    = (Int_t) hitTree->GetEntries();
    
    if (ntracks<=0) return;
    // Start loop on tracks in the hits containers
    for (Int_t track=0; track<ntracks;track++) {
      branch->GetEntry(track);
      Int_t nhits = hits->GetEntriesFast();
      for (Int_t ihit=0;ihit<nhits;ihit++) 
	{
	  AliT0hit  * startHit   = (AliT0hit*) hits->UncheckedAt(ihit);
	  if (!startHit) {
 	    AliError("The unchecked hit doesn't exist");
	    break;
	  }
	  Int_t pmt=startHit->Pmt();
	  FillHitsData(pmt-1,startHit->Time()) ;
	}
    }
  }
  //
  IncEvCountCycleHits();
  IncEvCountTotalHits();
  //
}

//____________________________________________________________________________
void AliT0QADataMaker::MakeDigits( TTree *digitsTree)
{
  //fills QA histos for Digits
 
  TArrayI *digCFD = new TArrayI(24);
  TArrayI *digLED = new TArrayI(24);
  TArrayI *digQT0 = new TArrayI(24);
  TArrayI *digQT1 = new TArrayI(24);
  Int_t refpoint=0;

   TBranch *brDigits=digitsTree->GetBranch("T0");
  AliT0digit *fDigits = new AliT0digit() ;
  if (brDigits) {
    brDigits->SetAddress(&fDigits);
  }else{
    AliError(Form("EXEC Branch T0 digits not found"));
    delete digCFD;
    delete digLED;
    delete digQT0;
    delete digQT1;
    return;
  }
  
  digitsTree->GetEvent(0);
  digitsTree->GetEntry(0);
  brDigits->GetEntry(0);
  fDigits->GetTimeCFD(*digCFD);
  fDigits->GetTimeLED(*digLED);
  fDigits->GetQT0(*digQT0);
  fDigits->GetQT1(*digQT1);
  refpoint = fDigits->RefPoint();
   for (Int_t i=0; i<24; i++)
    {
      if (digCFD->At(i)>0) {
	Int_t cfd=digCFD->At(i)- refpoint;
	FillDigitsData(i,cfd);
	FillDigitsData(i+24, digLED->At(i) - digCFD->At(i));
	FillDigitsData(i+48, digQT1->At(i) - digQT0->At(i));
      }
    }  
      
  delete digCFD;
  delete digLED;
  delete digQT0;
  delete digQT1;
  //
  IncEvCountCycleDigits();
  IncEvCountTotalDigits();
  //
}


//____________________________________________________________________________
void AliT0QADataMaker::MakeRaws( AliRawReader* rawReader)
{
  Int_t allData[110][5];
  for (Int_t i0=0; i0<105; i0++)
    {
      for (Int_t j0=0; j0<5; j0++) allData[i0][j0]=0;
    }
  //fills QA histos for RAW

    AliT0RawReader *start = new AliT0RawReader(rawReader);  
    start->Next();
    for (Int_t i=0; i<105; i++) 
      for (Int_t iHit=0; iHit<5; iHit++)
	allData[i][iHit]= start->GetData(i,iHit);
    
   
    for (Int_t ik = 0; ik<24; ik+=2){
      for (Int_t iHt=0; iHt<5; iHt++){
	Int_t cc = ik/2;
       if(allData[cc+1][iHt]!=0){
	 FillRawsData(cc,allData[cc+1][iHt]-allData[0][0]);
	if(allData[ik+25][iHt]!=0 && allData[ik+26][iHt]!=0)
	  FillRawsData(cc+48,allData[ik+26][iHt]-allData[ik+25][iHt]);
	if(allData[cc+13][iHt]!=0 )
	  FillRawsData(cc+24,allData[cc+13][iHt]-allData[cc+1][iHt]);
	}
      }
    }
    
    for (Int_t ik = 24; ik<48; ik+=2) {
      for (Int_t iHt=0; iHt<5; iHt++) {
	Int_t cc = ik/2;
	if(allData[cc+45][iHt]!=0) {
	  FillRawsData(cc,allData[cc+1][iHt]-allData[0][0]);
	  if(allData[ik+57][iHt]!=0 && allData[ik+58][iHt]!=0)
	    FillRawsData(cc+48,allData[ik+57][iHt]-allData[ik+58][iHt]);
	  if(allData[cc+57][iHt]!=0 )
	    FillRawsData(cc+48,allData[cc+57][iHt]-allData[cc+45][iHt]);
	}
      }
    }
    delete start;
    //
    IncEvCountCycleRaws();
    IncEvCountTotalRaws();
    //
}

//____________________________________________________________________________
void AliT0QADataMaker::MakeRecPoints(TTree * clustersTree)
{
  //fills QA histos for clusters

   AliT0RecPoint* frecpoints= new AliT0RecPoint ();
    if (!frecpoints) {
    AliError("Reconstruct Fill ESD >> no recpoints found");
    return;
  }
  TBranch *brRec =clustersTree ->GetBranch("T0");
  if (brRec) {
    brRec->SetAddress(&frecpoints);
  }else{
      AliError(Form("EXEC Branch T0 rec not found "));
      return;
  } 
    
  brRec->GetEntry(0);
  
  for ( Int_t i=0; i<24; i++) {
    FillRecPointsData(i,frecpoints -> GetTime(i)); 
    FillRecPointsData(i+24,frecpoints -> GetAmp(i));
    FillRecPointsData(i+48,frecpoints->AmpLED(i));
    //  if(frecpoints -> GetTime(i) > 0) fhRecEff->Fill(i);
  }
  FillRecPointsData(72,frecpoints->GetMeanTime());
  //
  IncEvCountCycleRecPoints();
  IncEvCountTotalRecPoints();
  //
}

//____________________________________________________________________________
void AliT0QADataMaker::MakeESDs(AliESDEvent * esd)
{
  //fills QA histos for ESD

  FillESDsData(0,esd->GetT0());
  FillESDsData(1,esd->GetT0zVertex());
  //
  IncEvCountCycleESDs();
  IncEvCountTotalESDs();
}