[u/mrichter/AliRoot.git] / PWG4 / JCORRAN / AliJPhoton.h

// $Id: AliJPhoton.h,v 1.1 2008/05/02 20:28:14 djkim Exp $

////////////////////////////////////////////////////
/*!
  \file AliJPhoton.h
  \author J. Rak, D.J.Kim, R.Diaz (University of Jyvaskyla)
  \email: djkim@jyu.fi
  \version $Revision: 1.1 $
  \date $Date: 2008/05/02 11:56:39 $
*/
////////////////////////////////////////////////////

#ifndef ALIJPHOTON_H
#define ALIJPHOTON_H

/*
#ifndef ROOT_TObject
#include <TObject.h>
#endif
*/
#include <iostream>
#include <fstream>
#include <stdlib.h>
#include <stdio.h>
#include "TMath.h"

#include "JConst.h"
//#include "AliPhJBaseTrack.h"
//#include "AliPhJPhoton.h"

class  TObject;
class  AliPhJBaseTrack;
class  AliPhJPhoton;

class AliJPhoton : public AliPhJPhoton {

public:
  
  enum AliJCaloType {
         kUndef = -1, 
		 kPHOSCalo,  
		 kEMCALCalo
  };
  
  enum AliJCaloPID{
    kElectronAli = 0,
    kMuonAli = 1,
    kPionAli = 2,
    kKaonAli = 3,
    kProtonAli = 4,
    kPhotonAli = 5,
    kPi0Ali = 6,
    kNeutronAli = 7,
    kKaon0Ali = 8,
    kEleConAli = 9,
    kUnknownAli = 10
  };


  AliJPhoton();	    	     // default constructor
  AliJPhoton(const AliJPhoton& a);
  ~AliJPhoton(){   //destructor  
    if(fCellsAbsId)       delete [] fCellsAbsId; 
    if(fCellsAmpFraction) delete [] fCellsAmpFraction;
  }		 
  
  Bool_t InPHOS()  const {return fCaloType==kPHOSCalo ? kTRUE:kFALSE ;}
  Bool_t InEMCAL() const {return fCaloType==kEMCALCalo ? kTRUE:kFALSE ;}
  // getters
  void GetPID(Double_t *pid) const {
    for(Int_t i=0; i<kUnknownAli+1; ++i) pid[i]=fCaloPID[i];
  }
  Double_t  GetDistToBadChannel() const {return fDistToBadChannel;}
  Double_t  GetDispersion()       const {return fDispersion;}
  Double_t  GetM20()              const {return fM20;}
  Double_t  GetM02()              const {return fM02;}
  Double_t  GetEmcCpvDist()       const {return fEmcCpvDist;}
  Double_t  GetNCells() const   { return fNCells;}
  UShort_t  *GetCellsAbsId() const {return  fCellsAbsId;}
  Int_t     GetCellAbsId(Int_t i) const {  
         if (fCellsAbsId && i >=0 && i < fNCells ) return fCellsAbsId[i];    
        else return -1;}
  Double32_t *GetCellsAmplitudeFraction() const {return  fCellsAmpFraction;}
  Double_t    GetCellAmplitudeFraction(Int_t i) const {  
              if (fCellsAmpFraction && i >=0 && i < fNCells ) return fCellsAmpFraction[i];    
              else return -1;}
  particleType GetParticleType();
 
  
  //setters
  void  SetCaloType(AliJCaloType calo) {fCaloType = calo;}
  void  SetDistToBadChannel(Double_t dist) {fDistToBadChannel = dist;}
  void  SetDispersion(Double_t disp) {fDispersion = disp;}
  void  SetM20(Double_t m20) {fM20 = m20;}
  void  SetM02(Double_t m02) {fM02 = m02;}
  void  SetEmcCpvDist(Double_t dist) {fEmcCpvDist = dist;} 
  void  SetPID(const Double_t *pid);
  void  SetNCells(Int_t n) { fNCells = n;}
  void  SetCellsAbsId(const UShort_t *array);
  void  SetCellsAmplitudeFraction(const Double32_t *array) ;

  Int_t    GetMCid() const { return fMCid; }
  void     SetMCid(Int_t id) { fMCid = id; }
  Int_t    GetSuperModuleID() const { return fSuperModuleId; }
  void     SetSuperModuleID(Int_t id) { fSuperModuleId = id; }

  
  AliJPhoton&  operator=(const  AliJPhoton& photon); 
  
private:

  AliJCaloType   fCaloType;              // PHOS or EMCAL photon
  Double_t       fCaloPID[kUnknownAli+1];          // [0.,1.,8] pointer to PID object
  Double_t       fDistToBadChannel;      // Distance to nearest bad channel
  Double_t       fDispersion;            // cluster dispersion, for shape analysis
  Double_t       fM20;                   // 2-nd moment along the main eigen axis
  Double_t       fM02;                   // 2-nd moment along the second eigen axis
  Double_t       fEmcCpvDist;            // the distance from PHOS EMC rec.point to the closest CPV rec.point
 
  Int_t          fNCells ;                 //number of cells
  Int_t          fSuperModuleId ;          //super module id
  UShort_t      *fCellsAbsId;            //[fNCells] array of cell absId numbers
  Double32_t    *fCellsAmpFraction;    //[fNCells][0.,1.,16] array with cell amplitudes fraction.

  Int_t      fMCid;                 // track label, points back to MC track

  
 ClassDef(AliJPhoton,1)
 
};

#endif
Commit	Line	Data
2f4cac02	1	// $Id: AliJPhoton.h,v 1.1 2008/05/02 20:28:14 djkim Exp $
	2
	3	////////////////////////////////////////////////////
	4	/*!
	5	\file AliJPhoton.h
	6	\author J. Rak, D.J.Kim, R.Diaz (University of Jyvaskyla)
	7	\email: djkim@jyu.fi
	8	\version $Revision: 1.1 $
	9	\date $Date: 2008/05/02 11:56:39 $
	10	*/
	11	////////////////////////////////////////////////////
	12
	13	#ifndef ALIJPHOTON_H
	14	#define ALIJPHOTON_H
	15
7d365db4	16	/*
2f4cac02	17	#ifndef ROOT_TObject
	18	#include <TObject.h>
	19	#endif
7d365db4	20	*/
2f4cac02	21	#include <iostream>
	22	#include <fstream>
	23	#include <stdlib.h>
	24	#include <stdio.h>
	25	#include "TMath.h"
	26
	27	#include "JConst.h"
7d365db4	28	//#include "AliPhJBaseTrack.h"
7d365db4	29	//#include "AliPhJPhoton.h"
2f4cac02	30
7d365db4	31	class TObject;
	32	class AliPhJBaseTrack;
	33	class AliPhJPhoton;
2f4cac02	34
	35	class AliJPhoton : public AliPhJPhoton {
	36
	37	public:
	38
	39	enum AliJCaloType {
	40	kUndef = -1,
	41	kPHOSCalo,
	42	kEMCALCalo
	43	};
	44
	45	enum AliJCaloPID{
	46	kElectronAli = 0,
	47	kMuonAli = 1,
	48	kPionAli = 2,
	49	kKaonAli = 3,
	50	kProtonAli = 4,
	51	kPhotonAli = 5,
	52	kPi0Ali = 6,
	53	kNeutronAli = 7,
	54	kKaon0Ali = 8,
	55	kEleConAli = 9,
	56	kUnknownAli = 10
	57	};
	58
	59
	60	AliJPhoton(); // default constructor
	61	AliJPhoton(const AliJPhoton& a);
7d365db4	62	~AliJPhoton(){ //destructor
	63	if(fCellsAbsId) delete [] fCellsAbsId;
	64	if(fCellsAmpFraction) delete [] fCellsAmpFraction;
	65	}
2f4cac02	66
	67	Bool_t InPHOS() const {return fCaloType==kPHOSCalo ? kTRUE:kFALSE ;}
	68	Bool_t InEMCAL() const {return fCaloType==kEMCALCalo ? kTRUE:kFALSE ;}
	69	// getters
	70	void GetPID(Double_t *pid) const {
	71	for(Int_t i=0; i<kUnknownAli+1; ++i) pid[i]=fCaloPID[i];
	72	}
	73	Double_t GetDistToBadChannel() const {return fDistToBadChannel;}
	74	Double_t GetDispersion() const {return fDispersion;}
	75	Double_t GetM20() const {return fM20;}
	76	Double_t GetM02() const {return fM02;}
	77	Double_t GetEmcCpvDist() const {return fEmcCpvDist;}
	78	Double_t GetNCells() const { return fNCells;}
	79	UShort_t *GetCellsAbsId() const {return fCellsAbsId;}
	80	Int_t GetCellAbsId(Int_t i) const {
	81	if (fCellsAbsId && i >=0 && i < fNCells ) return fCellsAbsId[i];
	82	else return -1;}
	83	Double32_t *GetCellsAmplitudeFraction() const {return fCellsAmpFraction;}
	84	Double_t GetCellAmplitudeFraction(Int_t i) const {
	85	if (fCellsAmpFraction && i >=0 && i < fNCells ) return fCellsAmpFraction[i];
	86	else return -1;}
	87	particleType GetParticleType();
	88
	89
	90	//setters
	91	void SetCaloType(AliJCaloType calo) {fCaloType = calo;}
	92	void SetDistToBadChannel(Double_t dist) {fDistToBadChannel = dist;}
	93	void SetDispersion(Double_t disp) {fDispersion = disp;}
	94	void SetM20(Double_t m20) {fM20 = m20;}
	95	void SetM02(Double_t m02) {fM02 = m02;}
	96	void SetEmcCpvDist(Double_t dist) {fEmcCpvDist = dist;}
	97	void SetPID(const Double_t *pid);
	98	void SetNCells(Int_t n) { fNCells = n;}
	99	void SetCellsAbsId(const UShort_t *array);
	100	void SetCellsAmplitudeFraction(const Double32_t *array) ;
	101
	102	Int_t GetMCid() const { return fMCid; }
	103	void SetMCid(Int_t id) { fMCid = id; }
	104	Int_t GetSuperModuleID() const { return fSuperModuleId; }
	105	void SetSuperModuleID(Int_t id) { fSuperModuleId = id; }
	106
	107
	108	AliJPhoton& operator=(const AliJPhoton& photon);
	109
	110	private:
	111
	112	AliJCaloType fCaloType; // PHOS or EMCAL photon
	113	Double_t fCaloPID[kUnknownAli+1]; // [0.,1.,8] pointer to PID object
	114	Double_t fDistToBadChannel; // Distance to nearest bad channel
	115	Double_t fDispersion; // cluster dispersion, for shape analysis
	116	Double_t fM20; // 2-nd moment along the main eigen axis
	117	Double_t fM02; // 2-nd moment along the second eigen axis
	118	Double_t fEmcCpvDist; // the distance from PHOS EMC rec.point to the closest CPV rec.point
	119
	120	Int_t fNCells ; //number of cells
	121	Int_t fSuperModuleId ; //super module id
	122	UShort_t *fCellsAbsId; //[fNCells] array of cell absId numbers
	123	Double32_t *fCellsAmpFraction; //[fNCells][0.,1.,16] array with cell amplitudes fraction.
	124
	125	Int_t fMCid; // track label, points back to MC track
	126
	127
	128
	129	ClassDef(AliJPhoton,1)
130
131	};
132
133	#endif
134