MiniTreeDataSet.hpp
Go to the documentation of this file.
#ifndef minitreedataset_h
#define minitreedataset_h
#include <string>
#include <tuple>
#include "filval/filval.hpp"
#include "filval_root/filval_root.hpp"
#include "MiniTree.hpp"
using namespace std;
using namespace fv;
using namespace fv::root;
class MiniTreeDataSet : public DataSet,
public MiniTree{
private:
// Maps filenames to data category. Either "signal" or "background"
std::map<std::string,std::string> input_categories;
// Maps filenames to data label, eg. "TTTT", or "TTZ"
std::map<std::string,std::string> input_labels;
std::string output_filename;
TFile* output_file;
long next_entry;
long nentries;
bool load_next(){
if (next_entry >= nentries) return false;
fChain->GetEntry(next_entry);
++next_entry;
return true;
}
int get_events(){
return nentries;
}
int get_current_event(){
return next_entry-1;
}
void save_event_count_and_xsection(){
std::map<std::string,int> event_counts;
std::map<std::string,float> xsecs;
string fname, label;
for(auto& p : input_labels){
std::tie(fname, label) = p;
TFile f(fname.c_str());
TH1D* count = (TH1D*)f.Get("Count");
event_counts[label] = (int)count->GetBinContent(1);
TTree* tree = (TTree*)f.Get("tree");
TBranch* b = tree->GetBranch("xsec");
float xsec;
b->SetAddress(&xsec);
b->GetEntry(1);
xsecs[label] = xsec;
}
output_file->cd();
gDirectory->WriteObjectAny(&event_counts, "std::map<std::string,int>", "_event_counts");
gDirectory->WriteObjectAny(&xsecs, "std::map<std::string,float>", "_xsecs");
}
public:
MiniTreeDataSet(const std::string& output_filename, const std::string input_filename, const std::string data_label)
:DataSet(),
input_categories({ {input_filename, "signal"} }),
input_labels({ {input_filename, data_label} }),
output_filename(output_filename),
next_entry(0) {
TChain* chain = new TChain("tree");
chain->Add(input_filename.c_str());
Init(chain);
nentries = fChain->GetEntries();
output_file = TFile::Open(output_filename.c_str(), "RECREATE");
this->fChain->SetBranchStatus("*", false);
}
MiniTreeDataSet(const std::string& output_filename, const std::map<std::string,std::string>& filenames_with_labels)
:DataSet(),
input_categories(filenames_with_labels),
output_filename(output_filename),
next_entry(0) {
TChain* chain = new TChain("tree");
for(auto& p : filenames_with_labels){
std::string filename;
std::tie(filename, std::ignore) = p;
chain->Add(filename.c_str());
}
Init(chain);
nentries = fChain->GetEntries();
output_file = TFile::Open(output_filename.c_str(), "RECREATE");
this->fChain->SetBranchStatus("*", false);
}
~MiniTreeDataSet(){
save_event_count_and_xsection();
output_file->Close();
}
const std::string& get_current_event_label() const{
TFile* file = fChain->GetFile();
std::string filename = file->GetName();
return input_categories.at(filename);
}
template <typename T>
Value<T>* track_branch(const std::string& bname){
TBranch* branch = fChain->GetBranch(bname.c_str());
if (branch == nullptr){
CRITICAL("Branch: " << bname << " does not exist in input tree.", -1);
}
T* bref = (T*) branch->GetAddress();
fChain->SetBranchStatus(bname.c_str(), true);
INFO("Registering branch \"" << bname
<< "\" with address " << bref
<< " and type " << typeid(bref).name());
return new ObservedValue<T>(bname, bref);
}
template <typename T>
Value<T*>* track_branch_ptr(const std::string& bname){
TBranch* branch = fChain->GetBranch(bname.c_str());
if (branch == nullptr){
CRITICAL("Branch: " << bname << " does not exist in input tree.", -1);
}
T* bref = (T*) branch->GetAddress();
fChain->SetBranchStatus(bname.c_str(), true);
INFO("Registering pointer branch \"" << bname
<< "\" with address " << bref
<< " and type " << typeid(bref).name());
return new PointerValue<T>(bname, bref);
}
template <typename T>
decltype(auto) track_branch_vec(const std::string& size_bname, const std::string& value_bname){
track_branch_ptr<T>(value_bname);
return wrapper_vector<T>(lookup<int>(size_bname), lookup<T*>(value_bname), value_bname);
}
void save_all(){
output_file->cd();
// Save the value names for each container to enable looking up
// what values are plotted
std::map<string,string> value_lookup = this->get_container_name_value_map();
gDirectory->WriteObjectAny(&value_lookup, "std::map<std::string,std::string>", "_value_lookup");
std::map<string,string> fn_impl_lookup = this->get_function_name_impl_map();
gDirectory->WriteObjectAny(&fn_impl_lookup, "std::map<std::string,std::string>", "_function_impl_lookup");
for(auto container : containers){
container.second->save_as("outfile", SaveOption::ROOT);
}
}
};
#endif // minitreedataset_h
std
STL namespace.
fv::ObservedValue
A value supplied by the dataset, not derived.
Definition: value.hpp:450
fv
The namespace containing all filval classes and functions.
Definition: api.hpp:46
fv::Value
A templated value.
Definition: value.hpp:265
fv::root
Definition: api.hpp:8
MiniTree.hpp
fv::PointerValue
A Value of a pointer.
Definition: value.hpp:1213