podio::ROOTWriter Class Reference

#include <ROOTWriter.h>

Public Member Functions
	ROOTWriter (const std::string &filename, EventStore *store)
	~ROOTWriter ()
	ROOTWriter (const ROOTWriter &)=delete
ROOTWriter &	operator= (const ROOTWriter &)=delete
bool	registerForWrite (const std::string &name)
void	writeEvent ()
void	finish ()

Detailed Description

Definition at line 22 of file ROOTWriter.h.

Constructor & Destructor Documentation

ROOTWriter() [1/2]

podio::ROOTWriter::ROOTWriter	(	const std::string &	filename,
		EventStore *	store
	)

Definition at line 14 of file ROOTWriter.cc.

                                                                   :
    m_filename(filename),
    m_store(store),
    m_file(new TFile(filename.c_str(), "RECREATE", "data file")),
    m_datatree(new TTree("events", "Events tree")),
    m_metadatatree(new TTree("metadata", "Metadata tree")),
    m_runMDtree(new TTree("run_metadata", "Run metadata tree")),
    m_evtMDtree(new TTree("evt_metadata", "Event metadata tree")),
    m_colMDtree(new TTree("col_metadata", "Collection metadata tree")) {
 
  m_evtMDtree->Branch("evtMD", "GenericParameters", m_store->eventMetaDataPtr());
}

~ROOTWriter()

podio::ROOTWriter::~ROOTWriter

(

)

Definition at line 27 of file ROOTWriter.cc.

                        {
  delete m_file;
}

ROOTWriter() [2/2]

podio::ROOTWriter::ROOTWriter ( const ROOTWriter &  )

delete

Member Function Documentation

finish()

void podio::ROOTWriter::finish

(

)

Definition at line 99 of file ROOTWriter.cc.

                        {
  // now we want to safe the metadata. This includes info about the
  // collections
  const auto collIDTable = m_store->getCollectionIDTable();
  m_metadatatree->Branch("CollectionIDs", collIDTable);
 
  // collectionID, collection type, subset collection
  std::vector<root_utils::CollectionInfoT> collectionInfo;
  collectionInfo.reserve(m_collectionsToWrite.size());
  for (const auto& name : m_collectionsToWrite) {
    const auto collID = collIDTable->collectionID(name);
    const podio::CollectionBase* coll{nullptr};
    // No check necessary, only registered collections possible
    m_store->get(name, coll);
    const auto collType = coll->getTypeName();
    // const auto collType = "std::vector<" + coll->getDataTypeName() + ">";
    collectionInfo.emplace_back(collID, std::move(collType), coll->isSubsetCollection());
  }
 
  m_metadatatree->Branch("CollectionTypeInfo", &collectionInfo);
 
  podio::version::Version podioVersion = podio::version::build_version;
  m_metadatatree->Branch("PodioVersion", &podioVersion);
 
  m_metadatatree->Fill();
 
  m_colMDtree->Branch("colMD", "std::map<int,podio::GenericParameters>", m_store->getColMetaDataMap());
  m_colMDtree->Fill();
  m_runMDtree->Branch("runMD", "std::map<int,podio::GenericParameters>", m_store->getRunMetaDataMap());
  m_runMDtree->Fill();
 
  m_file->Write();
  m_file->Close();
}

Referenced by testWithIO(), and writeCollection().

operator=()

ROOTWriter & podio::ROOTWriter::operator= ( const ROOTWriter &  )

delete

registerForWrite()

bool podio::ROOTWriter::registerForWrite

(

const std::string &

name

)

Definition at line 134 of file ROOTWriter.cc.

                                                       {
  const podio::CollectionBase* tmp_coll(nullptr);
  if (!m_store->get(name, tmp_coll)) {
    std::cerr << "no such collection to write, throw exception." << std::endl;
    return false;
  }
 
  m_collectionsToWrite.push_back(name);
  return true;
}

Referenced by testWithIO(), and writeCollection().

writeEvent()

void podio::ROOTWriter::writeEvent

(

)

Definition at line 31 of file ROOTWriter.cc.

                            {
  std::vector<StoreCollection> collections;
  collections.reserve(m_collectionsToWrite.size());
  for (const auto& name : m_collectionsToWrite) {
    const podio::CollectionBase* coll;
    m_store->get(name, coll);
    collections.emplace_back(name, const_cast<podio::CollectionBase*>(coll));
    collections.back().second->prepareForWrite();
  }
 
  if (m_firstEvent) {
    createBranches(collections);
    m_firstEvent = false;
  } else {
    setBranches(collections);
  }
 
  m_datatree->Fill();
  m_evtMDtree->Fill();
}

Referenced by testWithIO(), and writeCollection().

---

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

• ROOTWriter.h
• ROOTWriter.cc