TauPi0Cluster.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
*/

 
#ifndef TAUEVENT_TAUPI0CLUSTER_H
#define TAUEVENT_TAUPI0CLUSTER_H
 
#include <iostream>
 
//#include "CaloEvent/CaloCluster.h"
#include "CaloEvent/CaloClusterContainer.h"
#include "AthLinks/ElementLink.h"
#include "tauEvent/TauShot.h"
//#include "AthLinks/ElementLinkVector.h"
 
namespace Analysis {
 
  class TauPi0Cluster {
 
    public:
    TauPi0Cluster();
    
    TauPi0Cluster( const TauPi0Cluster& rhs );
    
    virtual ~TauPi0Cluster();

    TauPi0Cluster& operator = ( const TauPi0Cluster& rhs );

    bool operator==( const TauPi0Cluster& rhs ) const;

    bool operator!=( const TauPi0Cluster& rhs ) const;

    // public member accessors
    public:
    void dump(); //just for testing




    float deltaPhi() const;

    void setDeltaPhi(
    float deltaPhi);




    float deltaTheta() const;

    void setDeltaTheta(
    float deltaTheta);




    float engFracEM() const;

    void setEngFracEM(
    float engFracEM);




    float engFracMAX() const;

    void setEngFracMAX(
    float engFracMAX);




    float engFracCore() const;

    void setEngFracCore(
    float engFracCore);




    float secondEngDens() const;

    void setSecondEngDens(
    float secondEngDens);




    float LATERAL() const;

    void setLATERAL(
    float LATERAL);




    float LONGITUDINAL() const;

    void setLONGITUDINAL(
    float LONGITUDINAL); 




    float EM1CoreFrac() const;

    void setEM1CoreFrac(
    float EM1CoreFrac);




    float asymmetryInEM1WRTTrk() const;

    void setAsymmetryInEM1WRTTrk(
    float asymmetryInEM1WRTTrk);




    int NHitsInEM1() const;

    void setNHitsInEM1(
    int NHitsInEM1);




    int NPosECells_PS () const;

    void setNPosECells_PS(
    int NPosECells_PS);




    int NPosECells_EM1 () const;

    void setNPosECells_EM1(
    int NPosECells_EM1);




    int NPosECells_EM2 () const;

    void setNPosECells_EM2(
    int NPosECells_EM2);
 




    float firstEtaWRTClusterPosition_EM1() const;

    void setFirstEtaWRTClusterPosition_EM1(
    float firstEtaWRTClusterPosition_EM1);
 




    float firstEtaWRTClusterPosition_EM2() const;

    void setFirstEtaWRTClusterPosition_EM2(
    float firstEtaWRTClusterPosition_EM2);
 




    float secondEtaWRTClusterPosition_EM1() const;

    void setSecondEtaWRTClusterPosition_EM1(
    float secondEtaWRTClusterPosition_EM1);
 




    float secondEtaWRTClusterPosition_EM2() const;

    void setSecondEtaWRTClusterPosition_EM2(
    float secondEtaWRTClusterPosition_EM2);




    float BDTScore() const;

    void setBDTScore(
    float BDTScore); 
      
 
    
    void setExternalVars(float DELTA_PHI, float DELTA_THETA, float ENG_FRAC_EM, float ENG_FRAC_MAX,float ENG_FRAC_CORE, float SECOND_ENG_DENS, float LATERAL, float LONGITUDINAL,
    float EM1CoreFrac, float asymmetryInEM1WRTTrk, int NHitsInEM1 , int NPosECells_PS, int NPosECells_EM1, int NPosECells_EM2, 
    float firstEtaWRTClusterPosition_EM1, float firstEtaWRTClusterPosition_EM2, float secondEtaWRTClusterPosition_EM1, float secondEtaWRTClusterPosition_EM2, float BDTScore);
    


    const ElementLink<CaloClusterContainer>& clusterLink() const;
    
    ElementLink<CaloClusterContainer>& clusterLink();
        
    const CaloCluster* cluster() const;

    void setCluster(
        const CaloCluster *cluster,
        const CaloClusterContainer *cont );
 
    
    const std::vector<TauShot*>& tauShotVec() const;

    std::vector<TauShot*>& tauShotVec();

    void addTauShot(
           TauShot *tauShot );
    
 
    // alternative setCluster method
    // which can call addCluster(CaloCluster*, CaloClusterContainer*)
    
    void setPi0Cluster(
        const CaloCluster *cluster,
        const CaloClusterContainer *cont,
    std::vector<TauShot*> tauShotVec,
    float DELTA_PHI, 
    float DELTA_THETA, 
    float ENG_FRAC_EM,
    float ENG_FRAC_MAX, 
    float ENG_FRAC_CORE, 
    float SECOND_ENG_DENS,
    float LATERAL,
    float LONGITUDINAL,
        float EM1CoreFrac, 
    float asymmetryInEM1WRTTrk,
    int   NHitsInEM1, 
    int   NPosECells_PS, 
    int   NPosECells_EM1, 
    int   NPosECells_EM2,
    float firstEtaWRTClusterPosition_EM1, 
    float firstEtaWRTClusterPosition_EM2,
    float secondEtaWRTClusterPosition_EM1, 
    float secondEtaWRTClusterPosition_EM2,
    float BDTScore
    );
    
    

    // private data
    private:
    
    ElementLink<CaloClusterContainer> m_cluster;
 
    // shots
    std::vector<TauShot*> m_tauShotVector;
    
    float m_DELTA_PHI;
    float m_DELTA_THETA; 
    float m_ENG_FRAC_EM;
    float m_ENG_FRAC_MAX;
    float m_ENG_FRAC_CORE;
    float m_SECOND_ENG_DENS;
    float m_LATERAL;
    float m_LONGITUDINAL;
    float m_EM1CoreFrac; 
    float m_asymmetryInEM1WRTTrk;
    int   m_NHitsInEM1;
    int   m_NPosECells_PS;
    int   m_NPosECells_EM1;
    int   m_NPosECells_EM2;
    float m_firstEtaWRTClusterPosition_EM1;
    float m_firstEtaWRTClusterPosition_EM2;
    float m_secondEtaWRTClusterPosition_EM1;
    float m_secondEtaWRTClusterPosition_EM2;
    float m_BDTScore;
  };
 
} //end of namespace analysis

#include "TauPi0Cluster.icc"
 
#endif // TAUEVENT_TAUPI0CLUSTER_H