#ifndef TMBBCNNTagger_H
#define TMBBCNNTagger_H

//////////////////////////////////////////////////////////////////////////
//                                                                      //
// TMBTree class for the BCJet NN tagger                                //
//                                                                      //
// Created: 18-JUL-2002  A. Naumann (axel@fnal.gov)                     //
//                                                                      //
//////////////////////////////////////////////////////////////////////////

#ifndef TMBBCTagger_H
#include "tmb_tree/TMBBCTagger.hpp"
#endif

class TMBBCNNTagger : public TMBBCTagger {
public:
  typedef enum _ECNNParameter {
    paramBJetProb,
    paramCJetProb,
    numParameters
  } ECNNParameter;

  inline TMBBCNNTagger();
  inline virtual ~TMBBCNNTagger();
	
  // Accessors
  inline virtual Float_t GetBJetProbability();
  inline virtual Float_t GetCJetProbability();
 
private:
  ClassDef(TMBBCNNTagger, 0) // thumbnail data of muon bc jet id tagger 
};

TMBBCNNTagger::TMBBCNNTagger(): TMBBCTagger(numParameters) {}

TMBBCNNTagger::~TMBBCNNTagger(){}

Float_t TMBBCNNTagger::GetBJetProbability() {
    return GetParameterOutput(paramBJetProb);
}

Float_t TMBBCNNTagger::GetCJetProbability() {
    return GetParameterOutput(paramCJetProb);
}

#endif // ifndef TMBBCNNTaggger_H