Record the MVA variables. More...

Public Member Functions
	Impl (const std::function< void(TTree &)> &setBranches, const std::string &rootFileName, const std::string &treeName="recorded_tree")
	Constructor creating the TFile and TTree as well as setting up the branches with the given function.

	~Impl ()
	Destructor finalising the tree.

void	write ()
	Write all captured variables to disk.

void	capture ()
	Capture the registered variable values and write them out.

Private Attributes
TFile *	m_tFile
	Reference to the open TFile.

StoreObjPtr< RootMergeable< TTree > >	m_tTree
	Reference to the TTree.

Detailed Description

Record the MVA variables.

Definition at line 28 of file Recorder.cc.

Constructor & Destructor Documentation

◆ Impl()

Impl	(	const std::function< void(TTree &)> &	setBranches,
		const std::string &	rootFileName,
		const std::string &	treeName = "recorded_tree" )

Constructor creating the TFile and TTree as well as setting up the branches with the given function.

Definition at line 70 of file Recorder.cc.

  : m_tFile(nullptr)
  , m_tTree(treeName, DataStore::c_Persistent)
{
  TDirectory* ptrSavedCurrentTDirectory = gDirectory;
 
  m_tFile = new TFile(rootFileName.c_str(), "RECREATE");
  m_tFile->cd();
 
  m_tTree.registerInDataStore(DataStore::c_DontWriteOut | DataStore::c_ErrorIfAlreadyRegistered);
  m_tTree.construct(treeName.c_str(), treeName.c_str());
 
  if (m_tTree) {
    setBranches(m_tTree->get());
  }
 
  if (ptrSavedCurrentTDirectory) {
    ptrSavedCurrentTDirectory->cd();
  } else {
    gROOT->cd();
  }
}

◆ ~Impl()

~Impl ( )

Destructor finalising the tree.

Definition at line 95 of file Recorder.cc.

{
  if (!ProcHandler::parallelProcessingUsed() or ProcHandler::isOutputProcess()) {
    if (m_tTree) {
      m_tTree->get().SetDirectory(nullptr);
    }
    if (m_tFile) {
      m_tFile->Close();
    }
  }
}

Member Function Documentation

◆ capture()

void capture ( )

Capture the registered variable values and write them out.

Definition at line 133 of file Recorder.cc.

{
  m_tTree->get().Fill();
}

◆ write()

void write ( )

Write all captured variables to disk.

Definition at line 107 of file Recorder.cc.

{
  if (!ProcHandler::parallelProcessingUsed() or ProcHandler::isOutputProcess()) {
    if (m_tTree) {
      // Due to some weird behaviour we  have to cd to the file of the TTree
      // before saving in order to have the tree in the correct file.
      TDirectory* ptrSavedCurrentTDirectory = gDirectory;
 
      TFile* tFile = m_tTree->get().GetCurrentFile();
      if (tFile) {
        // We still own the TFile.
        tFile->cd();
        m_tTree->get().Write("", TObject::kOverwrite);
        m_tTree->get().SetDirectory(nullptr);
      }
 
      if (ptrSavedCurrentTDirectory) {
        ptrSavedCurrentTDirectory->cd();
      } else {
        gROOT->cd();
      }
 
    }
  }
}

Member Data Documentation

◆ m_tFile

TFile* m_tFile

private

Reference to the open TFile.

Definition at line 51 of file Recorder.cc.

◆ m_tTree

StoreObjPtr<RootMergeable<TTree> > m_tTree

private

Reference to the TTree.

Definition at line 54 of file Recorder.cc.

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

tracking/trackingUtilities/mva/src/Recorder.cc

Public Member Functions

Private Attributes