tracking::sim::PropagatorStepWriter Class Reference

Classes
struct	Config

Public Member Functions
	PropagatorStepWriter (const Config &cfg)
	Constructor with.
	~PropagatorStepWriter ()
bool	writeSteps (framework::Event &event, const std::vector< PropagationSteps > &stepCollection, const std::vector< ldmx::Measurement > &measurements, const Acts::Vector3 &start_pos, const Acts::Vector3 &start_mom)

Protected Attributes
Config	m_cfg_
	the configuration object
std::mutex	m_write_mutex_
	protect multi-threaded writes
TFile *	m_output_file_
	the output file name
TTree *	m_output_tree_
	the output tree
int	m_event_nr_
	the event number of
std::vector< int >	m_boundary_id_
	boundary identifier
std::vector< int >	m_layer_id_
	layer identifier if
std::vector< int >	m_approach_id_
	surface identifier
std::vector< int >	m_sensitive_id_
	surface identifier
std::vector< float >	m_x_
	global x_
std::vector< float >	m_y_
	global y_
std::vector< float >	m_z_
	global z_
std::vector< float >	m_dx_
	global direction x_
std::vector< float >	m_dy_
	global direction y_
std::vector< float >	m_dz_
	global direction z_
std::vector< int >	m_step_type_
	step type
std::vector< float >	m_step_acc_
	accuracy
std::vector< float >	m_step_act_
	actor check
std::vector< float >	m_step_abt_
	aborter
std::vector< float >	m_step_usr_
	user
std::vector< float >	m_hit_x_
	hit location X
std::vector< float >	m_hit_y_
	hit location Y
std::vector< float >	m_hit_z_
	hit location Z
std::vector< float >	m_start_pos_
	start position of the particle propagated
std::vector< float >	m_start_mom_
	start momentum of the particle propagated

Detailed Description

Definition at line 29 of file PropagatorStepWriter.h.

Constructor & Destructor Documentation

PropagatorStepWriter()

tracking::sim::PropagatorStepWriter::PropagatorStepWriter

(

const Config &

cfg

)

Constructor with.

Parameters

cfg	configuration struct
output	logging level

Definition at line 17 of file PropagatorStepWriter.cxx.

    : m_cfg_(cfg), m_output_file_(cfg.root_file_) {
  if (m_cfg_.tree_name_.empty()) {
    throw std::invalid_argument("Missing tree name");
  }
 
  // Setup ROOT I/O
  if (m_output_file_ == nullptr) {
    m_output_file_ =
        TFile::Open(m_cfg_.file_path_.c_str(), m_cfg_.file_mode_.c_str());
    if (m_output_file_ == nullptr) {
      throw std::ios_base::failure("Could not open '" + m_cfg_.file_path_);
    }
  }
  m_output_file_->cd();
 
  m_output_tree_ = new TTree(m_cfg_.tree_name_.c_str(),
                             "TTree from RootPropagationStepsWriter");
  if (m_output_tree_ == nullptr) throw std::bad_alloc();
 
  // Set the branches
  m_output_tree_->Branch("event_nr", &m_event_nr_);
  //  m_outputTree->Branch("volume_id", &m_volumeID);
  m_output_tree_->Branch("boundary_id", &m_boundary_id_);
  m_output_tree_->Branch("layer_id", &m_layer_id_);
  m_output_tree_->Branch("approach_id", &m_approach_id_);
  m_output_tree_->Branch("sensitive_id", &m_sensitive_id_);
  m_output_tree_->Branch("g_x", &m_x_);
  m_output_tree_->Branch("g_y", &m_y_);
  m_output_tree_->Branch("g_z", &m_z_);
  m_output_tree_->Branch("d_x", &m_dx_);
  m_output_tree_->Branch("d_y", &m_dy_);
  m_output_tree_->Branch("d_z", &m_dz_);
  m_output_tree_->Branch("type", &m_step_type_);
  m_output_tree_->Branch("step_acc", &m_step_acc_);
  m_output_tree_->Branch("step_act", &m_step_act_);
  m_output_tree_->Branch("step_abt", &m_step_abt_);
  m_output_tree_->Branch("step_usr", &m_step_usr_);
  m_output_tree_->Branch("hit_x", &m_hit_x_);
  m_output_tree_->Branch("hit_y", &m_hit_y_);
  m_output_tree_->Branch("hit_z", &m_hit_z_);
  m_output_tree_->Branch("start_pos", &m_start_pos_);
  m_output_tree_->Branch("start_mom", &m_start_mom_);
 
}  // constructor

References tracking::sim::PropagatorStepWriter::Config::file_mode_, tracking::sim::PropagatorStepWriter::Config::file_path_, m_approach_id_, m_boundary_id_, m_cfg_, m_dx_, m_dy_, m_dz_, m_event_nr_, m_hit_x_, m_hit_y_, m_hit_z_, m_layer_id_, m_output_file_, m_output_tree_, m_sensitive_id_, m_start_mom_, m_start_pos_, m_step_abt_, m_step_acc_, m_step_act_, m_step_type_, m_step_usr_, m_x_, m_y_, m_z_, and tracking::sim::PropagatorStepWriter::Config::tree_name_.

~PropagatorStepWriter()

tracking::sim::PropagatorStepWriter::~PropagatorStepWriter

(

)

Close the file if it's yours

Definition at line 64 of file PropagatorStepWriter.cxx.

                                            {
  if (m_cfg_.root_file_ == nullptr) {
    m_output_file_->cd();
    m_output_tree_->Write();
    m_output_file_->Close();
  }
}  // destructor

References m_cfg_, m_output_file_, m_output_tree_, and tracking::sim::PropagatorStepWriter::Config::root_file_.

Member Function Documentation

writeSteps()

bool tracking::sim::PropagatorStepWriter::writeSteps	(	framework::Event &	event,
		const std::vector< PropagationSteps > &	stepCollection,
		const std::vector< ldmx::Measurement > &	measurements,
		const Acts::Vector3 &	start_pos,
		const Acts::Vector3 &	start_mom )

Definition at line 73 of file PropagatorStepWriter.cxx.

                                                                {
  // Exclusive access to the tree while writing
  std::lock_guard<std::mutex> lock(m_write_mutex_);
 
  m_output_file_->cd();
 
  // we get the event number
  m_event_nr_ = event.getEventNumber();
 
  // fill the hits_
  m_hit_x_.clear();
  m_hit_y_.clear();
  m_hit_z_.clear();
  // m_hit_erru.clear();
  // m_hit_errv.clear();
 
  m_start_pos_.clear();
  m_start_mom_.clear();
 
  for (auto& meas : measurements) {
    m_hit_x_.push_back(meas.getGlobalPosition()[0]);
    m_hit_y_.push_back(meas.getGlobalPosition()[1]);
    m_hit_z_.push_back(meas.getGlobalPosition()[2]);
  }
 
  for (unsigned int i = 0; i < 3; i++) {
    m_start_pos_.push_back(start_pos(i));
    m_start_mom_.push_back(start_mom(i));
  }
 
  // loop over the step vector of each test propagation in this
  for (auto& steps : stepCollection) {
    // clear the vectors for each collection
    //    m_volumeID.clear();
    m_boundary_id_.clear();
    m_layer_id_.clear();
    m_approach_id_.clear();
    m_sensitive_id_.clear();
    m_x_.clear();
    m_y_.clear();
    m_z_.clear();
    m_dx_.clear();
    m_dy_.clear();
    m_dz_.clear();
    m_step_type_.clear();
    m_step_acc_.clear();
    m_step_act_.clear();
    m_step_abt_.clear();
    m_step_usr_.clear();
 
    // loop over single steps
    for (auto& step : steps) {
      // the identification of the step
      //      Acts::GeometryIdentifier::Value volumeID = 0;
      Acts::GeometryIdentifier::Value boundary_id = 0;
      Acts::GeometryIdentifier::Value layer_id = 0;
      Acts::GeometryIdentifier::Value approach_id = 0;
      Acts::GeometryIdentifier::Value sensitive_id = 0;
      // get the identification from the surface first
      if (step.surface) {
        auto geo_id = step.surface->geometryId();
        //        volumeID = geoID.volume();
        boundary_id = geo_id.boundary();
        layer_id = geo_id.layer();
        approach_id = geo_id.approach();
        sensitive_id = geo_id.sensitive();
      }
      // a current volume overwrites the surface tagged one
      // mg ... v36 Step does not include volume
      //      if (step.volume) {
      //    volumeID = step.volume->geometryId().volume();
      //   }
      // now fill
      m_sensitive_id_.push_back(sensitive_id);
      m_approach_id_.push_back(approach_id);
      m_layer_id_.push_back(layer_id);
      m_boundary_id_.push_back(boundary_id);
      // m_volumeID.push_back(volumeID);
 
      // kinematic information
      m_x_.push_back(step.position.x());
      m_y_.push_back(step.position.y());
      m_z_.push_back(step.position.z());
      auto direction = step.momentum.normalized();
      m_dx_.push_back(direction.x());
      m_dy_.push_back(direction.y());
      m_dz_.push_back(direction.z());
 
      //      double accuracy =
      //      step.stepSize.value(Acts::ConstrainedStep::accuracy());
      double accuracy = step.stepSize.accuracy();
      double actor = step.stepSize.value(Acts::ConstrainedStep::actor);
      double aborter = step.stepSize.value(Acts::ConstrainedStep::aborter);
      double user = step.stepSize.value(Acts::ConstrainedStep::user);
      double act2 = actor * actor;
      double acc2 = accuracy * accuracy;
      double abo2 = aborter * aborter;
      double usr2 = user * user;
 
      // todo - fold with direction
      if (act2 < acc2 && act2 < abo2 && act2 < usr2) {
        m_step_type_.push_back(0);
      } else if (acc2 < abo2 && acc2 < usr2) {
        m_step_type_.push_back(1);
      } else if (abo2 < usr2) {
        m_step_type_.push_back(2);
      } else {
        m_step_type_.push_back(3);
      }
 
      // step size information
      m_step_acc_.push_back(accuracy);
      m_step_act_.push_back(actor);
      m_step_abt_.push_back(aborter);
      m_step_usr_.push_back(user);
    }
    m_output_tree_->Fill();
  }
  return true;
}

Member Data Documentation

m_approach_id_

std::vector<int> tracking::sim::PropagatorStepWriter::m_approach_id_

protected

surface identifier

Definition at line 64 of file PropagatorStepWriter.h.

Referenced by PropagatorStepWriter().

m_boundary_id_

std::vector<int> tracking::sim::PropagatorStepWriter::m_boundary_id_

protected

boundary identifier

Definition at line 62 of file PropagatorStepWriter.h.

Referenced by PropagatorStepWriter().

m_cfg_

Config tracking::sim::PropagatorStepWriter::m_cfg_

protected

the configuration object

Definition at line 55 of file PropagatorStepWriter.h.

Referenced by PropagatorStepWriter(), and ~PropagatorStepWriter().

m_dx_

std::vector<float> tracking::sim::PropagatorStepWriter::m_dx_

protected

global direction x_

Definition at line 69 of file PropagatorStepWriter.h.

Referenced by PropagatorStepWriter().

m_dy_

std::vector<float> tracking::sim::PropagatorStepWriter::m_dy_

protected

global direction y_

Definition at line 70 of file PropagatorStepWriter.h.

Referenced by PropagatorStepWriter().

m_dz_

std::vector<float> tracking::sim::PropagatorStepWriter::m_dz_

protected

global direction z_

Definition at line 71 of file PropagatorStepWriter.h.

Referenced by PropagatorStepWriter().

m_event_nr_

int tracking::sim::PropagatorStepWriter::m_event_nr_

protected

the event number of

Definition at line 59 of file PropagatorStepWriter.h.

Referenced by PropagatorStepWriter().

m_hit_x_

std::vector<float> tracking::sim::PropagatorStepWriter::m_hit_x_

protected

hit location X

Definition at line 77 of file PropagatorStepWriter.h.

Referenced by PropagatorStepWriter().

m_hit_y_

std::vector<float> tracking::sim::PropagatorStepWriter::m_hit_y_

protected

hit location Y

Definition at line 78 of file PropagatorStepWriter.h.

Referenced by PropagatorStepWriter().

m_hit_z_

std::vector<float> tracking::sim::PropagatorStepWriter::m_hit_z_

protected

hit location Z

Definition at line 79 of file PropagatorStepWriter.h.

Referenced by PropagatorStepWriter().

m_layer_id_

std::vector<int> tracking::sim::PropagatorStepWriter::m_layer_id_

protected

layer identifier if

Definition at line 63 of file PropagatorStepWriter.h.

Referenced by PropagatorStepWriter().

m_output_file_

TFile* tracking::sim::PropagatorStepWriter::m_output_file_

protected

the output file name

Definition at line 57 of file PropagatorStepWriter.h.

Referenced by PropagatorStepWriter(), and ~PropagatorStepWriter().

m_output_tree_

TTree* tracking::sim::PropagatorStepWriter::m_output_tree_

protected

the output tree

Definition at line 58 of file PropagatorStepWriter.h.

Referenced by PropagatorStepWriter(), and ~PropagatorStepWriter().

m_sensitive_id_

std::vector<int> tracking::sim::PropagatorStepWriter::m_sensitive_id_

protected

surface identifier

Definition at line 65 of file PropagatorStepWriter.h.

Referenced by PropagatorStepWriter().

m_start_mom_

std::vector<float> tracking::sim::PropagatorStepWriter::m_start_mom_

protected

start momentum of the particle propagated

Definition at line 83 of file PropagatorStepWriter.h.

Referenced by PropagatorStepWriter().

m_start_pos_

std::vector<float> tracking::sim::PropagatorStepWriter::m_start_pos_

protected

start position of the particle propagated

Definition at line 81 of file PropagatorStepWriter.h.

Referenced by PropagatorStepWriter().

m_step_abt_

std::vector<float> tracking::sim::PropagatorStepWriter::m_step_abt_

protected

aborter

Definition at line 75 of file PropagatorStepWriter.h.

Referenced by PropagatorStepWriter().

m_step_acc_

std::vector<float> tracking::sim::PropagatorStepWriter::m_step_acc_

protected

accuracy

Definition at line 73 of file PropagatorStepWriter.h.

Referenced by PropagatorStepWriter().

m_step_act_

std::vector<float> tracking::sim::PropagatorStepWriter::m_step_act_

protected

actor check

Definition at line 74 of file PropagatorStepWriter.h.

Referenced by PropagatorStepWriter().

m_step_type_

std::vector<int> tracking::sim::PropagatorStepWriter::m_step_type_

protected

step type

Definition at line 72 of file PropagatorStepWriter.h.

Referenced by PropagatorStepWriter().

m_step_usr_

std::vector<float> tracking::sim::PropagatorStepWriter::m_step_usr_

protected

user

Definition at line 76 of file PropagatorStepWriter.h.

Referenced by PropagatorStepWriter().

m_write_mutex_

std::mutex tracking::sim::PropagatorStepWriter::m_write_mutex_

protected

protect multi-threaded writes

Definition at line 56 of file PropagatorStepWriter.h.

m_x_

std::vector<float> tracking::sim::PropagatorStepWriter::m_x_

protected

global x_

Definition at line 66 of file PropagatorStepWriter.h.

Referenced by PropagatorStepWriter().

m_y_

std::vector<float> tracking::sim::PropagatorStepWriter::m_y_

protected

global y_

Definition at line 67 of file PropagatorStepWriter.h.

Referenced by PropagatorStepWriter().

m_z_

std::vector<float> tracking::sim::PropagatorStepWriter::m_z_

protected

global z_

Definition at line 68 of file PropagatorStepWriter.h.

Referenced by PropagatorStepWriter().

---

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

• Tracking/include/Tracking/Sim/PropagatorStepWriter.h
• Tracking/src/Tracking/Sim/PropagatorStepWriter.cxx