VariantExampleTool.h

Go to the documentation of this file.

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

 
 
#ifndef COLUMNAR_EXAMPLE_TOOLS_VARIANT_EXAMPLE_TOOL_H
#define COLUMNAR_EXAMPLE_TOOLS_VARIANT_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>
#include <ColumnarVariant/VariantDef.h>
#include <ColumnarVariant/VariantObjectId.h>
#include <ColumnarVariant/VariantAccessor.h>
#include <ColumnarEgamma/EgammaDef.h>
#include <ColumnarMuon/MuonDef.h>
 
namespace ADD_CMODE(columnar)
{
 
  class VariantExampleTool final
    : public asg::AsgTool,
      public columnar::ColumnarTool<>
  {
  public:
 
    // Create a proper constructor for Athena
    ASG_TOOL_CLASS( VariantExampleTool, asg::IAsgTool )
 
    VariantExampleTool (const std::string& name);
 
    virtual StatusCode initialize () override;
 
    void callSingleEvent (columnar::ElectronRange electrons, columnar::MuonRange muons) const;
 
    virtual void callEvents (columnar::EventContextRange events) const override;
 

    columnar::ElectronAccessor<columnar::ObjectColumn> electronsHandle {*this, "AnalysisElectrons"};
    columnar::MuonAccessor<columnar::ObjectColumn> muonsHandle {*this, "AnalysisMuons"};
 

    using MyVariantDef = columnar::VariantContainerId<columnar::ParticleDef, columnar::ElectronDef, columnar::MuonDef>;
 

    columnar::ColumnAccessor<MyVariantDef,float> ptAcc {*this, "pt"};
    columnar::ColumnAccessor<MyVariantDef,float> etaAcc {*this, "eta"};
 

    columnar::ColumnDecorator<MyVariantDef,std::uint16_t> ptRankDec {*this, "ptRank"};

    columnar::ColumnDecorator<columnar::ElectronDef,std::uint16_t> etaRankSpecialDec {*this, "etaRank"};
  };
}
 
#endif