evgen::ActiveVolumeVertexSampler Class Reference

#include "ActiveVolumeVertexSampler.h"

Classes
struct	Config

Public Types
enum	vertex_type_t { vertex_type_t::kSampled, vertex_type_t::kFixed, vertex_type_t::kBox }
using	Name = fhicl::Name
using	Comment = fhicl::Comment

Public Member Functions
	ActiveVolumeVertexSampler (const fhicl::Table< Config > &conf, rndm::NuRandomService &rand_service, const geo::Geometry &geom, const std::string &generator_name)
	ActiveVolumeVertexSampler (const fhicl::ParameterSet &pset, rndm::NuRandomService &rand_service, const geo::Geometry &geom, const std::string &generator_name)
void	reconfigure (const fhicl::Table< Config > &conf, const geo::Geometry &geom)
TLorentzVector	sample_vertex_pos (const geo::Geometry &geom)

Protected Attributes
TLorentzVector	fVertexPosition
vertex_type_t	fVertexType
std::string	fGeneratorName
std::unique_ptr< std::discrete_distribution< size_t > >	fTPCDist
std::mt19937_64	fTPCEngine
double	fXmin
double	fYmin
double	fZmin
double	fXmax
double	fYmax
double	fZmax
bool	fCheckActive

Detailed Description

Definition at line 44 of file ActiveVolumeVertexSampler.h.

Member Typedef Documentation

using evgen::ActiveVolumeVertexSampler::Comment = fhicl::Comment

Definition at line 48 of file ActiveVolumeVertexSampler.h.

using evgen::ActiveVolumeVertexSampler::Name = fhicl::Name

Definition at line 47 of file ActiveVolumeVertexSampler.h.

Member Enumeration Documentation

enum evgen::ActiveVolumeVertexSampler::vertex_type_t

strong

Enumerator
kSampled
kFixed
kBox

Definition at line 85 of file ActiveVolumeVertexSampler.h.

{ kSampled, kFixed, kBox };

Constructor & Destructor Documentation

evgen::ActiveVolumeVertexSampler::ActiveVolumeVertexSampler	(	const fhicl::Table< Config > &	conf,
		rndm::NuRandomService &	rand_service,
		const geo::Geometry &	geom,
		const std::string &	generator_name
	)

Definition at line 22 of file ActiveVolumeVertexSampler.cxx.

References fGeneratorName, fTPCEngine, fhicl::Table< T, KeysToIgnore >::get_PSet(), reconfigure(), rndm::NuRandomService::registerEngine(), and seed.

  : fVertexType(vertex_type_t::kSampled), fGeneratorName(generator_name), fTPCDist(nullptr)
{
  // Configure the algorithm using the FHiCL parameters
  this->reconfigure(conf, geom);

  // Register the TPC sampling engine with the seed service. If you need the
  // seed later, get it from the seed service using the value of the variable
  // generator_name as the instance name.
  rndm::NuRandomService::seed_t tpc_seed = rand_service.registerEngine(
    [this](rndm::NuRandomService::EngineId const& /* unused */,
           rndm::NuRandomService::seed_t lar_seed) -> void {
      auto seed = static_cast<uint_fast64_t>(lar_seed);
      // Use the obtained seed to prepare the random number engine.  This is
      // an attempt to do a decent job, but optimally accomplishing this can
      // be tricky (see, for example,
      // http://www.pcg-random.org/posts/cpp-seeding-surprises.html)
      std::seed_seq seed_sequence{seed};
      fTPCEngine.seed(seed_sequence);
    },
    fGeneratorName,
    conf.get_PSet(),
    {"seed"});

  // TODO: resolve the other workaround mentioned in the MARLEYHelper
  // class, then fix this as well
  uint_fast64_t tpc_cast_seed = static_cast<uint_fast64_t>(tpc_seed);
  std::seed_seq tpc_seed_sequence{tpc_cast_seed};
  fTPCEngine.seed(tpc_seed_sequence);
}

evgen::ActiveVolumeVertexSampler::ActiveVolumeVertexSampler ( const fhicl::ParameterSet &  pset, rndm::NuRandomService &  rand_service, const geo::Geometry &  geom, const std::string &  generator_name  )

inline

Definition at line 93 of file ActiveVolumeVertexSampler.h.

      : ActiveVolumeVertexSampler(fhicl::Table<Config>(pset, {}),
                                  rand_service,
                                  geom,
                                  generator_name)
    {}

Member Function Documentation

void evgen::ActiveVolumeVertexSampler::reconfigure	(	const fhicl::Table< Config > &	conf,
		const geo::Geometry &	geom
	)

Definition at line 150 of file ActiveVolumeVertexSampler.cxx.

References fCheckActive, fGeneratorName, fTPCDist, fVertexPosition, fVertexType, fXmax, fXmin, fYmax, fYmin, fZmax, fZmin, geo::GeometryCore::Iterate(), kBox, kFixed, kSampled, and geo::GeometryCore::NTPC().

Referenced by ActiveVolumeVertexSampler().

{
  auto type = conf().type_();
  if (type == "sampled") {

    fVertexType = vertex_type_t::kSampled;

    // Get the active masses of each of the TPCs in the current geometry. Use
    // them as weights for sampling a TPC to use for the primary vertex.
    std::vector<double> tpc_masses;
    size_t num_tpcs = geom.NTPC();
    for (auto const& tpc : geom.Iterate<geo::TPCGeo>(geo::CryostatID{0})) {
      // For each TPC, use its active mass (returned in kg) as its sampling weight
      tpc_masses.push_back(tpc.ActiveMass());
    }

    // Load the discrete distribution used to sample TPCs with the up-to-date
    // set of weights
    fTPCDist.reset(new std::discrete_distribution<size_t>(tpc_masses.begin(), tpc_masses.end()));
  }
  else if (type == "fixed") {

    fVertexType = vertex_type_t::kFixed;

    auto vertex_pos = conf().position_();
    double Vx = vertex_pos.at(0);
    double Vy = vertex_pos.at(1);
    double Vz = vertex_pos.at(2);

    fVertexPosition.SetXYZT(Vx, Vy, Vz, 0.); // TODO: add time setting
  }
  else if (type == "box") {

    fVertexType = vertex_type_t::kBox;
    auto min_pos = conf().min_position_();
    auto max_pos = conf().max_position_();

    fXmin = min_pos.at(0);
    fYmin = min_pos.at(1);
    fZmin = min_pos.at(2);

    fXmax = max_pos.at(0);
    fYmax = max_pos.at(1);
    fZmax = max_pos.at(2);

    // By default, don't check whether each point is in a TPC active volume
    fCheckActive = false;
    // If the user specified this optional parameter, use that instead
    conf().check_active_(fCheckActive);
  }

  else
    throw cet::exception("ActiveVolumeVertexSampler " + fGeneratorName)
      << "Invalid vertex type '" << type << "' requested. Allowed values are"
      << " 'sampled' and 'fixed'";
}

TLorentzVector evgen::ActiveVolumeVertexSampler::sample_vertex_pos

(

const geo::Geometry &

geom

)

Definition at line 58 of file ActiveVolumeVertexSampler.cxx.

References fCheckActive, fGeneratorName, fTPCEngine, fVertexPosition, fVertexType, fXmax, fXmin, fYmax, fYmin, fZmax, fZmin, geo::GeometryCore::Iterate(), kBox, kSampled, maxY, MF_LOG_INFO, geo::BoxBoundedGeo::MinX(), minY, geo::GeometryCore::NTPC(), geo::GeometryCore::TPC(), x, y, and z.

{
  // sample a new position if needed
  if (fVertexType == vertex_type_t::kSampled) {

    // Sample a TPC index using the active masses as weights
    size_t tpc_index = (*fTPCDist)(fTPCEngine);

    // Get the dimensions of the chosen TPC's active volume
    const auto& tpc = geom.TPC(geo::TPCID(0, tpc_index));
    double minX = tpc.MinX();
    double maxX = tpc.MaxX();
    double minY = tpc.MinY();
    double maxY = tpc.MaxY();
    double minZ = tpc.MinZ();
    double maxZ = tpc.MaxZ();
    std::uniform_real_distribution<double>::param_type x_range(minX, maxX);
    std::uniform_real_distribution<double>::param_type y_range(minY, maxY);
    std::uniform_real_distribution<double>::param_type z_range(minZ, maxZ);

    // Sample a location uniformly over this volume
    std::uniform_real_distribution<double> uniform_dist;
    double x = uniform_dist(fTPCEngine, x_range);
    double y = uniform_dist(fTPCEngine, y_range);
    double z = uniform_dist(fTPCEngine, z_range);
    MF_LOG_INFO("ActiveVolumeVertexSampler " + fGeneratorName)
      << "Sampled primary vertex in TPC #" << tpc_index << ", x = " << x << ", y = " << y
      << ", z = " << z;

    // Update the vertex position 4-vector
    fVertexPosition.SetXYZT(x, y, z, 0.); // TODO: add time sampling
  }
  else if (fVertexType == vertex_type_t::kBox) {
    bool ok = false;
    int num_iterations = 0;

    // TODO: reduce code duplication here
    // Get the range of coordinates covered by the box
    std::uniform_real_distribution<double>::param_type x_range(fXmin, fXmax);
    std::uniform_real_distribution<double>::param_type y_range(fYmin, fYmax);
    std::uniform_real_distribution<double>::param_type z_range(fZmin, fZmax);

    // Sample a location uniformly over this volume
    std::uniform_real_distribution<double> uniform_dist;

    double x = 0.;
    double y = 0.;
    double z = 0.;
    while (!ok && num_iterations < MAX_BOX_ITERATIONS) {
      x = uniform_dist(fTPCEngine, x_range);
      y = uniform_dist(fTPCEngine, y_range);
      z = uniform_dist(fTPCEngine, z_range);

      // If we've been asked to check whether the vertex is within the
      // active volume of a TPC, verify that. Otherwise just accept the
      // first vertex position that was sampled unconditionally.
      if (fCheckActive) {
        size_t num_tpcs = geom.NTPC();
        for (auto const& tpc : geom.Iterate<geo::TPCGeo>(geo::CryostatID{0})) {
          double minX = tpc.MinX();
          double maxX = tpc.MaxX();
          double minY = tpc.MinY();
          double maxY = tpc.MaxY();
          double minZ = tpc.MinZ();
          double maxZ = tpc.MaxZ();
          if (x >= minX && x <= maxX && y >= minY && y <= maxY && z >= minZ && z <= maxZ) {
            ok = true;
            break;
          }
        }
      }
      else
        ok = true;
    }

    if (!ok)
      throw cet::exception("ActiveVolumeVertexSampler " + fGeneratorName)
        << "Failed to sample a vertex within a TPC active volume after " << MAX_BOX_ITERATIONS
        << " iterations";

    MF_LOG_INFO("ActiveVolumeVertexSampler " + fGeneratorName)
      << "Sampled primary vertex at x = " << x << ", y = " << y << ", z = " << z;

    // Update the vertex position 4-vector
    fVertexPosition.SetXYZT(x, y, z, 0.); // TODO: add time sampling
  }

  // if we're using a fixed vertex position, we don't need to do any sampling
  return fVertexPosition;
}

Member Data Documentation

bool evgen::ActiveVolumeVertexSampler::fCheckActive

protected

Definition at line 134 of file ActiveVolumeVertexSampler.h.

Referenced by reconfigure(), and sample_vertex_pos().

std::string evgen::ActiveVolumeVertexSampler::fGeneratorName

protected

Definition at line 116 of file ActiveVolumeVertexSampler.h.

Referenced by ActiveVolumeVertexSampler(), reconfigure(), and sample_vertex_pos().

std::unique_ptr<std::discrete_distribution<size_t> > evgen::ActiveVolumeVertexSampler::fTPCDist

protected

Definition at line 120 of file ActiveVolumeVertexSampler.h.

Referenced by reconfigure().

std::mt19937_64 evgen::ActiveVolumeVertexSampler::fTPCEngine

protected

Definition at line 123 of file ActiveVolumeVertexSampler.h.

Referenced by ActiveVolumeVertexSampler(), and sample_vertex_pos().

TLorentzVector evgen::ActiveVolumeVertexSampler::fVertexPosition

protected

Definition at line 112 of file ActiveVolumeVertexSampler.h.

Referenced by reconfigure(), and sample_vertex_pos().

vertex_type_t evgen::ActiveVolumeVertexSampler::fVertexType

protected

Definition at line 114 of file ActiveVolumeVertexSampler.h.

Referenced by reconfigure(), and sample_vertex_pos().

double evgen::ActiveVolumeVertexSampler::fXmax

protected

Definition at line 130 of file ActiveVolumeVertexSampler.h.

Referenced by reconfigure(), and sample_vertex_pos().

double evgen::ActiveVolumeVertexSampler::fXmin

protected

Definition at line 126 of file ActiveVolumeVertexSampler.h.

Referenced by reconfigure(), and sample_vertex_pos().

double evgen::ActiveVolumeVertexSampler::fYmax

protected

Definition at line 131 of file ActiveVolumeVertexSampler.h.

Referenced by reconfigure(), and sample_vertex_pos().

double evgen::ActiveVolumeVertexSampler::fYmin

protected

Definition at line 127 of file ActiveVolumeVertexSampler.h.

Referenced by reconfigure(), and sample_vertex_pos().

double evgen::ActiveVolumeVertexSampler::fZmax

protected

Definition at line 132 of file ActiveVolumeVertexSampler.h.

Referenced by reconfigure(), and sample_vertex_pos().

double evgen::ActiveVolumeVertexSampler::fZmin

protected

Definition at line 128 of file ActiveVolumeVertexSampler.h.

Referenced by reconfigure(), and sample_vertex_pos().

---

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

• larsim/v09_43_00/source/larsim/EventGenerator/MARLEY/ActiveVolumeVertexSampler.h
• larsim/v09_43_00/source/larsim/EventGenerator/MARLEY/ActiveVolumeVertexSampler.cxx