[u/mrichter/AliRoot.git] / STEER / AOD / AliAODPWG4Particle.cxx

/**************************************************************************
 * Copyright(c) 1998-2007, 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:   AliAODPWG4Particle.h $ */

//-------------------------------------------------------------------------
//     AOD class for photon and other particles storage and 
//     correlation studies
//     Author: Yves Schutz, CERN, Gustavo Conesa, INFN
//-------------------------------------------------------------------------

//-- ROOT system --

//-- Analysis system
#include "AliAODPWG4Particle.h"

ClassImp(AliAODPWG4Particle)


//______________________________________________________________________________
AliAODPWG4Particle::AliAODPWG4Particle() :
AliVParticle(),
fMomentum(0),fPdg(-1), fTag(0), fBtag(-1), fLabel(-1), fCaloLabel(), fTrackLabel(),
fDetector(""), fDisp(0), fTof(0), fCharged(0), fTagged(0), fBadDist(0), fFidArea(0), fInputFileIndex(0)
{
  // constructor
  fCaloLabel [0] = -1;
  fCaloLabel [1] = -1;
  fTrackLabel[0] = -1;
  fTrackLabel[1] = -1;
  fTrackLabel[2] = -1;
  fTrackLabel[3] = -1;
}

//______________________________________________________________________________
AliAODPWG4Particle::AliAODPWG4Particle(Double_t px, Double_t py, Double_t pz, Double_t e):
  AliVParticle(),
  fMomentum(0),fPdg(-1), fTag(0), fBtag(-1), fLabel(-1),fCaloLabel(), fTrackLabel(),
  fDetector(""), fDisp(0), fTof(0),fCharged(0), fTagged(0), fBadDist(0), fFidArea(0), fInputFileIndex(0)
{
  // constructor
  fMomentum = new TLorentzVector(px, py, pz, e);
  
  fCaloLabel [0] = -1;
  fCaloLabel [1] = -1;
  fTrackLabel[0] = -1;
  fTrackLabel[1] = -1;	
  fTrackLabel[2] = -1;
  fTrackLabel[3] = -1;	
}

//______________________________________________________________________________
AliAODPWG4Particle::AliAODPWG4Particle(TLorentzVector & p):
  AliVParticle(),
  fMomentum(0),fPdg(-1), fTag(0), fBtag(-1), fLabel(-1),fCaloLabel(), fTrackLabel(),
  fDetector(""), fDisp(0), fTof(0), fCharged(0), fTagged(0), fBadDist(0), fFidArea(0), fInputFileIndex(0)
{
  // constructor
  fMomentum = new TLorentzVector(p);
  
  fCaloLabel [0] = -1;
  fCaloLabel [1] = -1;
  fTrackLabel[0] = -1;
  fTrackLabel[1] = -1;
  fTrackLabel[2] = -1;
  fTrackLabel[3] = -1;
}


//______________________________________________________________________________
AliAODPWG4Particle::~AliAODPWG4Particle() 
{
  // destructor
    delete fMomentum;
}

//______________________________________________________________________________
void AliAODPWG4Particle::Clear(const Option_t* /*opt*/) 
{
  //clear
  delete fMomentum;
}

//______________________________________________________________________________
AliAODPWG4Particle::AliAODPWG4Particle(const AliAODPWG4Particle& part) :
  AliVParticle(part),
  fMomentum(0), fPdg(part.fPdg), fTag(part.fTag), fBtag(part.fBtag), fLabel(part.fLabel), 
  fCaloLabel(), fTrackLabel(), fDetector(part.fDetector),fDisp(part.fDisp), 
  fTof(part.fTof), fCharged(part.fCharged), fTagged(part.fTagged), fBadDist(part.fBadDist), 
  fFidArea(part.fFidArea), fInputFileIndex(part.fInputFileIndex)
{
  // Copy constructor
  fMomentum = new TLorentzVector(*part.fMomentum);
  
  fCaloLabel [0] = part.fCaloLabel[0];
  fCaloLabel [1] = part.fCaloLabel[1];
  fTrackLabel[0] = part.fTrackLabel[0];
  fTrackLabel[1] = part.fTrackLabel[1];
  fTrackLabel[2] = part.fTrackLabel[2];
  fTrackLabel[3] = part.fTrackLabel[3];
}

//______________________________________________________________________________
AliAODPWG4Particle& AliAODPWG4Particle::operator=(const AliAODPWG4Particle& part)
{
  // Assignment operator
  if(this!=&part) {
    
    fPdg   = part.fPdg;
    fTag   = part.fTag;
    fBtag  = part.fBtag;
    fLabel = part.fLabel;
	
    fCaloLabel [0] = part.fCaloLabel[0];
    fCaloLabel [1] = part.fCaloLabel[1];
    fTrackLabel[0] = part.fTrackLabel[0];
    fTrackLabel[1] = part.fTrackLabel[1];
	
    fDetector = part.fDetector;
    fDisp     = part.fDisp; 
    fTof      = part.fTof; 
    fCharged  = part.fCharged; 
	fTagged   = part.fTagged;
    fBadDist  = part.fBadDist;
	fFidArea  = part.fFidArea;
	fInputFileIndex =  part.fInputFileIndex;
	  
    if (fMomentum ) delete fMomentum;	
    fMomentum = new TLorentzVector(*part.fMomentum);
  }
  
  return *this;
}


//_______________________________________________________________
Bool_t AliAODPWG4Particle::IsPIDOK(const Int_t ipid, const Int_t pdgwanted) const{
  // returns true if particle satisfies given PID criterium
	switch(ipid){
	case 0: return kTRUE ; //No PID at all
	case 1: 
	  {
	    if (fPdg == pdgwanted) return kTRUE; 
	    else return kFALSE; //Overall PID calculated with bayesian methods.
	  }
	case 2: return fDisp ;   //only dispersion cut
	case 3: return fTof ;    //Only TOF cut
	case 4: return fCharged ;    //Only Charged cut
	case 5: return fDisp && fTof ;  //Dispersion and TOF
	case 6: return fDisp && fCharged ;  //Dispersion and Charged
	case 7: return fTof  && fCharged ;  //TOF and Charged
	case 8: return fDisp && fTof && fCharged ; // all 3 cuts
	default: return kFALSE ; //Not known combination
	}
}

//______________________________________________________________________________
void AliAODPWG4Particle::Print(Option_t* /*option*/) const 
{
  // Print information of all data members
  printf("Particle 4-vector:\n");
  printf("     E  = %13.3f", E() );
  printf("     Px = %13.3f", Px());
  printf("     Py = %13.3f", Py());
  printf("     Pz = %13.3f\n", Pz());
  printf("PID bits :\n");
  printf("     TOF        : %d",fTof);
  printf("     Charged    : %d",fCharged);
  printf("     Dispersion : %d\n",fDisp);
  printf("PDG       : %d\n",fPdg);
  printf("Tag       : %d\n",fTag); 
  printf("Btag      : %d\n",fBtag);  
  printf("Pi0 Tag   : %d\n",fTagged);  
  printf("Dist. to bad channel : %d\n",fBadDist);  
  printf("Fid Area  : %d\n",fFidArea);  
  printf("Input File Index : %d\n",fInputFileIndex);  
  printf("Detector  : %s\n",fDetector.Data());
  
}
Commit	Line	Data
a65a7e70	1	/**************************************************************************
	2	* Copyright(c) 1998-2007, 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
	16	/* $Id: AliAODPWG4Particle.h $ */
	17
	18	//-------------------------------------------------------------------------
	19	// AOD class for photon and other particles storage and
	20	// correlation studies
	21	// Author: Yves Schutz, CERN, Gustavo Conesa, INFN
	22	//-------------------------------------------------------------------------
	23
	24	//-- ROOT system --
	25
	26	//-- Analysis system
	27	#include "AliAODPWG4Particle.h"
	28
	29	ClassImp(AliAODPWG4Particle)
	30
	31
	32	//______________________________________________________________________________
	33	AliAODPWG4Particle::AliAODPWG4Particle() :
	34	AliVParticle(),
	35	fMomentum(0),fPdg(-1), fTag(0), fBtag(-1), fLabel(-1), fCaloLabel(), fTrackLabel(),
	36	fDetector(""), fDisp(0), fTof(0), fCharged(0), fTagged(0), fBadDist(0), fFidArea(0), fInputFileIndex(0)
	37	{
	38	// constructor
	39	fCaloLabel [0] = -1;
	40	fCaloLabel [1] = -1;
	41	fTrackLabel[0] = -1;
	42	fTrackLabel[1] = -1;
	43	fTrackLabel[2] = -1;
	44	fTrackLabel[3] = -1;
	45	}
	46
	47	//______________________________________________________________________________
	48	AliAODPWG4Particle::AliAODPWG4Particle(Double_t px, Double_t py, Double_t pz, Double_t e):
	49	AliVParticle(),
	50	fMomentum(0),fPdg(-1), fTag(0), fBtag(-1), fLabel(-1),fCaloLabel(), fTrackLabel(),
	51	fDetector(""), fDisp(0), fTof(0),fCharged(0), fTagged(0), fBadDist(0), fFidArea(0), fInputFileIndex(0)
	52	{
	53	// constructor
	54	fMomentum = new TLorentzVector(px, py, pz, e);
	55
	56	fCaloLabel [0] = -1;
	57	fCaloLabel [1] = -1;
	58	fTrackLabel[0] = -1;
	59	fTrackLabel[1] = -1;
	60	fTrackLabel[2] = -1;
	61	fTrackLabel[3] = -1;
	62	}
	63
	64	//______________________________________________________________________________
65	AliAODPWG4Particle::AliAODPWG4Particle(TLorentzVector & p):
66	AliVParticle(),
67	fMomentum(0),fPdg(-1), fTag(0), fBtag(-1), fLabel(-1),fCaloLabel(), fTrackLabel(),
68	fDetector(""), fDisp(0), fTof(0), fCharged(0), fTagged(0), fBadDist(0), fFidArea(0), fInputFileIndex(0)
69	{
70	// constructor
71	fMomentum = new TLorentzVector(p);
72
73	fCaloLabel [0] = -1;
74	fCaloLabel [1] = -1;
75	fTrackLabel[0] = -1;
76	fTrackLabel[1] = -1;
77	fTrackLabel[2] = -1;
78	fTrackLabel[3] = -1;
79	}
80
81
82	//______________________________________________________________________________
83	AliAODPWG4Particle::~AliAODPWG4Particle()
84	{
85	// destructor
86	delete fMomentum;
87	}
88
89	//______________________________________________________________________________
90	void AliAODPWG4Particle::Clear(const Option_t* /opt/)
91	{
92	//clear
93	delete fMomentum;
94	}
95
96	//______________________________________________________________________________
97	AliAODPWG4Particle::AliAODPWG4Particle(const AliAODPWG4Particle& part) :
98	AliVParticle(part),
99	fMomentum(0), fPdg(part.fPdg), fTag(part.fTag), fBtag(part.fBtag), fLabel(part.fLabel),
100	fCaloLabel(), fTrackLabel(), fDetector(part.fDetector),fDisp(part.fDisp),
101	fTof(part.fTof), fCharged(part.fCharged), fTagged(part.fTagged), fBadDist(part.fBadDist),
102	fFidArea(part.fFidArea), fInputFileIndex(part.fInputFileIndex)
103	{
104	// Copy constructor
105	fMomentum = new TLorentzVector(*part.fMomentum);
106
107	fCaloLabel [0] = part.fCaloLabel[0];
108	fCaloLabel [1] = part.fCaloLabel[1];
109	fTrackLabel[0] = part.fTrackLabel[0];
110	fTrackLabel[1] = part.fTrackLabel[1];
111	fTrackLabel[2] = part.fTrackLabel[2];
112	fTrackLabel[3] = part.fTrackLabel[3];
113	}
114
115	//______________________________________________________________________________
116	AliAODPWG4Particle& AliAODPWG4Particle::operator=(const AliAODPWG4Particle& part)
117	{
118	// Assignment operator
119	if(this!=&part) {
120
121	fPdg = part.fPdg;
122	fTag = part.fTag;
123	fBtag = part.fBtag;
124	fLabel = part.fLabel;
125
126	fCaloLabel [0] = part.fCaloLabel[0];
127	fCaloLabel [1] = part.fCaloLabel[1];
128	fTrackLabel[0] = part.fTrackLabel[0];
129	fTrackLabel[1] = part.fTrackLabel[1];
130
131	fDetector = part.fDetector;
132	fDisp = part.fDisp;
133	fTof = part.fTof;
134	fCharged = part.fCharged;
135	fTagged = part.fTagged;
136	fBadDist = part.fBadDist;
137	fFidArea = part.fFidArea;
138	fInputFileIndex = part.fInputFileIndex;
139
140	if (fMomentum ) delete fMomentum;
141	fMomentum = new TLorentzVector(*part.fMomentum);
142	}
143
144	return *this;
145	}
146
147
148	//_______________________________________________________________
149	Bool_t AliAODPWG4Particle::IsPIDOK(const Int_t ipid, const Int_t pdgwanted) const{
150	// returns true if particle satisfies given PID criterium
151	switch(ipid){
152	case 0: return kTRUE ; //No PID at all
153	case 1:
154	{
155	if (fPdg == pdgwanted) return kTRUE;
156	else return kFALSE; //Overall PID calculated with bayesian methods.
157	}
158	case 2: return fDisp ; //only dispersion cut
159	case 3: return fTof ; //Only TOF cut
160	case 4: return fCharged ; //Only Charged cut
161	case 5: return fDisp && fTof ; //Dispersion and TOF
162	case 6: return fDisp && fCharged ; //Dispersion and Charged
163	case 7: return fTof && fCharged ; //TOF and Charged
164	case 8: return fDisp && fTof && fCharged ; // all 3 cuts
165	default: return kFALSE ; //Not known combination
166	}
167	}
168
169	//______________________________________________________________________________
170	void AliAODPWG4Particle::Print(Option_t* /option/) const
171	{
172	// Print information of all data members
173	printf("Particle 4-vector:\n");
174	printf(" E = %13.3f", E() );
175	printf(" Px = %13.3f", Px());
176	printf(" Py = %13.3f", Py());
177	printf(" Pz = %13.3f\n", Pz());
178	printf("PID bits :\n");
179	printf(" TOF : %d",fTof);
180	printf(" Charged : %d",fCharged);
181	printf(" Dispersion : %d\n",fDisp);
182	printf("PDG : %d\n",fPdg);
183	printf("Tag : %d\n",fTag);
184	printf("Btag : %d\n",fBtag);
185	printf("Pi0 Tag : %d\n",fTagged);
186	printf("Dist. to bad channel : %d\n",fBadDist);
187	printf("Fid Area : %d\n",fFidArea);
188	printf("Input File Index : %d\n",fInputFileIndex);
189	printf("Detector : %s\n",fDetector.Data());
190
191	}