ITCSSelectionTool.h

Go to the documentation of this file.

#include "sbnobj/Common/Calibration/TrackCaloSkimmerObj.h"
#include "fhiclcpp/ParameterSet.h"

#include <stdlib.h>
#include <time.h> 

namespace sbn {

class ITCSSelectionTool {
public:
  ITCSSelectionTool(const fhicl::ParameterSet &p):
    fAllowT0(p.get<std::vector<bool>>("AllowT0", {})),
    fRequireT0(p.get<bool>("RequireT0", true)),
    fInvert(p.get<bool>("Invert", false)),
    fNPreScale(p.get<unsigned>("PreScale", 1)),
    fISelect(0) 
  {
      // Set ISelect to a random number
      // so that events get prescaled uniformly across runs
      srand (time(NULL)); // random seed
      fISelect = rand();
  }

  virtual ~ITCSSelectionTool() noexcept = default;

  /// For external modules to call: run the actual selection
  bool DoSelect(const TrackInfo &t) {

    bool has_good_t0 = t.whicht0 >= 0;
    // Check if the T0 is allowed
    if (!fAllowT0.empty()) {
      if (t.whicht0 >= 0 && ((unsigned)t.whicht0 >= fAllowT0.size() || !fAllowT0[t.whicht0])) {
        has_good_t0 = false;
      }
    }

    if (!has_good_t0 && fRequireT0) return false;

    bool selected = Select(t) == !fInvert /* implement inversion */;

    // update index
    fISelect += selected;

    return selected && (fISelect % fNPreScale == 0) /* implement prescale */;
  }

  unsigned GetPrescale() const { return fNPreScale; }

protected:
  /// For children to implement: Whether to select a given track
  virtual bool Select(const TrackInfo &t) = 0;

  std::vector<bool> fAllowT0;
  bool fRequireT0;
  bool fInvert;
  unsigned fNPreScale;
  unsigned fISelect;
};

} // end namespace sbn