dnn::SavePiMu Class Reference

Inheritance diagram for dnn::SavePiMu:

Public Member Functions
	SavePiMu (fhicl::ParameterSet const &ps)
void	saveImage (art::Event const &e, hep_hpc::hdf5::File &hdffile) override

Private Attributes
art::InputTag	fTrackModuleLabel
art::InputTag	fWireModuleLabel
art::InputTag	fMCTruthLabel

Detailed Description

Definition at line 24 of file SavePiMu_tool.cc.

Constructor & Destructor Documentation

dnn::SavePiMu::SavePiMu ( fhicl::ParameterSet const &  ps )

explicit

Definition at line 36 of file SavePiMu_tool.cc.

References fMCTruthLabel, fWireModuleLabel, and fhicl::ParameterSet::get().

    : fTrackModuleLabel{ps.get<art::InputTag>("TrackModuleLabel")}
    , fWireModuleLabel{ps.get<art::InputTag>("WireModuleLabel")}
    , fMCTruthLabel{ps.get<art::InputTag>("MCTruthLabel")}
  {}

Member Function Documentation

void dnn::SavePiMu::saveImage ( art::Event const &  e, hep_hpc::hdf5::File &  hdffile  )

overridevirtual

Implements dnn::ImageMaker.

Definition at line 42 of file SavePiMu_tool.cc.

References util::abs(), geo::GeometryCore::ChannelToWire(), DEFINE_ART_CLASS_TOOL, art::EventID::event(), fMCTruthLabel, fTrackModuleLabel, fWireModuleLabel, lar::sparse_vector< T >::get_ranges(), art::ProductRetriever::getHandle(), art::ProductRetriever::getProduct(), art::Event::id(), art::Event::run(), art::Event::subRun(), art::Event::time(), art::Timestamp::timeHigh(), and art::Timestamp::timeLow().

  {

    art::ServiceHandle<geo::Geometry> geom;

    static Ntuple<unsigned int, unsigned int, unsigned int, double> evtids(
      hdffile, "evtids", {"run", "subrun", "event", "evttime"});

    static auto image =
      make_ntuple({hdffile, "image", 1000},
                  make_scalar_column<unsigned short>("label"),
                  make_column<float, 2>(
                    "adc",                                // 2 means each element is a 2-d array
                    {50, 50},                             // extent of each array dimension
                    1024 * 1024 / (2500 * sizeof(float)), // chunk size
                    {PropertyList{H5P_DATASET_CREATE}(&H5Pset_shuffle)(&H5Pset_deflate, 6u)}));

    // Get event information
    double evttime;

    art::Timestamp ts = e.time();
    if (ts.timeHigh() == 0) {
      TTimeStamp tts(ts.timeLow());
      evttime = tts.AsDouble();
    }
    else {
      TTimeStamp tts(ts.timeHigh(), ts.timeLow());
      evttime = tts.AsDouble();
    }

    // Get image information
    // * tracks
    auto const& tracks = e.getProduct<std::vector<recob::Track>>(fTrackModuleLabel);
    auto trackListHandle = e.getHandle<std::vector<recob::Track>>(fTrackModuleLabel);
    art::FindManyP<recob::Hit, recob::TrackHitMeta> fmthm(trackListHandle, e, fTrackModuleLabel);

    // * wires
    auto const& wires = e.getProduct<std::vector<recob::Wire>>(fWireModuleLabel);

    // * MC truth
    auto const& mclist = e.getProduct<std::vector<simb::MCTruth>>(fMCTruthLabel);

    unsigned short flag = 999;
    if (!mclist.empty()) {
      auto particle = mclist[0].GetParticle(0);
      if (std::abs(particle.PdgCode()) == 13) flag = 0;
      if (std::abs(particle.PdgCode()) == 211) flag = 1;
    }

    if (!tracks.empty()) {
      auto trkend = tracks[0].End();
      if (trkend.X() > -360 + 50 && trkend.X() < 360 - 50 && trkend.Y() > 50 &&
          trkend.Y() < 610 - 50 && trkend.Z() > 50 && trkend.Z() < 710 - 50) {
        if (fmthm.isValid()) {
          auto vhit = fmthm.at(0);
          auto vmeta = fmthm.data(0);
          int ihit = -1;
          int maxindex = -1;
          for (size_t i = 0; i < vhit.size(); ++i) {
            if (static_cast<int>(vmeta[i]->Index()) == std::numeric_limits<int>::max()) {
              continue;
            }
            if (vhit[i]->WireID().Plane == 2) {
              if (int(vmeta[i]->Index()) > maxindex) {
                maxindex = vmeta[i]->Index();
                ihit = i;
              }
            }
          }
          if (ihit >= 0) {
            auto endwire = vhit[ihit]->WireID();
            float endtime = vhit[ihit]->PeakTime();
            float adc[50][50] = {{0.}};
            for (auto& wire : wires) {
              int channel = wire.Channel();
              auto wireids = geom->ChannelToWire(channel);
              if (wireids[0].Plane == 2 && wireids[0].TPC == endwire.TPC &&
                  wireids[0].Wire >= endwire.Wire - 25 && wireids[0].Wire < endwire.Wire + 25) {
                int idx = wireids[0].Wire - endwire.Wire + 25;
                const recob::Wire::RegionsOfInterest_t& signalROI = wire.SignalROI();
                int lasttick = 0;
                for (const auto& range : signalROI.get_ranges()) {
                  const auto& waveform = range.data();
                  // ROI start time
                  raw::TDCtick_t roiFirstBinTick = range.begin_index();
                  for (int i = lasttick; i < roiFirstBinTick; ++i) {
                    if (i >= int(endtime) - 25 && i < int(endtime) + 25) {
                      adc[idx][i - int(endtime) - 25] = 0;
                    }
                  }
                  lasttick = roiFirstBinTick;
                  for (size_t i = 0; i < waveform.size(); ++i) {
                    if (lasttick >= int(endtime) - 25 && lasttick < int(endtime) + 25) {
                      adc[idx][lasttick - int(endtime) + 25] = waveform[i];
                      ++lasttick;
                    }
                  }
                }
                for (int i = lasttick; i < 6000; ++i) {
                  if (i >= int(endtime) - 25 && i < int(endtime) + 25) {
                    adc[idx][i - int(endtime) + 25] = 0;
                  }
                }
              }
            } // Loop over all wire signals
            evtids.insert(e.run(), e.subRun(), e.id().event(), evttime);
            image.insert(flag, &adc[0][0]);
          }
        }
      }
    }

    return;
  }

Member Data Documentation

art::InputTag dnn::SavePiMu::fMCTruthLabel

private

Definition at line 33 of file SavePiMu_tool.cc.

Referenced by saveImage(), and SavePiMu().

art::InputTag dnn::SavePiMu::fTrackModuleLabel

private

Definition at line 31 of file SavePiMu_tool.cc.

Referenced by saveImage().

art::InputTag dnn::SavePiMu::fWireModuleLabel

private

Definition at line 32 of file SavePiMu_tool.cc.

Referenced by saveImage(), and SavePiMu().

---

The documentation for this class was generated from the following file:

• larrecodnn/v09_23_00/source/larrecodnn/ImageMaker/SavePiMu_tool.cc