/**************************************************************************
 * 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$ */

//_________________________________________________________________________
//  A singleton. This class should be used in the analysis stage to get 
//  reconstructed objects: Digits, RecPoints, TrackSegments and RecParticles,
//  instead of directly reading them from galice.root file. 
//                  
//  MvL Feb 2006:
//  The AliEMCALLoader now holds the TClonesArray and TObjArray for reading
//  Hits, Dgits, SDigits and RecPoints. Filling is managed in the GetEvent()
//  method.
//
//  Creation/writing of files is managed by the relevant parts of the 
//  reconstruction software (AliEMCALDigitiser etx)
//
//*-- Author: Yves Schutz (SUBATECH) & Dmitri Peressounko (RRC KI & SUBATECH)
//*--         Completely redesigned by Dmitri Peressounko March 2001  
//
//*-- YS June 2001 : renamed the original AliEMCALIndexToObject and make
//*--         systematic usage of TFolders without changing the interface
// 
//*-- Marco van Leeuwen, Jan 2006: complete revision to simplify reading
//*--         and fit better in general ALICE scheme
//
//////////////////////////////////////////////////////////////////////////////

// --- ROOT system ---
#include "TMath.h"
#include "TTree.h"

// --- Standard library ---

// --- AliRoot header files ---
#include "AliEMCALLoader.h"
#include "AliLog.h"
#include "AliCDBLocal.h"
#include "AliCDBStorage.h"
#include "AliCDBManager.h"
#include "AliCDBEntry.h"
#include "AliEMCALHit.h"

ClassImp(AliEMCALLoader)
  
const TString AliEMCALLoader::fgkECARecPointsBranchName("EMCALECARP");//Name for branch with ECA Reconstructed Points
AliEMCALCalibData* AliEMCALLoader::fgCalibData = 0; //calibation data
AliEMCALRecParam * AliEMCALLoader::fgRecParam  = 0; //reconstruction parameters

//____________________________________________________________________________ 
AliEMCALLoader::AliEMCALLoader()
  : fDebug(0),
    fHits(0),
    fDigits(0),
    fSDigits(0),
    fRecPoints(0)
{
  //Default constructor for EMCAL Loader Class

  fDebug = 0;
  fHits = new TClonesArray("AliEMCALHit");
  fDigits = new TClonesArray("AliEMCALDigit");
  fSDigits = new TClonesArray("AliEMCALDigit");
  fRecPoints = new TObjArray();
}

//____________________________________________________________________________ 
AliEMCALLoader::AliEMCALLoader(const Char_t *detname,const Char_t *eventfoldername)
  : AliLoader(detname,eventfoldername),
    fDebug(0),
    fHits(0),
    fDigits(0),
    fSDigits(0),
    fRecPoints(0)
{
  //Specific constructor for EMCAL Loader class

  fDebug=0;
  fHits = new TClonesArray("AliEMCALHit");
  fDigits = new TClonesArray("AliEMCALDigit");
  fSDigits = new TClonesArray("AliEMCALDigit");
  fRecPoints = new TObjArray();
}

//____________________________________________________________________________
AliEMCALLoader::AliEMCALLoader(const Char_t *name, TFolder *topfolder)
  : AliLoader(name,topfolder),
    fDebug(0),
    fHits(0),
    fDigits(0),
    fSDigits(0),
    fRecPoints(0)
{
  //Specific constructor for EMCAL Loader class

  fDebug=0;
  fHits = new TClonesArray("AliEMCALHit");
  fDigits = new TClonesArray("AliEMCALDigit");
  fSDigits = new TClonesArray("AliEMCALDigit");
  fRecPoints = new TObjArray();
}

//____________________________________________________________________________
AliEMCALLoader::AliEMCALLoader(const AliEMCALLoader & obj)
  : AliLoader(obj),
    fDebug(obj.fDebug),
    fHits(obj.fHits),
    fDigits(obj.fDigits),
    fSDigits(obj.fSDigits),
    fRecPoints(obj.fRecPoints)
{
  //copy ctor
}

//____________________________________________________________________________ 
AliEMCALLoader::~AliEMCALLoader()
{
  // Disconnect trees and remove arrays
  if (TreeH())
    TreeH()->SetBranchAddress(fDetectorName,0);
  if (TreeD())
    TreeD()->SetBranchAddress(fDetectorName,0);
  if (TreeS())
    TreeS()->SetBranchAddress(fDetectorName,0);
  if (TreeR())
    TreeR()->SetBranchAddress(fgkECARecPointsBranchName,0);
  if (fHits) {
    fHits->Delete();
    delete fHits;
  }
  delete fDigits;
  delete fSDigits;
  delete fRecPoints;
}

//____________________________________________________________________________ 
AliEMCALCalibData* AliEMCALLoader::CalibData()
{ 
  // Check if the instance of AliEMCALCalibData exists, if not, create it if 
  // the OCDB is available, and finally return it.

  if(!fgCalibData && (AliCDBManager::Instance()->IsDefaultStorageSet()))
    {
      AliCDBEntry *entry = (AliCDBEntry*) 
	AliCDBManager::Instance()->Get("EMCAL/Calib/Data");
      if (entry) fgCalibData =  (AliEMCALCalibData*) entry->GetObject();
    }
  
  if(!fgCalibData)
    AliFatal("Calibration parameters not found in CDB!");
  
  return fgCalibData;
  
}

//____________________________________________________________________________ 
AliEMCALRecParam* AliEMCALLoader::RecParam()
{ 
  // Check if the instance of AliEMCALRecParam exists, if not, create it if 
  // the OCDB is available, and finally return it.

  if(!fgRecParam && (AliCDBManager::Instance()->IsDefaultStorageSet())) {
    AliCDBEntry *entry = (AliCDBEntry*) 
      AliCDBManager::Instance()->Get("EMCAL/RecParam/Data");
    if (entry) fgRecParam =  (AliEMCALRecParam*) entry->GetObject();
  }
  
  if(!fgRecParam)
    AliWarning("Recostruction parameters not found in CDB!");
  
  return fgRecParam;
  
}

//____________________________________________________________________________ 
Int_t AliEMCALLoader::CalibrateRaw(Double_t energy, Int_t module, 
				   Int_t column, Int_t row)
{
  // Convert energy into digitized amplitude for a cell relId
  // It is a user responsilibity to open CDB and set
  // AliEMCALCalibData object by the following operators:
  // 
  // AliCDBLocal *loc = new AliCDBLocal("deCalibDB");
  // AliEMCALCalibData* clb = (AliEMCALCalibData*)AliCDBStorage::Instance()
  //    ->Get(path_to_calibdata,run_number);
  // AliEMCALGetter* gime = AliEMCALGetter::Instance("galice.root");
  // gime->SetCalibData(clb);

  if (CalibData() == 0)
    Warning("CalibrateRaw","Calibration DB is not initiated!");

  Float_t gainFactor = 0.00305;//0.0015; // width of one ADC channel in GeV
  Float_t pedestal   = 0.009;//0.005;  // pedestals

  if(CalibData()) {
    gainFactor = CalibData()->GetADCchannel (module,column,row);
    pedestal   = CalibData()->GetADCpedestal(module,column,row);
  }
  
  Int_t   amp = static_cast<Int_t>( (energy - pedestal) / gainFactor + 0.5 ) ; 
  return amp;
}

//____________________________________________________________________________ 
Int_t AliEMCALLoader::GetEvent() 
{
  //Method to load all of the data
  //members of the EMCAL for a given
  //event from the Trees

  AliLoader::GetEvent();  // First call AliLoader to do all the groundwork
  
  // Now connect and fill TClonesArray

  // Hits
   TTree *treeH = TreeH();
   
   if (treeH) {
     Int_t nEnt = treeH->GetEntries();  // TreeH has array of hits for every primary
     fHits->Clear();
     Int_t index = 0;
     TClonesArray *tempArr = 0x0;
     TBranch * branchH = treeH->GetBranch(fDetectorName);
     branchH->SetAddress(&tempArr);
     for (Int_t iEnt = 0; iEnt < nEnt; iEnt++) {
       treeH->GetEntry(iEnt);
       Int_t nHit = tempArr->GetEntriesFast();
       for (Int_t iHit = 0; iHit < nHit; iHit++) {
	 new ((*fHits)[index]) AliEMCALHit(*((AliEMCALHit*)tempArr->At(iHit)));
	 index++;
       }
     }
     branchH->ResetAddress();
     if (tempArr) {
       tempArr->Delete();
       delete tempArr;
     }
   }
  
   // SDigits
   TTree *treeS = TreeS();
   if (treeS) {
     TBranch * branchS = treeS->GetBranch(fDetectorName);
     branchS->ResetAddress();
     if (fSDigits) {
       fSDigits->Delete();
       delete fSDigits;
       fSDigits = 0x0;
     }
     branchS->SetAddress(&fSDigits);
     treeS->GetEvent(0);
   }
   
   // Digits
   TTree *treeD = TreeD();
   if (treeD) {
     TBranch * branchD = treeD->GetBranch(fDetectorName);
     branchD->ResetAddress();
     if (fDigits) {
       fDigits->Delete();
       delete fDigits;
       fDigits = 0x0;
     }
     branchD->SetAddress(&fDigits);
     treeD->GetEvent(0);
   }

   // RecPoints
   TTree *treeR = TreeR();
   if (treeR) {
     TBranch * branchR = treeR->GetBranch(fgkECARecPointsBranchName);
     branchR->ResetAddress();
     if (fRecPoints) {
       fRecPoints->Delete();
       delete fRecPoints;
       fRecPoints = 0x0;
     }
     branchR->SetAddress(&fRecPoints);
     treeR->GetEvent(0);
   }
   
   return 0;
}