HepMCWeightSvc.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef EVGENPRODTOOLS_HEPMCWEIGHTSVC_H
#define EVGENPRODTOOLS_HEPMCWEIGHTSVC_H
 
#include "AthenaBaseComps/AthService.h"
#include "xAODEventInfo/EventInfo.h"
#include "GenInterfaces/IHepMCWeightSvc.h"
#include "StoreGate/ReadHandleKey.h"
#include <string>
 
#include "IOVDbMetaDataTools/IIOVDbMetaDataTool.h"
 
#include "GaudiKernel/ToolHandle.h"
#include <vector>
#include <mutex>

class HepMCWeightSvc : public extends<AthService, IHepMCWeightSvc>
{
public:
  using base_class::base_class;
 

  virtual StatusCode initialize() override;
 

  virtual StatusCode
  setWeightNames (const WeightMap& weightNames,
                  const EventContext& ctx = Gaudi::Hive::currentContext()) override;
 

  virtual WeightMap
  weightNames (const EventContext& ctx = Gaudi::Hive::currentContext()) override;
 

  virtual std::vector<std::string>
  weightNameVec (const EventContext& ctx = Gaudi::Hive::currentContext()) override;
 
 
private:
  unsigned long getChanNum (const EventContext& ctx) const;
 

  unsigned int loadWeights (unsigned long chanNum);
 
  
  unsigned int getWeightIndex (const EventContext& ctx);
 

  PublicToolHandle<IIOVDbMetaDataTool> m_metaDataTool
    { this, "IOVDbMetaDataTool", "IOVDbMetaDataTool" };
 
 
  // Property: Allow disabling reading from metadata.
  Gaudi::Property<bool> m_enabled
    { this, "Enable", true, "If false, will return failure on loadWeights" };
 

  struct WeightInfo
  {
    unsigned long m_chanNum = 0;

    WeightMap m_weightNames;

    std::vector<std::string> m_weightNameVec;

    void fillVec();
  };

  static const unsigned int NWEIGHTS = 8;
  WeightInfo m_weights[NWEIGHTS];

  size_t m_nextWeight = 0;

  SG::ReadHandleKey<xAOD::EventInfo> m_eventInfoKey
    { this, "EventInfoKey", "EventInfo", "SG key for EventInfo object." };

  std::mutex m_mutex;
};
 
#endif