Subversion Repositories f9daq

#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, PHI, 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 GetTransverseMomentum() const { return sqrt(m_px*m_px+m_py*m_py); };

  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 , int same, float masslow, float massup, SIMPLEPID pid, TClonesArray *pout);

#endif