MCManager.h

Go to the documentation of this file.

 
#ifndef PANDORA_MC_MANAGER_H
#define PANDORA_MC_MANAGER_H 1
 
#include "Managers/InputObjectManager.h"
 
#include "Pandora/ObjectCreation.h"
#include "Pandora/PandoraInternal.h"
 
namespace pandora
{
 
class MCManager : public InputObjectManager<MCParticle>
{
public:
    MCManager(const Pandora *const pPandora);
 
    ~MCManager();
 
private:
    StatusCode Create(const object_creation::MCParticle::Parameters &parameters, const MCParticle *&pMCParticle,
        const ObjectFactory<object_creation::MCParticle::Parameters, object_creation::MCParticle::Object> &factory);
 
    StatusCode EraseAllContent();
 
    StatusCode SetMCParentDaughterRelationship(const Uid parentUid, const Uid daughterUid);
 
    StatusCode SetCaloHitToMCParticleRelationship(const Uid caloHitUid, const Uid mcParticleUid, const float mcParticleWeight);
 
    StatusCode SetTrackToMCParticleRelationship(const Uid trackUid, const Uid mcParticleUid, const float mcParticleWeight);
 
    StatusCode IdentifyPfoTargets();
 
    StatusCode SelectPfoTargets();
 
    StatusCode ApplyPfoSelectionRules(const MCParticle *const mcRootParticle, MCParticleSet &mcPfoSet) const;
 
    StatusCode SetPfoTargetInTree(const MCParticle *const pMCParticle, const MCParticle *const pPfoTarget, bool onlyDaughters = false) const;
 
    StatusCode CreateCaloHitToPfoTargetsMap(UidToMCParticleWeightMap &caloHitToPfoTargetsMap) const;
 
    StatusCode CreateTrackToPfoTargetsMap(UidToMCParticleWeightMap &trackToPfoTargetsMap) const;
 
    StatusCode AddMCParticleRelationships() const;
 
    StatusCode RemoveAllMCParticleRelationships();
 
    StatusCode RemoveMCParticleRelationships(const MCParticle *const pMCParticle) const;
 
    typedef std::unordered_map<Uid, float> UidToWeightMap;
    typedef std::unordered_map<Uid, UidToWeightMap> ObjectRelationMap;
    typedef std::unordered_multimap<Uid, Uid> MCParticleRelationMap;
 
    StatusCode SetUidToMCParticleRelationship(const Uid objectUid, const Uid mcParticleUid, const float mcParticleWeight,
        ObjectRelationMap &objectRelationMap) const;
 
    StatusCode CreateUidToPfoTargetsMap(UidToMCParticleWeightMap &uidToMCParticleWeightMap, const ObjectRelationMap &objectRelationMap) const;
 
    const std::string               m_selectedListName;                 
 
    UidToMCParticleMap              m_uidToMCParticleMap;               
    MCParticleRelationMap           m_parentDaughterRelationMap;        
    ObjectRelationMap               m_caloHitToMCParticleMap;           
    ObjectRelationMap               m_trackToMCParticleMap;             
 
    friend class PandoraApiImpl;
    friend class PandoraContentApiImpl;
    friend class PandoraImpl;
};
 
//------------------------------------------------------------------------------------------------------------------------------------------
 
inline StatusCode MCManager::SetCaloHitToMCParticleRelationship(const Uid caloHitUid, const Uid mcParticleUid, const float mcParticleWeight)
{
    return this->SetUidToMCParticleRelationship(caloHitUid, mcParticleUid, mcParticleWeight, m_caloHitToMCParticleMap);
}
 
//------------------------------------------------------------------------------------------------------------------------------------------
 
inline StatusCode MCManager::SetTrackToMCParticleRelationship(const Uid trackUid, const Uid mcParticleUid, const float mcParticleWeight)
{
    return this->SetUidToMCParticleRelationship(trackUid, mcParticleUid, mcParticleWeight, m_trackToMCParticleMap);
}
 
//------------------------------------------------------------------------------------------------------------------------------------------
 
inline StatusCode MCManager::CreateCaloHitToPfoTargetsMap(UidToMCParticleWeightMap &caloHitToPfoTargetsMap) const
{
    return this->CreateUidToPfoTargetsMap(caloHitToPfoTargetsMap, m_caloHitToMCParticleMap);
}
 
//------------------------------------------------------------------------------------------------------------------------------------------
 
inline StatusCode MCManager::CreateTrackToPfoTargetsMap(UidToMCParticleWeightMap &trackToPfoTargetsMap) const
{
    return this->CreateUidToPfoTargetsMap(trackToPfoTargetsMap, m_trackToMCParticleMap);
}
 
} // namespace pandora
 
#endif // #ifndef PANDORA_MC_MANAGER_H