Back to home page

Project CMSSW displayed by LXR

 
 

    


File indexing completed on 2021-10-08 22:26:30

0001 #ifndef __L1Trigger_L1THGCal_HGCalConcentratorAutoEncoderImpl_h__
0002 #define __L1Trigger_L1THGCal_HGCalConcentratorAutoEncoderImpl_h__
0003 
0004 #include "FWCore/ParameterSet/interface/ParameterSet.h"
0005 #include "DataFormats/L1THGCal/interface/HGCalTriggerCell.h"
0006 #include "DataFormats/L1THGCal/interface/HGCalConcentratorData.h"
0007 #include "L1Trigger/L1THGCal/interface/HGCalTriggerTools.h"
0008 #include <vector>
0009 
0010 #include "L1Trigger/L1THGCal/interface/HGCalTriggerGeometryBase.h"
0011 
0012 #include "PhysicsTools/TensorFlow/interface/TensorFlow.h"
0013 
0014 class HGCalConcentratorAutoEncoderImpl {
0015 public:
0016   HGCalConcentratorAutoEncoderImpl(const edm::ParameterSet& conf);
0017 
0018   void select(unsigned nLinks,
0019               const std::vector<l1t::HGCalTriggerCell>& trigCellVecInput,
0020               std::vector<l1t::HGCalTriggerCell>& trigCellVecOutput,
0021               std::vector<l1t::HGCalConcentratorData>& ae_EncodedOutput);
0022 
0023   void setGeometry(const HGCalTriggerGeometryBase* const geom) { triggerTools_.setGeometry(geom); }
0024 
0025 private:
0026   static constexpr int nTriggerCells_ = 48;
0027 
0028   static constexpr int maxAEInputSize_ = 192;
0029 
0030   static constexpr int nEncodedLayerNodes_ = 16;
0031 
0032   static constexpr unsigned int maxNumberOfLinks_ = 13;
0033 
0034   static constexpr int cellUVSize_ = 8;
0035 
0036   static constexpr int encoderTensorDims_ = 4;
0037 
0038   static constexpr int decoderTensorDims_ = 2;
0039 
0040   static constexpr int cellUVremap_[cellUVSize_][cellUVSize_] = {{47, 46, 45, 44, -1, -1, -1, -1},
0041                                                                  {16, 43, 42, 41, 40, -1, -1, -1},
0042                                                                  {20, 17, 39, 38, 37, 36, -1, -1},
0043                                                                  {24, 21, 18, 35, 34, 33, 32, -1},
0044                                                                  {28, 25, 22, 19, 3, 7, 11, 15},
0045                                                                  {-1, 29, 26, 23, 2, 6, 10, 14},
0046                                                                  {-1, -1, 30, 27, 1, 5, 9, 13},
0047                                                                  {-1, -1, -1, 31, 0, 4, 8, 12}};
0048   static constexpr int ae_outputCellU_[nTriggerCells_] = {7, 6, 5, 4, 7, 6, 5, 4, 7, 6, 5, 4, 7, 6, 5, 4,
0049                                                           1, 2, 3, 4, 2, 3, 4, 5, 3, 4, 5, 6, 4, 5, 6, 7,
0050                                                           3, 3, 3, 3, 2, 2, 2, 2, 1, 1, 1, 1, 0, 0, 0, 0};
0051   static constexpr int ae_outputCellV_[nTriggerCells_] = {4, 4, 4, 4, 5, 5, 5, 5, 6, 6, 6, 6, 7, 7, 7, 7,
0052                                                           0, 1, 2, 3, 0, 1, 2, 3, 0, 1, 2, 3, 0, 1, 2, 3,
0053                                                           6, 5, 4, 3, 5, 4, 3, 2, 4, 3, 2, 1, 3, 2, 1, 0};
0054 
0055   unsigned int nInputs_;
0056   std::vector<int> cellRemap_;
0057   std::vector<int> cellRemapNoDuplicates_;
0058   std::vector<uint> encoderShape_;
0059   std::vector<uint> decoderShape_;
0060   int bitsPerInput_;
0061   int maxBitsPerOutput_;
0062   std::vector<int> outputBitsPerLink_;
0063 
0064   std::vector<edm::ParameterSet> modelFilePaths_;
0065 
0066   std::string inputTensorName_encoder_;
0067   std::string outputTensorName_encoder_;
0068   std::unique_ptr<tensorflow::GraphDef> graphDef_encoder_;
0069   std::vector<std::unique_ptr<tensorflow::Session>> session_encoder_;
0070 
0071   std::string inputTensorName_decoder_;
0072   std::string outputTensorName_decoder_;
0073   std::unique_ptr<tensorflow::GraphDef> graphDef_decoder_;
0074   std::vector<std::unique_ptr<tensorflow::Session>> session_decoder_;
0075 
0076   std::vector<unsigned int> linkToGraphMap_;
0077 
0078   double zeroSuppresionThreshold_;
0079   bool bitShiftNormalization_;
0080   bool saveEncodedValues_;
0081   bool preserveModuleSum_;
0082 
0083   std::array<double, nEncodedLayerNodes_> ae_encodedLayer_;
0084 
0085   HGCalTriggerTools triggerTools_;
0086 };
0087 
0088 #endif