#include <EMTFContext.h>

Public Member Functions
	EMTFContext (const edm::ParameterSet &, edm::ConsumesCollector)

void	update (const edm::Event &, const edm::EventSetup &)

	~EMTFContext ()

Public Attributes
data::ActivationLut	activation_lut_

EMTFConfiguration	config_

tensorflow::GraphDef *	disp_graph_ptr_

tensorflow::Session *	disp_session_ptr_

algo::DuplicateRemovalLayer	duplicate_removal_layer_

GeometryTranslator	geometry_translator_

algo::HitmapLayer	hitmap_building_layer_

data::HostLut	host_lut_

EMTFModel	model_

algo::OutputLayer	output_layer_

algo::ParameterAssignmentLayer	parameter_assignment_layer_

algo::PatternMatchingLayer	pattern_matching_layer_

tensorflow::GraphDef *	prompt_graph_ptr_

tensorflow::Session *	prompt_session_ptr_

algo::RoadSortingLayer	road_sorting_layer_

data::SiteLut	site_lut_

data::TimeZoneLut	timezone_lut_

algo::TrackBuildingLayer	track_building_layer_

data::ZoneLut	zone_lut_

Detailed Description

Definition at line 26 of file EMTFContext.h.

Constructor & Destructor Documentation

◆ EMTFContext()

EMTFContext::EMTFContext	(	const edm::ParameterSet &	pset,
		edm::ConsumesCollector	i_consumes_collector
	)

Definition at line 13 of file EMTFContext.cc.

     :  // Helpers
       geometry_translator_(i_consumes_collector),
 
       // EMTF
       config_(pset),
       model_(*this),
 
       // Prompt Neural Network
       prompt_graph_ptr_(nullptr),
       prompt_session_ptr_(nullptr),
 
       // Displaced Neural Network
       disp_graph_ptr_(nullptr),
       disp_session_ptr_(nullptr),
 
       // Data
       site_lut_(),
       host_lut_(),
       zone_lut_(),
       timezone_lut_(),
       activation_lut_(),
 
       // Layers
       hitmap_building_layer_(*this),
       pattern_matching_layer_(*this),
       road_sorting_layer_(*this),
       track_building_layer_(*this),
       duplicate_removal_layer_(*this),
       parameter_assignment_layer_(*this),
       output_layer_(*this) {
   // Do Nothing
 }

◆ ~EMTFContext()

EMTFContext::~EMTFContext ( )

Definition at line 47 of file EMTFContext.cc.

References tensorflow::closeSession(), disp_graph_ptr_, disp_session_ptr_, prompt_graph_ptr_, and prompt_session_ptr_.

                           {
   // Delete Prompt Neural Network
   if (prompt_session_ptr_ != nullptr) {
     tensorflow::closeSession(prompt_session_ptr_);
     delete prompt_session_ptr_;
   }
 
   if (prompt_graph_ptr_ != nullptr) {
     delete prompt_graph_ptr_;
   }
 
   // Delete Displaced Neural Network
   if (disp_session_ptr_ != nullptr) {
     tensorflow::closeSession(disp_session_ptr_);
     delete disp_session_ptr_;
   }
 
   if (disp_graph_ptr_ != nullptr) {
     delete disp_graph_ptr_;
   }
 }

Member Function Documentation

◆ update()

void EMTFContext::update	(	const edm::Event &	i_event,
		const edm::EventSetup &	i_event_setup
	)

Definition at line 69 of file EMTFContext.cc.

References activation_lut_, L1TMuon::GeometryTranslator::checkAndUpdateGeometry(), config_, tensorflow::createSession(), disp_graph_ptr_, disp_session_ptr_, emtf::phase2::EMTFConfiguration::displ_graph_path_, edm::FileInPath::fullPath(), geometry_translator_, host_lut_, tensorflow::loadGraphDef(), emtf::phase2::EMTFConfiguration::prompt_graph_path_, prompt_graph_ptr_, prompt_session_ptr_, site_lut_, timezone_lut_, emtf::phase2::EMTFConfiguration::update(), emtf::phase2::data::ActivationLut::update(), emtf::phase2::data::ZoneLut::update(), emtf::phase2::data::TimeZoneLut::update(), emtf::phase2::data::HostLut::update(), emtf::phase2::data::SiteLut::update(), and zone_lut_.

Referenced by progressbar.ProgressBar::__next__(), MatrixUtil.Matrix::__setitem__(), MatrixUtil.Steps::__setitem__(), progressbar.ProgressBar::finish(), MatrixUtil.Steps::overwrite(), and emtf::phase2::TrackFinder::process().

                                                                                     {
   // Update Helpers
   geometry_translator_.checkAndUpdateGeometry(i_event_setup);
 
   // Update Config
   config_.update(i_event, i_event_setup);
 
   // Update Prompt Neural Network
   if (prompt_session_ptr_ != nullptr) {
     delete prompt_session_ptr_;
   }
 
   if (prompt_graph_ptr_ != nullptr) {
     delete prompt_graph_ptr_;
   }
 
   prompt_graph_ptr_ = tensorflow::loadGraphDef(edm::FileInPath(config_.prompt_graph_path_).fullPath());
 
   prompt_session_ptr_ = tensorflow::createSession(prompt_graph_ptr_);
 
   // Update Displaced Neural Network
   if (disp_session_ptr_ != nullptr) {
     delete disp_session_ptr_;
   }
 
   if (disp_graph_ptr_ != nullptr) {
     delete disp_graph_ptr_;
   }
 
   disp_graph_ptr_ = tensorflow::loadGraphDef(edm::FileInPath(config_.displ_graph_path_).fullPath());
 
   disp_session_ptr_ = tensorflow::createSession(disp_graph_ptr_);
 
   // Update Data
   site_lut_.update(i_event, i_event_setup);
   host_lut_.update(i_event, i_event_setup);
   zone_lut_.update(i_event, i_event_setup);
   timezone_lut_.update(i_event, i_event_setup);
   activation_lut_.update(i_event, i_event_setup);
 }