KFParticle_DST.h

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

#ifndef KFPARTICLESPHENIX_KFPARTICLEDST_H
#define KFPARTICLESPHENIX_KFPARTICLEDST_H

#include <KFParticle.h>

#include <string>
#include <vector>

class KFParticle_Container;
class PHCompositeNode;
class SvtxTrack;
class SvtxTrackMap;

class KFParticle_DST
{
 public:
  KFParticle_DST() = default;

  virtual ~KFParticle_DST() = default;

  int createParticleNode(PHCompositeNode* topNode);

  void fillParticleNode(PHCompositeNode* topNode, const KFParticle& motherParticle,
                        const std::vector<KFParticle>& daughters,
                        const std::vector<KFParticle>& intermediates);

  void fillParticleNode_Track(PHCompositeNode* topNode, const KFParticle& motherParticle,
                              std::vector<KFParticle> daughters,
                              std::vector<KFParticle> intermediates);

  void fillParticleNode_Particle(PHCompositeNode* topNode, const KFParticle& motherParticle,
                                 std::vector<KFParticle> daughters,
                                 std::vector<KFParticle> intermediates);

  void printNode(PHCompositeNode* topNode);

 protected:
  bool m_has_intermediates_DST = false;
  bool m_write_track_container = true;
  bool m_write_particle_container = true;
  std::string m_container_name;

 private:
  SvtxTrackMap* m_recoTrackMap = nullptr;
  KFParticle_Container* m_recoParticleMap = nullptr;
  SvtxTrack* buildSvtxTrack(const KFParticle& particle);
};

#endif  // KFPARTICLESPHENIX_KFPARTICLEDST_H