PixelMapProducer.h

Go to the documentation of this file.

#ifndef LCVN_PIXELMAPPRODUCER_H
#define LCVN_PIXELMAPPRODUCER_H

#include <array>
#include <vector>

// Framework includes
#include "art/Framework/Principal/Handle.h"
#include "fhiclcpp/ParameterSet.h"

#include "lardataobj/RecoBase/Hit.h"
#include "lardataobj/RecoBase/SpacePoint.h"
#include "lardataobj/RecoBase/Wire.h"
#include "lardataobj/Simulation/SimChannel.h"
#include "larrecodnn/CVN/func/Boundary.h"
#include "larrecodnn/CVN/func/PixelMap.h"

#include "art/Framework/Services/Registry/ServiceHandle.h"
#include "larcore/Geometry/Geometry.h"
#include "larcore/Geometry/WireReadout.h"
#include "larcorealg/Geometry/GeometryCore.h"
#include "lardata/DetectorInfoServices/DetectorPropertiesService.h"

namespace lcvn {
  typedef std::vector<std::map<double, double>> Waveform;

  class HitHelper {
  public:
    HitHelper(recob::Hit hit, double thresh = 0.) : fHit(hit), fThreshold(thresh)
    {
      fGeometry = &*(art::ServiceHandle<geo::Geometry>());
    }

    virtual Waveform GetWaveform();
    virtual geo::WireID GetID();

  protected:
    recob::Hit fHit;
    double fThreshold;
    geo::GeometryCore const* fGeometry;
  };

  class WireHelper {
  public:
    WireHelper(recob::Wire wire, double thresh = 0.) : fWire(wire), fThreshold(thresh) {}

    virtual Waveform GetWaveform();
    virtual geo::WireID GetID();

  protected:
    recob::Wire fWire;
    double fThreshold;
    geo::WireReadoutGeom const* fWireReadoutGeom{
      &art::ServiceHandle<geo::WireReadout const>()->Get()};
  };

  class SimChannelHelper {
  public:
    SimChannelHelper(sim::SimChannel simchan, double thresh = 0.)
      : fSimchan(simchan), fThreshold(thresh)
    {}

    virtual Waveform GetWaveform();
    virtual geo::WireID GetID();

  protected:
    sim::SimChannel fSimchan;
    double fThreshold;
    geo::WireReadoutGeom const* fWireReadoutGeom{
      &art::ServiceHandle<geo::WireReadout const>()->Get()};
  };

  template <class T, class U>
  class PixelMapProducer {
  public:
    PixelMapProducer(unsigned int nWire, unsigned int nTdc, double tRes, double threshold = 0.);
    PixelMapProducer() = default;

    // overload constructor for inputs from fcl
    PixelMapProducer(const fhicl::ParameterSet& pset);

    unsigned int NWire() const { return fNWire; }
    unsigned int NTdc() const { return fNTdc; }
    double TRes() const { return fTRes; }
    unsigned int TotHits() const { return fTotHits; }
    double Threshold() const { return fThreshold; }
    bool MultipleDrifts() const { return fMultipleDrifts; }

    void SetTotHits(unsigned int tothits) { fTotHits = tothits; }
    void SetMultipleDrifts(bool multipledrifts = true) { fMultipleDrifts = multipledrifts; }

    virtual Boundary DefineBoundary(detinfo::DetectorPropertiesData const& detProp,
                                    const std::vector<const T*>& cluster);

    virtual void ConvertLocaltoGlobal(geo::WireID wireid,
                                      unsigned int& globalWire,
                                      unsigned int& globalPlane) const;

    virtual void ConvertLocaltoGlobalTDC(geo::WireID wireid,
                                         double localTDC,
                                         unsigned int& globalWire,
                                         unsigned int& globalPlane,
                                         double& globalTDC) const;

    virtual PixelMap CreateMap(detinfo::DetectorPropertiesData const& detProp,
                               const std::vector<art::Ptr<T>>& cluster);
    virtual PixelMap CreateMap(detinfo::DetectorPropertiesData const& detProp,
                               const std::vector<const T*>& cluster);

    virtual PixelMap CreateMapGivenBoundary(detinfo::DetectorPropertiesData const& detProp,
                                            const std::vector<const T*>& cluster,
                                            const Boundary& bound);

  private:
    unsigned int fNWire;   
    unsigned int fNTdc;    
    double fTRes;          
    unsigned int fTotHits; 
    double fThreshold;     
    bool
      fMultipleDrifts; 
  };

  extern template class PixelMapProducer<recob::Hit, lcvn::HitHelper>;
  extern template class PixelMapProducer<recob::Wire, lcvn::WireHelper>;
  extern template class PixelMapProducer<sim::SimChannel, lcvn::SimChannelHelper>;

  typedef PixelMapProducer<recob::Hit, lcvn::HitHelper> PixelMapHitProducer;
  typedef PixelMapProducer<recob::Wire, lcvn::WireHelper> PixelMapWireProducer;
  typedef PixelMapProducer<sim::SimChannel, lcvn::SimChannelHelper> PixelMapSimProducer;

}

#endif // CVN_PIXELMAPPRODUCER_H