KFParticle_eventReconstruction.h

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

#ifndef KFPARTICLESPHENIX_KFPARTICLEEVENTRECONSTRUCTION_H
#define KFPARTICLESPHENIX_KFPARTICLEEVENTRECONSTRUCTION_H

#include "KFParticle_Tools.h"

#include <KFParticle.h>

#include <vector>

class PHCompositeNode;

class KFParticle_eventReconstruction : public KFParticle_Tools
{
 public:
  KFParticle_eventReconstruction();

  ~KFParticle_eventReconstruction() override = default;

  void createDecay(PHCompositeNode* topNode, std::vector<KFParticle>& selectedMother, std::vector<KFParticle>& selectedVertex,
                   std::vector<std::vector<KFParticle>>& selectedDaughters,
                   std::vector<std::vector<KFParticle>>& selectedIntermediates,
                   int& nPVs, int& multiplicity);

  void buildBasicChain(std::vector<KFParticle>& selectedMotherBasic,
                       std::vector<KFParticle>& selectedVertexBasic,
                       std::vector<std::vector<KFParticle>>& selectedDaughtersBasic,
                       const std::vector<KFParticle>& daughterParticlesBasic,
                       const std::vector<int>& goodTrackIndexBasic,
                       const std::vector<KFParticle>& primaryVerticesBasic);

  void buildChain(std::vector<KFParticle>& selectedMotherAdv,
                  std::vector<KFParticle>& selectedVertexAdv,
                  std::vector<std::vector<KFParticle>>& selectedDaughtersAdv,
                  std::vector<std::vector<KFParticle>>& selectedIntermediatesAdv,
                  const std::vector<KFParticle>& daughterParticlesAdv,
                  const std::vector<int>& goodTrackIndexAdv,
                  const std::vector<KFParticle>& primaryVerticesAdv);

  void getCandidateDecay(std::vector<KFParticle>& selectedMotherCand,
                         std::vector<KFParticle>& selectedVertexCand,
                         std::vector<std::vector<KFParticle>>& selectedDaughtersCand,
                         std::vector<KFParticle> daughterParticlesCand,
                         std::vector<std::vector<int>> goodTracksThatMeetCand,
                         std::vector<KFParticle> primaryVerticesCand,
                         int n_track_start, int n_track_stop,
                         bool isIntermediate, int intermediateNumber, bool constrainMass);

  int selectBestCombination(bool PVconstraint, bool isAnInterMother,
                            std::vector<KFParticle> possibleCandidates,
                            std::vector<KFParticle> possibleVertex);

  KFParticle createFakePV();

 protected:
  bool m_constrain_to_vertex;
  bool m_constrain_int_mass;
  bool m_use_fake_pv;

  // private:
};

#endif  // KFPARTICLESPHENIX_KFPARTICLEEVENTRECONSTRUCTION_H