Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
23 changes: 22 additions & 1 deletion sbncode/CAFMaker/CAFMaker_module.cc
Original file line number Diff line number Diff line change
Expand Up @@ -41,6 +41,7 @@
#endif

#include "ifdh_art/IFDHService/IFDH_service.h"
#include "dk2nu/tree/dk2nu.h"

// ROOT includes
#include "TFile.h"
Expand Down Expand Up @@ -1469,6 +1470,14 @@ void CAFMaker::produce(art::Event& evt) noexcept {
art::fill_ptr_vector(mcfluxes, mcflux_handle);
}

art::Handle<std::vector<bsim::Dk2Nu>> dk2nu_handle;
GetByLabelStrict(evt, std::string("generator"), dk2nu_handle);

std::vector<art::Ptr<bsim::Dk2Nu>> dk2nus;
if (dk2nu_handle.isValid()) {
art::fill_ptr_vector(dk2nus, dk2nu_handle);
}

// get the MCReco for the fake-reco
art::Handle<std::vector<sim::MCTrack>> mctrack_handle;
GetByLabelStrict(evt, std::string("mcreco"), mctrack_handle);
Expand Down Expand Up @@ -1552,9 +1561,13 @@ void CAFMaker::produce(art::Event& evt) noexcept {
// holder for invalid MCFlux
simb::MCFlux badflux; // default constructor gives nonsense values

// holder for invalid Dk2Nu (default-constructed ancestor vector is empty -- treated as "absent")
bsim::Dk2Nu baddk2nu;

for (size_t i=0; i<mctruths.size(); i++) {
auto const& mctruth = mctruths.at(i);
const simb::MCFlux &mcflux = (mcfluxes.size()) ? *mcfluxes.at(i) : badflux;
const bsim::Dk2Nu &dk2nu = (dk2nus.size()) ? *dk2nus.at(i) : baddk2nu;

simb::GTruth gtruth;
bool ok = GetAssociatedProduct(fmp_gtruth, i, gtruth);
Expand All @@ -1569,7 +1582,15 @@ void CAFMaker::produce(art::Event& evt) noexcept {

if ( !isRealData ){

FillTrueNeutrino(mctruth, mcflux, gtruth, true_particles, id_to_truehit_map, srtruthbranch.nu.back(), i, fActiveVolumes);
FillTrueNeutrino(mctruth, mcflux, dk2nu, gtruth, true_particles, id_to_truehit_map, srtruthbranch.nu.back(), i, fActiveVolumes);

if (ok) {
srtruthbranch.nu.back().generator = caf::kGENIE;
} else if (i < mevprtl_truths.size()) {
srtruthbranch.nu.back().generator = caf::kMeVPrtl;
} else {
srtruthbranch.nu.back().generator = caf::kUnknownGenerator;
}

srtruthbranch.nu.back().genie_evtrec_idx = fGenieEventCounter;

Expand Down
1 change: 1 addition & 0 deletions sbncode/CAFMaker/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,7 @@ art_make_library( LIBRARY_NAME sbncode_CAFMaker
ROOT::EG
CLHEP::Random
caf_RecoUtils
dk2nu::Tree
lardataobj::AnalysisBase
lardataobj::RecoBase
lardataobj::Simulation
Expand Down
7 changes: 7 additions & 0 deletions sbncode/CAFMaker/FillTrue.cxx
Original file line number Diff line number Diff line change
@@ -1,5 +1,7 @@
#include "FillTrue.h"

#include "dk2nu/tree/dk2nu.h"

#include "larcorealg/GeoAlgo/GeoAlgo.h"
#include "larcorealg/Geometry/GeometryCore.h"
#include "larcorealg/Geometry/WireReadoutGeom.h"
Expand Down Expand Up @@ -385,6 +387,7 @@ namespace caf {
srtruth.position.y = truth.decay_pos.Y();
srtruth.position.z = truth.decay_pos.Z();
srtruth.time = truth.decay_pos.T();
srtruth.prod_time = truth.meson_dpos_beamcoord.T(); //Meson time from flux file.

srtruth.momentum.x = truth.mevprtl_mom.X();
srtruth.momentum.y = truth.mevprtl_mom.Y();
Expand Down Expand Up @@ -457,6 +460,7 @@ namespace caf {
//------------------------------------------------
void FillTrueNeutrino(const art::Ptr<simb::MCTruth> mctruth,
const simb::MCFlux &mcflux,
const bsim::Dk2Nu &dk2nu,
const simb::GTruth& gtruth,
const std::vector<caf::SRTrueParticle> &srparticles,
const std::map<int, std::vector<art::Ptr<recob::Hit>>> &id_to_truehit_map,
Expand Down Expand Up @@ -549,6 +553,9 @@ namespace caf {
// Set the MCFlux stuff
srneutrino.initpdg = mcflux.fntype;
srneutrino.baseline = mcflux.fdk2gen + mcflux.fgen2vtx;
srneutrino.dk2gen = mcflux.fdk2gen;
srneutrino.prod_time = (!dk2nu.ancestor.empty())
? dk2nu.ancestor.back().startt : std::numeric_limits<float>::signaling_NaN();
srneutrino.parent_pdg = mcflux.fptype;
srneutrino.parent_dcy_mode = mcflux.fndecay;
srneutrino.prod_vtx.x = mcflux.fvx;
Expand Down
3 changes: 3 additions & 0 deletions sbncode/CAFMaker/FillTrue.h
Original file line number Diff line number Diff line change
Expand Up @@ -33,6 +33,8 @@
#include "sbnanaobj/StandardRecord/StandardRecord.h"
#include "sbnanaobj/StandardRecord/SRMeVPrtl.h"

namespace bsim { class Dk2Nu; }

namespace caf
{
struct HitsEnergy {
Expand Down Expand Up @@ -86,6 +88,7 @@ namespace caf

void FillTrueNeutrino(const art::Ptr<simb::MCTruth> mctruth,
const simb::MCFlux &mcflux,
const bsim::Dk2Nu &dk2nu,
const simb::GTruth& gtruth,
const std::vector<caf::SRTrueParticle> &srparticles,
const std::map<int, std::vector<art::Ptr<recob::Hit>>> &id_to_truehit_map,
Expand Down
Loading