RNtupleTreeMakerAlg.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
#ifndef ASGANALYSISALGORITHMS_RNTUPLETREEMAKERALG_H
#define ASGANALYSISALGORITHMS_RNTUPLETREEMAKERALG_H
 
#include "AnaAlgorithm/AnaAlgorithm.h"
#include "AsgTools/PropertyWrapper.h"
#include "AsgTools/ToolHandle.h"
#include "PATInterfaces/SystematicSet.h"
 
// RNTuple include(s):
#include <ROOT/RNTupleModel.hxx>
#include <ROOT/RNTupleWriter.hxx>
#include <TFile.h>
 
// Local include(s):
#include "AsgAnalysisAlgorithms/RNtupleFieldHelpers.h"
 
#include <string>
#include <vector>
#include <memory>
#include <unordered_map>
 
namespace CP {

   class RNtupleTreeMakerAlg : public EL::AnaAlgorithm {
 
   public:
      using EL::AnaAlgorithm::AnaAlgorithm;

      virtual StatusCode initialize() override;

      virtual StatusCode execute(const EventContext& ctx) override;

      virtual StatusCode finalize() override;
 
   private:
      StatusCode setupTree();

 
      Gaudi::Property<std::string> m_modelName {
         this, "TreeName", "physics", "Name of the RNTuple model/tree"};
 
      Gaudi::Property<std::string> m_outputStreamName {
         this, "OutputStreamName", "ANALYSIS", "Name of the output stream"};
 
      Gaudi::Property<std::vector<std::string>> m_branches {
         this, "Branches", {}, "List of branches to create (format: Object.Var -> BranchName)"};
 
      Gaudi::Property<std::vector<std::string>> m_nonContainers {
         this, "NonContainers", {}, "List of objects to treat as non-containers"};


      ServiceHandle<ISystematicsSvc> m_systematicsService {
         this, "SystematicsSvc", "SystematicsSvc", "Systematics service"};

 
      std::unique_ptr<ROOT::RNTupleModel> m_model;
      std::unique_ptr<ROOT::RNTupleWriter> m_writer;
 
      CP::RNtupleFieldHelpers::ProcessorList m_processorList{this};
 
      bool m_isInitialized = false;
 
#ifndef XAOD_STANDALONE
      std::unique_ptr<TFile> m_outputFile;
#endif

 
   }; // class RNtupleTreeMakerAlg
 
} // namespace CP
 
#endif // ASGANALYSISALGORITHMS_RNTUPLETREEMAKERALG_H