PUReweighter.h

#pragma once
 
#include <cstdint>
#include <map>
#include <memory>
 
#include <TH1.h>
 
#include <FWCore/ParameterSet/interface/ParameterSet.h>
 
namespace Framework {
    enum PUProfile : uint8_t {
        Spring15_50ns,
        Spring15_25ns,
        Fall15_25ns,
        Spring16_25ns,
        Moriond17_25ns
    };
 
    PUProfile stringToPUProfile(const std::string&);
 
    class PUReweighter {
        public:
            PUReweighter(const edm::ParameterSet&, PUProfile mc_pu_profile);
 
            float getWeight(float n_interactions);
 
        private:
            std::shared_ptr<TH1> m_pu_weights;
 
            std::map<PUProfile, std::vector<float>> m_pu_profiles {
                {
                    // From https://github.com/cms-sw/cmssw/blob/CMSSW_8_0_X/SimGeneral/MixingModule/python/mix_2016_25ns_Moriond17MC_PoissonOOTPU_cfi.py
                    Moriond17_25ns,
                    {
                        1.78653e-05, 
                        2.56602e-05, 
                        5.27857e-05, 
                        8.88954e-05, 
                        0.000109362, 
                        0.000140973, 
                        0.000240998, 
                        0.00071209, 
                        0.00130121, 
                        0.00245255, 
                        0.00502589, 
                        0.00919534, 
                        0.0146697, 
                        0.0204126, 
                        0.0267586, 
                        0.0337697, 
                        0.0401478, 
                        0.0450159, 
                        0.0490577, 
                        0.0524855, 
                        0.0548159, 
                        0.0559937, 
                        0.0554468, 
                        0.0537687, 
                        0.0512055, 
                        0.0476713, 
                        0.0435312, 
                        0.0393107, 
                        0.0349812, 
                        0.0307413, 
                        0.0272425, 
                        0.0237115, 
                        0.0208329, 
                        0.0182459, 
                        0.0160712, 
                        0.0142498, 
                        0.012804, 
                        0.011571, 
                        0.010547, 
                        0.00959489, 
                        0.00891718, 
                        0.00829292, 
                        0.0076195, 
                        0.0069806, 
                        0.0062025, 
                        0.00546581, 
                        0.00484127, 
                        0.00407168, 
                        0.00337681, 
                        0.00269893, 
                        0.00212473, 
                        0.00160208, 
                        0.00117884, 
                        0.000859662, 
                        0.000569085, 
                        0.000365431, 
                        0.000243565, 
                        0.00015688, 
                        9.88128e-05, 
                        6.53783e-05, 
                        3.73924e-05, 
                        2.61382e-05, 
                        2.0307e-05, 
                        1.73032e-05, 
                        1.435e-05, 
                        1.36486e-05, 
                        1.35555e-05, 
                        1.37491e-05, 
                        1.34255e-05, 
                        1.33987e-05, 
                        1.34061e-05, 
                        1.34211e-05, 
                        1.34177e-05, 
                        1.32959e-05, 
                        1.33287e-05
                    }
                },
                {
                    // From https://github.com/cms-sw/cmssw/blob/CMSSW_8_0_X/SimGeneral/MixingModule/python/mix_2016_25ns_SpringMC_PUScenarioV1_PoissonOOTPU_cfi.py
                    Spring16_25ns,
                    {
                        0.000829312873542,
                        0.00124276120498,
                        0.00339329181587,
                        0.00408224735376,
                        0.00383036590008,
                        0.00659159288946,
                        0.00816022734493,
                        0.00943640833116,
                        0.0137777376066,
                        0.017059392038,
                        0.0213193035468,
                        0.0247343174676,
                        0.0280848773878,
                        0.0323308476564,
                        0.0370394341409,
                        0.0456917721191,
                        0.0558762890594,
                        0.0576956187107,
                        0.0625325287017,
                        0.0591603758776,
                        0.0656650815128,
                        0.0678329011676,
                        0.0625142146389,
                        0.0548068448797,
                        0.0503893295063,
                        0.040209818868,
                        0.0374446988111,
                        0.0299661572042,
                        0.0272024759921,
                        0.0219328403791,
                        0.0179586571619,
                        0.0142926728247,
                        0.00839941654725,
                        0.00522366397213,
                        0.00224457976761,
                        0.000779274977993,
                        0.000197066585944,
                        7.16031761328e-05,
                        0.0,
                        0.0,
                        0.0,
                        0.0,
                        0.0,
                        0.0,
                        0.0,
                        0.0,
                        0.0,
                        0.0,
                        0.0,
                        0.0
                    }
                },
                {
                    // From https://github.com/cms-sw/cmssw/blob/CMSSW_7_6_X/SimGeneral/MixingModule/python/mix_2015_25ns_FallMC_matchData_PoissonOOTPU_cfi.py
                    Fall15_25ns,
                    {
                        0.000108643,
                        0.000388957,
                        0.000332882,
                        0.00038397,
                        0.000549167,
                        0.00105412,
                        0.00459007,
                        0.0210314,
                        0.0573688,
                        0.103986,
                        0.142369,
                        0.157729,
                        0.147685,
                        0.121027,
                        0.08855,
                        0.0582866,
                        0.0348526,
                        0.019457,
                        0.0107907,
                        0.00654313,
                        0.00463195,
                        0.00370927,
                        0.0031137,
                        0.00261141,
                        0.00215499,
                        0.00174491,
                        0.00138268,
                        0.00106731,
                        0.000798828,
                        0.00057785,
                        0.00040336,
                        0.00027161,
                        0.000176535,
                        0.00011092,
                        6.75502e-05,
                        4.00323e-05,
                        2.32123e-05,
                        1.32585e-05,
                        7.51611e-06,
                        4.25902e-06,
                        2.42513e-06,
                        1.39077e-06,
                        8.02452e-07,
                        4.64159e-07,
                        2.67845e-07,
                        1.5344e-07,
                        8.68966e-08,
                        4.84931e-08,
                        2.6606e-08,
                        1.433e-08
                    }
                },
                {
                    Spring15_25ns,
                    {
                        4.8551E-07,
                        1.74806E-06,
                        3.30868E-06,
                        1.62972E-05,
                        4.95667E-05,
                        0.000606966,
                        0.003307249,
                        0.010340741,
                        0.022852296,
                        0.041948781,
                        0.058609363,
                        0.067475755,
                        0.072817826,
                        0.075931405,
                        0.076782504,
                        0.076202319,
                        0.074502547,
                        0.072355135,
                        0.069642102,
                        0.064920999,
                        0.05725576,
                        0.047289348,
                        0.036528446,
                        0.026376131,
                        0.017806872,
                        0.011249422,
                        0.006643385,
                        0.003662904,
                        0.001899681,
                        0.00095614,
                        0.00050028,
                        0.000297353,
                        0.000208717,
                        0.000165856,
                        0.000139974,
                        0.000120481,
                        0.000103826,
                        8.88868E-05,
                        7.53323E-05,
                        6.30863E-05,
                        5.21356E-05,
                        4.24754E-05,
                        3.40876E-05,
                        2.69282E-05,
                        2.09267E-05,
                        1.5989E-05,
                        4.8551E-06,
                        2.42755E-06,
                        4.8551E-07,
                        2.42755E-07,
                        1.21378E-07,
                        4.8551E-08
                    }
                }
            };
    };
};