HepMC3 event record library
ReaderRootTree.h
Go to the documentation of this file.
1// -*- C++ -*-
2//
3// This file is part of HepMC
4// Copyright (C) 2014-2019 The HepMC collaboration (see AUTHORS for details)
5//
6#ifndef HEPMC3_READERROOTTREE_H
7#define HEPMC3_READERROOTTREE_H
8/**
9 * @file ReaderRootTree.h
10 * @brief Definition of \b class ReaderRootTree
11 *
12 * @class HepMC3::ReaderRootTree
13 * @brief GenEvent I/O parsing and serialization for root files based on root TTree
14 *
15 * If HepMC was compiled with path to ROOT available, this class can be used
16 * for root file I/O in the same manner as with HepMC::ReaderAscii class.
17 *
18 * @ingroup IO
19 *
20 */
21#include "HepMC3/Reader.h"
22#include "HepMC3/GenEvent.h"
25
26// ROOT header files
27#include "TFile.h"
28#include "TTree.h"
29#include "TBranch.h"
30
31namespace HepMC3
32{
33
34class ReaderRootTree : public Reader
35{
36//
37// Constructors
38//
39public:
40 /** @brief Default constructor */
41 ReaderRootTree(const std::string &filename);
42 /** @brief Constructor with tree name*/
43 ReaderRootTree(const std::string &filename,const std::string &treename,const std::string &branchname);
44
45//
46// Functions
47//
48public:
49
50 /// @brief skip events
51 bool skip(const int) override;
52
53 /** @brief Read event from file
54 *
55 * @param[out] evt Contains parsed event
56 */
57 bool read_event(GenEvent &evt) override;
58
59 /** @brief Close file */
60 void close() override;
61
62 /** @brief Get file error state */
63 bool failed() override;
64
65private:
66 /** @brief init routine */
67 bool init();
68//
69// Fields
70//
71private:
72 TFile* m_file; //!< File handler
73public:
74 TTree* m_tree;//!< Tree handler. Public to allow simple access, e.g. custom branches.
75private:
76 int m_events_count; //!< Events count. Needed to read the tree
77 GenEventData* m_event_data; //!< Pointer to structure that holds event data
78 GenRunInfoData* m_run_info_data; //!< Pointer to structure that holds run info data
79 std::string m_tree_name; //!< Name of TTree
80 std::string m_branch_name; //!< Name of TBranch in TTree
81};
82
83} // namespace HepMC3
84
85#endif
Definition of struct GenEventData.
Definition of class GenEvent.
Definition of struct GenRunInfoData.
Definition of interface Reader.
Stores event-related information.
Definition: GenEvent.h:41
GenEvent I/O parsing and serialization for root files based on root TTree.
bool read_event(GenEvent &evt) override
Read event from file.
TTree * m_tree
Tree handler. Public to allow simple access, e.g. custom branches.
ReaderRootTree(const std::string &filename)
Default constructor.
bool failed() override
Get file error state.
bool skip(const int) override
skip events
GenEventData * m_event_data
Pointer to structure that holds event data.
int m_events_count
Events count. Needed to read the tree.
void close() override
Close file.
std::string m_tree_name
Name of TTree.
std::string m_branch_name
Name of TBranch in TTree.
TFile * m_file
File handler.
GenRunInfoData * m_run_info_data
Pointer to structure that holds run info data.
bool init()
init routine
Base class for all I/O readers.
Definition: Reader.h:25
HepMC3 main namespace.
Stores serializable event information.
Definition: GenEventData.h:26
Stores serializable run information.