[u/mrichter/AliRoot.git] / PWG4 / PartCorrDep / 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.                  *
 **************************************************************************/
/* $Id:  $ */

//_________________________________________________________________________
// 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() : 
    AliAnaPartCorrBaseClass(),   
    fhDeltaEtaNearParton(0), fhDeltaPhiNearParton(0), 
    fhDeltaPtNearParton(0), fhPtRatNearParton(0),
    fhDeltaEtaAwayParton(0), fhDeltaPhiAwayParton(0), 
    fhDeltaPtAwayParton(0), fhPtRatAwayParton(0)
{
  //Default Ctor

  //Initialize parameters
  InitParameters();
}

//____________________________________________________________________________
AliAnaParticlePartonCorrelation::AliAnaParticlePartonCorrelation(const AliAnaParticlePartonCorrelation & g) :   
  AliAnaPartCorrBaseClass(g),   
  fhDeltaEtaNearParton(g.fhDeltaEtaNearParton), fhDeltaPhiNearParton(g.fhDeltaPhiNearParton), 
  fhDeltaPtNearParton(g.fhDeltaPtNearParton), fhPtRatNearParton(g.fhPtRatNearParton),
  fhDeltaEtaAwayParton(g.fhDeltaEtaAwayParton), fhDeltaPhiAwayParton(g.fhDeltaPhiAwayParton), 
  fhDeltaPtAwayParton(g.fhDeltaPtAwayParton), fhPtRatAwayParton(g.fhPtRatAwayParton)
{
  // cpy ctor

}

//_________________________________________________________________________
AliAnaParticlePartonCorrelation & AliAnaParticlePartonCorrelation::operator = (const AliAnaParticlePartonCorrelation & source)
{
  // assignment operator

  if(this == &source)return *this;
  ((AliAnaPartCorrBaseClass *)this)->operator=(source);

  fhDeltaEtaAwayParton = source.fhDeltaEtaAwayParton;
  fhDeltaPhiAwayParton = source.fhDeltaPhiAwayParton;
  fhDeltaPtAwayParton = source.fhDeltaPtAwayParton;
  fhPtRatAwayParton = source.fhPtRatAwayParton;
  fhDeltaEtaNearParton = source.fhDeltaEtaNearParton;
  fhDeltaPhiNearParton = source.fhDeltaPhiNearParton;
  fhDeltaPtNearParton = source.fhDeltaPtNearParton;
  fhPtRatNearParton = source.fhPtRatNearParton;

  return *this;

}


//________________________________________________________________________
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() ) ;
  AliAnaPartCorrBaseClass::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
    TParticle * parton = new TParticle ;

    //Array with reference to partons, initialize
    TObjArray * objarray  = new TObjArray;

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