PHG4EventActionClearZeroEdep.cc

Go to the documentation of this file.

#include "PHG4EventActionClearZeroEdep.h"
#include <g4main/PHG4HitContainer.h>

#include <phool/getClass.h>

#include <Geant4/G4Event.hh>

#include <boost/foreach.hpp>

using namespace std;

//___________________________________________________
PHG4EventActionClearZeroEdep::PHG4EventActionClearZeroEdep( PHCompositeNode *node, const string &name ):
  topNode(node)
{
  AddNode(name);
}


void
PHG4EventActionClearZeroEdep::AddNode(const std::string &name)
{
  nodename_set.insert(name);
}

//___________________________________________________
void PHG4EventActionClearZeroEdep::EndOfEventAction(const G4Event* evt)
{
  BOOST_FOREACH(string node, nodename_set)
    {
      PHG4HitContainer* generic_hits =  findNode::getClass<PHG4HitContainer>( topNode , node.c_str());
      if (generic_hits)
        {
          generic_hits->RemoveZeroEDep();
        }
    }
}