TrackCalorimetryAlg.h

Go to the documentation of this file.

#ifndef TRACKCALORIMETRYALG_H
#define TRACKCALORIMETRYALG_H

#include "larcorealg/Geometry/fwd.h"
#include "larreco/Calorimetry/CalorimetryAlg.h"

#include "fhiclcpp/ParameterSet.h"

#include "TVector3.h"

#include <iostream>
#include <set>

namespace anab {
  class Calorimetry;
}
namespace detinfo {
  class DetectorClocksData;
  class DetectorPropertiesData;
}
namespace geo {
  struct PlaneID;
  struct WireID;
}
namespace recob {
  class Hit;
  class Track;
}

namespace calo {
  class TrackCalorimetryAlg;
}

class calo::TrackCalorimetryAlg {
public:
  TrackCalorimetryAlg(fhicl::ParameterSet const& p);

  void ExtractCalorimetry(detinfo::DetectorClocksData const& clock_data,
                          detinfo::DetectorPropertiesData const& det_prop,
                          std::vector<recob::Track> const&,
                          std::vector<recob::Hit> const&,
                          std::vector<std::vector<size_t>> const&,
                          std::vector<anab::Calorimetry>&,
                          std::vector<size_t>&,
                          geo::WireReadoutGeom const&);

private:
  CalorimetryAlg caloAlg;
  unsigned int fNHitsToDetermineStart;

  struct HitProperties {
    HitProperties() = default;
    HitProperties(float q, float dqdx, float dedx, float p, TVector3 pos, float pf)
      : charge(q), dQdx(dqdx), dEdx(dedx), pitch(p), xyz(pos), path_fraction(pf)
    {}
    float charge;
    float dQdx;
    float dEdx;
    float pitch;
    TVector3 xyz;
    float path_fraction;
    void Print() const
    {
      std::cout << "\tCharge " << charge << "  dQdx " << dQdx << "  dEdx " << dEdx << "  pitch "
                << pitch << "  (x,y,z) (" << xyz.X() << "," << xyz.Y() << "," << xyz.Z() << ")"
                << " path_fraction " << path_fraction << std::endl;
    }
  };
  struct HitPropertySorter {
    bool operator()(HitProperties const& i, HitProperties const& j) const
    {
      return i.path_fraction < j.path_fraction;
    }
  };

  using HitPropertiesMultiset_t = std::multiset<HitProperties, HitPropertySorter>;

  std::vector<float> CreatePathLengthFractionVector(recob::Track const& track);

  void AnalyzeHit(detinfo::DetectorClocksData const&,
                  detinfo::DetectorPropertiesData const&,
                  recob::Hit const&,
                  recob::Track const&,
                  std::vector<std::pair<geo::WireID, float>> const&,
                  std::vector<float> const&,
                  HitPropertiesMultiset_t&,
                  geo::WireReadoutGeom const&);

  bool IsInvertedTrack(HitPropertiesMultiset_t const&);

  void MakeCalorimetryObject(HitPropertiesMultiset_t const& hpm,
                             recob::Track const& track,
                             size_t const& i_track,
                             std::vector<anab::Calorimetry>& caloVector,
                             std::vector<size_t>& assnTrackCaloVector,
                             geo::PlaneID const& planeID);

  void PrintHitPropertiesMultiset(HitPropertiesMultiset_t const& hpm);
};

#endif