TestPerformanceFWLite_ES.cc

Go to the documentation of this file.

#include <vector>
#include <iostream>

#include "TFile.h"
#include "TTree.h"
#include "FWCore/FWLite/interface/FWLiteEnabler.h"

#include "PhysicsTools/CondLiteIO/interface/RecordWriter.h"
#include "DataFormats/FWLite/interface/Record.h"
#include "DataFormats/FWLite/interface/EventSetup.h"
#include "DataFormats/FWLite/interface/ESHandle.h"
#include "CondFormats/PhysicsToolsObjects/interface/BinningPointByMap.h"
#include "RecoBTag/PerformanceDB/interface/BtagPerformance.h"

int main(int argc, char** argv) {
  FWLiteEnabler::enable();

  std::cout << "Test!!!" << std::endl << std::endl;
  TFile f("performance_ssvm.root", "READ");

  fwlite::EventSetup es(&f);

  if (es.exists("BTagPerformanceRecord")) {
    std::cout << "Got the right tree" << std::endl;
  } else {
    std::cout << "Can't find tree" << std::endl;
  }

  fwlite::RecordID testRecID = es.recordID("BTagPerformanceRecord");

  int index = 1001;
  es.syncTo(edm::EventID(index, 0, 0), edm::Timestamp());

  std::cout << "Got record ID " << testRecID << es.get(testRecID).startSyncValue().eventID() << std::endl;

  fwlite::ESHandle<PerformancePayload> plHandle;
  es.get(testRecID).get(plHandle, "MCPfTCHEMb");
  fwlite::ESHandle<PerformanceWorkingPoint> wpHandle;
  es.get(testRecID).get(wpHandle, "MCPfTCHEMb");

  if (plHandle.isValid() && wpHandle.isValid()) {
    BtagPerformance perf(*plHandle, *wpHandle);

    std::cout << "Values: " << PerformanceResult::BTAGNBEFF << " " << PerformanceResult::MUERR << " " << std::endl;

    // check beff, berr for eta=.6, et=55;
    BinningPointByMap p;

    //     std::cout <<" My Performance Object is indeed a "<<typeid(perf).name()<<std::endl;

    std::cout << " test eta=0.6, et=55" << std::endl;

    p.insert(BinningVariables::JetAbsEta, 0.6);
    p.insert(BinningVariables::JetEt, 55);
    std::cout << " nbeff/nberr ?" << perf.isResultOk(PerformanceResult::BTAGNBEFF, p) << "/"
              << perf.isResultOk(PerformanceResult::BTAGNBERR, p) << std::endl;
    std::cout << " beff/berr ?" << perf.isResultOk(PerformanceResult::BTAGBEFF, p) << "/"
              << perf.isResultOk(PerformanceResult::BTAGBERR, p) << std::endl;
    std::cout << " beff/berr =" << perf.getResult(PerformanceResult::BTAGBEFF, p) << "/"
              << perf.getResult(PerformanceResult::BTAGBERR, p) << std::endl;

    std::cout << " test eta=1.9, et=33" << std::endl;
    p.insert(BinningVariables::JetAbsEta, 1.9);
    p.insert(BinningVariables::JetEt, 33);
    std::cout << " beff/berr ?" << perf.isResultOk(PerformanceResult::BTAGBEFF, p) << "/"
              << perf.isResultOk(PerformanceResult::BTAGBERR, p) << std::endl;
    std::cout << " beff/berr =" << perf.getResult(PerformanceResult::BTAGBEFF, p) << "/"
              << perf.getResult(PerformanceResult::BTAGBERR, p) << std::endl;

    std::cout << " The WP is defined by a cut at " << perf.workingPoint().cut() << std::endl;
    std::cout << " Discriminant is " << perf.workingPoint().discriminantName() << std::endl;

    std::cout << " now I ask for a calibration but I do not set eta in the binning point ---> should return all not "
                 "available "
              << std::endl;
    p.reset();
    p.insert(BinningVariables::JetNTracks, 3);
    p.insert(BinningVariables::JetEt, 55);
    std::cout << " beff/berr ?" << perf.isResultOk(PerformanceResult::BTAGBEFF, p) << "/"
              << perf.isResultOk(PerformanceResult::BTAGBERR, p) << std::endl;
    std::cout << " beff/berr =" << perf.getResult(PerformanceResult::BTAGBEFF, p) << "/"
              << perf.getResult(PerformanceResult::BTAGBERR, p) << std::endl;

    //  std::cout <<" now I ask for a calibration which is not present ---> should throw an exception "<<std::endl;

    //  edm::ESHandle<BtagPerformance> perfH2;
    //  iSetup.get<BTagPerformanceRecord>().get("TrackCountingHighEff_tight",perfH2);

  } else {
    std::cout << "invalid handle: workingPoint " << wpHandle.isValid() << " payload " << plHandle.isValid()
              << std::endl;
    try {
      *wpHandle;
      *plHandle;
    } catch (std::exception& iE) {
      std::cout << iE.what() << std::endl;
    }
  }
}