LArEventTopology.h

Go to the documentation of this file.

#ifndef LAR_EVENT_TOPOLOGY_H
#define LAR_EVENT_TOPOLOGY_H 1

#include "Objects/MCParticle.h"

#include <map>

namespace lar_content
{

class LArEventTopology
{
private:
    typedef std::map<const pandora::MCParticle *, pandora::CaloHitList> MCHitMap;

    class Particle
    {
    public:
        Particle(const pandora::MCParticle *const pRoot);

        Particle(const pandora::MCParticle *const pRoot, const pandora::CaloHitList &caloHitList);

        virtual ~Particle();

        void AddChild(Particle *pChild);

        void GetVertices(pandora::CartesianPointVector &vertices) const;

        void Parse(const MCHitMap &mcHitMap);

        void Prune();

        const std::string Print(const MCHitMap &mcHitMap, const std::string &indent) const;

    private:
        pandora::MCParticleList m_particles;
        pandora::CaloHitList m_caloHits;
        std::list<Particle *> m_children;
        bool m_fold;
    };

public:
    LArEventTopology(const pandora::CaloHitList &caloHitList2D);

    virtual ~LArEventTopology();

    void ConstructVisibleHierarchy();

    void GetVertices(pandora::CartesianPointVector &vertices) const;

    void PruneHierarchy();

    void Print() const;

private:
    void ConstructVisibleHierarchy(Particle *pParticle, const pandora::MCParticle *const pRootMC);

    MCHitMap m_mcHitMap;
    const pandora::MCParticle *m_pRoot;
    Particle *m_pParticle;
};

} // namespace lar_content

#endif // #ifndef LAR_EVENT_TOPOLOGY_H