NTupler-PEC-interface/Electron.cc at master · mjagkow/NTupler-PEC-interface · 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
81
82
83
#include "Electron.h"

#include <algorithm>
#include <iterator>
#include <stdexcept>
#include <type_traits>


unsigned const pec::Electron::contIdSize;


pec::Electron::Electron() noexcept:
    Lepton(),
    etaSC(0.f),
    cutBasedId(0)
{
    std::fill(std::begin(mvaId), std::end(mvaId), 0);
}


void pec::Electron::Reset()
{
    Lepton::Reset();

    etaSC = 0.f;
    cutBasedId = 0;
    std::fill(std::begin(mvaId), std::end(mvaId), 0);
}


void pec::Electron::SetBooleanID(unsigned bitIndex, bool value /*= true*/)
{
    if (bitIndex >= 8)
        throw std::runtime_error("pec::Electron::SetBooleanID: Given index exceeds the maximal "
         "allowed value.");

    if (value)
        cutBasedId |= (1 << bitIndex);
    else
        cutBasedId &= ~(1 << bitIndex);
}


void pec::Electron::SetContinuousID(unsigned index, float mva)
{
    if (index >= contIdSize)
        throw std::runtime_error("pec::Electron::SetContinuousID: Given index exceeds the maximal "
         "allowed value. Consider increasing constant pec::Electron::contIdSize.");

    mvaId[index] = mva;
}


void pec::Electron::SetEtaSC(float etaSC_)
{
    etaSC = etaSC_;
}


bool pec::Electron::BooleanID(unsigned bitIndex) const
{
    if (bitIndex >= 8)
        throw std::runtime_error("pec::Electron::CutBasedId: Given index exceeds the maximal "
         "allowed value.");

    return (cutBasedId & (1 << bitIndex));
}


float pec::Electron::ContinuousID(unsigned index) const
{
    if (index >= contIdSize)
        throw std::runtime_error("pec::Electron::SetMvaId: Given index exceeds the maximal allowed "
         "value.");

    return mvaId[index];
}


float pec::Electron::EtaSC() const
{
    return etaSC;
}