cnnscore

Author: Mun Jung Jung

sbnobj/Common/Reco/CNNScore.cc

@@ -0,0 +1,10 @@
+#include "sbnobj/Common/Reco/CNNScore.h"
+
+sbn::PFPCNNScore::PFPCNNScore(float pfpTrackScore, float pfpShowerScore, float pfpNoiseScore, float pfpEndMichelScore, int nClusters)
+  : pfpTrackScore(pfpTrackScore)
+  , pfpShowerScore(pfpShowerScore)
+  , pfpNoiseScore(pfpNoiseScore)
+  , pfpEndMichelScore(pfpEndMichelScore)
+  , nClusters(nClusters)
+{
+}

sbnobj/Common/Reco/CNNScore.h

@@ -0,0 +1,22 @@
+#include <limits>
+
+#ifndef sbnobj_CNNScore_HH
+#define sbnobj_CNNScore_HH
+
+namespace sbn {
+
+  constexpr float nan_float = std::numeric_limits<float>::signaling_NaN();
+
+  class PFPCNNScore {
+  public:
+    PFPCNNScore(float pfpTrackScore = nan_float, float pfpShowerScore = nan_float, float pfpNoiseScore = nan_float, float pfpEndMichelScore = nan_float, int nClusters = 0);
+
+    float pfpTrackScore; //!< shower score for PFP, average over all associated hits
+    float pfpShowerScore; //!< shower score for PFP, average over all associated hits
+    float pfpNoiseScore; //!< shower score for PFP, average over all associated hits
+    float pfpEndMichelScore; //!< michel score for PFP, average over 48*48 area around the endpoint
+    int nClusters; //!< number of clusters used for calculating average score
+  };    
+}
+
+#endif

sbnobj/Common/Reco/classes.h

@@ -26,6 +26,7 @@
 #include "sbnobj/Common/Reco/MVAPID.h"
 #include "sbnobj/Common/Reco/CRUMBSResult.h"
 #include "sbnobj/Common/Reco/OpT0FinderResult.h"
+#include "sbnobj/Common/Reco/CNNScore.h"
 
 #include <utility>
 #include <vector>

sbnobj/Common/Reco/classes_def.xml

@@ -92,6 +92,21 @@
   <class name="art::Wrapper<art::Assns<recob::OpFlash, sbn::OpT0Finder>>" />
   <class name="art::Wrapper<art::Assns<sbn::OpT0Finder, recob::OpFlash, void>>" />
 
+  <class name="sbn::PFPCNNScore" ClassVersion="14">
+   <version ClassVersion="14" checksum="3262687246"/>
+   <version ClassVersion="13" checksum="587801397"/>
+   <version ClassVersion="12" checksum="2317961223"/>
+  </class>
+  <class name="std::vector<sbn::PFPCNNScore>" />
+  <class name="art::Wrapper<sbn::PFPCNNScore>" />
+  <class name="art::Wrapper<std::vector<sbn::PFPCNNScore>>" />
+
+  <class name="art::Assns<recob::PFParticle,sbn::PFPCNNScore,void>" />
+  <class name="art::Wrapper<art::Assns<recob::PFParticle,sbn::PFPCNNScore,void>>" />
+
+  <class name="art::Assns<sbn::PFPCNNScore,recob::PFParticle,void>" />
+  <class name="art::Wrapper<art::Assns<sbn::PFPCNNScore,recob::PFParticle,void>>" />
+
   <class name="sbn::ShowerDensityFit" ClassVersion="10">
    <version ClassVersion="10" checksum="942453570"/>
   </class>