ClusterMergeHelper.cxx

Go to the documentation of this file.

//
//  ClusterMergeHelper source
//

#include "ClusterMergeHelper.h"

#include "art/Framework/Principal/Event.h"
#include "art/Framework/Principal/Handle.h"
#include "art/Framework/Services/Registry/ServiceHandle.h"
#include "canvas/Persistency/Common/FindManyP.h"

#include "larcore/Geometry/Geometry.h"
#include "lardata/Utilities/AssociationUtil.h"
#include "larreco/RecoAlg/ClusterRecoUtil/ClusterParams.h"
#include "larreco/RecoAlg/ClusterRecoUtil/ClusterParamsAlg.h"
#include "larreco/RecoAlg/ClusterRecoUtil/LazyClusterParamsAlg.h"

namespace cluster {

  //####################################################################################################
  void ClusterMergeHelper::SetClusters(
    util::GeometryUtilities const& gser,
    const std::vector<std::vector<art::Ptr<recob::Hit>>>& clusters)
  //####################################################################################################
  {
    fInputClusters.clear();
    fOutputClusters.clear();

    std::vector<std::vector<util::PxHit>> px_clusters(clusters.size(), std::vector<util::PxHit>());

    fInputClusters.resize(clusters.size(), std::vector<art::Ptr<recob::Hit>>());

    for (size_t cluster_index = 0; cluster_index < clusters.size(); ++cluster_index) {

      px_clusters.at(cluster_index).resize(clusters.at(cluster_index).size(), util::PxHit());

      fInputClusters.at(cluster_index).resize(clusters.at(cluster_index).size());

      for (size_t hit_index = 0; hit_index < clusters.at(cluster_index).size(); ++hit_index) {

        px_clusters.at(cluster_index).at(hit_index).plane =
          clusters.at(cluster_index).at(hit_index)->WireID().Plane;
        px_clusters.at(cluster_index).at(hit_index).w =
          clusters.at(cluster_index).at(hit_index)->WireID().Wire * fGeoU.WireToCm();
        px_clusters.at(cluster_index).at(hit_index).t =
          clusters.at(cluster_index).at(hit_index)->PeakTime() * fGeoU.TimeToCm();
        px_clusters.at(cluster_index).at(hit_index).charge =
          clusters.at(cluster_index).at(hit_index)->Integral();

        fInputClusters.at(cluster_index).at(hit_index) = clusters.at(cluster_index).at(hit_index);
      }
    }

    SetClusters(gser, px_clusters);
  }

  //##################################################################################################
  void ClusterMergeHelper::SetClusters(util::GeometryUtilities const& gser,
                                       const art::Event& evt,
                                       const std::string& cluster_module_label)
  //##################################################################################################
  {

    art::Handle<std::vector<recob::Cluster>> clusters_h;
    evt.getByLabel(cluster_module_label, clusters_h);

    if (!(clusters_h.isValid()))

      throw cet::exception(__FUNCTION__)
        << "\033[93m"
        << " Failed to retrieve recob::Cluster with label: " << cluster_module_label.c_str()
        << "\033[00m" << std::endl;

    std::vector<std::vector<art::Ptr<recob::Hit>>> cluster_hits_v;
    cluster_hits_v.reserve(clusters_h->size());

    art::FindManyP<recob::Hit> hit_m(clusters_h, evt, cluster_module_label);

    for (size_t i = 0; i < clusters_h->size(); ++i)
      cluster_hits_v.push_back(hit_m.at(i));

    SetClusters(gser, cluster_hits_v);
  }

  //################################
  void ClusterMergeHelper::Process(util::GeometryUtilities const& gser)
  //################################
  {
    if (fMgr.GetClusters().size())
      throw cet::exception(__PRETTY_FUNCTION__)
        << "\033[93m"
        << "Merged cluster set not empty... Called Process() twice?"
        << "\033[00m" << std::endl;

    fMgr.Process(gser);

    // Now create output clusters
    auto res = fMgr.GetBookKeeper();

    std::vector<std::vector<unsigned short>> out_clusters = res.GetResult();
    fOutputClusters.clear();

    fOutputClusters.reserve(out_clusters.size());

    for (auto const& cluster_index_v : out_clusters) {

      std::vector<art::Ptr<recob::Hit>> out_cluster;

      for (auto const& cluster_index : cluster_index_v) {
        out_cluster.reserve(out_cluster.size() + fInputClusters.at(cluster_index).size());
        for (auto const& hit_ptr : fInputClusters.at(cluster_index))
          out_cluster.push_back(hit_ptr);
      }
      fOutputClusters.push_back(out_cluster);
    }
  }

  //######################################################################################################
  const std::vector<std::vector<art::Ptr<recob::Hit>>>& ClusterMergeHelper::GetMergedClusterHits()
    const
  //######################################################################################################
  {
    if (!fOutputClusters.size())
      throw cet::exception(__FUNCTION__)
        << "\033[93m"
        << "You must call Process() before calling " << __FUNCTION__ << " to retrieve result."
        << "\033[00m" << std::endl;

    return fOutputClusters;
  }

  //#####################################################################################
  const std::vector<cluster::ClusterParamsAlg>& ClusterMergeHelper::GetMergedCPAN() const
  //#####################################################################################
  {
    if (!fOutputClusters.size())
      throw cet::exception(__FUNCTION__)
        << "\033[93m"
        << "You must call Process() before calling " << __FUNCTION__ << " to retrieve result."
        << "\033[00m" << std::endl;

    return fMgr.GetClusters();
  }

  void ClusterMergeHelper::AppendResult(util::GeometryUtilities const& gser,
                                        art::Event& ev,
                                        std::vector<recob::Cluster>& out_clusters,
                                        art::Assns<recob::Cluster, recob::Hit>& assns) const
  {

    if (!fOutputClusters.size())

      throw cet::exception(__FUNCTION__)
        << "\033[93m"
        << "You must call Process() before calling " << __FUNCTION__ << " to retrieve result."
        << "\033[00m" << std::endl;

    art::ServiceHandle<geo::Geometry const> geo;

    // Store output
    for (size_t out_index = 0; out_index < GetMergedCPAN().size(); ++out_index) {

      // To save typing let's just retrieve const cluster_params instance
      const cluster_params& res = GetMergedCPAN()[out_index].GetParams();

      // this "algo" is actually parroting its cluster_params
      LazyClusterParamsAlg algo(res);

      std::vector<art::Ptr<recob::Hit>> const& hits = GetMergedClusterHits().at(out_index);

      // the full plane needed but not a part of cluster_params...
      // get the one from the first hit
      geo::PlaneID plane; // invalid by default
      if (!hits.empty()) plane = hits.front()->WireID().planeID();

      // View_t needed but not a part of cluster_params, so retrieve it here
      geo::View_t view_id = geo->Plane(plane).View();

      // Push back a new cluster data product with parameters copied from cluster_params
      out_clusters.emplace_back(res.start_point.w / fGeoU.WireToCm(), // start_wire
                                0.,                                   // sigma_start_wire
                                res.start_point.t / fGeoU.TimeToCm(), // start_tick
                                0.,                                   // sigma_start_tick
                                algo.StartCharge(gser).value(),
                                algo.StartAngle().value(),
                                algo.StartOpeningAngle().value(),
                                res.end_point.w / fGeoU.WireToCm(), // end_wire
                                0.,                                 // sigma_end_wire
                                res.end_point.t / fGeoU.TimeToCm(), // end_tick
                                0.,                                 // sigma_end_tick
                                algo.EndCharge(gser).value(),
                                algo.EndAngle().value(),
                                algo.EndOpeningAngle().value(),
                                algo.Integral().value(),
                                algo.IntegralStdDev().value(),
                                algo.SummedADC().value(),
                                algo.SummedADCStdDev().value(),
                                algo.NHits(),
                                algo.MultipleHitDensity(),
                                algo.Width(gser),
                                out_clusters.size(), // ID
                                view_id,
                                plane,
                                recob::Cluster::Sentry);

      util::CreateAssn(ev, out_clusters, GetMergedClusterHits().at(out_index), assns);
    }
  }

} // namespace cluster