L1TMuonOverlapPhase2/interface/PtAssignmentNNRegression.h

0001 /*
0002  * PtAssignmentNN.h
0003  *
0004  *  Created on: May 8, 2020
0005  *      Author: kbunkow
0006  */
0007
0008 #ifndef L1Trigger_L1TMuonOverlapPhase2_PtAssigmentNNRegression_h
0009 #define L1Trigger_L1TMuonOverlapPhase2_PtAssigmentNNRegression_h
0010
0011 #include "L1Trigger/L1TMuonOverlapPhase1/interface/Omtf/PtAssignmentBase.h"
0012 #include "L1Trigger/L1TMuonOverlapPhase2/interface/LutNetworkFixedPointRegression2Outputs.h"
0013
0014 class PtAssignmentNNRegression : public PtAssignmentBase {
0015 public:
0016   PtAssignmentNNRegression(const edm::ParameterSet& edmCfg,
0017                            const OMTFConfiguration* omtfConfig,
0018                            std::string networkFile);
0019   ~PtAssignmentNNRegression() override = default;
0020
0021   std::vector<float> getPts(AlgoMuons::value_type& algoMuon,
0022                             std::vector<std::unique_ptr<IOMTFEmulationObserver> >& observers) override;
0023
0024 private:
0025   unique_ptr<lutNN::LutNetworkFixedPointRegressionBase> lutNetworkFP;
0026 };
0027
0028 #endif /* L1Trigger_L1TMuonOverlapPhase2_PtAssigmentNNRegression_h */