evdb_tool::DrawSimEnergyDeposit3D Class Reference

Inheritance diagram for evdb_tool::DrawSimEnergyDeposit3D:

Public Member Functions
	DrawSimEnergyDeposit3D (const fhicl::ParameterSet &pset)
void	Draw (const art::Event &, evdb::View3D *) const override
Public Member Functions inherited from evdb_tool::ISim3DDrawer
virtual	~ISim3DDrawer () noexcept=default

Private Member Functions
void	drawMCPartAssociated (const art::Event &, evdb::View3D *) const
void	drawAll (const art::Event &, evdb::View3D *) const

Private Attributes
bool	fDrawAllSimEnergy

Detailed Description

Definition at line 27 of file DrawSimEnergyDeposit3D_tool.cc.

Constructor & Destructor Documentation

evdb_tool::DrawSimEnergyDeposit3D::DrawSimEnergyDeposit3D ( const fhicl::ParameterSet &  pset )

explicit

Definition at line 42 of file DrawSimEnergyDeposit3D_tool.cc.

  {
    fDrawAllSimEnergy = pset.get<bool>("DrawAllSimEnergyDeposits", false);
  }

Member Function Documentation

void evdb_tool::DrawSimEnergyDeposit3D::Draw ( const art::Event &  evt, evdb::View3D *  view  ) const

overridevirtual

Implements evdb_tool::ISim3DDrawer.

Definition at line 48 of file DrawSimEnergyDeposit3D_tool.cc.

  {
    art::ServiceHandle<evd::SimulationDrawingOptions const> drawOpt;

    // If the option is turned off, there's nothing to do
    if (!drawOpt->fShowSimEnergyInfo) return;

    // Split here if drawing all vs drawing MC associated only
    if (fDrawAllSimEnergy)
      drawAll(evt, view);
    else
      drawMCPartAssociated(evt, view);

    return;
  }

void evdb_tool::DrawSimEnergyDeposit3D::drawAll ( const art::Event &  evt, evdb::View3D *  view  ) const

private

Definition at line 186 of file DrawSimEnergyDeposit3D_tool.cc.

  {
    art::ServiceHandle<evd::SimulationDrawingOptions const> drawOpt;

    // NOTE: In this mode we cannot correct the voxel positions for time offsets since we have nothing to offset with
    // The voxels are drawn in the x,y,z locations given by the SimEnergyDeposit objects

    // Recover the simchannels
    art::Handle<std::vector<sim::SimEnergyDeposit>> simEnergyDepositHandle;

    evt.getByLabel(drawOpt->fSimEnergyLabel, simEnergyDepositHandle);

    if (simEnergyDepositHandle.isValid() && simEnergyDepositHandle->size() > 0) {
      mf::LogDebug("SimEnergyDeposit3DDrawer")
        << "Starting loop over " << simEnergyDepositHandle->size() << " SimEnergyDeposits, "
        << std::endl;

      // Get the geometry service and its friends
      auto const clockData =
        art::ServiceHandle<detinfo::DetectorClocksService const>()->DataFor(evt);
      auto const detProp =
        art::ServiceHandle<detinfo::DetectorPropertiesService const>()->DataFor(evt, clockData);
      art::ServiceHandle<geo::Geometry const> geom;

      // Would like to draw the deposits as markers with colors given by particle id
      // So we make two passes, first to fill a map with color the key and positions for the markers
      std::map<int, std::vector<sim::SimEnergyDeposit::Point_t>> colorToPositionMap;

      // Go through the SimEnergyDeposits and populate the map
      for (const auto& simEnergyDeposit : *simEnergyDepositHandle) {
        // If we have cosmic rays then we need to get the offset which allows translating from
        // when they were generated vs when they were tracked.
        // Note that this also explicitly checks that they are in a TPC volume
        try {
          sim::SimEnergyDeposit::Point_t point = simEnergyDeposit.MidPoint();
          double depTime = simEnergyDeposit.T();
          geo::TPCID tpcID = geom->PositionToTPCID(point);
          geo::PlaneID planeID(tpcID, 0);
          double g4Ticks = clockData.TPCG4Time2Tick(depTime) - trigger_offset(clockData);
          double xPosMinTick = detProp.ConvertTicksToX(0, planeID);
          double xOffset = detProp.ConvertTicksToX(g4Ticks, planeID) - xPosMinTick;

          colorToPositionMap[evd::Style::ColorFromPDG(simEnergyDeposit.PdgCode())].emplace_back(
            sim::SimEnergyDeposit::Point_t(point.X() + xOffset, point.Y(), point.Z()));
        }
        catch (...) {
          continue;
        }
      }

      // Now we can do some drawing
      for (const auto& pair : colorToPositionMap) {
        int colorIdx(pair.first);
        int markerIdx(kFullDotMedium);
        int markerSize(2);

        TPolyMarker3D& pm = view->AddPolyMarker3D(1, colorIdx, markerIdx, markerSize);

        // Import positions into an array
        std::vector<double> posArrayVec;
        int hitCount(0);

        posArrayVec.resize(3 * pair.second.size());

        for (const auto& point : pair.second) {
          posArrayVec[3 * hitCount] = point.X();
          posArrayVec[3 * hitCount + 1] = point.Y();
          posArrayVec[3 * hitCount + 2] = point.Z();
          hitCount++;
        }

        pm.SetPolyMarker(hitCount, posArrayVec.data(), markerIdx);
      }
    }

    return;
  }

void evdb_tool::DrawSimEnergyDeposit3D::drawMCPartAssociated ( const art::Event &  evt, evdb::View3D *  view  ) const

private

Definition at line 65 of file DrawSimEnergyDeposit3D_tool.cc.

  {
    art::ServiceHandle<evd::SimulationDrawingOptions const> drawOpt;

    // Recover a handle to the collection of MCParticles
    // We need these so we can determine the offset (if any)
    art::Handle<std::vector<simb::MCParticle>> mcParticleHandle;

    evt.getByLabel(drawOpt->fG4ModuleLabel, mcParticleHandle);

    if (!mcParticleHandle.isValid()) return;

    // Create a mapping between track ID's and MCParticles
    using TrackToMcParticleMap = std::map<int, const simb::MCParticle*>;

    TrackToMcParticleMap trackToMcParticleMap;

    for (const auto& mcParticle : *mcParticleHandle)
      trackToMcParticleMap[mcParticle.TrackId()] = &mcParticle;

    // Now recover the simchannels
    art::Handle<std::vector<sim::SimEnergyDeposit>> simEnergyDepositHandle;

    evt.getByLabel(drawOpt->fSimEnergyLabel, simEnergyDepositHandle);

    auto const clockData = art::ServiceHandle<detinfo::DetectorClocksService const>()->DataFor(evt);
    auto const detProp =
      art::ServiceHandle<detinfo::DetectorPropertiesService const>()->DataFor(evt, clockData);

    if (simEnergyDepositHandle.isValid() && simEnergyDepositHandle->size() > 0) {
      mf::LogDebug("SimEnergyDeposit3DDrawer")
        << "Starting loop over " << simEnergyDepositHandle->size() << " SimEnergyDeposits, "
        << std::endl;

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

      // First step is to create a map between MCParticle and SimEnergyDeposit objects...
      using MCPartToSimEnergyMap =
        std::map<const simb::MCParticle*, std::vector<const sim::SimEnergyDeposit*>>;

      MCPartToSimEnergyMap mcPartToSimEnergyMap;

      // Go through the SimEnergyDeposits and populate the map
      for (const auto& simEnergyDeposit : *simEnergyDepositHandle) {
        TrackToMcParticleMap::const_iterator trackMCItr =
          trackToMcParticleMap.find(simEnergyDeposit.TrackID());

        if (trackMCItr == trackToMcParticleMap.end()) continue;

        mcPartToSimEnergyMap[trackMCItr->second].push_back(&simEnergyDeposit);
      }

      // Would like to draw the deposits as markers with colors given by particle id
      // So we make two passes, first to fill a map with color the key and positions for the markers
      std::map<int, std::vector<sim::SimEnergyDeposit::Point_t>> colorToPositionMap;

      // Now we loop through and build the mapping of color to positions
      for (const auto& mcPartToSimEnergy : mcPartToSimEnergyMap) {
        // The first task we need to take on is to find the offset for the
        // energy deposit This is for the case of "out of time" particles...
        // (e.g. cosmic rays)
        double g4Ticks(clockData.TPCG4Time2Tick(mcPartToSimEnergy.first->T()) -
                       trigger_offset(clockData));
        double xOffset(0.);
        double xPosMinTick(0.);
        double xPosMaxTick(std::numeric_limits<double>::max());

        for (const auto& simEnergyDeposit : mcPartToSimEnergy.second) {
          sim::SimEnergyDeposit::Point_t point = simEnergyDeposit->MidPoint();

          // If we have cosmic rays then we need to get the offset which allows translating from
          // when they were generated vs when they were tracked.
          // Note that this also explicitly checks that they are in a TPC volume
          try {
            geo::TPCID tpcID = geom->PositionToTPCID(point);
            geo::PlaneID planeID(tpcID, 0);

            xPosMinTick = detProp.ConvertTicksToX(0, planeID);
            xPosMaxTick = detProp.ConvertTicksToX(detProp.NumberTimeSamples(), planeID);
            xOffset = detProp.ConvertTicksToX(g4Ticks, planeID) - xPosMinTick;

            if (xPosMaxTick < xPosMinTick) std::swap(xPosMinTick, xPosMaxTick);
          }
          catch (...) {
            continue;
          }

          colorToPositionMap[evd::Style::ColorFromPDG(simEnergyDeposit->PdgCode())].emplace_back(
            sim::SimEnergyDeposit::Point_t(point.X() + xOffset, point.Y(), point.Z()));
        }
      }

      // Now we can do some drawing
      for (const auto& pair : colorToPositionMap) {
        int colorIdx(pair.first);
        int markerIdx(kFullDotMedium);
        int markerSize(2);

        TPolyMarker3D& pm = view->AddPolyMarker3D(1, colorIdx, markerIdx, markerSize);

        // Import positions into an array
        std::vector<double> posArrayVec;
        int hitCount(0);

        posArrayVec.resize(3 * pair.second.size());

        for (const auto& point : pair.second) {
          posArrayVec[3 * hitCount] = point.X();
          posArrayVec[3 * hitCount + 1] = point.Y();
          posArrayVec[3 * hitCount + 2] = point.Z();
          hitCount++;
        }

        pm.SetPolyMarker(hitCount, posArrayVec.data(), markerIdx);
      }
    }

    return;
  }

Member Data Documentation

bool evdb_tool::DrawSimEnergyDeposit3D::fDrawAllSimEnergy

private

Definition at line 37 of file DrawSimEnergyDeposit3D_tool.cc.

---

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

• DrawSimEnergyDeposit3D_tool.cc