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?
  bool m_purgeUnstableWithoutEndVtx; // Remove unstable particles without decay vertex?
  bool m_ignoreSemiDisconnected; // Ignore semi-disconnected particles (normal in Sherpa)
  std::string m_forced_momentum{""}; // Force momentum unit for the event
  std::string m_forced_length{""}; // Force length unit for the event
  bool m_unitsFix;    // Attempt to identify and fix momentum units problems


  long m_loopKilled;
  long m_pdg0Killed;
  long m_decayCleaned;
  long m_unstablePurged;
  long m_totalSeen;
  long m_replacedPIDs;
   
  std::map<int,int> m_pidmap; 
  std::map<int,int> m_replacedpid_counts; 
 
 
  MC::Loops<HepMC::GenEvent,HepMC::ConstGenParticlePtr,HepMC::ConstGenVertexPtr> m_looper; 
 
 
};
 
 
#endif
 
#endif