TMBBTag.cpp

Go to the documentation of this file.

//
// Name: TMBBTag.cpp
//
// Purpose: Implementation for class TMBBTag.
//
// Created: 25-Jan-2005
//

#include "tmb_tree/TMBBTag.hpp"
#include "tmb_tree/TMBMuon.hpp"
#include "tmb_tree/TMBJet.hpp"
#include "TTree.h"
#include "TBranch.h"
#include <iostream>

TMBBTag::TMBBTag() :
  _taggable(kFALSE),
  _tagged(kFALSE),
  _neg_tagged(kFALSE),
  _trf_data(0),
  _trf_data_err(0),
  _trf_mc(0),
  _trf_sf(0),
  _mc_flavor(-1),
  _num_tracks(0),
  _num_neg_tracks(0)
  //
  // Purpose: Default constructor.
  //
{}

TMBBTag::TMBBTag(Bool_t taggable,
                 Bool_t tagged,
                 Bool_t neg_tagged,
                 Int_t mc_flavor,
                 const TMBJet* jet,
                 const TMBMuon* muon,
                 const TRefArray* tracks,
                 const TRefArray* neg_tracks) :
  _taggable(taggable),
  _tagged(tagged),
  _neg_tagged(neg_tagged),
  _trf_data(0),
  _trf_data_err(0),
  _trf_mc(0),
  _trf_sf(0),
  _mc_flavor(mc_flavor)
  //
  // Purpose: Initializing constructor.
  //
  // Arguments: taggable   - Taggable flag.
  //            tagged     - Tag decision (positive).
  //            neg_tagged - Tag decision (negative).
  //            mc_flavor  - MC jet flavor (-1=unknown, 0=light, 4=c, 5=b).
  //            jet        - Associated reconstructed jet (TMBJet).
  //            muon       - Associated reconstructed muon (TMBMuon).
  //            tracks     - Associated tracks (TMBTrack).
  //            neg_tracks - Associated negative tracks (TMBTrack).
  //
{
  _jet = (TMBJet*)jet;
  _muon = (TMBMuon*)muon;

  //std::cout<<"Make BTag..."<<std::endl;
  //if (!_jet.IsValid()) std::cout<<"invalid jet!"<<std::endl;

  _tracks.Clear();
  if (tracks) {
    for (Int_t i=0; i<tracks->GetLast()+1; i++) {
      _tracks.Add( tracks->At(i) );
    }
  }

  _neg_tracks.Clear();
  if (neg_tracks) {
    for (Int_t i=0; i<neg_tracks->GetLast()+1; i++) {
      _neg_tracks.Add( neg_tracks->At(i) );
    }
  }
  
  //_tracks = *tracks;
  //_neg_tracks = *neg_tracks;
  
  _num_tracks = _tracks.GetEntriesFast();
  _num_neg_tracks = _neg_tracks.GetEntriesFast();
}

TMBBTag::~TMBBTag()
//
// Purpose: Destructor.
//
{}

bool TMBBTag::print() const
{
  bool ret=true;
  std::cout<<"BTag:";
  if (GetJet()) {
    if (GetJet()->isEM()) std::cout<<" isEM ";
    if (GetJet()->isBad()) std::cout<<" isBad ";
    std::cout<<" taggable: "<<_taggable<<"="<<GetJet()->taggable();
    if (_taggable!=GetJet()->taggable()) {std::cout<<" - mismatch!"; ret=false;}
    std::cout<<" id: "<<GetId()<<"="<<GetJet()->btagdictcheck(_algo);
    if (GetId()!=GetJet()->btagdictcheck(_algo)) {std::cout<<" - mismatch!"; ret=false;}
  }
  else {std::cout<<" bad associated jet"; ret=false;}
  
  std::cout<<std::endl;
  return ret;
}

const TMBJet* TMBBTag::GetJet() const {
  //First make sure we read the jet branch
  TTree *tree = (TTree*)gROOT->FindObject("TMBTree");
  if (tree){
    //std::cout<<"finding branch "<<_algo<<std::endl;
    TBranch *b =tree->GetBranch("BTag_"+_algo);
    if (b) {
      //std::cout<<"finding current entry"<<std::endl;
      Long64_t current = b->GetReadEntry();
      //std::cout<<"entry="<<current<<std::endl;
      b =tree->GetBranch(_jetbranch);
      if (current>=0 && b) {
        Long64_t other_current = b->GetReadEntry();
        //std::cout<<"other entry="<<other_current<<std::endl;
        if (other_current!=current) {
          //std::cout<<"reading entry "<<current<<" from branch "<<_jetbranch<<std::endl;
          b->GetEntry(current);
        }
      }
    }
  }
  return _jet.IsValid() ? (const TMBJet *)_jet.GetObject() : 0;
}

const TMBMuon* TMBBTag::GetMuon() const {
  //First make sure we read the muon branch
  TTree *tree = (TTree*)gROOT->FindObject("TMBTree");
  if (tree){
    //std::cout<<"finding branch "<<_algo<<std::endl;
    TBranch *b =tree->GetBranch("BTag_"+_algo);
    if (b) {
      //std::cout<<"finding current entry"<<std::endl;
      Long64_t current = b->GetReadEntry();
      //std::cout<<"entry="<<current<<std::endl;
      b =tree->GetBranch(_muonbranch);
      if (current>=0 && b) {
        Long64_t other_current = b->GetReadEntry();
        //std::cout<<"other entry="<<other_current<<std::endl;
        if (other_current!=current) {
          //std::cout<<"reading entry "<<current<<" from branch "<<_muonbranch<<std::endl;
          b->GetEntry(current);
        }
      }
    }
  }
  return _muon.IsValid() ? (const TMBMuon *)_muon.GetObject() : 0;
}

---