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 
31 namespace HepMC3
32 {
33 
34 class ReaderRootTree : public Reader
35 {
36 //
37 // Constructors
38 //
39 public:
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 //
48 public:
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 
65 private:
66  /** @brief init routine */
67  bool init();
68 //
69 // Fields
70 //
71 private:
72  TFile* m_file; //!< File handler
73 public:
74  TTree* m_tree;//!< Tree handler. Public to allow simple access, e.g. custom branches.
75 private:
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 interface Reader.
GenRunInfoData * m_run_info_data
Pointer to structure that holds run info data.
Definition of struct GenEventData.
TTree * m_tree
Tree handler. Public to allow simple access, e.g. custom branches.
TFile * m_file
File handler.
bool skip(const int) override
skip events
bool init()
init routine
Stores event-related information.
Definition: GenEvent.h:41
Stores serializable event information.
Definition: GenEventData.h:26
Stores serializable run information.
int m_events_count
Events count. Needed to read the tree.
void close() override
Close file.
std::string m_tree_name
Name of TTree.
GenEvent I/O parsing and serialization for root files based on root TTree.
bool read_event(GenEvent &evt) override
Read event from file.
bool failed() override
Get file error state.
Definition of class GenEvent.
GenEventData * m_event_data
Pointer to structure that holds event data.
std::string m_branch_name
Name of TBranch in TTree.
Base class for all I/O readers.
Definition: Reader.h:25
ReaderRootTree(const std::string &filename)
Default constructor.
Definition of struct GenRunInfoData.