SliceRemover.cxx

Go to the documentation of this file.

////////////////////////////////////////////////////////////////////////
// \file:   SliceRemover.cxx
// \brief   Remove hits associated with slices.
// \author  Justin Vasel <justin.vasel@gmail.com>
// \date    2020-01-21
////////////////////////////////////////////////////////////////////////

// ART includes
#include "art/Framework/Principal/Event.h"
#include "art/Framework/Principal/Handle.h"
#include "art/Framework/Services/Registry/ServiceHandle.h"
#include "canvas/Persistency/Common/Ptr.h"
#include "fhiclcpp/ParameterSet.h"

// ROOT includes
#include "TVector3.h"

// NOvASOFT includes
#include "RecoBase/CellHit.h"
#include "RecoBase/Cluster.h"
#include "Supernova/BackgroundRejection/HitVetoMap.h"
#include "Supernova/BackgroundRejection/SliceRemover.h"

// ............................................................................
sn::SliceRemover::SliceRemover(sn::HitVetoMap& vetoMap):
fVetoMap(vetoMap),
fNumRemoved(0)
{
}

// ............................................................................
sn::SliceRemover::~SliceRemover() {}

// ............................................................................
void sn::SliceRemover::remove(std::vector<art::Ptr<rb::CellHit>>& hits, std::vector<art::Ptr<rb::Cluster>>& clusters)
{
  std::cout << "Cleaning up " << clusters.size() << " clusters." << std::endl;
  for (size_t clusterId=0; clusterId<clusters.size(); ++clusterId) {
    art::Ptr<rb::Cluster> cluster = clusters.at(clusterId);

    if (cluster->IsNoise()) continue;

    art::PtrVector<rb::CellHit> clusterHits = cluster->AllCells();
    for (art::Ptr<rb::CellHit> clusterHit : clusterHits) {
      this->fVetoMap.AddHit(*clusterHit);
      ++this->fNumRemoved;
    }
  }

  return;
}