PndSmpCand.h

Go to the documentation of this file.

#ifndef PndSmpCand_H
#define PndSmpCand_H

#include "TLorentzVector.h"
#include <iostream>

using std::cout;
using std::endl;

// -------------------------------------------------------------------------

class PndSmpCand {
 public:
  // Constructors
  PndSmpCand(){};

  PndSmpCand(TLorentzVector lv, Float_t chrg, Int_t pdg = 0, Int_t uid = -1);

  PndSmpCand(TLorentzVector *lv, Float_t chrg, Int_t pdg = 0, Int_t uid = -1);

  PndSmpCand(Int_t pdg, PndSmpCand &c0, PndSmpCand &c1) { CombCand(pdg, &c0, &c1); }
  PndSmpCand(Int_t pdg, PndSmpCand &c0, PndSmpCand &c1, PndSmpCand &c2) { CombCand(pdg, &c0, &c1, &c2); }
  PndSmpCand(Int_t pdg, PndSmpCand &c0, PndSmpCand &c1, PndSmpCand &c2, PndSmpCand &c3) { CombCand(pdg, &c0, &c1, &c2, &c3); }
  PndSmpCand(Int_t pdg, PndSmpCand &c0, PndSmpCand &c1, PndSmpCand &c2, PndSmpCand &c3, PndSmpCand &c4) { CombCand(pdg, &c0, &c1, &c2, &c3, &c4); }

  PndSmpCand(Int_t pdg, PndSmpCand *c0, PndSmpCand *c1, PndSmpCand *c2 = nullptr, PndSmpCand *c3 = nullptr, PndSmpCand *c4 = nullptr) { CombCand(pdg, c0, c1, c2, c3, c4); }

  // Modifiers
  void SetP4(TLorentzVector p4) { fLV = p4; }
  void SetM(Double_t m) { fLV.SetVectM(fLV.Vect(), m); }
  void SetCharge(Float_t ch) { fCharge = ch; }

  // Methods
  Bool_t operator==(const PndSmpCand &c1) const { return (c1.fMarker == fMarker) && (c1.fPdgCode == fPdgCode); }
  Bool_t operator!=(const PndSmpCand &c1) const { return !(*this == c1); }

  void CombCand(Int_t pdg, PndSmpCand *c0, PndSmpCand *c1, PndSmpCand *c2 = nullptr, PndSmpCand *c3 = nullptr, PndSmpCand *c4 = nullptr);

  TLorentzVector P4() { return fLV; }
  Float_t Charge() { return fCharge; }
  Int_t Pdg() { return fPdgCode; }
  Int_t NDau() { return fNDau; }
  ULong_t Marker() { return fMarker; }
  PndSmpCand *Dau(Int_t idx)
  {
    if (idx >= 0 && idx < fNDau)
      return fDaughter[idx];
    else
      return nullptr;
  }
  Bool_t Overlap(PndSmpCand *c) { return fMarker & c->fMarker; }
  Bool_t Overlap(PndSmpCand &c) { return fMarker & c.fMarker; }
  void Print();

 private:
  // Private methods
  void PrintBinary(ULong_t x)
  {
    ULong_t bit = ULong_t(1) << 63;
    while (bit) {
      cout << ((x & bit) ? 1 : 0);
      bit >>= 1;
      if (bit & 0x8080808080808080)
        cout << " ";
    }
  }

  // Member variables
  TLorentzVector fLV;       // the Lorentz vector
  Float_t fCharge;          // charge
  Int_t fPdgCode;           // pdg code
  Int_t fUniqueID;          // unique id, only applied for initial (non-composite) candidates
  ULong_t fMarker;          // a bit marker used for overlap check; limits number of initial candidate to 64 (long = 64 bits)
  Int_t fNDau;              // number of daughters of composites (up to 5)
  PndSmpCand *fDaughter[5]; // pointers to daughters
};

// -------------------------------------------------------------------------

#endif