TruthParticleClassificationFillerTool.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
*/
 
#include "TruthParticleClassificationFillerTool.h"
#include "AthenaKernel/errorcheck.h"
 
 
namespace D3PD {
 
 
TruthParticleClassificationFillerTool::TruthParticleClassificationFillerTool
  (const std::string& type,
   const std::string& name,
   const IInterface* parent)
 : Base(type, name, parent),
   m_classifier ("MCTruthClassifier")
{
  declareProperty ("Classifier", m_classifier, "Classifier tool instance.");
 
  book().ignore(); // Avoid coverity warnings.
}
 
 
StatusCode TruthParticleClassificationFillerTool::initialize()
{
  CHECK( m_classifier.retrieve() );
  return Base::initialize();
}
 
 
StatusCode TruthParticleClassificationFillerTool::book()
{
  CHECK( addVariable ("type",  m_type, "MC particle type, from classifier tool" )  );
  CHECK( addVariable ("origin",  m_origin, "MC particle origin, from classifier tool" )  );
 
  return StatusCode::SUCCESS;
}
 
 
StatusCode
TruthParticleClassificationFillerTool::fill (const xAOD::TruthParticle& p)
{
  auto res = m_classifier->particleTruthClassifier(&p);
  *m_type   = res.first;
  *m_origin = res.second;
  
  return StatusCode::SUCCESS;
}
 
 
} // namespace D3PD