/cvmfs/atlas.cern.ch/repo/sw/ASG/AnalysisBase/2.4.31/InDetTrackSystematicsTools/InDetTrackSystematicsTools/InDetTrackBiasingTool.h

// -*- c++ -*-
#pragma once

#include "InDetTrackSystematicsTools/IInDetTrackBiasingTool.h"
#include "AsgTools/AsgTool.h"
#include "PATInterfaces/CorrectionTool.h"
#include "InDetTrackSystematicsTools/InDetTrackSystematicsTool.h"
#include "xAODTracking/TrackParticleContainer.h"
#include <string>
#include <vector>

#include <TH2.h>

#include "TRandom3.h"

class TFile;

namespace InDet {


  class InDetTrackBiasingTool
    : public virtual IInDetTrackBiasingTool
    , public virtual InDetTrackSystematicsTool
    , public virtual CP::CorrectionTool< xAOD::TrackParticleContainer >
  {

    ASG_TOOL_CLASS( InDetTrackBiasingTool,
            InDet::IInDetTrackBiasingTool )

    public:

    InDetTrackBiasingTool (const std::string& name);
    virtual ~InDetTrackBiasingTool();

    virtual StatusCode initialize() override;
    virtual void prepare() override {};
    virtual StatusCode finalize() override;

    virtual CP::CorrectionCode applyCorrection(xAOD::TrackParticle& track) override;
    virtual CP::CorrectionCode correctedCopy( const xAOD::TrackParticle& in,
                          xAOD::TrackParticle*& out );
    virtual CP::CorrectionCode applyContainerCorrection( xAOD::TrackParticleContainer& cont );


    virtual bool isAffectedBySystematic( const CP::SystematicVariation& ) const;
    virtual CP::SystematicSet affectingSystematics() const;
    virtual CP::SystematicSet recommendedSystematics() const;
    virtual CP::SystematicCode applySystematicVariation( const CP::SystematicSet& );

  protected:
    
    StatusCode initHistograms(int runNumber);
    StatusCode firstCall();

    float readHistogram(float fDefault, TH2* histogram, float phi, float eta) const;
    //    float readEtaHistogram(float fDefault, TH1* histogram, float eta) const;

    float m_biasD0 = 0.f;
    float m_biasZ0 = 0.f;
    float m_biasQoverPsagitta = 0.f;    

    TH2* m_biasD0Histogram = nullptr; 
    TH2* m_biasZ0Histogram = nullptr; 
    TH2* m_biasQoverPsagittaHistogram = nullptr; 
    TH2* m_biasD0HistError = nullptr; 
    TH2* m_biasZ0HistError = nullptr; 
    TH2* m_biasQoverPsagittaHistError = nullptr; 

    // if neither of these is set manually, then the tool will try to get this from EventInfo in the event store.
    // these options exist only as a manual override.
    bool m_isData = false;
    bool m_isSimulation = false;
    uint32_t m_runNumber = 0;

    bool m_doD0Bias = true;
    bool m_doZ0Bias = true;
    bool m_doQoverPBias = true;

  }; // class InDetTrackBiasingTool

} // namespace InDet

---