BootstrapGeneratorAlg.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/

 
#ifndef ASG_ANALYSIS_ALGORITHMS__BOOTSTRAP_GENERATOR_ALG_H
#define ASG_ANALYSIS_ALGORITHMS__BOOTSTRAP_GENERATOR_ALG_H
 
#include <AnaAlgorithm/AnaAlgorithm.h>
#include <SystematicsHandles/SysListHandle.h>
#include <SystematicsHandles/SysReadHandle.h>
#include <SystematicsHandles/SysWriteDecorHandle.h>
#include <xAODEventInfo/EventInfo.h>
#include <AsgTools/PropertyWrapper.h>
#include <TRandomGen.h>
 
namespace CP
{
  class BootstrapGenerator
  {
  public:
    BootstrapGenerator() {};

  public:
    std::uint64_t fnv1a_64(const void *buffer, size_t size, std::uint64_t offset_basis);

  public:
    void setSeed(std::uint64_t eventNumber, std::uint32_t runNumber, std::uint32_t mcChannelNumber);

  public:
    std::uint64_t generateSeed(std::uint64_t eventNumber, std::uint32_t runNumber, std::uint32_t mcChannelNumber);

  public:
    std::uint8_t getBootstrap() { return m_rng.Poisson(1); };

  private:
    static constexpr std::uint64_t m_offset = 14695981039346656037u;
    static constexpr std::uint64_t m_prime = 1099511628211u;

  private:
    TRandomRanluxpp m_rng;
  };
 

  class BootstrapGeneratorAlg final : public EL::AnaAlgorithm
  {
  public:
    BootstrapGeneratorAlg(const std::string &name,
                          ISvcLocator *pSvcLocator);
 
  public:
    StatusCode initialize() override;
 
  public:
    StatusCode execute(const EventContext& ctx) override;

  private:
    SysListHandle m_systematicsList{this};

  private:
    SysReadHandle<xAOD::EventInfo> m_eventInfoHandle{
      this, "eventInfo", "EventInfo", "the EventInfo container"};

  private:
    Gaudi::Property<int> m_nReplicas {this, "nReplicas", 1000, "number of bootstrapped weights (toys) to generate"};

  private:
    Gaudi::Property<bool> m_data {this, "isData", false, "whether we are running on data"};

  private:
    BootstrapGenerator m_bootstrap;

  private:
    std::vector<std::uint8_t> m_weights;

  private:
    SysWriteDecorHandle<std::vector<std::uint8_t>> m_decoration{
      this, "decorationName", "bootstrapWeights_%SYS%", "decoration name for the vector of bootstrapped weights"};
  };
} // namespace CP
 
#endif