EventSaverxAODNext.h

Go to the documentation of this file.

/*
   Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
 */
 
#ifndef EVENTSAVERXAODNEXT_H_
#define EVENTSAVERXAODNEXT_H_
 
#include "TopAnalysis/EventSaverBase.h"
 
// Framework include(s):
#include "AsgTools/AsgMetadataTool.h"
 
class TFile;
 
namespace top {
  class EventSaverxAODNext: public top::EventSaverBase, public asg::AsgMetadataTool {
  public:
    EventSaverxAODNext();
 
    virtual ~EventSaverxAODNext();
 
    virtual void initialize(std::shared_ptr<top::TopConfig> config, TFile* file,
                            const std::vector<std::string>& extraBranches);
 
    //Keep the asg::AsgTool happy
    virtual StatusCode initialize() {return StatusCode::SUCCESS;}
 
 
    virtual void saveEventToxAOD();
 
    virtual void finalize();
  private:
    std::shared_ptr<top::TopConfig> m_config;
 
    TFile* m_outputFile;
 
    bool m_saveAllObjects;
 
    // helper typedef
    typedef std::map<std::size_t, std::map<unsigned int, unsigned int> > ThinningMap_t;
    typedef ThinningMap_t::const_iterator ThinningMap_Itr;
 
    std::shared_ptr<ThinningMap_t> savePhotons(const bool saveEventObjects);
    std::shared_ptr<ThinningMap_t> saveElectrons(const bool saveEventObjects);
    std::shared_ptr<ThinningMap_t> saveMuons(const bool saveEventObjects);
    std::shared_ptr<ThinningMap_t> saveTaus(const bool saveEventObjects);
    std::shared_ptr<ThinningMap_t> saveJets(const bool saveEventObjects);
    std::shared_ptr<ThinningMap_t> saveLargeRJets(const bool saveEventObjects);
    std::shared_ptr<ThinningMap_t> saveTrackJets(const bool saveEventObjects);
 
    std::vector<unsigned int> thinObjectSelection(const std::size_t hashValue,
                                                  const std::vector<unsigned int>& objectList,
                                                  const std::shared_ptr<ThinningMap_t>& thinningMap) const;
 
 
    ClassDef(top::EventSaverxAODNext, 0);
  };
}
 
#endif