source: huonglan/ANA2011/Particle.C

#include "Particle.h"
#include <cstdio>
#include <math.h>

using namespace std;

Particle::Particle()
{
  TVector3 v3(0,0,0);
  TLorentzVector v4(v3,0);
  this-> m_v4 = v4;
  this-> m_pdg = 0;
  this-> m_stat = 0;
  this-> m_pol = 0;
  this-> m_jetC.clear();
  this-> m_lepC.clear();
}

Particle::Particle(TLorentzVector vec4, int pdg, short stat, short pol)
{
  this->m_v4 = vec4;
  this->m_pdg = pdg;
  this->m_stat = stat;
  this->m_pol = pol;
  this->m_bcode = 0;
  this->m_mbcode = 0;
  this-> m_jetC.clear();
  this-> m_lepC.clear();
}

int Particle::PDG() const
{
  return this->m_pdg;
}

short Particle::STAT() const
{
  return this->m_stat;
}

short Particle::POL() const
{
  return this->m_pol;
}

void Particle::barcode(int b)
{
  this->m_bcode = b;
}

int Particle::barcode() const
{
  return m_bcode;
}

void Particle::motherbarcode(int mb)
{
  this->m_mbcode = mb;
}

int Particle::motherbarcode() const
{
  return m_mbcode;
}

TLorentzVector Particle::vector4() const
{
  return this->m_v4;
}

const std::set<short>& Particle::jetC()
{
  return m_jetC;
}

const std::set<short>& Particle::lepC()
{
  return m_lepC;
}

void Particle::jetC(std::set<short> s)
{
  m_jetC = s;
}

void Particle::lepC(std::set<short> s)
{
  m_lepC = s;
}

void Particle::child_index(std::vector<int> vchild_index)
{
  for (unsigned int i = 0; i < vchild_index.size(); i++){
    m_child_index.push_back(vchild_index.at(i));
  }
}

Particle& Particle::operator = (Particle& par)
{
  TLorentzVector vec = par.vector4();
  int pdg = par.PDG();
  short stat = par.STAT();
  short pol = par.POL();
  vector<int> childindex = par.child_index();

  this->m_v4 = vec;
  this->m_pdg = pdg;
  this->m_stat = stat;
  this->m_pol = pol;
  
  for (unsigned int i = 0; i < childindex.size(); i++){
    this->m_child_index.push_back(childindex.at(i));
  }

  return *this;
}
//********************* Print operator ************************
/*std::ostream& operator<<(std::ostream& out, const Particle& par)
{
  out << " " << par.barcode() << " " << par.motherbarcode()
      << " " << par.PDG() 
      << " " << par.STAT()
      << " " <<  par.vector4()
      << std::endl;

 return out;
 }*/