File indexing completed on 2024-04-06 12:21:00
0001 #ifndef L1Trigger_L1TMuonEndCapPhase2_EMTFContext_h
0002 #define L1Trigger_L1TMuonEndCapPhase2_EMTFContext_h
0003
0004 #include "PhysicsTools/TensorFlow/interface/TensorFlow.h"
0005
0006 #include "L1Trigger/L1TMuonEndCapPhase2/interface/EMTFfwd.h"
0007 #include "L1Trigger/L1TMuonEndCapPhase2/interface/EMTFConfiguration.h"
0008 #include "L1Trigger/L1TMuonEndCapPhase2/interface/EMTFModel.h"
0009 #include "L1Trigger/L1TMuonEndCapPhase2/interface/EMTFTypes.h"
0010 #include "L1Trigger/L1TMuonEndCapPhase2/interface/Algo/HitmapLayer.h"
0011 #include "L1Trigger/L1TMuonEndCapPhase2/interface/Algo/PatternMatchingLayer.h"
0012 #include "L1Trigger/L1TMuonEndCapPhase2/interface/Algo/RoadSortingLayer.h"
0013 #include "L1Trigger/L1TMuonEndCapPhase2/interface/Algo/TrackBuildingLayer.h"
0014 #include "L1Trigger/L1TMuonEndCapPhase2/interface/Algo/DuplicateRemovalLayer.h"
0015 #include "L1Trigger/L1TMuonEndCapPhase2/interface/Algo/ParameterAssignmentLayer.h"
0016 #include "L1Trigger/L1TMuonEndCapPhase2/interface/Algo/OutputLayer.h"
0017 #include "L1Trigger/L1TMuonEndCapPhase2/interface/Data/ActivationLut.h"
0018 #include "L1Trigger/L1TMuonEndCapPhase2/interface/Data/HostLut.h"
0019 #include "L1Trigger/L1TMuonEndCapPhase2/interface/Data/SiteLut.h"
0020 #include "L1Trigger/L1TMuonEndCapPhase2/interface/Data/ZoneLut.h"
0021 #include "L1Trigger/L1TMuonEndCapPhase2/interface/Data/TimeZoneLut.h"
0022
0023 namespace emtf::phase2 {
0024
0025
0026 class EMTFContext {
0027 public:
0028 EMTFContext(const edm::ParameterSet&, edm::ConsumesCollector);
0029
0030 ~EMTFContext();
0031
0032
0033 void update(const edm::Event&, const edm::EventSetup&);
0034
0035
0036 GeometryTranslator geometry_translator_;
0037
0038
0039 EMTFConfiguration config_;
0040 EMTFModel model_;
0041
0042
0043 tensorflow::GraphDef* prompt_graph_ptr_;
0044 tensorflow::Session* prompt_session_ptr_;
0045
0046
0047 tensorflow::GraphDef* disp_graph_ptr_;
0048 tensorflow::Session* disp_session_ptr_;
0049
0050
0051 data::SiteLut site_lut_;
0052 data::HostLut host_lut_;
0053 data::ZoneLut zone_lut_;
0054 data::TimeZoneLut timezone_lut_;
0055 data::ActivationLut activation_lut_;
0056
0057
0058 algo::HitmapLayer hitmap_building_layer_;
0059 algo::PatternMatchingLayer pattern_matching_layer_;
0060 algo::RoadSortingLayer road_sorting_layer_;
0061 algo::TrackBuildingLayer track_building_layer_;
0062 algo::DuplicateRemovalLayer duplicate_removal_layer_;
0063 algo::ParameterAssignmentLayer parameter_assignment_layer_;
0064 algo::OutputLayer output_layer_;
0065 };
0066
0067 }
0068
0069 #endif