TruthParticleClassificationFillerTool.h

Go to the documentation of this file.

// This file's extension implies that it's C, but it's really -*- C++ -*-.
 
/*
  Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
*/

 
#ifndef TRUTHD3PDMAKER_TRUTHPARTICLECLASSIFICATIONFILLERTOOL_H
#define TRUTHD3PDMAKER_TRUTHPARTICLECLASSIFICATIONFILLERTOOL_H
 
 
#include "D3PDMakerUtils/BlockFillerTool.h"
#include "MCTruthClassifier/IMCTruthClassifier.h"
#include "xAODTruth/TruthParticle.h"
#include "GaudiKernel/ToolHandle.h"
 
 
namespace D3PD {
  
 
class TruthParticleClassificationFillerTool
  : public BlockFillerTool<xAOD::TruthParticle>
{
public:
  typedef BlockFillerTool<xAOD::TruthParticle> Base;
 

  TruthParticleClassificationFillerTool (const std::string& type,
                                         const std::string& name,
                                         const IInterface* parent);
 

  StatusCode initialize() override;
 

  virtual StatusCode book() final;
 

  virtual StatusCode fill (const xAOD::TruthParticle& p) override;
 
 
  private:
  ToolHandle<IMCTruthClassifier> m_classifier;

  int* m_type;

  int* m_origin;
};
 
 
} // namespace D3PD
 
 
#endif // TRUTHD3PDMAKER_TRUTHPARTICLECLASSIFICATIONFILLERTOOL_H