/cvmfs/atlas.cern.ch/repo/sw/ASG/AnalysisBase/2.4.31/ParticleJetTools/ParticleJetTools/CopyTruthJetParticles.h

// this file is -*- C++ -*-
#ifndef COPYTRUTHJETPARTICLES_H
#define COPYTRUTHJETPARTICLES_H

#include "ParticleJetTools/CopyTruthParticles.h"
#include "AsgTools/ToolHandle.h"
#include "xAODTruth/TruthParticle.h"
#include "MCTruthClassifier/MCTruthClassifierDefs.h"
#include "MCTruthClassifier/IMCTruthClassifier.h"
#include <vector>
#include <map>

class CopyTruthJetParticles : public CopyTruthParticles {
ASG_TOOL_CLASS2(CopyTruthJetParticles, CopyTruthParticles, IJetExecuteTool)
public:

  CopyTruthJetParticles(const std::string& name);

  virtual int execute() const;

  bool classifyJetInput(const xAOD::TruthParticle* tp, int barcodeOffset,
                        std::vector<const xAOD::TruthParticle*>& promptLeptons,
            std::map<const xAOD::TruthParticle*,MCTruthPartClassifier::ParticleOrigin>& originMap) const;


  bool classify(const xAOD::TruthParticle* ) const {return false;}

private:
  // Options for storate
  bool m_includeNu; 
  bool m_includeMu; 
  bool m_includePromptLeptons; 
  //  bool m_includeTau; //!< Include particles from tau decays

  bool isPrompt( const xAOD::TruthParticle* tp,
         std::map<const xAOD::TruthParticle*,MCTruthPartClassifier::ParticleOrigin>& originMap ) const;
  // bool fromTau( const xAOD::TruthParticle* tp,
  //        std::map<const xAOD::TruthParticle*,MCTruthPartClassifier::ParticleOrigin>& originMap ) const;


  MCTruthPartClassifier::ParticleOrigin getPartOrigin(const xAOD::TruthParticle* tp,
                              std::map<const xAOD::TruthParticle*,MCTruthPartClassifier::ParticleOrigin>& originMap) const;

  float m_maxAbsEta;

  int m_barcodeOffset;
  
  int m_barcodeFromMetadata;

  float m_photonCone;

  mutable ToolHandle<IMCTruthClassifier> m_classif;
};


#endif

---