[u/mrichter/AliRoot.git] / PWGCF / Correlations / JCORRAN / AliJMCTrack.h

/* Copyright(c) 1998-2014, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice */

// Short comment describing what this class does needed!

// $Id: AliJMCTrack.h,v 1.3 2008/05/08 15:19:52 djkim Exp $

////////////////////////////////////////////////////
/*!
  \file AliJMCTrack.h
  \brief
  \author J. Rak, D.J.Kim, R.Diaz (University of Jyvaskyla)
  \email: djkim@jyu.fi
  \version $Revision: 1.3 $
  \date $Date: 2008/05/08 15:19:52 $
  */
////////////////////////////////////////////////////

#ifndef ALIJMCTRACK_H
#define ALIJMCTRACK_H

// include ROOT lib
#ifndef ROOT_TObject
#include <TObject.h>
#endif
#include <TDatabasePDG.h>

// include Inharitance
#include "AliJBaseTrack.h"

class AliJMCTrack : public AliJBaseTrack {

 public:
  enum { kPrimary=AliJBaseTrack::kNFlag, kPHOS, kEMCAL, kTPC, kInjected, kNFlag };//For ALICE
  enum { kFinal=AliJBaseTrack::kNFlag };// for MC
  //usage : this->SetFlag( kPrimary, kTRUE );
  //usage : this->IsTrue( kFinal );

  AliJMCTrack();      //default constructor
  AliJMCTrack(const AliJMCTrack& a);  //copy constructor

  ~AliJMCTrack(){;}    //destructor

  Int_t  GetPdgCode()    const {return fPdgCode;}
  Int_t  GetMother  (Int_t i) const {return fMother[i];}
  Int_t  GetDaughter(Int_t i) const {return fDaughter[i];}
  Double32_t GetVx()    const{return fVx;}
  Double32_t  GetVy()    const{return fVy;}
  Double32_t  GetVz()    const{return fVz;}

  const TParticlePDG& GetPDGData() const ;
  void SetPdgCode(Int_t icode);// Set PDG and E,charge
  void SetMother  (int m0, int m1){ fMother[0] = m0;fMother[1]=m1; }
  void SetDaughter(int d0, int d1){ fDaughter[0]=d0;fDaughter[1]=d1; }
  void SetProductionVertex(Double_t vx, Double_t vy, Double_t vz)
  {fVx=vx; fVy=vy; fVz=vz;}


  AliJMCTrack& operator=(const AliJMCTrack& trk);


  //Extra
  Bool_t IsFinal() const { return IsTrue( kFinal ); }
  void  SetIsFinal(Bool_t t){ SetFlag( kFinal, t );}
  //TODO
  Bool_t IsHadron() const;
  Bool_t IsCharged() const { return GetCharge(); }
  Bool_t IsParton()  const {return ( fPdgCode > -7 && fPdgCode < 7 && fPdgCode !=0 );}

 private:

  Short_t         fPdgCode;              // PDG code of the particle
  Short_t         fMother[2];            // Index of the mother particles
  Short_t         fDaughter[2];          // Indices of the daughter particles

  Double32_t      fVx;                   //[0.,0.,12] x of production vertex
  Double32_t      fVy;                   //[0.,0.,12] y of production vertex
  Double32_t      fVz;                   //[0.,0.,12] z of production vertex

  ClassDef(AliJMCTrack,2)
};

#endif
Commit	Line	Data
9dc4f671	1	/* Copyright(c) 1998-2014, ALICE Experiment at CERN, All rights reserved. *
	2	* See cxx source for full Copyright notice */
	3
	4	// Short comment describing what this class does needed!
	5
37dde34e	6	// $Id: AliJMCTrack.h,v 1.3 2008/05/08 15:19:52 djkim Exp $
	7
	8	////////////////////////////////////////////////////
	9	/*!
	10	\file AliJMCTrack.h
	11	\brief
	12	\author J. Rak, D.J.Kim, R.Diaz (University of Jyvaskyla)
	13	\email: djkim@jyu.fi
	14	\version $Revision: 1.3 $
	15	\date $Date: 2008/05/08 15:19:52 $
	16	*/
	17	////////////////////////////////////////////////////
	18
	19	#ifndef ALIJMCTRACK_H
	20	#define ALIJMCTRACK_H
	21
	22	// include ROOT lib
	23	#ifndef ROOT_TObject
	24	#include <TObject.h>
	25	#endif
	26	#include <TDatabasePDG.h>
	27
	28	// include Inharitance
	29	#include "AliJBaseTrack.h"
	30
	31	class AliJMCTrack : public AliJBaseTrack {
	32
	33	public:
9dc4f671	34	enum { kPrimary=AliJBaseTrack::kNFlag, kPHOS, kEMCAL, kTPC, kInjected, kNFlag };//For ALICE
37dde34e	35	enum { kFinal=AliJBaseTrack::kNFlag };// for MC
	36	//usage : this->SetFlag( kPrimary, kTRUE );
	37	//usage : this->IsTrue( kFinal );
	38
9dc4f671	39	AliJMCTrack(); //default constructor
9dc4f671	40	AliJMCTrack(const AliJMCTrack& a); //copy constructor
37dde34e	41
9dc4f671	42	~AliJMCTrack(){;} //destructor
37dde34e	43
	44	Int_t GetPdgCode() const {return fPdgCode;}
	45	Int_t GetMother (Int_t i) const {return fMother[i];}
	46	Int_t GetDaughter(Int_t i) const {return fDaughter[i];}
	47	Double32_t GetVx() const{return fVx;}
	48	Double32_t GetVy() const{return fVy;}
	49	Double32_t GetVz() const{return fVz;}
	50
	51	const TParticlePDG& GetPDGData() const ;
	52	void SetPdgCode(Int_t icode);// Set PDG and E,charge
	53	void SetMother (int m0, int m1){ fMother[0] = m0;fMother[1]=m1; }
	54	void SetDaughter(int d0, int d1){ fDaughter[0]=d0;fDaughter[1]=d1; }
	55	void SetProductionVertex(Double_t vx, Double_t vy, Double_t vz)
	56	{fVx=vx; fVy=vy; fVz=vz;}
	57
	58
	59	AliJMCTrack& operator=(const AliJMCTrack& trk);
	60
	61
	62	//Extra
	63	Bool_t IsFinal() const { return IsTrue( kFinal ); }
	64	void SetIsFinal(Bool_t t){ SetFlag( kFinal, t );}
	65	//TODO
	66	Bool_t IsHadron() const;
	67	Bool_t IsCharged() const { return GetCharge(); }
9dc4f671	68	Bool_t IsParton() const {return ( fPdgCode > -7 && fPdgCode < 7 && fPdgCode !=0 );}
37dde34e	69
	70	private:
	71
	72	Short_t fPdgCode; // PDG code of the particle
	73	Short_t fMother[2]; // Index of the mother particles
	74	Short_t fDaughter[2]; // Indices of the daughter particles
	75
	76	Double32_t fVx; //[0.,0.,12] x of production vertex
	77	Double32_t fVy; //[0.,0.,12] y of production vertex
	78	Double32_t fVz; //[0.,0.,12] z of production vertex
	79
	80	ClassDef(AliJMCTrack,2)
	81	};
	82
	83	#endif