PfoCreator.h

#ifndef PFO_CREATOR_H
#define PFO_CREATOR_H 1

#include "ClusterShapes.h"
#include "Api/PandoraApi.h"

namespace IMPL { class ClusterImpl; class ReconstructedParticleImpl; }
namespace EVENT { class LCEvent; }

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

class PfoCreator
{
public:
    class Settings
    {
    public:
        Settings();

        std::string     m_clusterCollectionName;                
        std::string     m_pfoCollectionName;                    
        std::string     m_startVertexCollectionName;            
        std::string     m_startVertexAlgName;                   
        float           m_emStochasticTerm;                     
        float           m_hadStochasticTerm;                    
        float           m_emConstantTerm;                       
        float           m_hadConstantTerm;                      
    };

     PfoCreator(const Settings &settings, const pandora::Pandora *const pPandora);

     ~PfoCreator();

    pandora::StatusCode CreateParticleFlowObjects(EVENT::LCEvent *pLCEvent);

private:
    enum Index
    {
        ECAL_INDEX = 0,
        HCAL_INDEX = 1,
        YOKE_INDEX = 2,
        LCAL_INDEX = 3,
        LHCAL_INDEX = 4,
        BCAL_INDEX = 5
    };

    void InitialiseSubDetectorNames(pandora::StringVector &subDetectorNames) const;

    void SetClusterSubDetectorEnergies(const pandora::StringVector &subDetectorNames, IMPL::ClusterImpl *const pLcioCluster,
        const pandora::CaloHitList &pandoraCaloHitList, pandora::FloatVector &hitE, pandora::FloatVector &hitX, pandora::FloatVector &hitY,
        pandora::FloatVector &hitZ) const;

    void SetClusterEnergyAndError(const pandora::ParticleFlowObject *const pPandoraPfo, const pandora::Cluster *const pPandoraCluster, 
        IMPL::ClusterImpl *const pLcioCluster, float &clusterCorrectEnergy) const;

    void SetClusterPositionAndError(const unsigned int nHitsInCluster, pandora::FloatVector &hitE, pandora::FloatVector &hitX, 
        pandora::FloatVector &hitY, pandora::FloatVector &hitZ, IMPL::ClusterImpl *const pLcioCluster, pandora::CartesianVector &clusterPositionVec) const;

    pandora::StatusCode CalculateTrackBasedReferencePoint(const pandora::ParticleFlowObject *const pPandoraPfo, pandora::CartesianVector &referencePoint) const;

    void SetRecoParticleReferencePoint(const pandora::CartesianVector &referencePoint, IMPL::ReconstructedParticleImpl *const pReconstructedParticle) const;

    void AddTracksToRecoParticle(const pandora::ParticleFlowObject *const pPandoraPfo, IMPL::ReconstructedParticleImpl *const pReconstructedParticle) const;

    void SetRecoParticlePropertiesFromPFO(const pandora::ParticleFlowObject *const pPandoraPfo, IMPL::ReconstructedParticleImpl *const pReconstructedParticle) const;

    bool IsValidParentTrack(const pandora::Track *const pPandoraTrack, const pandora::TrackList &allTrackList) const;

    bool HasValidSiblingTrack(const pandora::Track *const pPandoraTrack, const pandora::TrackList &allTrackList) const;

    bool IsClosestTrackToIP(const pandora::Track *const pPandoraTrack, const pandora::TrackList &allTrackList) const;

    bool AreAnyOtherSiblingsInList(const pandora::Track *const pPandoraTrack, const pandora::TrackList &allTrackList) const;

    const Settings              m_settings;                         
    const pandora::Pandora      *m_pPandora;                        
};

#endif // #ifndef PFO_CREATOR_H