#ifndef BPARTICLE_H
#define BPARTICLE_H
//+
// File : BParticle.h
// Description : class to contain particle info.
//
// Author : Ryosuke Itoh, IPNS, KEK
// Date : 28 - Jan - 2004
//-
#include "TObject.h"
#include "TClonesArray.h"
enum SIMPLEPID {PHOTON, ELECTRON, PION, MUON, KAON, PROTON, JPSI, D, DSTAR, B,KS, ALL };
class BParticle : public TObject {
private:
float m_px;
float m_py;
float m_pz;
float m_e;
float m_charge;
SIMPLEPID m_pid;
public:
BParticle(){};
BParticle ( float px, float py, float pz, float e, float c, SIMPLEPID pid );
~BParticle(){};
float px() const { return m_px; };
float py() const { return m_py; };
float pz() const { return m_pz; };
float e() const { return m_e; };
float GetMomentum() const { return sqrt(m_px*m_px+m_py*m_py+m_pz*m_pz); };
float charge() const { return m_charge; };
SIMPLEPID pid() const { return m_pid; };
float GetMass(SIMPLEPID pid);
float GetMass();
void SetEnergyFromMass (float mass);
void SetEnergyFromPid ();
void SetPid(SIMPLEPID pid){ m_pid = pid; };
int InMassRange(float mlower,float mupper){ float m = GetMass(); if (m>=mlower && m<=mupper) return 1; else return 0; };
BParticle operator+(const BParticle& b){
BParticle particle(
px() + b.px(),
py() + b.py(),
pz() + b.pz(),
e() + b.e(),
charge() + b.charge(),
PHOTON // wrong
);
return particle;
};
BParticle & operator=(const BParticle &p)
{
//if ( this != &p){
m_px = p.px();
m_py = p.py();
m_pz = p.pz();
m_e = p.e();
m_charge = p.charge();
m_pid = p.pid();
//}
return *this;
};
ClassDef ( BParticle, 1 ) // Simple particle class
};
int SelectParticles(TClonesArray *pin , int charge, SIMPLEPID type, TClonesArray *pout);
int CombineParticles(TClonesArray *plist1 ,TClonesArray *plist2 , float masslow, float massup, SIMPLEPID pid, TClonesArray *pout);
#endif