ToFClusterAlg.h

Go to the documentation of this file.

#pragma once

#include <vector>
#include <map>
#include <string>

#include "canvas/Persistency/Common/Ptr.h"
#include "canvas/Utilities/Exception.h"
#include "BeamlineRecoBase/BeamlineDigit.h"
#include "BeamlineUtils/BeamlineGeometry.h"

namespace beamlinereco {

  struct ToFPulseCluster {
    beamlinegeo::ToFCounter counter;
    std::vector<art::Ptr<brb::BeamlineDigit>> pulses;
    double time;
  };

  class ToFClusterAlg {
  public:
    ToFClusterAlg() { Reset(); }
    void Reset();

    void                         FindClusters(std::vector<art::Ptr<brb::BeamlineDigit>> digits, double clusterWindow);
    std::vector<ToFPulseCluster> GetClusters(beamlinegeo::ToFCounter counter) const;

  private:
    std::vector<ToFPulseCluster> _dsClusters;
    std::vector<ToFPulseCluster> _usClusters;
    std::vector<ToFPulseCluster> _sipmClusters;
    double                       _clusterWindow;

  };

  void ToFClusterAlg::Reset() {
    _usClusters.clear();
    _dsClusters.clear();
    _sipmClusters.clear();
    _clusterWindow = -999;
  }

  void ToFClusterAlg::FindClusters(std::vector<art::Ptr<brb::BeamlineDigit>> digits, double clusterWindow) {
    _clusterWindow = clusterWindow;

    for (auto d : digits) {
      // Get correct cluster list for digit
      std::vector<ToFPulseCluster> *clusters;
      switch (d->ChannelID().Detector) {
      case beamlinegeo::ToFCounter::US:     clusters = &_usClusters; break;
      case beamlinegeo::ToFCounter::DS:     clusters = &_dsClusters; break;
      case beamlinegeo::ToFCounter::DSSiPM: clusters = &_sipmClusters; break;
      default: continue;
      }
      // Find cluster containing pulse (if any)
      double starttime = d->StartTimeInNanoSec();
      auto clusterit = std::find_if(clusters->begin(), clusters->end(),
        [starttime,clusterWindow](ToFPulseCluster &cluster){
          return fabs(starttime - cluster.time) < clusterWindow;
        });

      if (clusterit == clusters->end()) { // Could not find one
        // insert new cluster
        ToFPulseCluster cluster;
        cluster.time = starttime;
        cluster.counter = beamlinegeo::ToFCounter(d->ChannelID().Detector);
        cluster.pulses.push_back(d);
        clusters->push_back(cluster);
      } else { // Found cluster
        clusterit->pulses.push_back(d);
      }
    }
  }

  std::vector<ToFPulseCluster> ToFClusterAlg::GetClusters(beamlinegeo::ToFCounter counter) const {
    switch(counter) {
    case beamlinegeo::ToFCounter::US:     return _usClusters;
    case beamlinegeo::ToFCounter::DS:     return _dsClusters;
    case beamlinegeo::ToFCounter::DSSiPM: return _sipmClusters;
    default: throw art::Exception(art::errors::StdException)
               << "Bad time of flight counter: " << counter << "." << std::endl;
    }
  }
}