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 columnar
{
 
  class VariantExampleTool final
    : public asg::AsgTool,
      public 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 (ElectronRange electrons, MuonRange muons) const;
 
    virtual void callEvents (EventContextRange events) const override;
 

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

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

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

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

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