BTagJetBrowsable.cpp

Go to the documentation of this file.

#include "tmb_tree/BTagJetBrowsable.hpp"
#include "tmb_tree/TMBJet.hpp"

#include "TBranch.h"
#include "TTree.h"
#include "TBrowser.h"
#include "TPad.h"

#include <iostream>
#include <sstream>
ClassImp(BTagJetBrowsable)

// Wrapper for the registration method. Needed against MSVC, which 
// assigned different addr to the same method, depending on what
// translation unit you're in...
void BTagJetBrowsable::Register() {
  TVirtualBranchBrowsable::RegisterGenerator(GetBrowsables);
}

// Wrapper for the registration method. Needed against MSVC, which 
// assigned different addr to the same method, depending on what
// translation unit you're in...
void BTagJetBrowsable::Unregister() {
  TVirtualBranchBrowsable::UnregisterGenerator(GetBrowsables);
}

Int_t BTagJetBrowsable::GetBrowsables(TList& li, const TBranch* branch,
                                                                          const TVirtualBranchBrowsable* parent)
{

  TClass* clContained=0;
  TClass* clCollection=GetCollectionContainedType(branch, parent, clContained);
  if (!clCollection || !clContained) return 0;

  if (clContained != TMBJet::Class()) {
        return 0;
  }


  TString scope;
  if (parent) {
        parent->GetScope(scope);
        branch=parent->GetBranch();
  } else if (branch){
        scope=branch->GetName();
        scope+=".";
        const TBranch* mother=branch;
        while (mother != mother->GetMother() && (mother=mother->GetMother())) {
          TString nameMother(mother->GetName());
          if (!nameMother.EndsWith(".")) {
                scope.Prepend(".");
                scope.Prepend(nameMother);
          } else {
                if (mother != mother->GetMother()) {
                  // If the mother is the top level mother
                  // and its ends ends with a ., the name is already
                  // embedded!
                  scope.Prepend(nameMother);
                }
          }
        }
  } else {

        return 0;
  }

  // remove trailing delimiters
  if (scope.EndsWith(".")) scope.Remove(scope.Length()-1, 1);
  else if (scope.EndsWith("->")) scope.Remove(scope.Length()-2, 2);


  typedef std::vector<std::pair<std::string, std::string> > algs_to_do_type;
  algs_to_do_type algs_to_do;
  algs_to_do.push_back (std::make_pair ("SVT", "TIGHT"));
  algs_to_do.push_back (std::make_pair ("SVT", "MEDIUM"));
  algs_to_do.push_back (std::make_pair ("SVT", "LOOSE"));
  algs_to_do.push_back (std::make_pair ("JLIP", "TIGHT"));
  algs_to_do.push_back (std::make_pair ("JLIP", "MEDIUM"));
  algs_to_do.push_back (std::make_pair ("JLIP", "LOOSE"));
  algs_to_do.push_back (std::make_pair ("NN", "TIGHT"));
  algs_to_do.push_back (std::make_pair ("NN", "MEDIUM"));
  algs_to_do.push_back (std::make_pair ("NN", "LOOSE"));

  for (algs_to_do_type::const_iterator itr = algs_to_do.begin(); itr != algs_to_do.end(); itr++) {
        std::ostringstream thecall, thename, thetitle;
        thecall << scope.Data() << ".GetBTag("
          << "\"" << itr->first << "\","
          << "\"" << itr->second << "\")"
          << "->is_tagged()";

        thename << "@Tagged_" << itr->first << "_" << itr->second;
        thetitle << "Jet has a " << itr->first << " " << itr->second << " tag";

        BTagJetBrowsable *bb = new BTagJetBrowsable (thename.str().c_str(), thetitle.str().c_str(),
          thecall.str().c_str(), branch, parent);
        li.Add (bb);
  }


  return li.GetEntries();
}

void BTagJetBrowsable::Browse(TBrowser *b) {
  GetBranch()->GetTree()->Draw(GetDraw(), "", b ? b->GetDrawOption() : "");
  if (gPad) gPad->Update();
}

---