-
Notifications
You must be signed in to change notification settings - Fork 0
/
TupleReader.h
51 lines (41 loc) · 1.45 KB
/
TupleReader.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
// This class accepts a vector of strings containing variable names
// for integer columns in the ROOT files.
// It allocates appropriate memories using SetBranchAddress and
// retains a map from the name to memory location.
// It also provides a next_record() method that calls GetEntry(i),
// which loads the specified variables from the ROOT file into the
// memory allocated previously.
// Lastly, it provides a get function, which accepts the name of
// the variables and returns a pointer to the memory, to be passed
// to the inserter to the database.
#ifndef TUPLE_READER_H
#define TUPLE_READER_H
#include <vector>
#include <string>
#include <map>
//ROOT Headers
#include <TDirectory.h>
#include <TFile.h>
#include <TTree.h>
class TupleReader {
public:
TupleReader(std::vector<std::string> var_types,
std::map<std::string, std::vector<std::string>> var_names,
std::string root_filename,
std::string root_treename);
~TupleReader();
bool next_record();
int GetVarInt(std::string var_name) const;
float GetVarFloat(std::string var_name) const;
private:
TFile *root_file_;
TTree *root_tree_;
std::vector<std::string> var_types_;
std::map<std::string, std::vector<std::string>> var_names_;
std::map<std::string, int> var_values_int_;
std::map<std::string, float> var_values_float_;
int current_event_idx_ = 0;
int num_events_ = 0;
void SetAddresses();
};
#endif