Corrections/MuonFsrRecoveryProvider.h at main · cms-flaf/Corrections · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
#pragma once

#include "correction.h"
#include "corrections.h"

namespace correction {
    namespace MuonFsrRecoveryProvider {
        LorentzVectorM fsr_corrected_p4(float mu_pt,
                                        float mu_eta,
                                        float mu_phi,
                                        float mu_mass,
                                        int mu_fsrIdx,
                                        const RVecF& fsr_pt,
                                        const RVecF& fsr_eta,
                                        const RVecF& fsr_phi,
                                        const RVecF& fsr_dROverEt2,
                                        const RVecF& fsr_relIso03,
                                        const RVecF& fsr_electronIdx) {
            LorentzVectorM res{mu_pt, mu_eta, mu_phi, mu_mass};

            if (mu_fsrIdx == -1) {
                return res;
            }

            float deltaR_mu_fsr = ROOT::VecOps::DeltaR(mu_eta, fsr_eta[mu_fsrIdx], mu_phi, fsr_phi[mu_fsrIdx]);

            if (!((deltaR_mu_fsr > 0.0001) && (deltaR_mu_fsr < 0.5))) {
                return res;
            }

            if (fsr_electronIdx[mu_fsrIdx] != -1) {
                return res;
            }

            if (fsr_pt[mu_fsrIdx] / mu_pt > 0.4) {
                return res;
            }

            if (fsr_dROverEt2[mu_fsrIdx] > 0.012) {
                return res;
            }

            if (fsr_relIso03[mu_fsrIdx] / mu_pt > 1.8) {
                return res;
            }

            res += LorentzVectorM{fsr_pt[mu_fsrIdx], fsr_eta[mu_fsrIdx], fsr_phi[mu_fsrIdx], 0.0};

            return res;
        }

        RVecLV fsr_corrected_p4(const RVecF& mu_pt,
                                const RVecF& mu_eta,
                                const RVecF& mu_phi,
                                const RVecF& mu_mass,
                                const RVecI& mu_fsrIdx,
                                const RVecF& fsr_pt,
                                const RVecF& fsr_eta,
                                const RVecF& fsr_phi,
                                const RVecF& fsr_dROverEt2,
                                const RVecF& fsr_relIso03,
                                const RVecF& fsr_electronIdx) {
            RVecLV res_vec(mu_pt.size());
            for (size_t i = 0; i < mu_pt.size(); ++i) {
                res_vec[i] = fsr_corrected_p4(mu_pt[i],
                                              mu_eta[i],
                                              mu_phi[i],
                                              mu_mass[i],
                                              mu_fsrIdx[i],
                                              fsr_pt,
                                              fsr_eta,
                                              fsr_phi,
                                              fsr_dROverEt2,
                                              fsr_relIso03,
                                              fsr_electronIdx);
            }
            return res_vec;
        }
    }  // namespace MuonFsrRecoveryProvider
}  // namespace correction