EventInfoBranch.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
#ifndef MUONTESTERTREE_EVENTINFOBRANCH_H
#define MUONTESTERTREE_EVENTINFOBRANCH_H
#include <MuonTesterTree/ScalarBranch.h>
#include <StoreGate/ReadHandleKey.h>
#include <StoreGate/ReadDecorHandleKey.h>
#include <xAODEventInfo/EventInfo.h>
#include <MuonTesterTree/MuonTesterBranch.h>
#include <MuonTesterTree/MuonTesterTree.h>
#include <atomic>
 
namespace MuonVal {
class EventInfoBranch : public MuonTesterBranch {
public:
    
    enum WriteOpts {
        isMC = 1,       
        writePileUp = 1 << 1,
        writePRW = 1 << 2,
        writeBeamSpot = 1 << 3,
        writeTrigger = 1 << 4,
        writeLHE = 1<<5
    };
  
    EventInfoBranch(MuonTesterTree& tree, unsigned int write_mask,
                    const std::string& evtKey="EventInfo");
 
    bool fill(const EventContext& ctx) override final;
    bool init() override final;
    
    static void setNumLHE(unsigned int numLHE) ATLAS_THREAD_SAFE;
    static unsigned int getNumLHE();
  
private:
    static std::atomic<unsigned int> s_num_lhe ATLAS_THREAD_SAFE;
    SG::ReadHandleKey<xAOD::EventInfo> m_key{"EventInfo"};
    using EvtInfoDecor = SG::ReadDecorHandleKey<xAOD::EventInfo>;
    std::vector<EvtInfoDecor> m_prwKeys{};
    unsigned int m_writemask{0};
 
    
    ScalarBranch<unsigned long long>& m_evtNumber{parent().newScalar<unsigned long long>("eventNumber")};
    ScalarBranch<uint32_t>& m_runNumber{parent().newScalar<uint32_t>("runNumber")};
 
    ScalarBranch<uint32_t>& m_lbNumber{parent().newScalar<uint32_t>("lbNumber")};
    ScalarBranch<uint32_t>& m_bcid{parent().newScalar<uint32_t>("bcid")};
    
    ScalarBranch<uint32_t>& m_l1id{parent().newScalar<uint32_t>("l1id")};

    ScalarBranch<float>& m_average_mu{parent().newScalar<float>("average_mu")};
    ScalarBranch<float>& m_actual_mu{parent().newScalar<float>("actual_mu")};
    ScalarBranch<uint32_t>& m_mcChannel{parent().newScalar<uint32_t>("mcChannelNumber")};
    ScalarBranch<double>& m_weight{parent().newScalar<double>("mcEventWeight")};    
    std::map<unsigned int, std::shared_ptr<ScalarBranch<double>>> m_lhe_weights{};

    ScalarBranch<double>& m_prwWeight{parent().newScalar<double>("prwWeight")};
    ScalarBranch<uint32_t>& m_rnd_run{parent().newScalar<uint32_t>("randomRunNumber")};
    ScalarBranch<uint32_t>& m_rnd_lumi_block{parent().newScalar<uint32_t>("randomLumiBlockNumber")};
    ScalarBranch<double>& m_beamSpotWeight{parent().newScalar<double>("BeamSpotWeight")};
};
}
#endif