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

//-----------------------------------------------------------------------
// Example of task (running locally, on AliEn and CAF),
// which provides standard way of calculating acceptance and efficiency
// between different steps of the procedure.
// The ouptut of the task is a AliCFContainer from which the efficiencies
// can be calculated
//-----------------------------------------------------------------------
// Author : R. Vernet, Consorzio Cometa - Catania (it)
//-----------------------------------------------------------------------
// Modification done by X. Lopez - LPC Clermont (fr)
//-----------------------------------------------------------------------

#ifndef ALICFMUONSINGLETASK1_CXX
#define ALICFMUONSINGLETASK1_CXX

#include "AliCFMuonSingleTask1.h"
#include "AliHeader.h"
#include "AliESDHeader.h"
#include "AliStack.h"
#include "TParticle.h"
#include "TLorentzVector.h"
#include "TH1I.h"
#include "AliMCEvent.h"
#include "AliAnalysisManager.h"
#include "AliESDEvent.h"
#include "AliAODEvent.h"
#include "AliCFManager.h"
#include "AliCFCutBase.h"
#include "AliCFContainer.h"
#include "TChain.h"
#include "AliESDtrack.h"
#include "AliLog.h"
#include "AliESDMuonTrack.h"
#include "AliESDtrack.h"
#include "AliESDInputHandler.h"
#include "TCanvas.h"

ClassImp(AliCFMuonSingleTask1)

//__________________________________________________________________________
AliCFMuonSingleTask1::AliCFMuonSingleTask1() :
  fReadAODData(0),
  fCFManager(0x0),
  fQAHistList(0x0),
  fHistEventsProcessed(0x0),
  fNevt(0),
  fIsMC(kFALSE)
{
  //
  //Default ctor
  //
}
//___________________________________________________________________________
AliCFMuonSingleTask1::AliCFMuonSingleTask1(const Char_t* name) :
  AliAnalysisTaskSE(name),
  fReadAODData(0),
  fCFManager(0x0),
  fQAHistList(0x0),
  fHistEventsProcessed(0x0),
  fNevt(0),
  fIsMC(kFALSE)
{
  //
  // Constructor. Initialization of Inputs and Outputs
  //
  Info("AliCFMuonSingleTask1","Calling Constructor");

  fHistEventsProcessed = new TH1I("fHistEventsProcessed","",1,0,1) ;

  DefineOutput(1,TH1I::Class());
  DefineOutput(2,AliCFContainer::Class());

}

//___________________________________________________________________________
AliCFMuonSingleTask1& AliCFMuonSingleTask1::operator=(const AliCFMuonSingleTask1& c) 
{
  //
  // Assignment operator
  //
  if (this!=&c) {
    AliAnalysisTaskSE::operator=(c) ;
    fReadAODData = c.fReadAODData ;
    fCFManager  = c.fCFManager;
    fQAHistList = c.fQAHistList ;
    fHistEventsProcessed = c.fHistEventsProcessed;
    fNevt = c.fNevt ;
  }
  return *this;
}

//___________________________________________________________________________
AliCFMuonSingleTask1::AliCFMuonSingleTask1(const AliCFMuonSingleTask1& c) :
  AliAnalysisTaskSE(c),
  fReadAODData(c.fReadAODData),
  fCFManager(c.fCFManager),
  fQAHistList(c.fQAHistList),
  fHistEventsProcessed(c.fHistEventsProcessed),
  fNevt(c.fNevt),
  fIsMC(kFALSE)
{
  //
  // Copy Constructor
  //
}

//___________________________________________________________________________
AliCFMuonSingleTask1::~AliCFMuonSingleTask1() {
  //
  //destructor
  //
  Info("~AliCFMuonSingleTask1","Calling Destructor");
  if (fCFManager)           delete fCFManager ;
  if (fHistEventsProcessed) delete fHistEventsProcessed ;
  if (fQAHistList) {fQAHistList->Clear(); delete fQAHistList;}
}

//_________________________________________________
void AliCFMuonSingleTask1::UserExec(Option_t *)
{
  //
  // Main loop function
  //  

  Info("UserExec"," ") ;
  if (!fMCEvent && fIsMC) {  
    Error("UserExec","NO MC EVENT FOUND!");
    return;
  }

  fNevt++;
  fCFManager->SetEventInfo(fMCEvent);
  Double_t containerInput[18] ;

////////
//// MC
////////

// loop on the MC part

  if(fIsMC){

      for (Int_t ipart=0; ipart<fMCEvent->GetNumberOfTracks(); ipart++) { 
	  
	  AliMCParticle *mcPart  = (AliMCParticle*) fMCEvent->GetTrack(ipart);
	  TParticle *part = mcPart->Particle();

	  // rapidity and Pt cuts (default -4<y<-2.5 et 0<pt<20)
	  if (!fCFManager->CheckParticleCuts(AliCFManager::kPartAccCuts,mcPart)) continue;

	  // Selection of mu
	  Int_t pdg=part->GetPdgCode();
	  if(TMath::Abs(pdg)==13){

	      Float_t emc = part->Energy();
	      Float_t pzmc = part->Pz();           
	      Float_t rapmc = Rap(emc,pzmc);
	      Float_t phimc = part->Phi(); 
	      phimc = Phideg(phimc);
	      Float_t ptmc = part->Pt();
	      Float_t pmc = part->P();    
	      Float_t zmc = part->Vz();
	      Float_t etamc = part->Eta();
	      Float_t chargemc=0;
	      if(pdg==13) chargemc=-1;
	      if(pdg==-13) chargemc=1;
	      Float_t thetamc;
	      thetamc = TMath::Abs(-TMath::Pi()+TMath::ASin(ptmc/pmc)); // convention for EVGEN
	      Float_t ymc = part->Vy();
	      Float_t xmc = part->Vx();
	      Float_t rmc = TMath::Sqrt(xmc*xmc+ymc*ymc);   
      
	      containerInput[0] = etamc ;   
	      containerInput[1] = rapmc ;   
	      containerInput[2] = phimc ;   
	      containerInput[3] = ptmc ;   
	      containerInput[4] = pmc ;
	      containerInput[14] = zmc ;
	      containerInput[16] = chargemc ;
	      containerInput[17] = thetamc ;

	      // 10 var calculated only for ESD .i.e set at 1 in MC step
	      for(Int_t i=5; i<=13; i++){
		  containerInput[i] = 1;
	      }

	      containerInput[7] = 3;
	      containerInput[10] = rmc ; // radius at production      
	      containerInput[15] = 175. ;  // rabsmc
	      
	      // fill the container at the first step
	      fCFManager->GetParticleContainer()->Fill(containerInput,0);
	  }
      }
  }
  	      

////////
//// ESD
////////

  Int_t trig1MuL = 1 ,trig1MuH = 2, trigUSL = 4, trigUSH = 8, trigLSL = 16, trigLSH = 32;
  Int_t trig = 0;
 
  AliESDEvent *fESD; 
  AliESDInputHandler *esdH = dynamic_cast<AliESDInputHandler*>
      (AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler());
  if ( ! esdH) {
    AliError("Cannot get input event handler");    
    return;
  }    
  fESD = esdH->GetEvent();

// trigger 

  if (fESD->GetTriggerMask() & trig1MuL) trig = 1;
  if (fESD->GetTriggerMask() & trig1MuH) trig = 2;
  if (fESD->GetTriggerMask() & trigUSL)  trig = 3;
  if (fESD->GetTriggerMask() & trigUSH)  trig = 4;
  if (fESD->GetTriggerMask() & trigLSL)  trig = 5;
  if (fESD->GetTriggerMask() & trigLSH)  trig = 6;
 
// vertex 

  Float_t vx = -200 , vy = -200 , vz = -200, vt=-200;
  
  Double_t Ncont = 0.; 
  AliESDVertex* vertex = (AliESDVertex*) fESD->GetVertex();
  if (vertex->GetNContributors()) {
      vz = vertex->GetZ();
      vy = vertex->GetY();
      vx = vertex->GetX();
      vt=TMath::Sqrt(vx*vx+vy*vy);
      Ncont=vertex->GetNContributors();
  }

// tracks

  Int_t mult1 = fESD->GetNumberOfMuonTracks() ;

    for (Int_t j = 0; j<mult1; j++) {
	AliESDMuonTrack* mu1 = new AliESDMuonTrack(*(fESD->GetMuonTrack(j)));

	if (!mu1->ContainTrackerData()) { mu1=0; continue; }

	Float_t charger = mu1->Charge();
	Float_t er = mu1->E();
	Float_t pzr = mu1->Pz();
	Float_t rapr=Rap(er,pzr);
	Float_t phir = mu1->Phi(); 
	phir = Phideg(phir);
	Float_t ptr = mu1->Pt();
	Float_t pr = mu1->P();
	Float_t hit = mu1->GetNHit();	
	Float_t chi2 = mu1->GetChi2() / (2.0 * hit - 5);
	Int_t matchtrig = mu1->GetMatchTrigger();
	Float_t chi2match = mu1->GetChi2MatchTrigger();
	Float_t dcar = mu1->GetDCA();
	Float_t zr = mu1->GetZ();
	Float_t etar = mu1->Eta();
	Float_t rabs = mu1->GetRAtAbsorberEnd();
	Float_t thetar ;
	thetar = TMath::Abs(-TMath::Pi()+TMath::ASin(ptr/pr)); 

// rapidity and Pt cuts (default -4<y<-2.5 et 0<pt<20)
	if (!fCFManager->CheckParticleCuts(AliCFManager::kPartAccCuts,mu1)) continue;

        Float_t thetaAbs = 180.*(1.-TMath::ATan(rabs/505.)/TMath::Pi());
        if (thetaAbs<171 || thetaAbs>178) continue;

	containerInput[0] = etar ;   
	containerInput[1] = rapr ;   
	containerInput[2] = phir ;   
	containerInput[3] = ptr ;
	containerInput[4] = pr ;
	containerInput[5] = hit ;   
	containerInput[6] = chi2 ;   
	containerInput[7] = matchtrig ;   
	containerInput[8] = chi2match ;
	containerInput[9] = Ncont ;   
	containerInput[10] = vt ;   
	containerInput[11] = vz ;   
	containerInput[12] = trig ;
	containerInput[13] = dcar ;
	containerInput[14] = zr ;
	containerInput[15] = thetaAbs;
	containerInput[16] = charger ;
	containerInput[17] = thetar ;

// fill the container at the second step (for simu, first for data)
	if(fIsMC){
	    fCFManager->GetParticleContainer()->Fill(containerInput,1);
	}
	else{
	    fCFManager->GetParticleContainer()->Fill(containerInput,0);
	}
	

    }      // mu Loop

//  ----------
  fHistEventsProcessed->Fill(0);
  PostData(1,fHistEventsProcessed) ;
  PostData(2,fCFManager->GetParticleContainer()) ;

}
//________________________________________________________________________
Float_t AliCFMuonSingleTask1::Rap(Float_t e, Float_t pz) 
{
// calculate rapidity
    Float_t rap;
    if(e!=pz){
	rap = 0.5*TMath::Log((e+pz)/(e-pz));
	return rap;
    }
    else{
	rap = -200;
	return rap;
    }
}
//________________________________________________________________________
Float_t AliCFMuonSingleTask1::Phideg(Float_t phi) 
{
// calculate Phi in range [-180,180] 
    Float_t phideg;
    
	phideg = phi-TMath::Pi();
	phideg = phideg*57.296;
	return phideg;
}
//________________________________________________________________________
void AliCFMuonSingleTask1::Terminate(Option_t *) 
{
  // project pt (var[3]) from the two steps MC(0) and ESD(1)

    AliCFContainer *cont = dynamic_cast<AliCFContainer*> (GetOutputData(2));
    if ( ! cont ) {
      AliError("Cannot find container in file");
      return;
    }

    TH1D *kpt = cont->ShowProjection(16,0);
    TH1D *rpt = cont->ShowProjection(16,1);

    TCanvas *c1 = new TCanvas("AliCFMuonSingleTask1"," MC & ESD",10,10,510,510);
    c1->Divide(1,2);
    c1->cd(1);
    kpt->Draw("HIST");
    c1->cd(2);
    rpt->Draw("HIST");

}
//________________________________________________________________________

#endif