FixHepMC.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2019 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef XAOD_ANALYSIS
 
#ifndef FIX_HEPMC
#define FIX_HEPMC
 
#include "GeneratorModules/GenBase.h"
#include "AtlasHepMC/GenParticle.h"
#include "TruthUtils/Loops.h"
#include <set>
#include <memory>
 
 
class FixHepMC : public GenBase {
public:
 
  FixHepMC(const std::string& name, ISvcLocator* pSvcLocator);
  StatusCode execute();
  StatusCode finalize();
 
 
private:
 
 
  bool isPID0(const HepMC::ConstGenParticlePtr& p) const;
  bool isNonTransportableInDecayChain(const HepMC::ConstGenParticlePtr& p) const;
  bool isSimpleLoop(const HepMC::ConstGenParticlePtr& p) const;
  bool fromDecay(const HepMC::ConstGenParticlePtr& p, std::shared_ptr<std::set<int> >& storage) const;
 
 
  bool m_killLoops;   // Kill loops?
  bool m_killPDG0;    // Kill PDG0 particles?
  bool m_cleanDecays; // Clean decays?
 
 
  long m_loopKilled;
  long m_pdg0Killed;
  long m_decayCleaned;
  long m_totalSeen;
  long m_replacedPIDs;
   
  std::map<int,int> m_pidmap; 
 
 
  MC::Loops<HepMC::GenEvent,HepMC::ConstGenParticlePtr,HepMC::ConstGenVertexPtr> m_looper; 
 
 
};
 
 
#endif
 
#endif