DataMcScaleFactorProducers.h

Go to the documentation of this file.

#pragma once

#include "HiggsAnalysis/KITHiggsToTauTau/interface/HttTypes.h"
#include "HiggsAnalysis/KITHiggsToTauTau/interface/HttEnumTypes.h"


class DataMcScaleFactorProducerBase: public ProducerBase<HttTypes> {
public:

        DataMcScaleFactorProducerBase(std::vector<std::string>& (setting_type::*GetEfficiencyData)(void) const,
                                      std::vector<std::string>& (setting_type::*GetEfficiencyMc)(void) const,
                                      std::string (setting_type::*GetEfficiencyHistogram)(void) const,
                                      std::string (setting_type::*GetEfficiencyMode)(void) const,
                                      std::string const& weightName);
        
        virtual void Init(setting_type const& settings, metadata_type& metadata) override;

        virtual void Produce(event_type const& event, product_type& product,
                             setting_type const& settings, metadata_type const& metadata) const override;


private:
        std::vector<std::string>& (setting_type::*GetEfficiencyData)(void) const;
        std::vector<std::string>& (setting_type::*GetEfficiencyMc)(void) const;
        std::string (setting_type::*GetEfficiencyHistogram)(void) const;
        std::string (setting_type::*GetEfficiencyMode)(void) const;
        std::string m_weightName;
        
        std::map<std::string, std::vector<TH2F*> > efficienciesDataByHltName;
        std::map<size_t, std::vector<TH2F*> > efficienciesDataByIndex;
        std::map<std::string, std::vector<TH2F*> > efficienciesMcByHltName;
        std::map<size_t, std::vector<TH2F*> > efficienciesMcByIndex;
        
        HttEnumTypes::DataMcScaleFactorProducerMode m_scaleFactorMode = HttEnumTypes::DataMcScaleFactorProducerMode::NONE;
        
        std::vector<double> GetEfficienciesFromHistograms(std::vector<TH2F*> const& histograms, KLepton* lepton) const;
        
        std::vector<std::vector<double> > GetEfficiencies(
                        std::map<std::string, std::vector<TH2F*> > const& efficienciesByHltName,
                        std::map<size_t, std::vector<TH2F*> > const& efficienciesByIndex,
                        event_type const& event, product_type const& product,
                        setting_type const& settings, metadata_type const& metadata
        ) const;

};


class TriggerWeightProducer: public DataMcScaleFactorProducerBase {
public:

        virtual std::string GetProducerId() const override {
                return "TriggerWeightProducer";
        }
        
        TriggerWeightProducer();
        
};


class IdentificationWeightProducer: public DataMcScaleFactorProducerBase {
public:

        virtual std::string GetProducerId() const override {
                return "IdentificationWeightProducer";
        }
        
        IdentificationWeightProducer();
        
};


class LepTauScaleFactorWeightProducer: public ProducerBase<HttTypes> {
public:

        LepTauScaleFactorWeightProducer(std::vector<std::string>& (setting_type::*GetWeightFiles)(void) const,
                                        std::vector<std::string>& (setting_type::*GetWeightHistograms)(void) const);
        
        virtual void Init(setting_type const& settings, metadata_type& metadata) override;

        virtual void Produce(event_type const& event, product_type& product,
                             setting_type const& settings, metadata_type const& metadata) const override;

private:
        std::vector<std::string>& (setting_type::*GetWeightFiles)(void) const;
        std::vector<std::string>& (setting_type::*GetWeightHistograms)(void) const;
        
        std::map<size_t, std::vector<TH1F*> > weightsByIndex;
};

class EleTauFakeRateWeightProducer: public LepTauScaleFactorWeightProducer {
public:

        virtual std::string GetProducerId() const override {
                return "EleTauFakeRateWeightProducer";
        }
        
        EleTauFakeRateWeightProducer();
};

class MuonTauFakeRateWeightProducer: public LepTauScaleFactorWeightProducer {
public:

        virtual std::string GetProducerId() const override {
                return "MuonTauFakeRateWeightProducer";
        }
        
        MuonTauFakeRateWeightProducer();
};