IBTaggingTruthTaggingTool.h

Go to the documentation of this file.

// IBTaggingTruthTaggingTool.h, (c) ATLAS Detector software
 
#ifndef CPIBTAGGINGTRUTHTAGGINGTOOL_H
#define CPIBTAGGINGTRUTHTAGGINGTOOL_H
 
#include "AsgTools/IAsgTool.h"
#include "PATInterfaces/ISystematicsTool.h"
#include "PATInterfaces/CorrectionCode.h"
#include "AsgMessaging/StatusCode.h"
#include "PATInterfaces/SystematicSet.h"
 
#include <string>
#include <vector>
 
#include "CalibrationDataInterface/CalibrationDataVariables.h"
 
#include "xAODJet/JetContainer.h"
#include "xAODJet/Jet.h"
 
#include "PATInterfaces/CorrectionCode.h"
#include "AsgMessaging/StatusCode.h"
#include "PATInterfaces/SystematicSet.h"
 
namespace Analysis {
  class TruthTagResults;
}
 
class IBTaggingTruthTaggingTool : virtual public CP::ISystematicsTool {
 
  ASG_TOOL_INTERFACE( IBTagTruthTaggingTool )
 
  public:
 
  virtual StatusCode setEffMapIndex(const std::string& flavour, unsigned int index)=0 ;
 
  virtual StatusCode CalculateResults( std::vector<float>& pt, std::vector<float>& eta, std::vector<int>& flav, std::vector<float>& tagw, Analysis::TruthTagResults& results,int rand_seed=-1)=0 ;
    
  virtual StatusCode CalculateResults( const xAOD::JetContainer& jets, Analysis::TruthTagResults& results,int rand_seed=-1)=0;
 
  virtual StatusCode CalculateResultsONNX( const std::vector<std::vector<float>>& node_feat, std::vector<float>& tagw,  Analysis::TruthTagResults& results, int rand_seed=-1)=0 ;
    
  virtual StatusCode CalculateResultsONNX( const xAOD::JetContainer& jets, const std::vector<std::vector<float>>& node_feat, Analysis::TruthTagResults& results, int rand_seed=-1)=0;
 
};
#endif // CPIBTAGGINGTRUTHTAGGINGTOOL_H