IJetProvider.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2019 CERN for the benefit of the ATLAS collaboration
*/
 
 
#ifndef JETINTERFACE_IJETPROVIDER_H
#define JETINTERFACE_IJETPROVIDER_H

 
#include <utility>
#include <memory>
#include "AsgDataHandles/WriteHandle.h"
#include "AsgTools/IAsgTool.h"
#include "xAODJet/JetContainer.h"
 
namespace SG {
  class IAuxStore;
}
 
class IJetProvider : virtual public asg::IAsgTool {
ASG_TOOL_INTERFACE(IJetProvider)
 
public:

  virtual ~IJetProvider() { };

  virtual std::pair<std::unique_ptr<xAOD::JetContainer>,std::unique_ptr<SG::IAuxStore> > getJets() const = 0;

  virtual StatusCode getAndRecordJets(SG::WriteHandle<xAOD::JetContainer>& jetHandle) const = 0;

  virtual StatusCode initWithOutput(const SG::WriteHandleKey<xAOD::JetContainer>&) {return StatusCode::SUCCESS;};
 
};
 

template <typename CONCRETEAUX> class JetProvider
  : virtual public IJetProvider
{
 
  public:
  
  StatusCode getAndRecordJets(SG::WriteHandle<xAOD::JetContainer>& jetHandle) const {
    std::unique_ptr<xAOD::JetContainer> jets(nullptr);
    std::unique_ptr<SG::IAuxStore> auxCont(nullptr);
    std::tie(jets, auxCont) = getJets();
    if(jets.get()==nullptr || auxCont.get()==nullptr) {return StatusCode::FAILURE;}
 
    // Convert the aux container pointer to the known base class
    std::unique_ptr<CONCRETEAUX> auxCont_derived(static_cast<CONCRETEAUX*>(auxCont.release()));
 
    // Write out JetContainer and JetAuxContainer
    return jetHandle.record(std::move(jets), std::move(auxCont_derived));
  }
 
};
    
 
#endif