Merge pull request #493 from yangtj207/feature/tjyang_cvn

Save CVN scores in CAFMaker.

Author: ibsafa

sbncode/CAFMaker/CAFMakerParams.h

@@ -442,6 +442,12 @@ namespace caf
       true
     };
 
+    Atom<string> CVNLabel {
+      Name("CVNLabel"),
+      Comment("Label of CVN scores."),
+      "cvn" 
+    };
+    
   };
 }
 

sbncode/CAFMaker/CAFMaker_module.cc

@@ -1680,6 +1680,12 @@ void CAFMaker::produce(art::Event& evt) noexcept {
     const sbn::TPCPMTBarycenterMatch *barycenterMatch
       = foTPCPMTBarycenterMatch.isValid()? foTPCPMTBarycenterMatch.at(0).get(): nullptr;
 
+    art::FindOneP<lcvn::Result> foCVNResult =
+      FindOnePStrict<lcvn::Result>(sliceList, evt,
+          fParams.CVNLabel() + slice_tag_suff);
+    const lcvn::Result *cvnResult
+      = foCVNResult.isValid()? foCVNResult.at(0).get(): nullptr;
+    
     art::FindManyP<larpandoraobj::PFParticleMetadata> fmPFPMeta =
       FindManyPStrict<larpandoraobj::PFParticleMetadata>(fmPFPart, evt,
                fParams.PFParticleLabel() + slice_tag_suff);
@@ -1905,7 +1911,8 @@ void CAFMaker::produce(art::Event& evt) noexcept {
     FillSliceOpT0Finder(slcOpT0, recslc);
     FillSliceBarycenter(slcHits, slcSpacePoints, recslc);
     FillTPCPMTBarycenterMatch(barycenterMatch, recslc);
-
+    FillCVNScores(cvnResult, recslc);
+    
     // select slice
     if (!SelectSlice(recslc, fParams.CutClearCosmic())) continue;
 

sbncode/CAFMaker/CMakeLists.txt

@@ -25,6 +25,7 @@ art_make_library( LIBRARY_NAME sbncode_CAFMaker
                   caf_RecoUtils
                   lardataobj::AnalysisBase
                   lardataobj::RecoBase
+                  larrecodnn::CVN_func
                   larcorealg::Geometry
                   larcore::Geometry_Geometry_service
                   larcorealg::GeoAlgo

sbncode/CAFMaker/FillReco.cxx

@@ -1003,6 +1003,24 @@ namespace caf
     }
   }
 
+  void FillCVNScores(const lcvn::Result *cvnResult,
+                           caf::SRSlice& slice)
+  { 
+
+    if ( cvnResult != nullptr ) {
+      auto const & cvn = cvnResult->fOutput;
+      if (cvn.size()==1 && cvn[0].size()==4){
+        slice.cvn.numuscore    = cvn[0][0];
+        slice.cvn.nuescore     = cvn[0][1];
+        slice.cvn.cosmicscore  = cvn[0][2];
+        slice.cvn.ncscore      = cvn[0][3];
+      }
+      else{
+        std::cout<<"CVN result does not have the correct dimensions."<<std::endl;
+      }
+    }
+  }
+  
   //......................................................................
 
   void FillPFPRazzled(const art::Ptr<sbn::MVAPID> razzled,

sbncode/CAFMaker/FillReco.h

@@ -24,6 +24,7 @@
 #include "lardataobj/AnalysisBase/T0.h"
 #include "lardataobj/RecoBase/PFParticleMetadata.h"
 #include "lardataobj/RecoBase/MCSFitResult.h"
+#include "larrecodnn/CVN/func/Result.h"
 #include "sbnobj/Common/Reco/Stub.h"
 #include "sbnobj/Common/Reco/RangeP.h"
 #include "sbnobj/Common/Reco/ShowerSelectionVars.h"
@@ -228,6 +229,9 @@ namespace caf
   void FillTPCPMTBarycenterMatch(const sbn::TPCPMTBarycenterMatch *matchInfo,
                            caf::SRSlice& slice);
 
+  void FillCVNScores(const lcvn::Result *cvnResult,
+                     caf::SRSlice& slice);
+
   void FillPFPRazzled(const art::Ptr<sbn::MVAPID> razzled,
                       caf::SRPFP& srpfp,
                       bool allowEmpty = false);