Cerati/feature nugraph2

CMakeLists.txt

@@ -69,6 +69,7 @@ find_package(lardata REQUIRED )
 find_package(larsim REQUIRED )
 find_package(larevt REQUIRED )
 find_package(larreco REQUIRED )
+find_package(larrecodnn REQUIRED )
 find_package(larana REQUIRED )
 find_package(larpandora REQUIRED )
 find_package(larpandoracontent REQUIRED )

fcl/caf/cafmaker_defs.fcl

@@ -319,6 +319,9 @@ cafmaker.SystWeightLabels: ["genieweight", "fluxweight"]
 cafmaker.SaveGENIEEventRecord: true # save GENIE event record by default. Turn this off for data cafmaker fcl
 cafmaker.TPCPMTBarycenterMatchLabel: "tpcpmtbarycentermatch"
 cafmaker.TrackHitFillRREndCut: 30 # include entire PID region
+cafmaker.NuGraphSliceHitLabel: "nuslhits"
+cafmaker.NuGraphFilterLabel: "NuGraph:filter"
+cafmaker.NuGraphSemanticLabel: "NuGraph:semantic"
 
 # Add CAFMaker to the list of producers
 caf_preprocess_producers.cafmaker: @local::cafmaker

fcl/reco/Definitions/stage1_icarus_defs.fcl

@@ -22,6 +22,7 @@
 #include "supera_modules.fcl"
 #include "crtpmtmatching_parameters.fcl"
 #include "tpcpmtbarycentermatch_config.fcl"
+#include "nugraph_icarus.fcl"
 
 BEGIN_PROLOG
 
@@ -80,6 +81,13 @@ icarus_stage1_producers:
 
   tpcpmtbarycentermatchCryoE: @local::data_tpcpmtbarycentermatchproducer_east
   tpcpmtbarycentermatchCryoW: @local::data_tpcpmtbarycentermatchproducer_west
+
+  ## NuGraph
+  nuslhitsCryoE: @local::nuslhitsCryoE
+  nuslhitsCryoW: @local::nuslhitsCryoW
+  NuGraphCryoE: @local::NuGraphCryoE
+  NuGraphCryoW: @local::NuGraphCryoW
+
 }
 
 icarus_stage1_filters:
@@ -224,13 +232,19 @@ icarus_reco_cluster3DCryoW:        [ cluster3DCryoW ]
 
 icarus_reco_cluster3DCryoE:        [ cluster3DCryoE ]
 
-icarus_reco_pandoraGausCryoW:      [ pandoraGausCryoW,
+icarus_reco_pandoraGausCryoW:      [ pandoraGausCryoW ]
+
+icarus_reco_pandoraGausCryoE:      [ pandoraGausCryoE ]
+
+icarus_reco_postPandoraGausCryoW:  [ nuslhitsCryoW,
+                                     NuGraphCryoW,
                                      pandoraTrackGausCryoW,
                                      pandoraKalmanTrackGausCryoW,
                                      SBNShowerGausCryoW
                                    ]
 
-icarus_reco_pandoraGausCryoE:      [ pandoraGausCryoE,
+icarus_reco_postPandoraGausCryoE:  [ nuslhitsCryoE,
+                                     NuGraphCryoE,
                                      pandoraTrackGausCryoE,
                                      pandoraKalmanTrackGausCryoE,
                                      SBNShowerGausCryoE
@@ -265,6 +279,11 @@ icarus_pandora_Gauss:              [
                                      @sequence::icarus_reco_pandoraGausCryoW
                                    ]
 
+icarus_postPandora_Gauss:          [
+                                     @sequence::icarus_reco_postPandoraGausCryoE,
+                                     @sequence::icarus_reco_postPandoraGausCryoW
+                                   ]
+
 #Add flash matching
 icarus_reco_fm:                    [ fmatchCryoE,
                                      fmatchCryoW,

fcl/reco/Stage1/data/stage1_run2_icarus.fcl

@@ -6,6 +6,7 @@ physics.reco: [ @sequence::icarus_filter2D_cluster3D,
                 @sequence::icarus_pandora_Gauss,
                 @sequence::icarus_reco_fm,
                 @sequence::icarus_tpcpmtbarycentermatch,
+                @sequence::icarus_postPandora_Gauss,
                 @sequence::icarus_crttrack,
                 @sequence::icarus_crtt0tagging,
                 caloskimCalorimetryCryoE, caloskimCalorimetryCryoW]

fcl/reco/Stage1/mc/stage1_run2_1d_icarus_MC.fcl

@@ -6,6 +6,7 @@ physics.reco: [
                 @sequence::icarus_reco_Gauss1D_CryoW ,
                 @sequence::icarus_reco_fm,
                 @sequence::icarus_tpcpmtbarycentermatch,
+                @sequence::icarus_postPandora_Gauss,
                 @sequence::icarus_crttrack,
                 @sequence::icarus_crtt0tagging,
                 caloskimCalorimetryCryoE, caloskimCalorimetryCryoW,

fcl/reco/Stage1/mc/stage1_run2_icarus_MC.fcl

@@ -53,6 +53,7 @@ physics.reco: [
                 @sequence::icarus_reco_Gauss2D_CryoW ,
                 @sequence::icarus_reco_fm,
                 @sequence::icarus_tpcpmtbarycentermatch,
+                @sequence::icarus_postPandora_Gauss,
                 @sequence::icarus_crttrack,
                 @sequence::icarus_crtt0tagging,
                 caloskimCalorimetryCryoE, caloskimCalorimetryCryoW,

icaruscode/TPC/CMakeLists.txt

@@ -9,3 +9,4 @@ add_subdirectory(Simulation)
 add_subdirectory(Tracking)
 add_subdirectory(ICARUSWireCell)
 add_subdirectory(SPAna)
+add_subdirectory(NuGraph)

icaruscode/TPC/NuGraph/CMakeLists.txt

@@ -0,0 +1,58 @@
+cet_enable_asserts()
+
+set( nugraph_tool_lib_list
+       lardataobj::RecoBase
+       larrecodnn::NuGraphBaseTools
+       TorchScatter::TorchScatter
+       art::Framework_Core
+       hep_concurrency::hep_concurrency
+       art_plugin_types::tool
+)
+
+include_directories($ENV{HEP_HPC_INC})
+
+cet_build_plugin(ICARUSNuGraphLoader art::tool
+  LIBRARIES PRIVATE
+  ${nugraph_tool_lib_list}
+)
+
+simple_plugin(ICARUSNuSliceHitsProducer "module"
+    LIBRARIES PRIVATE
+    art::Framework_Core
+    lardataobj::RecoBase
+    sbnobj::Common_Reco
+)
+
+simple_plugin(ICARUSTrueNuSliceHitsProducer "module"
+    LIBRARIES PRIVATE
+    art::Framework_Core
+    nusimdata::SimulationBase
+    lardataobj::RecoBase
+    lardataalg::DetectorInfo
+    sbnobj::Common_Reco
+    larsim::MCCheater_BackTrackerService_service           
+    larsim::MCCheater_ParticleInventoryService_service
+    lardata::DetectorClocksService
+)
+
+simple_plugin(ICARUSHDF5Maker "module"
+    LIBRARIES PRIVATE
+    art::Framework_Core
+    nusimdata::SimulationBase
+    larsim::MCCheater_BackTrackerService_service           
+    larsim::MCCheater_ParticleInventoryService_service
+    hep_hpc_hdf5
+    lardataobj::RecoBase
+    lardataalg::DetectorInfo
+    lardata::DetectorClocksService
+    larcorealg::Geometry
+    larcore::WireReadout
+    ${HDF5_LIBRARIES}
+)
+
+add_subdirectory(scripts)
+
+install_headers()
+install_fhicl()
+install_source()
+