ModularExampleTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
 
 
#ifndef COLUMNAR_EXAMPLE_TOOLS_MODULAR_EXAMPLE_TOOL_H
#define COLUMNAR_EXAMPLE_TOOLS_MODULAR_EXAMPLE_TOOL_H
 
#include <AsgTools/AsgTool.h>
#include <AsgTools/PropertyWrapper.h>
#include <ColumnarCore/ColumnAccessor.h>
#include <ColumnarCore/ColumnarTool.h>
#include <ColumnarCore/ObjectColumn.h>
#include <ColumnarCore/ParticleDef.h>
 
namespace columnar
{
 
  class ModularExampleTool final
    : public asg::AsgTool,
      public ColumnarTool<>
  {
  public:
 
    // Create a proper constructor for Athena
    ASG_TOOL_CLASS( ModularExampleTool, asg::IAsgTool )
 
    ModularExampleTool (const std::string& name);
 
    virtual StatusCode initialize () override;
 
    virtual void callEvents (EventContextRange events) const override;
 
 
    Gaudi::Property<float> m_ptCut {this, "ptCut", 10e3, "pt cut (in MeV)"};
 
    Gaudi::Property<float> m_etaCut {this, "etaCut", 2.47, "max eta cut"};
 
 
    ParticleAccessor<ObjectColumn> particlesHandle {*this, "Particles"};
 
 
    ParticleDecorator<char> selectionDec {*this, "selection"};
 
 
    struct SubtoolPt : public ColumnarTool<>
    {
      SubtoolPt (float val_cutValue);
      bool select (ParticleId particle) const;
 
      // this accessor isn't actually used, but it needs to be declared
      // so that in columnar mode the accessors know the name of the
      // object for each column.  it the subtool is connected before the
      // accessors get connected (i.e. via the base class constructor),
      // this isn't needed.
      ParticleAccessor<ObjectColumn> particlesHandle {*this, "Particles"};
 
      ParticleAccessor<float> ptAcc {*this, "pt"};
      float m_cutValue = 0;
    };
    std::unique_ptr<SubtoolPt> m_subtoolPt;
 
 
 
 
    struct SubtoolEta : public ColumnarTool<>
    {
      SubtoolEta (ColumnarTool<>* parent, float val_cutValue);
      bool select (ParticleId particle) const;
 
      ParticleAccessor<float> etaAcc {*this, "eta"};
      float m_cutValue = 0;
    };
    std::unique_ptr<SubtoolEta> m_subtoolEta;
  };
}
 
#endif