ClearParticleLists.cc

Go to the documentation of this file.

#include "ClearParticleLists.h"
#include <utl/Reader.h>
#include <fwk/CentralConfig.h>
#include <utl/ErrorLogger.h>

#include <evt/Event.h>
#include <sevt/SEvent.h>
#include <mevt/MEvent.h>

using namespace std;
using namespace utl;
using namespace fwk;


namespace ClearParticleLists {

  VModule::ResultFlag
  ClearParticleLists::Init()
  {
    Branch topB = CentralConfig::GetInstance()->GetTopBranch("ClearParticleLists");
    fClearSEvent = topB.GetChild("ClearSEvent").Get<bool>();
    fClearMEvent = topB.GetChild("ClearMEvent").Get<bool>();
    ostringstream info;
    info << "ClearSEvent: " << fClearSEvent << ", ClearMEvent: " << fClearMEvent;
    INFO(info);
    return eSuccess;
  }


  VModule::ResultFlag
  ClearParticleLists::Run(evt::Event& event)
  {
    int nLists = 0;
    if (fClearSEvent && event.HasSEvent()) {
      auto& sEvent = event.GetSEvent();
      for (auto& s : sEvent.StationsRange())
        if (s.HasSimData()) {
          s.GetSimData().ClearParticleList();
          ++nLists;
        }
    }
    if (fClearMEvent && event.HasMEvent()) {
      auto& mEvent = event.GetMEvent();
      for (auto& m : mEvent.CountersRange())
        if (m.HasSimData()) {
          auto& simData = m.GetSimData();
          simData.ClearGrdParticleList();
          simData.ClearUGrdParticleList();
          ++nLists;
        }
    }
    ostringstream info;
    info << "Cleared " << nLists << " particle lists.";
    INFO(info);
    return eSuccess;
  }

}