ITruthClassificationTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef ITRUTHCLASSIFICATIONTOOL_H_
#define ITRUTHCLASSIFICATIONTOOL_H_
 
#include <iostream>
#include <type_traits>
 
#include <AsgAnalysisInterfaces/IClassificationTool.h>
#include <xAODBase/IParticle.h>
 
 
namespace Truth
{

enum class Type {
  Unknown,
  KnownUnknown,
  IsoElectron,
  ChargeFlipIsoElectron,
  PromptMuon,
  PromptPhotonConversion,
  ElectronFromMuon,
  TauDecay,
  BHadronDecay,
  CHadronDecay,
  LightFlavorDecay,
  ChargeFlipMuon,
  NonMuonlike,
  TauDecayLike = 70,
  BHadronDecayLike = 80,
  CHadronDecayLike = 90,
};
 
} // namespace Truth
 
inline std::ostream &operator<<(std::ostream &os, const Truth::Type &obj)
{
  os << static_cast<std::underlying_type<Truth::Type>::type>(obj);
  return os;
}
 
 

class ITruthClassificationTool : virtual public CP::IClassificationTool
{
  ASG_TOOL_INTERFACE(ITruthClassificationTool)
 
public:
  virtual ~ITruthClassificationTool() = default;

  virtual StatusCode classify(const xAOD::IParticle &particle,
                              unsigned int &classification) const = 0;

  virtual StatusCode classify(const xAOD::IParticle &particle,
                              Truth::Type &classification) const = 0;
};
 
#endif // ITRUTHCLASSIFICATIONTOOL_H_