/cvmfs/atlas.cern.ch/repo/sw/ASG/AnalysisBase/2.4.31/xAODMuon/xAODMuon/versions/Muon_v1.h

// Dear emacs, this is -*- c++ -*-
// $Id: Muon_v1.h 731310 2016-03-21 01:15:58Z wleight $
#ifndef XAODMUON_VERSIONS_MUON_V1_H
#define XAODMUON_VERSIONS_MUON_V1_H

// Core include(s):
#include "AthContainers/AuxElement.h"
#include "AthLinks/ElementLink.h"

//xAOD Primitives
#include "xAODPrimitives/IsolationCorrection.h"
#include "xAODPrimitives/IsolationType.h"
#include "xAODPrimitives/IsolationFlavour.h"

// Misc xAOD include(s):
#include "xAODBase/IParticle.h"
#include "xAODTracking/TrackingPrimitives.h" 
#include "xAODTracking/TrackParticleContainer.h" 
#include "xAODCaloEvent/CaloClusterContainer.h" 
#include "xAODMuon/MuonSegmentContainer.h" 

#include <bitset>
#include <stdint.h>


namespace xAOD {
  class Muon_v1 : public IParticle {

  public:
    Muon_v1();
    virtual double           pt() const;
    virtual double           eta() const;
    virtual double           phi() const;
    virtual double           m() const;
    virtual double           e() const;
    virtual double           rapidity() const;

    void setP4(double pt, double eta, double phi);
        
    typedef IParticle::FourMom_t FourMom_t;

    virtual const FourMom_t& p4() const;
        
    virtual Type::ObjectType type() const;

    float charge() const;    
    
    void setCharge(float charge);

    enum Author {
      unknown=0, 
      MuidCo,
      STACO, 
      MuTag, 
      MuTagIMO,
      MuidSA,
      MuGirl,
      MuGirlLowBeta,
      CaloTag,
      CaloLikelihood,
      ExtrapolateMuonToIP,
      NumberOfMuonAuthors // increase this guy when adding
    };
    Author author() const;
    bool isAuthor ( const Author author ) const;
    void setAuthor(Author auth);
    uint16_t allAuthors() const;
    void setAllAuthors(uint16_t authors);
    void addAllAuthor( const Author author ); 
 

    enum MuonType {
      Combined, MuonStandAlone, SegmentTagged, CaloTagged, SiliconAssociatedForwardMuon
    };
    MuonType muonType() const;
    void setMuonType(MuonType type);

    bool summaryValue(uint8_t& value, const SummaryType information) const;      
    void setSummaryValue(uint8_t value, const SummaryType information);  
    bool summaryValue(float& value, const SummaryType information) const;        
    bool summaryValue(uint8_t& value, const MuonSummaryType information) const;
    void setSummaryValue(uint8_t value, const MuonSummaryType information); 
     
    float floatSummaryValue(const SummaryType information) const;
    uint8_t uint8SummaryValue(const SummaryType information) const;
    float uint8MuonSummaryValue(const MuonSummaryType information) const;
        
    enum ParamDef {
      spectrometerFieldIntegral=0  , //<! B-field integral in MS
      scatteringCurvatureSignificance=1 , //<! Scattering angle significance: curvature
      scatteringNeighbourSignificance=2 , //<! Scattering angle significance: neighbours
      momentumBalanceSignificance=3, //<! momentum balance significance
        
      segmentDeltaEta=4 , 
      segmentDeltaPhi=5 ,
      segmentChi2OverDoF=6 ,
      t0=7        ,
      beta=8      ,
      annBarrel=9 ,
      annEndCap=10,
      innAngle=11 ,
      midAngle=12 ,
      msInnerMatchChi2=13, 
      msInnerMatchDOF=14, 
      msOuterMatchChi2=15, 
      msOuterMatchDOF=16, 
      meanDeltaADCCountsMDT=17, 

      CaloLRLikelihood=18, 
      CaloMuonIDTag=19, 
      FSR_CandidateEnergy=20, 
      EnergyLoss=21, 
      ParamEnergyLoss=22, 
      MeasEnergyLoss=23, 
      EnergyLossSigma=24, 
      ParamEnergyLossSigmaPlus=25, 
      ParamEnergyLossSigmaMinus=26, 
      MeasEnergyLossSigma=27, 
      /*Perigee parameters of the original (MS-only) extrapolated track*/
      d0_sa=28,
      z0_sa=29,
      phi0_sa=30,
      theta_sa=31,
      qOverP_sa=32,
      Eloss_sa=33
    };
        
    bool parameter(float& value, const ParamDef parameter) const;

    void setParameter(float value, const ParamDef parameter);

    float floatParameter(const ParamDef parameter) const;

    bool parameter(int& value, const ParamDef parameter) const;

    void setParameter(int value, const ParamDef parameter);

    int intParameter(const ParamDef parameter) const;

    enum Quality {Tight, 
                  Medium, 
                  Loose, 
                  VeryLoose}; 
    Quality quality() const;
    void setQuality(Quality);
          
    bool passesIDCuts() const;

    void setPassesIDCuts(bool);

    bool passesHighPtCuts() const;

    void setPassesHighPtCuts(bool);
              
    

    bool isolation(float& value,   const Iso::IsolationType information) const;

    float isolation(const Iso::IsolationType information) const;

    void setIsolation(float value, const Iso::IsolationType information);

    bool isolationCaloCorrection(float& value, const Iso::IsolationFlavour flavour, const Iso::IsolationCaloCorrection type,
                                 const Iso::IsolationCorrectionParameter param) const;

    float isolationCaloCorrection(const Iso::IsolationFlavour flavour, const Iso::IsolationCaloCorrection type,
                                  const Iso::IsolationCorrectionParameter param) const;

    bool setIsolationCaloCorrection(float value, const Iso::IsolationFlavour flavour, const Iso::IsolationCaloCorrection type,
                                    const Iso::IsolationCorrectionParameter param);


    bool isolationTrackCorrection(float& value, const Iso::IsolationFlavour flavour , const Iso::IsolationTrackCorrection type ) const;

    float isolationTrackCorrection(const Iso::IsolationFlavour flavour , const Iso::IsolationTrackCorrection type) const;

    bool setIsolationTrackCorrection(float value, const Iso::IsolationFlavour flavour , const Iso::IsolationTrackCorrection type);


    bool isolationCorrectionBitset(std::bitset<32>& value, const Iso::IsolationFlavour flavour ) const;

    std::bitset<32> isolationCorrectionBitset(const Iso::IsolationFlavour flavour ) const;

    bool setIsolationCorrectionBitset(uint32_t value, const Iso::IsolationFlavour flavour );


    const ElementLink< TrackParticleContainer >& primaryTrackParticleLink() const;
        
    const TrackParticle* primaryTrackParticle() const; 

    const ElementLink< TrackParticleContainer >& inDetTrackParticleLink() const;
    const ElementLink< TrackParticleContainer >& muonSpectrometerTrackParticleLink() const;
    const ElementLink< TrackParticleContainer >& combinedTrackParticleLink() const;
    const ElementLink< TrackParticleContainer >& extrapolatedMuonSpectrometerTrackParticleLink() const;
        
    enum TrackParticleType {
      Primary, InnerDetectorTrackParticle, MuonSpectrometerTrackParticle, CombinedTrackParticle, ExtrapolatedMuonSpectrometerTrackParticle
    };
    const ElementLink< TrackParticleContainer >& trackParticleLink( TrackParticleType type) const;
    void setTrackParticleLink(TrackParticleType type, const ElementLink< TrackParticleContainer >& link);
    const TrackParticle* trackParticle( TrackParticleType type) const; 
       
    const ElementLink<CaloClusterContainer>& clusterLink() const; 
    void setClusterLink(const ElementLink<CaloClusterContainer>& link);
    const CaloCluster* cluster() const;
        
    enum EnergyLossType { Parametrized=0, 
      NotIsolated=1,  
      MOP=2,          
      Tail=3,         
      FSRcandidate=4  
    };
    EnergyLossType energyLossType (void) const;
    void setEnergyLossType (EnergyLossType type) ;
        

    const std::vector< ElementLink< MuonSegmentContainer > > & muonSegmentLinks() const;
    void setMuonSegmentLinks(const std::vector< ElementLink< MuonSegmentContainer > >& segments) ;
    size_t nMuonSegments() const;
    const MuonSegment* muonSegment( size_t i ) const;
    const ElementLink< MuonSegmentContainer >& muonSegmentLink( size_t i ) const;
        
  private:
    mutable FourMom_t m_p4;
    mutable bool m_p4Cached1;
  }; // class xAOD::Muon

} // namespace xAOD

// Set up a CLID for the class:
#include "xAODCore/CLASS_DEF.h"
CLASS_DEF( xAOD::Muon_v1, 239636954, 1 )

  // Declare IParticle as a base class of Muon_v1:
#include "AthContainers/DataVector.h"
  DATAVECTOR_BASE( xAOD::Muon_v1, xAOD::IParticle );

#endif // XAODMUON_VERSIONS_Muon_v1_H

---