/cvmfs/atlas.cern.ch/repo/sw/ASG/AnalysisBase/2.4.31/tauRecTools/src/TauPi0ClusterCreator.h

#ifndef TAUREC_TAUPI0CLUSTERCREATOR_H
#define TAUREC_TAUPI0CLUSTERCREATOR_H

#include <string>
#include <vector>
#include "tauRecTools/TauRecToolBase.h"
#include "xAODPFlow/PFOAuxContainer.h"

using std::vector;

class TauPi0ClusterCreator : virtual public TauRecToolBase {
public:
    TauPi0ClusterCreator(const std::string& name) ;
    ASG_TOOL_CLASS2(TauPi0ClusterCreator, TauRecToolBase, ITauToolBase);
    virtual ~TauPi0ClusterCreator();

    virtual StatusCode initialize();
    virtual StatusCode finalize();
    virtual StatusCode eventInitialize();
    virtual StatusCode execute(xAOD::TauJet& pTau);
    virtual StatusCode eventFinalize();

    virtual void print() const { }

    
private:


    float getEM1CoreFrac( const xAOD::CaloCluster* /*pi0Candidate*/);
    
    vector<int> getNPosECells( const xAOD::CaloCluster* /*pi0Candidate*/);

    std::map<unsigned, xAOD::CaloCluster*> getClusterToShotMap(
        const std::vector<const xAOD::PFO*> shotVector,
        const xAOD::CaloClusterContainer& pPi0ClusterContainer,
        const xAOD::TauJet &pTau);

    std::vector<unsigned> getShotsMatchedToCluster(
        const std::vector<const xAOD::PFO*> shotVector,
        std::map<unsigned, xAOD::CaloCluster*> clusterToShotMap,
        xAOD::CaloCluster* pPi0Cluster);

    int getNPhotons( const std::vector<const xAOD::PFO*> /*shotVector*/, 
                     std::vector<unsigned> /*shotsInCluster*/);

    vector<float> get1stEtaMomWRTCluster( const xAOD::CaloCluster* /*pi0Candidate*/);

    vector<float> get2ndEtaMomWRTCluster(const xAOD::CaloCluster* /*pi0Candidate*/);

    bool setHadronicClusterPFOs(xAOD::TauJet& pTau);

    // TODO: input cluster container name
    std::string m_inputPi0ClusterContainerName;
    
    // TODO: output cluster container name
    std::string m_outputPi0ClusterContainerName;

    xAOD::PFOContainer* m_neutralPFOContainer;
    std::string m_neutralPFOContainerName;
    xAOD::PFOAuxContainer* m_neutralPFOAuxStore;

    xAOD::PFOContainer* m_hadronicClusterPFOContainer;
    std::string m_hadronicClusterPFOContainerName;
    xAOD::PFOAuxContainer* m_hadronicClusterPFOAuxStore;
    

    double m_clusterEtCut;

    bool m_AODmode;

    xAOD::CaloClusterContainer*  m_pOutputPi0CaloClusterContainer;

};

#endif  /* TAUPI0CLUSTERCREATOR_H */

---