DecayFinder.h

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file DecayFinder.h

#ifndef DECAYFINDER_DECAYFINDER_H
#define DECAYFINDER_DECAYFINDER_H

//sPHENIX stuff
#include <fun4all/SubsysReco.h>

#include <cstddef>              // for NULL
#include <string>
#include <vector>
#include <utility>               // for pair

class DecayFinderContainer_v1;
class PHCompositeNode;
class PHHepMCGenEvent;
class PHHepMCGenEventMap;
namespace HepMC { class GenParticle; }

class DecayFinder : public SubsysReco
{
 public:
  typedef std::vector<std::pair<int, int>> Decay;

  DecayFinder();

  explicit DecayFinder(const std::string &name);

  virtual ~DecayFinder() {}

  int Init(PHCompositeNode *topNode);

  int process_event(PHCompositeNode *topNode);

  int End(PHCompositeNode *topNode);

  int parseDecayDescriptor();

  bool findDecay(PHCompositeNode *topNode);

  bool findParticle(std::string particle);

  int checkIfCorrectParticle(HepMC::GenParticle *particle, bool &trackFailedPT, bool &trackFailedETA);

  int deleteElement(int arr[], int n, int x);

  void multiplyVectorByScalarAndSort(std::vector<int> &v, int k);

  int get_pdgcode(std::string name);

  int get_charge(std::string name);

  int createDecayNode(PHCompositeNode *topNode);

  void fillDecayNode(PHCompositeNode* topNode, Decay decay);

  void printInfo();

  void printNode(PHCompositeNode *topNode);

  //User configuration
  void setDecayDescriptor(const std::string &decayDescriptor) { m_decayDescriptor = decayDescriptor; }
  void triggerOnDecay(bool trigger) { m_triggerOnDecay = trigger; }
  void allowPhotons(bool allow) { m_allowPhotons = allow; }
  void allowPi0(bool allow) { m_allowPi0 = allow; }
  void saveDST(bool save) { m_save_dst = save; }
  void setNodeName(const std::string &name) { m_container_name = name; }

 private:
  PHHepMCGenEventMap *m_geneventmap = nullptr;
  PHHepMCGenEvent *m_genevt = nullptr;

  int m_counter = 0;
  int m_nCandFail_pT = 0;
  int m_nCandFail_eta = 0;
  int m_nCandFail_pT_and_eta = 0;
  int m_nCandReconstructable = 0;

  bool m_getChargeConjugate = false;

  std::string m_decayDescriptor;
  bool m_triggerOnDecay = false;
  bool m_allowPi0 = false;
  bool m_allowPhotons = false;

  int m_mother_ID = 0;
  std::vector<int> m_intermediates_ID;
  std::vector<int> m_daughters_ID;

  int m_nTracksFromMother = 0;
  std::vector<int> m_nTracksFromIntermediates;

  std::vector<int> m_motherDecayProducts;

  bool m_save_dst;
  DecayFinderContainer_v1 *m_decayMap = nullptr;
  Decay decayChain;
  std::string m_nodeName;
  std::string m_container_name;
};

#endif  //DECAYFINDER_DECAYFINDER_H