SecondaryVertex/interface/CandidateBoostedDoubleSecondaryVertexComputer.h

0001 #ifndef RecoBTag_SecondaryVertex_CandidateBoostedDoubleSecondaryVertexComputer_h
0002 #define RecoBTag_SecondaryVertex_CandidateBoostedDoubleSecondaryVertexComputer_h
0003
0004 #include "CommonTools/MVAUtils/interface/TMVAEvaluator.h"
0005 #include "CondFormats/DataRecord/interface/GBRWrapperRcd.h"
0006 #include "FWCore/Framework/interface/ESConsumesCollector.h"
0007 #include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"
0008 #include "FWCore/ParameterSet/interface/ParameterSet.h"
0009 #include "FWCore/ParameterSet/interface/ParameterSetDescription.h"
0010 #include "RecoBTau/JetTagComputer/interface/JetTagComputer.h"
0011
0012 class CandidateBoostedDoubleSecondaryVertexComputer : public JetTagComputer {
0013 public:
0014   struct Tokens {
0015     Tokens(const edm::ParameterSet &parameters, edm::ESConsumesCollector &&cc);
0016     edm::ESGetToken<GBRForest, GBRWrapperRcd> gbrForest_;
0017   };
0018
0019   CandidateBoostedDoubleSecondaryVertexComputer(const edm::ParameterSet &parameters, Tokens tokens);
0020
0021   void initialize(const JetTagComputerRecord &) override;
0022   float discriminator(const TagInfoHelper &tagInfos) const override;
0023
0024   static void fillPSetDescription(edm::ParameterSetDescription &desc);
0025
0026 private:
0027   const edm::FileInPath weightFile_;
0028   const bool useGBRForest_;
0029   const bool useAdaBoost_;
0030   const Tokens tokens_;
0031
0032   std::unique_ptr<TMVAEvaluator> mvaID;
0033 };
0034
0035 #endif  // RecoBTag_SecondaryVertex_CandidateBoostedDoubleSecondaryVertexComputer_h