[u/mrichter/AliRoot.git] / PWGGA / CaloTrackCorrelations / AliAnaParticlePartonCorrelation.cxx

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

//_________________________________________________________________________
// Class for the analysis of particle-parton correlations
// Particle (for example direct gamma) must be found in a previous analysis 
// -- Author: Gustavo Conesa (LNF-INFN) 
//////////////////////////////////////////////////////////////////////////////

// --- ROOT system ---
//#include "Riostream.h"
#include "TH2F.h"
#include "TParticle.h"
#include "TClass.h"

//---- ANALYSIS system ----
#include "AliAnaParticlePartonCorrelation.h" 
#include "AliStack.h"  
#include "AliAODPWG4ParticleCorrelation.h"

ClassImp(AliAnaParticlePartonCorrelation)


//________________________________________________________________
AliAnaParticlePartonCorrelation::AliAnaParticlePartonCorrelation() : 
AliAnaCaloTrackCorrBaseClass(),   
fhDeltaEtaNearParton(0), fhDeltaPhiNearParton(0), 
fhDeltaPtNearParton(0),  fhPtRatNearParton(0),
fhDeltaEtaAwayParton(0), fhDeltaPhiAwayParton(0), 
fhDeltaPtAwayParton(0),  fhPtRatAwayParton(0)
{
  //Default Ctor
  
  //Initialize parameters
  InitParameters();
}

//________________________________________________________________
TList *  AliAnaParticlePartonCorrelation::GetCreateOutputObjects()
{  
  // Create histograms to be saved in output file 
  
  TList * outputContainer = new TList() ; 
  outputContainer->SetName("ParticlePartonHistos") ; 
  
  fhDeltaPhiNearParton  = new TH2F
  ("DeltaPhiNearParton","#phi_{particle} - #phi_{parton} vs p_{T particle}",
   200,0,120,200,0,6.4); 
  fhDeltaPhiNearParton->SetYTitle("#Delta #phi");
  fhDeltaPhiNearParton->SetXTitle("p_{T particle} (GeV/c)");
  outputContainer->Add(fhDeltaPhiNearParton);
  
  fhDeltaEtaNearParton  = new TH2F
  ("DeltaEtaNearParton","#eta_{particle} - #eta_{parton} vs p_{T particle}",
   200,0,120,200,-2,2); 
  fhDeltaEtaNearParton->SetYTitle("#Delta #eta");
  fhDeltaEtaNearParton->SetXTitle("p_{T particle} (GeV/c)");
  outputContainer->Add(fhDeltaEtaNearParton);
  
  fhDeltaPtNearParton  = new TH2F
  ("DeltaPtNearParton","#p_{T particle} - #p_{T parton} vs p_{T particle}",
   200,0,120,100,-10,10); 
  fhDeltaPtNearParton->SetYTitle("#Delta #p_{T}");
  fhDeltaPtNearParton->SetXTitle("p_{T particle} (GeV/c)"); 
  outputContainer->Add(fhDeltaPtNearParton);
  
  fhPtRatNearParton  = new TH2F
  ("PtRatNearParton","#p_{T parton} / #p_{T particle} vs p_{T particle}",
   200,0,120,200,0,5); 
  fhPtRatNearParton->SetYTitle("ratio");
  fhPtRatNearParton->SetXTitle("p_{T particle} (GeV/c)");
  outputContainer->Add(fhPtRatNearParton);
  
  fhDeltaPhiAwayParton  = new TH2F
  ("DeltaPhiAwayParton","#phi_{particle} - #phi_{parton} vs p_{T particle}",
   200,0,120,200,0,6.4); 
  fhDeltaPhiAwayParton->SetYTitle("#Delta #phi");
  fhDeltaPhiAwayParton->SetXTitle("p_{T particle} (GeV/c)");
  outputContainer->Add(fhDeltaPhiAwayParton);
  
  fhDeltaEtaAwayParton  = new TH2F
  ("DeltaEtaAwayParton","#eta_{particle} - #eta_{parton} vs p_{T particle}",
   200,0,120,200,-2,2); 
  fhDeltaEtaAwayParton->SetYTitle("#Delta #eta");
  fhDeltaEtaAwayParton->SetXTitle("p_{T particle} (GeV/c)");
  outputContainer->Add(fhDeltaEtaAwayParton);
  
  fhDeltaPtAwayParton  = new TH2F
  ("DeltaPtAwayParton","#p_{T particle} - #p_{T parton} vs p_{T particle}",
   200,0,120,100,-10,10); 
  fhDeltaPtAwayParton->SetYTitle("#Delta #p_{T}");
  fhDeltaPtAwayParton->SetXTitle("p_{T particle} (GeV/c)"); 
  outputContainer->Add(fhDeltaPtAwayParton);
  
  fhPtRatAwayParton  = new TH2F
  ("PtRatAwayParton","#p_{T parton} / #p_{T particle} vs p_{T particle}",
   200,0,120,200,0,5); 
  fhPtRatAwayParton->SetYTitle("ratio");
  fhPtRatAwayParton->SetXTitle("p_{T particle} (GeV/c)");
  outputContainer->Add(fhPtRatAwayParton);
  
  return outputContainer;
  
}

//____________________________________________________
void AliAnaParticlePartonCorrelation::InitParameters()
{
  
  //Initialize the parameters of the analysis.
  SetInputAODName("PWG4Particle");
  SetAODObjArrayName("Partons");  
  AddToHistogramsName("AnaPartonCorr_");
  
}

//_____________________________________________________________________
void AliAnaParticlePartonCorrelation::Print(const Option_t * opt) const
{
  
  //Print some relevant parameters set for the analysis
  if(! opt)
    return;
  
  printf("**** Print %s %s ****\n", GetName(), GetTitle() ) ;
  AliAnaCaloTrackCorrBaseClass::Print(" ");
  
} 

//__________________________________________________________
void  AliAnaParticlePartonCorrelation::MakeAnalysisFillAOD()  
{
  //Particle-Parton Correlation Analysis, create AODs
  //Add partons to the reference list of the trigger particle
  //Partons are considered those in the first eight possitions in the stack
  //being 0, and 1 the 2 protons, and 6 and 7 the outgoing final partons.
  if(!GetInputAODBranch()){
    printf("AliAnaParticlePartonCorrelation::MakeAnalysisFillAOD() - No input particles in AOD with name branch < %s > \n",GetInputAODName().Data());
    abort();	
  }
	
  if(strcmp(GetInputAODBranch()->GetClass()->GetName(), "AliAODPWG4ParticleCorrelation")){
    printf("AliAnaParticlePartonCorrelation::MakeAnalysisFillAOD() - Wrong type of AOD object, change AOD class name in input AOD: It should be <AliAODPWG4ParticleCorrelation> and not <%s> \n",GetInputAODBranch()->GetClass()->GetName());
    abort();
  }	
	
  if(GetDebug() > 1){
    printf("AliAnaParticlePartonCorrelation::MakeAnalysisFillAOD() - Begin fill AODs \n");
    printf("AliAnaParticlePartonCorrelation::MakeAnalysisFillAOD() - In particle branch aod entries %d\n", GetInputAODBranch()->GetEntriesFast());
  }
  
  //Loop on stored AOD particles
  Int_t naod = GetInputAODBranch()->GetEntriesFast();
  for(Int_t iaod = 0; iaod < naod ; iaod++){
    AliAODPWG4ParticleCorrelation* particle =  (AliAODPWG4ParticleCorrelation*) (GetInputAODBranch()->At(iaod));
    
    AliStack * stack =  GetMCStack() ;
    if(!stack){ 
      printf("AliAnaParticlePartonCorrelation::MakeAnalysisFillAOD() - No Stack available, STOP\n");
      abort();
    }
    if(stack->GetNtrack() < 8) {
      printf("AliAnaParticlePartonCorrelation::MakeAnalysisFillAOD() *** small number of particles, not a PYTHIA simulation? ***:  n tracks %d \n", stack->GetNprimary());
      continue ;
    }
    
    //Fill AOD reference only with partons
    
    //Array with reference to partons, initialize
    TObjArray * objarray  = NULL;
    Int_t nrefs = 0;
    
    TParticle * parton    = NULL ;
    for(Int_t ipr = 0;ipr < 8; ipr ++ ){
      parton = stack->Particle(ipr) ;
      nrefs++;
      if(nrefs==1){
        objarray = new TObjArray(0);
        objarray->SetName(GetAODObjArrayName());
        objarray->SetOwner(kFALSE);
      }
      objarray->Add(parton);
    }//parton loop
    
    if(objarray->GetEntriesFast() > 0) particle->AddObjArray(objarray);
    
  }//Aod branch loop
  
  if(GetDebug() > 1) printf("AliAnaParticlePartonCorrelation::MakeAnalysisFillAOD() - End fill AODs \n");
}

//_________________________________________________________________
void  AliAnaParticlePartonCorrelation::MakeAnalysisFillHistograms() 
{
  //Particle-Parton Correlation Analysis, fill histograms
  if(!GetInputAODBranch()){
    printf("AliAnaParticlePartonCorrelation::MakeAnalysisFillHistograms() - No input particles in AOD with name branch < %s > \n",GetInputAODName().Data());
    abort();	
  }
  if(GetDebug() > 1){
    printf("AliAnaParticlePartonCorrelation::MakeAnalysisFillHistograms() - Begin parton correlation analysis, fill histograms \n");
    printf("AliAnaParticlePartonCorrelation::MakeAnalysisFillHistograms() - In particle branch aod entries %d\n", GetInputAODBranch()->GetEntriesFast());
  }
  
  AliStack * stack =  GetMCStack() ;
  if(!stack) {
    printf("AliAnaParticlePartonCorrelation::MakeAnalysisFillHistograms() - No Stack available, STOP\n");
    abort();
  }
  
  //Loop on stored AOD particles
  Int_t naod = GetInputAODBranch()->GetEntriesFast();
  TParticle *  mom = NULL ;
  
  for(Int_t iaod = 0; iaod < naod ; iaod++){
    AliAODPWG4ParticleCorrelation* particle =  (AliAODPWG4ParticleCorrelation*) (GetInputAODBranch()->At(iaod));
    
    Float_t ptTrigg  = particle->Pt();
    Float_t phiTrigg = particle->Phi();
    Float_t etaTrigg = particle->Eta(); 
    Int_t imom = particle->GetLabel();
    Int_t iparent  = 2000;
    Int_t iawayparent = -1;
    
    TObjArray * objarray = particle->GetObjArray(GetAODObjArrayName());
    if(!(objarray) || (objarray->GetEntriesFast() < 7) ) {
      printf("AliAnaParticlePartonCorrelation::MakeAnalysisFillHistograms() - Reference list with partons not filled, STOP analysis\n");
      abort();
    }
    
    //Check and get indeces of mother and parton    
    if(imom < 8 ) iparent = imom ;   //mother is already a parton
    else if (imom <  stack->GetNtrack()) {
      mom =  stack->Particle(imom);
      if(mom){
        iparent=mom->GetFirstMother();
        //cout<<" iparent "<<iparent<<endl;
        while(iparent > 7 ){
          mom = stack->Particle(iparent);
          if (mom) {
            imom = iparent ; //Mother label is of the inmediate parton daughter
            iparent = mom->GetFirstMother();
          }
          else iparent = -1;
          //cout<<" while iparent "<<iparent<<endl;
        } 
      }
    }
    
    if(GetDebug() > 1) printf("AliAnaParticlePartonCorrelation::MakeAnalysisFillHistograms() - N reference partons %d; labels:  mother %d, parent %d \n", objarray->GetEntriesFast(), imom, iparent);
    
    
    if(iparent < 0 || iparent > 8) { 
      if(GetDebug() > 0 ) printf("AliAnaParticlePartonCorrelation::MakeAnalysisFillHistograms() - Failed to find appropriate parton, index %d", iparent);
      continue ;
    }
    
    //Near parton is the parton that fragmented and created the mother    
    TParticle * nearParton = (TParticle*) objarray->At(iparent);
    Float_t  ptNearParton    = nearParton->Pt();
    Float_t  phiNearParton   = nearParton->Phi() ;
    Float_t  etaNearParton   = nearParton->Eta() ;
    
    fhDeltaEtaNearParton->Fill(ptTrigg,etaTrigg-etaNearParton);
    fhDeltaPhiNearParton->Fill(ptTrigg,phiTrigg-phiNearParton);
    fhDeltaPtNearParton->Fill(ptTrigg,ptTrigg-ptNearParton);
    fhPtRatNearParton->Fill(ptTrigg,ptNearParton/ptTrigg);
    
    if(iparent == 7) iawayparent =6;
    else if(iparent == 6) iawayparent =7;
    else{
      printf("AliAnaParticlePartonCorrelation::MakeAnalysisFillHistograms() - Parent parton is not final state, skip \n");
      continue;
    }
    
    //Away parton is the other final parton.
    TParticle * awayParton = (TParticle*) objarray->At(iawayparent);
    Float_t  ptAwayParton    = awayParton->Pt();
    Float_t  phiAwayParton   = awayParton->Phi() ;
    Float_t  etaAwayParton   = awayParton->Eta() ;
    fhDeltaEtaAwayParton->Fill(ptTrigg,etaTrigg-etaAwayParton);
    fhDeltaPhiAwayParton->Fill(ptTrigg,phiTrigg-phiAwayParton);
    fhDeltaPtAwayParton->Fill(ptTrigg,ptTrigg-ptAwayParton);
    fhPtRatAwayParton->Fill(ptTrigg,ptAwayParton/ptTrigg);
    
  }
  
  if(GetDebug() > 1) printf("AliAnaParticlePartonCorrelation::MakeAnalysisFillHistograms() - End fill histograms \n");
  
}
Commit	Line	Data
1c5acb87	1	/**************************************************************************
	2	* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	3	* *
	4	* Author: The ALICE Off-line Project. *
	5	* Contributors are mentioned in the code where appropriate. *
	6	* *
	7	* Permission to use, copy, modify and distribute this software and its *
	8	* documentation strictly for non-commercial purposes is hereby granted *
	9	* without fee, provided that the above copyright notice appears in all *
	10	* copies and that both the copyright notice and this permission notice *
	11	* appear in the supporting documentation. The authors make no claims *
	12	* about the suitability of this software for any purpose. It is *
	13	* provided "as is" without express or implied warranty. *
	14	**************************************************************************/
1c5acb87	15
	16	//_________________________________________________________________________
	17	// Class for the analysis of particle-parton correlations
	18	// Particle (for example direct gamma) must be found in a previous analysis
	19	// -- Author: Gustavo Conesa (LNF-INFN)
	20	//////////////////////////////////////////////////////////////////////////////
	21
	22	// --- ROOT system ---
	23	//#include "Riostream.h"
	24	#include "TH2F.h"
	25	#include "TParticle.h"
9415d854	26	#include "TClass.h"
1c5acb87	27
	28	//---- ANALYSIS system ----
	29	#include "AliAnaParticlePartonCorrelation.h"
1c5acb87	30	#include "AliStack.h"
	31	#include "AliAODPWG4ParticleCorrelation.h"
	32
745913ae	33	ClassImp(AliAnaParticlePartonCorrelation)
745913ae	34
1c5acb87	35
745913ae	36	//________________________________________________________________
	37	AliAnaParticlePartonCorrelation::AliAnaParticlePartonCorrelation() :
	38	AliAnaCaloTrackCorrBaseClass(),
	39	fhDeltaEtaNearParton(0), fhDeltaPhiNearParton(0),
	40	fhDeltaPtNearParton(0), fhPtRatNearParton(0),
	41	fhDeltaEtaAwayParton(0), fhDeltaPhiAwayParton(0),
	42	fhDeltaPtAwayParton(0), fhPtRatAwayParton(0)
1c5acb87	43	{
1c5acb87	44	//Default Ctor
745913ae	45
1c5acb87	46	//Initialize parameters
	47	InitParameters();
	48	}
1c5acb87	49
745913ae	50	//________________________________________________________________
1c5acb87	51	TList * AliAnaParticlePartonCorrelation::GetCreateOutputObjects()
	52	{
	53	// Create histograms to be saved in output file
745913ae	54
1c5acb87	55	TList * outputContainer = new TList() ;
	56	outputContainer->SetName("ParticlePartonHistos") ;
	57
	58	fhDeltaPhiNearParton = new TH2F
745913ae	59	("DeltaPhiNearParton","#phi_{particle} - #phi_{parton} vs p_{T particle}",
745913ae	60	200,0,120,200,0,6.4);
1c5acb87	61	fhDeltaPhiNearParton->SetYTitle("#Delta #phi");
	62	fhDeltaPhiNearParton->SetXTitle("p_{T particle} (GeV/c)");
	63	outputContainer->Add(fhDeltaPhiNearParton);
	64
	65	fhDeltaEtaNearParton = new TH2F
745913ae	66	("DeltaEtaNearParton","#eta_{particle} - #eta_{parton} vs p_{T particle}",
745913ae	67	200,0,120,200,-2,2);
1c5acb87	68	fhDeltaEtaNearParton->SetYTitle("#Delta #eta");
	69	fhDeltaEtaNearParton->SetXTitle("p_{T particle} (GeV/c)");
	70	outputContainer->Add(fhDeltaEtaNearParton);
	71
	72	fhDeltaPtNearParton = new TH2F
745913ae	73	("DeltaPtNearParton","#p_{T particle} - #p_{T parton} vs p_{T particle}",
745913ae	74	200,0,120,100,-10,10);
1c5acb87	75	fhDeltaPtNearParton->SetYTitle("#Delta #p_{T}");
	76	fhDeltaPtNearParton->SetXTitle("p_{T particle} (GeV/c)");
	77	outputContainer->Add(fhDeltaPtNearParton);
	78
	79	fhPtRatNearParton = new TH2F
745913ae	80	("PtRatNearParton","#p_{T parton} / #p_{T particle} vs p_{T particle}",
745913ae	81	200,0,120,200,0,5);
1c5acb87	82	fhPtRatNearParton->SetYTitle("ratio");
	83	fhPtRatNearParton->SetXTitle("p_{T particle} (GeV/c)");
	84	outputContainer->Add(fhPtRatNearParton);
	85
	86	fhDeltaPhiAwayParton = new TH2F
745913ae	87	("DeltaPhiAwayParton","#phi_{particle} - #phi_{parton} vs p_{T particle}",
745913ae	88	200,0,120,200,0,6.4);
1c5acb87	89	fhDeltaPhiAwayParton->SetYTitle("#Delta #phi");
	90	fhDeltaPhiAwayParton->SetXTitle("p_{T particle} (GeV/c)");
	91	outputContainer->Add(fhDeltaPhiAwayParton);
	92
	93	fhDeltaEtaAwayParton = new TH2F
745913ae	94	("DeltaEtaAwayParton","#eta_{particle} - #eta_{parton} vs p_{T particle}",
745913ae	95	200,0,120,200,-2,2);
1c5acb87	96	fhDeltaEtaAwayParton->SetYTitle("#Delta #eta");
	97	fhDeltaEtaAwayParton->SetXTitle("p_{T particle} (GeV/c)");
	98	outputContainer->Add(fhDeltaEtaAwayParton);
	99
	100	fhDeltaPtAwayParton = new TH2F
745913ae	101	("DeltaPtAwayParton","#p_{T particle} - #p_{T parton} vs p_{T particle}",
745913ae	102	200,0,120,100,-10,10);
1c5acb87	103	fhDeltaPtAwayParton->SetYTitle("#Delta #p_{T}");
	104	fhDeltaPtAwayParton->SetXTitle("p_{T particle} (GeV/c)");
	105	outputContainer->Add(fhDeltaPtAwayParton);
	106
	107	fhPtRatAwayParton = new TH2F
745913ae	108	("PtRatAwayParton","#p_{T parton} / #p_{T particle} vs p_{T particle}",
745913ae	109	200,0,120,200,0,5);
1c5acb87	110	fhPtRatAwayParton->SetYTitle("ratio");
	111	fhPtRatAwayParton->SetXTitle("p_{T particle} (GeV/c)");
	112	outputContainer->Add(fhPtRatAwayParton);
	113
	114	return outputContainer;
745913ae	115
1c5acb87	116	}
1c5acb87	117
745913ae	118	//____________________________________________________
1c5acb87	119	void AliAnaParticlePartonCorrelation::InitParameters()
	120	{
	121
	122	//Initialize the parameters of the analysis.
a3aebfff	123	SetInputAODName("PWG4Particle");
591cc579	124	SetAODObjArrayName("Partons");
a3aebfff	125	AddToHistogramsName("AnaPartonCorr_");
745913ae	126
1c5acb87	127	}
1c5acb87	128
745913ae	129	//_____________________________________________________________________
1c5acb87	130	void AliAnaParticlePartonCorrelation::Print(const Option_t * opt) const
	131	{
	132
	133	//Print some relevant parameters set for the analysis
	134	if(! opt)
	135	return;
	136
a3aebfff	137	printf("** Print %s %s **\n", GetName(), GetTitle() ) ;
745913ae	138	AliAnaCaloTrackCorrBaseClass::Print(" ");
745913ae	139
1c5acb87	140	}
1c5acb87	141
745913ae	142	//__________________________________________________________
1c5acb87	143	void AliAnaParticlePartonCorrelation::MakeAnalysisFillAOD()
	144	{
	145	//Particle-Parton Correlation Analysis, create AODs
	146	//Add partons to the reference list of the trigger particle
	147	//Partons are considered those in the first eight possitions in the stack
	148	//being 0, and 1 the 2 protons, and 6 and 7 the outgoing final partons.
477d6cee	149	if(!GetInputAODBranch()){
a3aebfff	150	printf("AliAnaParticlePartonCorrelation::MakeAnalysisFillAOD() - No input particles in AOD with name branch < %s > \n",GetInputAODName().Data());
477d6cee	151	abort();
477d6cee	152	}
9415d854	153
9415d854	154	if(strcmp(GetInputAODBranch()->GetClass()->GetName(), "AliAODPWG4ParticleCorrelation")){
745913ae	155	printf("AliAnaParticlePartonCorrelation::MakeAnalysisFillAOD() - Wrong type of AOD object, change AOD class name in input AOD: It should be <AliAODPWG4ParticleCorrelation> and not <%s> \n",GetInputAODBranch()->GetClass()->GetName());
745913ae	156	abort();
9415d854	157	}
9415d854	158
1c5acb87	159	if(GetDebug() > 1){
a3aebfff	160	printf("AliAnaParticlePartonCorrelation::MakeAnalysisFillAOD() - Begin fill AODs \n");
a3aebfff	161	printf("AliAnaParticlePartonCorrelation::MakeAnalysisFillAOD() - In particle branch aod entries %d\n", GetInputAODBranch()->GetEntriesFast());
1c5acb87	162	}
	163
	164	//Loop on stored AOD particles
	165	Int_t naod = GetInputAODBranch()->GetEntriesFast();
	166	for(Int_t iaod = 0; iaod < naod ; iaod++){
	167	AliAODPWG4ParticleCorrelation* particle = (AliAODPWG4ParticleCorrelation*) (GetInputAODBranch()->At(iaod));
	168
	169	AliStack * stack = GetMCStack() ;
477d6cee	170	if(!stack){
a3aebfff	171	printf("AliAnaParticlePartonCorrelation::MakeAnalysisFillAOD() - No Stack available, STOP\n");
477d6cee	172	abort();
477d6cee	173	}
1c5acb87	174	if(stack->GetNtrack() < 8) {
a3aebfff	175	printf("AliAnaParticlePartonCorrelation::MakeAnalysisFillAOD() * small number of particles, not a PYTHIA simulation? *: n tracks %d \n", stack->GetNprimary());
1c5acb87	176	continue ;
	177	}
	178
	179	//Fill AOD reference only with partons
745913ae	180
a3aebfff	181	//Array with reference to partons, initialize
f8006433	182	TObjArray * objarray = NULL;
	183	Int_t nrefs = 0;
	184
	185	TParticle * parton = NULL ;
1c5acb87	186	for(Int_t ipr = 0;ipr < 8; ipr ++ ){
1c5acb87	187	parton = stack->Particle(ipr) ;
f8006433	188	nrefs++;
	189	if(nrefs==1){
	190	objarray = new TObjArray(0);
	191	objarray->SetName(GetAODObjArrayName());
	192	objarray->SetOwner(kFALSE);
	193	}
	194	objarray->Add(parton);
a3aebfff	195	}//parton loop
745913ae	196
591cc579	197	if(objarray->GetEntriesFast() > 0) particle->AddObjArray(objarray);
745913ae	198
1c5acb87	199	}//Aod branch loop
477d6cee	200
a3aebfff	201	if(GetDebug() > 1) printf("AliAnaParticlePartonCorrelation::MakeAnalysisFillAOD() - End fill AODs \n");
1c5acb87	202	}
a3aebfff	203
745913ae	204	//_________________________________________________________________
745913ae	205	void AliAnaParticlePartonCorrelation::MakeAnalysisFillHistograms()
1c5acb87	206	{
1c5acb87	207	//Particle-Parton Correlation Analysis, fill histograms
477d6cee	208	if(!GetInputAODBranch()){
a3aebfff	209	printf("AliAnaParticlePartonCorrelation::MakeAnalysisFillHistograms() - No input particles in AOD with name branch < %s > \n",GetInputAODName().Data());
477d6cee	210	abort();
477d6cee	211	}
1c5acb87	212	if(GetDebug() > 1){
a3aebfff	213	printf("AliAnaParticlePartonCorrelation::MakeAnalysisFillHistograms() - Begin parton correlation analysis, fill histograms \n");
a3aebfff	214	printf("AliAnaParticlePartonCorrelation::MakeAnalysisFillHistograms() - In particle branch aod entries %d\n", GetInputAODBranch()->GetEntriesFast());
1c5acb87	215	}
477d6cee	216
1c5acb87	217	AliStack * stack = GetMCStack() ;
477d6cee	218	if(!stack) {
a3aebfff	219	printf("AliAnaParticlePartonCorrelation::MakeAnalysisFillHistograms() - No Stack available, STOP\n");
477d6cee	220	abort();
477d6cee	221	}
745913ae	222
1c5acb87	223	//Loop on stored AOD particles
1c5acb87	224	Int_t naod = GetInputAODBranch()->GetEntriesFast();
164a1d84	225	TParticle * mom = NULL ;
477d6cee	226
1c5acb87	227	for(Int_t iaod = 0; iaod < naod ; iaod++){
	228	AliAODPWG4ParticleCorrelation* particle = (AliAODPWG4ParticleCorrelation*) (GetInputAODBranch()->At(iaod));
	229
	230	Float_t ptTrigg = particle->Pt();
	231	Float_t phiTrigg = particle->Phi();
	232	Float_t etaTrigg = particle->Eta();
	233	Int_t imom = particle->GetLabel();
	234	Int_t iparent = 2000;
	235	Int_t iawayparent = -1;
745913ae	236
591cc579	237	TObjArray * objarray = particle->GetObjArray(GetAODObjArrayName());
591cc579	238	if(!(objarray) \|\| (objarray->GetEntriesFast() < 7) ) {
a3aebfff	239	printf("AliAnaParticlePartonCorrelation::MakeAnalysisFillHistograms() - Reference list with partons not filled, STOP analysis\n");
477d6cee	240	abort();
477d6cee	241	}
745913ae	242
1c5acb87	243	//Check and get indeces of mother and parton
	244	if(imom < 8 ) iparent = imom ; //mother is already a parton
	245	else if (imom < stack->GetNtrack()) {
	246	mom = stack->Particle(imom);
164a1d84	247	if(mom){
	248	iparent=mom->GetFirstMother();
	249	//cout<<" iparent "<<iparent<<endl;
	250	while(iparent > 7 ){
	251	mom = stack->Particle(iparent);
	252	if (mom) {
	253	imom = iparent ; //Mother label is of the inmediate parton daughter
	254	iparent = mom->GetFirstMother();
	255	}
	256	else iparent = -1;
745913ae	257	//cout<<" while iparent "<<iparent<<endl;
164a1d84	258	}
164a1d84	259	}
1c5acb87	260	}
1c5acb87	261
591cc579	262	if(GetDebug() > 1) printf("AliAnaParticlePartonCorrelation::MakeAnalysisFillHistograms() - N reference partons %d; labels: mother %d, parent %d \n", objarray->GetEntriesFast(), imom, iparent);
477d6cee	263
1c5acb87	264
1c5acb87	265	if(iparent < 0 \|\| iparent > 8) {
a3aebfff	266	if(GetDebug() > 0 ) printf("AliAnaParticlePartonCorrelation::MakeAnalysisFillHistograms() - Failed to find appropriate parton, index %d", iparent);
1c5acb87	267	continue ;
1c5acb87	268	}
745913ae	269
1c5acb87	270	//Near parton is the parton that fragmented and created the mother
591cc579	271	TParticle * nearParton = (TParticle*) objarray->At(iparent);
1c5acb87	272	Float_t ptNearParton = nearParton->Pt();
	273	Float_t phiNearParton = nearParton->Phi() ;
	274	Float_t etaNearParton = nearParton->Eta() ;
	275
	276	fhDeltaEtaNearParton->Fill(ptTrigg,etaTrigg-etaNearParton);
	277	fhDeltaPhiNearParton->Fill(ptTrigg,phiTrigg-phiNearParton);
	278	fhDeltaPtNearParton->Fill(ptTrigg,ptTrigg-ptNearParton);
	279	fhPtRatNearParton->Fill(ptTrigg,ptNearParton/ptTrigg);
	280
	281	if(iparent == 7) iawayparent =6;
	282	else if(iparent == 6) iawayparent =7;
	283	else{
a3aebfff	284	printf("AliAnaParticlePartonCorrelation::MakeAnalysisFillHistograms() - Parent parton is not final state, skip \n");
1c5acb87	285	continue;
1c5acb87	286	}
745913ae	287
1c5acb87	288	//Away parton is the other final parton.
591cc579	289	TParticle * awayParton = (TParticle*) objarray->At(iawayparent);
1c5acb87	290	Float_t ptAwayParton = awayParton->Pt();
	291	Float_t phiAwayParton = awayParton->Phi() ;
	292	Float_t etaAwayParton = awayParton->Eta() ;
	293	fhDeltaEtaAwayParton->Fill(ptTrigg,etaTrigg-etaAwayParton);
	294	fhDeltaPhiAwayParton->Fill(ptTrigg,phiTrigg-phiAwayParton);
	295	fhDeltaPtAwayParton->Fill(ptTrigg,ptTrigg-ptAwayParton);
	296	fhPtRatAwayParton->Fill(ptTrigg,ptAwayParton/ptTrigg);
	297
1c5acb87	298	}
745913ae	299
a3aebfff	300	if(GetDebug() > 1) printf("AliAnaParticlePartonCorrelation::MakeAnalysisFillHistograms() - End fill histograms \n");
1c5acb87	301
1c5acb87	302	}