EventAction.cc

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2010-2011  Belle II Collaboration                         *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Guofu Cao, Andreas Moll, Martin Ritter                   *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#include <simulation/kernel/EventAction.h>
#include <framework/datastore/RelationArray.h>
#include <mdst/dataobjects/MCParticleGraph.h>
#include <simulation/kernel/SensitiveDetectorBase.h>
 
#include <string>
 
using namespace std;
using namespace Belle2;
using namespace Belle2::Simulation;
 
 
EventAction::EventAction(const std::string& mcCollectionName, MCParticleGraph& mcParticleGraph):
  G4UserEventAction(), m_mcCollectionName(mcCollectionName),  m_mcParticleGraph(mcParticleGraph)
{
  if (false) {
    G4Event* event;
    BeginOfEventAction(event);
    EndOfEventAction(event);
  }
}
 
EventAction::~EventAction()
{
 
}
 
void EventAction::BeginOfEventAction(const G4Event*)
{
  //Enable recording of Hits
  SensitiveDetectorBase::setActive(true);
}
 
 
void EventAction::EndOfEventAction(const G4Event*)
{
  //Disable recording of Hits
  SensitiveDetectorBase::setActive(false);
 
  //Create the final MCParticle list and update the indices of the MCParticle graph particles.
  m_mcParticleGraph.generateList(m_mcCollectionName, MCParticleGraph::c_clearParticles);
 
  //Build a TrackID to MCParticle index list
  vector<std::pair<unsigned int, bool> > indices;
  indices.resize(m_mcParticleGraph.size() + 1);
  for (unsigned int iParticle = 0; iParticle < m_mcParticleGraph.size(); ++iParticle) {
    MCParticleGraph::GraphParticle& currParticle = m_mcParticleGraph[iParticle];
    //assert(currParticle.getTrackID()<indices.size());
    indices[currParticle.getTrackID()] = std::make_pair(currParticle.getIndex() - 1, currParticle.getIgnore());
  }
  RelationArray::ReplaceVec<> indexReplacement(indices);
 
  //Update all registered MCParticle Relations and replace the TrackID by the final MCParticle id
  const std::map<std::string, RelationArray::EConsolidationAction>& relations = SensitiveDetectorBase::getMCParticleRelations();
  for (std::map<std::string, RelationArray::EConsolidationAction>::const_iterator it = relations.begin(); it != relations.end();
       ++it) {
    RelationArray mcPartRelation(it->first);
    if (mcPartRelation) mcPartRelation.consolidate(indexReplacement, RelationArray::Identity(), it->second);
  }
}