Replace throughgoing selection and add exiting selection as previous

Author: henrylay97

sbncode/Calibration/CMakeLists.txt

@@ -70,6 +70,11 @@ cet_build_plugin(TrackCaloSkimmerSelectAllTrack art::tool
                             ROOT::X3d
 )
 
+cet_build_plugin(TrackCaloSkimmerSelectThroughgoingTrack art::tool
+                      LIBRARIES
+                            larcore::Geometry_Geometry_service
+)
+
 cet_build_plugin ( DataSelect art::module
            LIBRARIES
                ifdh_art::IFDH_service

sbncode/Calibration/TrackCaloSkimmerSelectThroughgoingTrack_tool.cc

@@ -0,0 +1,79 @@
+#include "art/Utilities/ToolMacros.h"
+#include "fhiclcpp/ParameterSet.h"
+
+#include "larcore/Geometry/Geometry.h"
+#include "larcore/CoreUtils/ServiceUtil.h"
+#include "larcorealg/Geometry/GeometryCore.h"
+
+#include "ITCSSelectionTool.h"
+
+namespace sbn {
+
+  class TrackCaloSkimmerSelectThroughgoingTrack: public ITCSSelectionTool {
+  public:
+
+    TrackCaloSkimmerSelectThroughgoingTrack(const fhicl::ParameterSet &p);
+    ~TrackCaloSkimmerSelectThroughgoingTrack() {}
+
+    bool Select(const TrackInfo &t) override;
+
+  private:
+    double fFVInsetMinX;
+    double fFVInsetMaxX;
+    double fFVInsetMinY;
+    double fFVInsetMaxY;
+    double fFVInsetMinZ;
+    double fFVInsetMaxZ;
+    bool   fCheckFiducialX;
+
+    std::vector<geo::BoxBoundedGeo> fTPCVolumes;
+    geo::BoxBoundedGeo fActiveVolume, fFiducialVolume;
+  };
+
+  TrackCaloSkimmerSelectThroughgoingTrack::TrackCaloSkimmerSelectThroughgoingTrack(const fhicl::ParameterSet &p):
+    ITCSSelectionTool(p),
+    fFVInsetMinX(p.get<double>("FVInsetMinX")),
+    fFVInsetMaxX(p.get<double>("FVInsetMaxX")),
+    fFVInsetMinY(p.get<double>("FVInsetMinY")),
+    fFVInsetMaxY(p.get<double>("FVInsetMaxY")),
+    fFVInsetMinZ(p.get<double>("FVInsetMinZ")),
+    fFVInsetMaxZ(p.get<double>("FVInsetMaxZ")),
+    fCheckFiducialX(p.get<bool>("CheckFiducialX"))
+  {
+    const geo::GeometryCore *geometry = lar::providerFrom<geo::Geometry>();
+
+    for(auto const &cryoid: geometry->Iterate<geo::CryostatID>())
+      {
+        for(auto const& TPC : geometry->Iterate<geo::TPCGeo>(cryoid))
+          fTPCVolumes.push_back(TPC.ActiveBoundingBox());
+      }
+
+    double XMin = std::min_element(fTPCVolumes.begin(), fTPCVolumes.end(), [](auto &lhs, auto &rhs) { return lhs.MinX() < rhs.MinX(); })->MinX();
+    double YMin = std::min_element(fTPCVolumes.begin(), fTPCVolumes.end(), [](auto &lhs, auto &rhs) { return lhs.MinY() < rhs.MinY(); })->MinY();
+    double ZMin = std::min_element(fTPCVolumes.begin(), fTPCVolumes.end(), [](auto &lhs, auto &rhs) { return lhs.MinZ() < rhs.MinZ(); })->MinZ();
+
+    double XMax = std::max_element(fTPCVolumes.begin(), fTPCVolumes.end(), [](auto &lhs, auto &rhs) { return lhs.MaxX() < rhs.MaxX(); })->MaxX();
+    double YMax = std::max_element(fTPCVolumes.begin(), fTPCVolumes.end(), [](auto &lhs, auto &rhs) { return lhs.MaxY() < rhs.MaxY(); })->MaxY();
+    double ZMax = std::max_element(fTPCVolumes.begin(), fTPCVolumes.end(), [](auto &lhs, auto &rhs) { return lhs.MaxZ() < rhs.MaxZ(); })->MaxZ();
+
+    fActiveVolume = geo::BoxBoundedGeo(XMin, XMax, YMin, YMax, ZMin, ZMax);
+
+    fFiducialVolume = geo::BoxBoundedGeo(fActiveVolume.MinX() + fFVInsetMinX, fActiveVolume.MaxX() - fFVInsetMaxX,
+                                         fActiveVolume.MinY() + fFVInsetMinY, fActiveVolume.MaxY() - fFVInsetMaxY,
+                                         fActiveVolume.MinZ() + fFVInsetMinZ, fActiveVolume.MaxZ() - fFVInsetMaxZ);
+  }
+
+  bool TrackCaloSkimmerSelectThroughgoingTrack::Select(const TrackInfo &t)
+  {
+    geo::Point_t start {t.start.x, t.start.y, t.start.z};
+    bool start_is_non_fid = fCheckFiducialX ? !fFiducialVolume.ContainsPosition(start) : !fFiducialVolume.ContainsYZ(start.Y(), start.Z());
+
+    geo::Point_t end {t.end.x, t.end.y, t.end.z};
+    bool end_is_non_fid = fCheckFiducialX ? !fFiducialVolume.ContainsPosition(end) : !fFiducialVolume.ContainsYZ(end.Y(), end.Z());
+
+    return start_is_non_fid && end_is_non_fid;
+  }
+
+  DEFINE_ART_CLASS_TOOL(TrackCaloSkimmerSelectThroughgoingTrack)
+
+}

sbncode/Calibration/fcl/sbnd/sbnd_trackcalo_skimmer.fcl

@@ -56,10 +56,10 @@ stopping_selection: {
   EndMediandQdxCut: 1600
   NumberTimeSamples: 3400
   MediandQdxRRMax: 5.
-  CheckFiducialX: false
+  CheckFiducialX: true
 }
 
-throughgoing_selection: {
+exiting_selection: {
   tool_type: TrackCaloSkimmerSelectStoppingTrack
   Invert: true
   PreScale: 1
@@ -74,15 +74,26 @@ throughgoing_selection: {
   EndMediandQdxCut: -1
   RequireDownwards: false
   NumberTimeSamples: 3400
-  CheckFiducialX: false
+  CheckFiducialX: true
 }
 
 a2c_selection: {
   tool_type: TrackCaloSkimmerSelectAnode2CathodeTrack
   TickCut: 2500
 }
 
+throughgoing_selection: {
+  tool_type: TrackCaloSkimmerSelectThroughgoingTrack
+  FVInsetMinX: 5
+  FVInsetMaxX: 5
+  FVInsetMinY: 5
+  FVInsetMaxY: 5
+  FVInsetMinZ: 5
+  FVInsetMaxZ: 5
+  CheckFiducialX: true
+}
+
 caloskim_nodigits_goldentracks: @local::caloskim_nodigits
-caloskim_nodigits_goldentracks.SelectionTools: [@local::stopping_selection, @local::a2c_selection, @local::throughgoing_selection]
+caloskim_nodigits_goldentracks.SelectionTools: [@local::stopping_selection, @local::a2c_selection, @local::throughgoing_selection, @local::exiting_selection]
 
 END_PROLOG