/cvmfs/atlas.cern.ch/repo/sw/ASG/AnalysisBase/2.4.28/MuonEfficiencyCorrections/MuonEfficiencyCorrections/EfficiencyScaleFactor.h

/*
 * EfficiencyScaleFactor.h
 *
 *  Created on: Apr 10, 2014
 *      Author: goblirsc
 */

#ifndef EFFICIENCYSCALEFACTOR_H_
#define EFFICIENCYSCALEFACTOR_H_

// EDM include(s):
#include "xAODMuon/Muon.h"
// Infrastructure include(s):
#ifdef ROOTCORE
#   include "xAODRootAccess/Init.h"
#   include "xAODRootAccess/TEvent.h"
#endif // ROOTCORE

// EDM include(s):
#include "xAODEventInfo/EventInfo.h"
#include "xAODMuon/MuonContainer.h"

#include <PATInterfaces/CorrectionCode.h>
#include <MuonEfficiencyCorrections/HistHandler.h>
#include <MuonEfficiencyCorrections/MuonEfficiencyType.h>
#include <MuonEfficiencyCorrections/KinematicSystHandler.h>
#include <PATInterfaces/ISystematicsTool.h>

// further ROOT includes
#include <TFile.h>
#include <TDirectory.h>

// STL includes
#include <string>
#include <iostream>
#include <exception>
#include <map>
#include <cmath>

namespace CP {
    class SystematicSet;
    class EfficiencyScaleFactor {


        public:

            EfficiencyScaleFactor();
            EfficiencyScaleFactor(std::string file, std::string time_unit, SystematicSet sys, CP::MuonEfficiencyType effType, bool is_lowpt = false, bool hasPtDepSys = false);

            EfficiencyScaleFactor(const EfficiencyScaleFactor & other);
            EfficiencyScaleFactor & operator =(const EfficiencyScaleFactor & other);

            bool ReadFromFile(std::string file, std::string time_unit);
            HistHandler* ReadHistFromFile(std::string name, TFile* f, std::string time_unit);

            // mangle the histograms as required by the systematic being run
            void ApplySysVariation();

            bool CheckConsistency();


            CorrectionCode ScaleFactor(const xAOD::Muon& mu, float & SF);
            CorrectionCode ScaleFactorReplicas(const xAOD::Muon& mu, std::vector<float> & SF);

            CorrectionCode DataEfficiency(const xAOD::Muon& mu, float & Eff);
            CorrectionCode DataEfficiencyReplicas(const xAOD::Muon& mu, std::vector<float> & eff);

            CorrectionCode MCEfficiency(const xAOD::Muon& mu, float & Eff);
            CorrectionCode MCEfficiencyReplicas(const xAOD::Muon& mu, std::vector<float> & eff);

            void DebugPrint();

            virtual ~EfficiencyScaleFactor();

            // a string name assigned to each sys variation
            std::string sysname();

            // directly interact with a histo
            HistHandler* get_sf() {
                return m_sf;
            }
            HistHandler* get_eff() {
                return m_eff;
            }
            HistHandler* get_sf_sys() {
                return m_sf_sys;
            }
            HistHandler* get_eff_sys() {
                return m_eff_sys;
            }
            HistHandler* get_mc_eff() {
                return m_mc_eff;
            }
            HistHandler* get_mc_eff_sys() {
                return m_mc_eff_sys;
            }
        private:

            // use some maps for easy histo loading / arithmetics by name

            typedef std::vector<HistHandler*> SFvec;
            typedef std::vector<HistHandler*>::iterator iSFvec;
            typedef std::vector<HistHandler*>::const_iterator ciSFvec;

            CorrectionCode GetContentFromHist(HistHandler* Hist, IKinematicSystHandler* PtDepHist, const xAOD::Muon& mu, float & SF, bool PtDepHistNeeded);
            CorrectionCode GetContentReplicasFromHist(EfficiencyScaleFactor::SFvec &replicas, const xAOD::Muon& mu, std::vector<float> & SF);

            // adds weight*staterr to each bin (used for systematics)
            void AddStatErrors(float weight);
            void AddStatErrors_histo(HistHandler* h, float weight);
            // adds weight*syserr to each bin (used for systematics)
            void AddSysErrors(float weight);
            void AddSysErrors_histo(HistHandler* h, HistHandler* hsys, float weight);
            void AddSysErrors_vector(EfficiencyScaleFactor::SFvec &Vec, HistHandler* hsys, float weight);
            // package a TH1 in a HistHandler
            HistHandler *package_histo(TH1* h);

            void AddHistos(HistHandler* & add_to, HistHandler* add_this, float weight);
            void ScaleHisto(HistHandler* & h, float weight);

            // replica generation
            void GenerateReplicas(int nrep, int seed);
            void GenerateReplicasFromHist(HistHandler* h, int nrep, int seed, EfficiencyScaleFactor::SFvec &repVector);

            //Some functions to clean up and copy the data
            void Clear();
            void DeleteOldReplicas(EfficiencyScaleFactor::SFvec &Vec, bool ClearVec = false);
            void CopyContent(const EfficiencyScaleFactor &other);
            void CopyHistHandler(HistHandler* &own, const HistHandler* other);
            void CopyReplicaVec(EfficiencyScaleFactor::SFvec &own, const EfficiencyScaleFactor::SFvec &other);
            HistHandler* m_sf;
            HistHandler* m_eff;
            HistHandler* m_mc_eff;
            HistHandler* m_sf_sys;
            HistHandler* m_eff_sys;
            HistHandler* m_mc_eff_sys;
            // these are for the continuous pt dependent systematic if we have one
            IKinematicSystHandler* m_sf_KineDepsys;
            IKinematicSystHandler* m_eff_KineDepsys;

            // replicas, in case we use them
            SFvec m_sf_replicas;
            SFvec m_eff_replicas;
            SFvec m_mc_eff_replicas;

            // utility class to work with a fine eta phi binning
            fineEtaPhiBinning m_etaphi;

            // the systematic variation associated with this instance
            SystematicSet m_sys;

            // states that this SF should respond to low pt systematics rather than high pt ones
            bool m_is_lowpt;
            // steers the pt dependent systematics
            bool m_respond_to_kineDepSyst;
            // default efficiency value (in case of OutOfValidityRange)
            float m_default_eff;
            // default TTVA efficiency value (in case of MuonStandAlone for |eta|>2.5)
            float m_default_eff_ttva;
            CP::MuonEfficiencyType m_Type;

    };
} /* namespace CP */

#endif /* EFFICIENCYSCALEFACTOR_H_ */

---