PrimaryTruthClassifier.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
*/

// PrimaryTruthClassifier.h
//   Header file for PrimaryTruthClassification tool
// (c) ATLAS Detector software
 
#ifndef TRK_PRIMARYTRUTHCLASSIFICATIONTOOL_H
#define TRK_PRIMARYTRUTHCLASSIFICATIONTOOL_H
 
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "TrkValInterfaces/ITrackTruthClassifier.h"
#include "TrkValEvent/TruthClassificationDefs.h"
 
#include "AtlasHepMC/GenParticle.h"
#include "AtlasHepMC/SimpleVector.h"
 
namespace Trk {

 
  class PrimaryTruthClassifier : virtual public ITrackTruthClassifier,
    public AthAlgTool {
  public:
 
      PrimaryTruthClassifier(const std::string& type, const std::string& name,
                                     const IInterface* parent);
      ~PrimaryTruthClassifier() {}
      virtual StatusCode initialize();
      virtual StatusCode finalize();

      virtual void initClassification(const McEventCollection&,
                                      const std::vector<HepMC::ConstGenParticlePtr>*) const;
 
      virtual unsigned int classify(HepMC::ConstGenParticlePtr) const;
 
      virtual std::string nameOfClassifier() const;
 
      virtual std::string classificationAsString(unsigned int) const;
 
      virtual unsigned int numberOfClassifiers() const;
 
  private:
      float             m_maxRStartPrimary;
      float             m_maxZStartPrimary;
      float             m_minREndPrimary;
      float             m_minZEndPrimary;
      float             m_maxRStartSecondary;
      float             m_maxZStartSecondary;
      float             m_minREndSecondary;
      float             m_minZEndSecondary;
 
  };
 
  inline unsigned int Trk::PrimaryTruthClassifier::numberOfClassifiers() const {
    return Trk::TruthClassification::NumberOfPrimaryClassifiers;
  }
 
  inline std::string Trk::PrimaryTruthClassifier::nameOfClassifier() const {
    return "reconstructableForInDet";
  }
 
  inline std::string Trk::PrimaryTruthClassifier::classificationAsString(unsigned int i) const {
    Trk::TruthClassification::PrimaryClassifierStrings PC;
    return PC.names[i];
  }
 
} // end namespace
#endif