Back to home page

Project CMSSW displayed by LXR

 
 

    


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   // Class
0026   class EMTFContext {
0027   public:
0028     EMTFContext(const edm::ParameterSet&, edm::ConsumesCollector);
0029 
0030     ~EMTFContext();
0031 
0032     // Event configuration
0033     void update(const edm::Event&, const edm::EventSetup&);
0034 
0035     // Helpers
0036     GeometryTranslator geometry_translator_;
0037 
0038     // EMTF
0039     EMTFConfiguration config_;
0040     EMTFModel model_;
0041 
0042     // Prompt Neural Network
0043     tensorflow::GraphDef* prompt_graph_ptr_;
0044     tensorflow::Session* prompt_session_ptr_;
0045 
0046     // Displaced Neural Network
0047     tensorflow::GraphDef* disp_graph_ptr_;
0048     tensorflow::Session* disp_session_ptr_;
0049 
0050     // Data
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     // Algorithm
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 }  // namespace emtf::phase2
0068 
0069 #endif  // L1Trigger_L1TMuonEndCapPhase2_EMTFContext_h not defined