Subversion Repositories f9daq

//+
// File : BParticle.cc
// Description : Implementation of BTrack class //
//-

#include "BParticle.h"

ClassImp (BParticle)


 BParticle::BParticle ( float px, float py, float pz, float e,
                       float charge, SIMPLEPID pid ) {
   m_px = px;
   m_py = py;
   m_pz = pz;
   m_e = e;
   m_charge = charge;
   m_pid = pid;
 }


float BParticle::GetMass(SIMPLEPID pid){

switch (pid){
        case PHOTON:   return 0;
        case ELECTRON: return 0.51;
        case PION:     return 0.139;
        case MUON:     return 0.105;
        case KAON:     return 0.497;
        case PROTON:   return 0.938;
        case JPSI:     return 3.1;
        case D:        return 1.86;
        case DSTAR:    return 2.01;
        case B:        return 5.27;
        case PHI:      return 1.02;
        default: return 0;
}      

}

 float BParticle::GetMass ()
 {
   float m2 = m_e * m_e-m_px * m_px-m_py * m_py-m_pz * m_pz;
   if (m2<0) m2=0;
   return sqrt(m2);
 }  
 
 void BParticle::SetEnergyFromMass (float mass)
 {
   m_e = sqrt( mass * mass + m_px * m_px +m_py * m_py +m_pz * m_pz );
   
 }
 
void BParticle::SetEnergyFromPid(){
        SetEnergyFromMass(GetMass(m_pid));
}

int SelectParticles(TClonesArray *pin , int charge, SIMPLEPID type, TClonesArray *pout){
  pout->Clear();
  int nprt=0;
 
  for(TIter next(pin); BParticle * p =(BParticle *) next();) {
        if (p->charge()== charge && p->pid()== type ) {
          TClonesArray& list = *pout;
          new (list[nprt++]) BParticle ( *p );
        }
  }      
  return nprt;  
}

int CombineParticles(TClonesArray *plist1 ,TClonesArray *plist2 , int same, float masslow, float massup, SIMPLEPID pid, TClonesArray *pout){
 // Loop over all the particles in both lists.
 pout->Clear();
 int nprt=0;
 
 for(TIter next1(plist1);BParticle * p1 =(BParticle *) next1();) {
    // the second loop
   for( TIter next2 = (plist1!=plist2 && same==0) ?  TIter(plist2): TIter(next1) ; BParticle * p2 =(BParticle *) next2();) {  
      BParticle  p = *p1 + *p2; // Combine two particles into a new particle  
      if (p.InMassRange(masslow, massup)){
            p.SetPid(pid);
                TClonesArray& list = *pout;
            new (list[nprt++]) BParticle ( p ); // create a new entry in kslist list of particles
      }
       
   }
               
 }
 return nprt;
}