Analysis::TagNtupleDumper Class Reference

#include <TagNtupleDumper.h>

Inheritance diagram for Analysis::TagNtupleDumper:

Collaboration diagram for Analysis::TagNtupleDumper:

Public Member Functions
	TagNtupleDumper (const std::string &name, const std::string &n, const IInterface *)
virtual	~TagNtupleDumper ()
virtual StatusCode	initialize () override
virtual StatusCode	finalize () override
virtual void	assignProbability (xAOD::BTagging *BTag, const std::map< std::string, double > &inputs, const std::string &jetauthor) const override

Private Attributes
ServiceHandle< ITHistSvc >	m_hist_svc
std::string	m_stream
std::map< std::pair< std::string, std::string >, float * > m_features	ATLAS_THREAD_SAFE
std::map< std::string, TTree * > m_trees	ATLAS_THREAD_SAFE
std::mutex m_mutex	ATLAS_THREAD_SAFE

Detailed Description

Definition at line 37 of file TagNtupleDumper.h.

Constructor & Destructor Documentation

TagNtupleDumper()

Analysis::TagNtupleDumper::TagNtupleDumper	(	const std::string &	name,
		const std::string &	n,
		const IInterface *	p
	)

Definition at line 29 of file TagNtupleDumper.cxx.

                                                                                                :
    base_class(name, n,p),
    m_hist_svc("THistSvc", name)
  {
    // setup stream
    declareProperty("Stream", m_stream = "FTAG");
  }

~TagNtupleDumper()

Analysis::TagNtupleDumper::~TagNtupleDumper ( )

virtual

Definition at line 37 of file TagNtupleDumper.cxx.

                                    {
    for (auto& pair: m_features) {
      delete pair.second;
    }
    // no idea how the hist service handles trees, do I have to delete them?
    // for (auto& tree: m_trees) {
    //   delete tree.second;
    // }
  }

Member Function Documentation

assignProbability()

void Analysis::TagNtupleDumper::assignProbability ( xAOD::BTagging *  BTag, const std::map< std::string, double > &  inputs, const std::string &  jetauthor  ) const

overridevirtual

Definition at line 56 of file TagNtupleDumper.cxx.

  {
    std::lock_guard<std::mutex> lock (m_mutex);
 
    TTree* & tree = m_trees[jetauthor];
    if (tree == nullptr) {
      tree = new TTree(jetauthor.c_str(), "who cares");
      if (m_hist_svc->regTree("/" + m_stream + "/" + jetauthor,
                              tree).isFailure()) {
        ATH_MSG_ERROR("Cannot register tree " + m_stream + "/" + jetauthor);
        return;
      }
    }
    for (const auto& in: inputs) {
      const auto idx = std::make_pair(in.first, jetauthor);
      // add this branch if it doesn't exist
      float* & feature = m_features[idx];
      if (feature == nullptr) {
        feature = new float;
        m_trees.at(jetauthor)->Branch(in.first.c_str(), feature);
        ATH_MSG_INFO("Added Branch " + jetauthor + ": " + in.first );
      }
      *feature = in.second;
    }
    // fill tree
    m_trees.at(jetauthor)->Fill();
  } //end assign_probability

finalize()

StatusCode Analysis::TagNtupleDumper::finalize ( )

overridevirtual

Definition at line 52 of file TagNtupleDumper.cxx.

                                       { // all taken care of in destructor
    return StatusCode::SUCCESS;
  }

initialize()

StatusCode Analysis::TagNtupleDumper::initialize ( )

overridevirtual

Definition at line 47 of file TagNtupleDumper.cxx.

                                         {
    CHECK( m_hist_svc.retrieve() );
    return StatusCode::SUCCESS;
  }

Member Data Documentation

ATLAS_THREAD_SAFE [1/3]

std::map< std::pair<std::string, std::string>, float* > m_features Analysis::TagNtupleDumper::ATLAS_THREAD_SAFE

mutableprivate

Definition at line 57 of file TagNtupleDumper.h.

ATLAS_THREAD_SAFE [2/3]

std::map<std::string, TTree*> m_trees Analysis::TagNtupleDumper::ATLAS_THREAD_SAFE

mutableprivate

Definition at line 58 of file TagNtupleDumper.h.

ATLAS_THREAD_SAFE [3/3]

std::mutex m_mutex Analysis::TagNtupleDumper::ATLAS_THREAD_SAFE

mutableprivate

Definition at line 60 of file TagNtupleDumper.h.

m_hist_svc

ServiceHandle<ITHistSvc> Analysis::TagNtupleDumper::m_hist_svc

private

Definition at line 54 of file TagNtupleDumper.h.

m_stream

std::string Analysis::TagNtupleDumper::m_stream

private

Definition at line 55 of file TagNtupleDumper.h.

---

The documentation for this class was generated from the following files:

• TagNtupleDumper.h
• TagNtupleDumper.cxx