AccessorHelpers.h

Go to the documentation of this file.

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

 
#ifndef ACCESSORHELPERS_H
#define ACCESSORHELPERS_H
 
#include <AthContainers/Accessor.h>
#include <Math/Vector4D.h>
#include <TMath.h>
 
#include <map>
#include <memory>
#include <string>
#include <vector>
 
namespace CP {

struct PartonAccessor {
  std::map<std::string, std::unique_ptr<SG::Accessor<float>>> floatAccessors;
  std::map<std::string, std::unique_ptr<SG::Accessor<int>>> intAccessors;

  void initializeFloatAccessor(const std::string& name) {
    floatAccessors[name] = std::make_unique<SG::Accessor<float>>(name);
  }

  void initializeFloatAccessor(const std::vector<std::string>& names) {
    for (const auto& name : names) {
      initializeFloatAccessor(name);
    }
  }

  void initializeIntAccessor(const std::string& name) {
    intAccessors[name] = std::make_unique<SG::Accessor<int>>(name);
  }

  void initializeIntAccessor(const std::vector<std::string>& names) {
    for (const auto& name : names) {
      initializeIntAccessor(name);
    }
  }

  void initializePtEtaPhiMAccessor(const std::string& prefix) {
    initializeFloatAccessor(prefix + "_pt");
    initializeFloatAccessor(prefix + "_eta");
    initializeFloatAccessor(prefix + "_phi");
    initializeFloatAccessor(prefix + "_m");
  }

  template <typename T>
  float getFloat(const std::string& name, T* object) {
    return (*floatAccessors.at(name))(*object);
  }

  template <typename T>
  int getInt(const std::string& name, T* object) {
    return (*intAccessors.at(name))(*object);
  }

  template <typename T>
  void getPtEtaPhiMVector(const std::string& prefix,
                          ROOT::Math::PtEtaPhiMVector& p, T* object) {
    p.SetCoordinates(
        getFloat(prefix + "_pt", object), getFloat(prefix + "_eta", object),
        getFloat(prefix + "_phi", object), getFloat(prefix + "_m", object));
  }
};
 
}  // namespace CP
 
#endif  // ACCESSORHELPERS_H