// $Id: AliHLTTriggerEmcalElectron.cxx 50471 2011-07-07 09:50:47Z fronchet $
//**************************************************************************
//* This file is property of and copyright by the ALICE HLT Project        * 
//* ALICE Experiment at CERN, All rights reserved.                         *
//*                                                                        *
//* Primary Authors: marcelfigueredo@gmail.com                             *
//*                  for The ALICE HLT Project.                            *
//*                                                                        *
//* 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.                  *
//**************************************************************************

#include "AliHLTTriggerEmcalElectron.h"
#include "AliESDEvent.h"
#include "AliVCluster.h" 
#include "AliESDCaloCluster.h"
#include "AliHLTTriggerDecision.h"
#include "AliHLTDomainEntry.h"
#include "AliHLTCaloClusterReader.h"
#include "AliHLTCaloClusterDataStruct.h"
#include "TRefArray.h"
#include "TString.h"
#include "TMap.h"
#include "AliESDtrack.h"
#include "AliHLTScalars.h"
/** ROOT macro for the implementation of ROOT specific class methods */
ClassImp(AliHLTTriggerEmcalElectron)

AliHLTTriggerEmcalElectron::AliHLTTriggerEmcalElectron() : 
  AliHLTTrigger(),
  fEThreshold(0.0),
  fEoverPThreshold(0.),
  fEoverPLimit(0.),
  fdEta(1.),
  fdPhi(1.),
  
//   fClustersRefs(NULL),
//   fClusterReader(NULL),
  fOCDBEntry("HLT/ConfigHLT/EmcalElectronTrigger"),
//   fOCDBEntry(""), 
  fDetector("EMCAL"),
  fInputDataType(),
  fMakeStats(kFALSE)
{
  // see header file for class documentation
  // or
  // refer to README to build package
  // or
  // visit http://web.ift.uib.no/~kjeks/doc/alice-hlts

  //if ( fMakeStats ) AliHLTScalars scalars;
  //AliHLTScalars scalars;
//   fClusterReader = new AliHLTCaloClusterReader();
//   fClustersRefs = new TRefArray();

}


AliHLTTriggerEmcalElectron::~AliHLTTriggerEmcalElectron() {
  // see header file for class documentation
//   if (fClusterReader)
//     delete fClusterReader;
//   fClusterReader = NULL;
// 
//   if(fClustersRefs)
//     delete fClustersRefs;
//   fClustersRefs = NULL;
}

//Trigger name
const char* AliHLTTriggerEmcalElectron::GetTriggerName() const {
  // see header file for class documentation
  return "EmcalElectronTrigger";
}

AliHLTComponent* AliHLTTriggerEmcalElectron::Spawn() {
  // see header file for class documentation
  return new AliHLTTriggerEmcalElectron;
}

Int_t AliHLTTriggerEmcalElectron::DoTrigger() {
  // see header file for class documentation
  
  Int_t iResult = 0;
  AliHLTScalars scalars;

  if ( GetFirstInputBlock( kAliHLTDataTypeSOR ) || GetFirstInputBlock( kAliHLTDataTypeEOR ) )
    return 0;

  //Try the caloclusterstruct input

/*  
  for (const AliHLTComponentBlockData* pBlock=GetFirstInputBlock(fInputDataType); pBlock!=NULL; pBlock=GetNextInputBlock()) {
    AliHLTCaloClusterHeaderStruct *caloClusterHeader = reinterpret_cast<AliHLTCaloClusterHeaderStruct*>(pBlock->fPtr);
    fClusterReader->SetMemory(caloClusterHeader);
    
    AliHLTCaloClusterDataStruct * caloClusterStruct;
    while( (caloClusterStruct = fClusterReader->NextCluster()) != 0) {
      if (TriggerOnEoverP(caloClusterStruct)) {
	return iResult;
      }
    }
  }*/

  //Try the ESD input
  const TObject* obj = GetFirstInputObject(kAliHLTAllDataTypes, "AliESDEvent");
  AliESDEvent* esd = dynamic_cast<AliESDEvent*>(const_cast<TObject*>(obj));
  
  if (esd != NULL) {
    esd->GetStdContent();

//     Int_t ncc = GetClustersFromEsd(esd, fClustersRefs); //marcel test
    Int_t ncc = esd->GetNumberOfCaloClusters();  
//     Int_t ncc = GetClustersFromEsd(esd, fClustersRefs); 
    
    for (Int_t i = 0; i < ncc ; i++) {
      
//       AliESDCaloCluster * cluster = static_cast<AliESDCaloCluster*>(fClustersRefs->At(i));
      AliVCluster *cluster = (AliVCluster*) esd->GetCaloCluster(i); //MARCEL test
      if(TriggerOnEoverP(cluster,esd)) { 
	return iResult;
      }
    }
  }

  // If we got to this point then we did not find any clusters with E > fEThreshold
  // generate negative trigger decision
  TString description;
  description.Form("No %s clusters corresponding to Energy >  %.02f GeV and %.02f < E/P < %.02f", fDetector.Data(),fEThreshold,fEoverPThreshold,fEoverPLimit);
  SetDescription(description.Data());
  TriggerEvent(false);
  
  if ( fMakeStats ) iResult = PushBack(&scalars, kAliHLTDataTypeEventStatistics|kAliHLTDataOriginHLT);
  return iResult;

}


template <class T>
Bool_t AliHLTTriggerEmcalElectron::TriggerOnEoverP(T* cluster,AliESDEvent *esd) {
  
  AliHLTScalars scalars;
  
  if (cluster->E() > fEThreshold) {    

      Int_t trackindex=cluster->GetTrackMatchedIndex(); 
      if(trackindex<0)return kFALSE;

      Double_t dEta=cluster->GetTrackDz();
      Double_t dPhi=cluster->GetTrackDx();

      AliESDtrack* track = esd->GetTrack(trackindex);
      if(!track)return kFALSE;     
      Double_t EoverP=cluster->E()/track->P(); 
        
      if ( fMakeStats ) {
	Double_t deltaR=TMath::Sqrt(dEta*dEta+dPhi*dPhi);
        scalars.Add("dR","Residuals dR of track matching", deltaR);
	scalars.Add("dEta","Residuals dEta of track matching", dEta);
	scalars.Add("dPhi","Residuals dPhi of track matching", dPhi);
        }
      
      if(TMath::Abs(dEta)>fdEta)return kFALSE;
      if(TMath::Abs(dPhi)>fdPhi)return kFALSE;

      if ( fMakeStats ) {  
        scalars.Add("TracksPt","TPC tracks pT", track->Pt());
	scalars.Add("ClusterEn","Cluster Energy",cluster->E());
	scalars.Add("EoverP","EoverP for matched tracks",EoverP);
	}
	    
      if(EoverP<fEoverPThreshold)return kFALSE;
      if(EoverP>fEoverPLimit)return kFALSE; 
 
      //We have a cluster satisfying trigger criteria
    TString description;
    description.Form("Event contains at least one %s cluster with energy greater than %.02f corresponding to  %.02f < E/P < %.02f, residuals: dPhi < %.02f dEta < %.02f", fDetector.Data(),fEThreshold,fEoverPThreshold,fEoverPLimit,fdPhi,fdEta);
    SetDescription(description.Data());
    
    // Enable the detectors for readout.

    //GetReadoutList().Enable(AliHLTReadoutList::kPHOS);
//     SetCaloReadoutList("EMCAL");  //FR
//     SetCaloReadoutList();  //FR
     
    // Add the available HLT information for readout too.
    GetTriggerDomain().Add(kAliHLTAnyDataTypeID, fDetector.Data());
    
    //Set trigger decision
    TriggerEvent(kTRUE);
    
    return kTRUE;
  } 


  return kFALSE;

}

int AliHLTTriggerEmcalElectron::DoInit(int argc, const char** argv) {
  // see header file for class documentation
//   return 0;// marcel test
  // first configure the default
  int iResult=ConfigureFromCDBTObjString(fOCDBEntry);

  // configure from the command line parameters if specified
  if (iResult>=0 && argc>0) {
    iResult=ConfigureFromArgumentString(argc, argv);
    HLTImportant("Trigger threshold set from argument string:  %.02f GeV: E/P:%.02f and %.02f, residuals: dPhi:%.02f dEta:%.02f", fDetector.Data(),fEThreshold,fEoverPThreshold,fEoverPLimit,fdPhi,fdEta);   
  } else if ( iResult >=0 ) {
    HLTImportant("Trigger threshold set from OCDB database entry:  %.02f GeV: E/P:%.02f and %.02f, residuals: dPhi:%.02f dEta:%.02f", fDetector.Data(),fEThreshold,fEoverPThreshold,fEoverPLimit,fdPhi,fdEta);
  }
  return iResult;
}

int AliHLTTriggerEmcalElectron::DoDeinit() {

  // see header file for class documentation
 
  return 0;
}

int AliHLTTriggerEmcalElectron::Reconfigure(const char* cdbEntry, const char* /*chainId*/) {
  // see header file for class documentation

  // configure from the specified antry or the default one
  const char* entry=cdbEntry;
  if (!entry || entry[0]==0) entry=fOCDBEntry;

  return ConfigureFromCDBTObjString(entry);
}

int AliHLTTriggerEmcalElectron::ScanConfigurationArgument(int argc, const char** argv) {
  // see header file for class documentation
  if (argc<=0) return 0;
  int i=0;
  TString argument=argv[i];

  // -maxpt
  if (argument.CompareTo("-energy")==0) {
    if (++i>=argc) return -EPROTO;
    argument=argv[i];
    fEThreshold=argument.Atof(); // 
    return 2;
  }    

  if (argument.CompareTo("-minEoverP")==0) {
    if (++i>=argc) return -EPROTO;
    argument=argv[i];
    fEoverPThreshold=argument.Atof(); // 
    return 2;
  } 
  
    if (argument.CompareTo("-maxEoverP")==0) {
    if (++i>=argc) return -EPROTO;
    argument=argv[i];
    fEoverPLimit=argument.Atof(); // 
    return 2;
  }
  
    if (argument.CompareTo("-dEta")==0) {
    if (++i>=argc) return -EPROTO;
    argument=argv[i];
    fdEta=argument.Atof(); // 
    return 2;
  } 
  
    if (argument.CompareTo("-dPhi")==0) {
    if (++i>=argc) return -EPROTO;
    argument=argv[i];
    fdPhi=argument.Atof(); // 
    return 2;
  } 
  
    if (argument.CompareTo("-makestats")==0) {
    fMakeStats = kTRUE;
    return 2;
  }
  
// unknown argument
  return -EINVAL;
}

//______________________________________________________________

void AliHLTTriggerEmcalElectron::GetOutputDataTypes(AliHLTComponentDataTypeList &list) const {
  // return the output data types generated
  
  list.push_back(kAliHLTDataTypeTriggerDecision);
  list.push_back(kAliHLTDataTypeEventStatistics|kAliHLTDataOriginHLT);
}
//_______________________________________________________________
// 
void AliHLTTriggerEmcalElectron::GetOutputDataSize(unsigned long& constBase, double& inputMultiplier) {
  // see header file for documentation
  constBase = sizeof(AliHLTTriggerDecision) + sizeof(AliHLTDomainEntry)*14;
  inputMultiplier = 1;
}


void AliHLTTriggerEmcalElectron::GetOCDBObjectDescription( TMap* const targetMap) {
  
  // Get a list of OCDB object description.
  if (!targetMap) return;
  targetMap->Add(new TObjString(fOCDBEntry),
		 new TObjString(Form("%s threshold trigger OCDB object", fDetector.Data()) ) 
		 );
}