CandidateBoostedDoubleSecondaryVertexComputer.h

Go to the documentation of this file.

#ifndef RecoBTag_SecondaryVertex_CandidateBoostedDoubleSecondaryVertexComputer_h
#define RecoBTag_SecondaryVertex_CandidateBoostedDoubleSecondaryVertexComputer_h
 
#include "FWCore/Framework/interface/ESConsumesCollector.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "CommonTools/MVAUtils/interface/TMVAEvaluator.h"
#include "RecoBTau/JetTagComputer/interface/JetTagComputer.h"
#include "CondFormats/DataRecord/interface/GBRWrapperRcd.h"
 
class CandidateBoostedDoubleSecondaryVertexComputer : public JetTagComputer {
public:
  struct Tokens {
    Tokens(const edm::ParameterSet &parameters, edm::ESConsumesCollector &&cc);
    edm::ESGetToken<GBRForest, GBRWrapperRcd> gbrForest_;
  };
 
  CandidateBoostedDoubleSecondaryVertexComputer(const edm::ParameterSet &parameters, Tokens tokens);
 
  void initialize(const JetTagComputerRecord &) override;
  float discriminator(const TagInfoHelper &tagInfos) const override;
 
private:
  const edm::FileInPath weightFile_;
  const bool useGBRForest_;
  const bool useAdaBoost_;
  const Tokens tokens_;
 
  std::unique_ptr<TMVAEvaluator> mvaID;
};
 
#endif  // RecoBTag_SecondaryVertex_CandidateBoostedDoubleSecondaryVertexComputer_h