|
@@ -46,10 +46,9 @@ using namespace fv::root;
|
|
|
template<typename TREE_CLASS>
|
|
|
class TreeDataSet : public DataSet{
|
|
|
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;
|
|
|
+ // Maps filenames to data category. Either "signal" or "background"
|
|
|
std::map<std::string,std::string> input_categories;
|
|
|
std::string output_filename;
|
|
|
std::vector<fv::util::DataFileDescriptor> dfds;
|
|
@@ -99,32 +98,6 @@ class TreeDataSet : public DataSet{
|
|
|
|
|
|
public:
|
|
|
|
|
|
- /* TreeDataSet(const std::string& output_filename, */
|
|
|
- /* const std::vector<std::string>& input_filenames, */
|
|
|
- /* const std::string& data_label, */
|
|
|
- /* const std::string& tree_name="tree") */
|
|
|
- /* :DataSet(), */
|
|
|
- /* input_labels({ {input_filename, data_label} }), */
|
|
|
- /* output_filename(output_filename), */
|
|
|
- /* next_entry(0) { */
|
|
|
- /* TChain* chain = new TChain(tree_name.c_str()); */
|
|
|
- /* chain->Add(input_filename.c_str()); */
|
|
|
- /* tree_obj = new TREE_CLASS(chain); */
|
|
|
- /* nentries = tree_obj->fChain->GetEntries(); */
|
|
|
- /* output_file = TFile::Open(output_filename.c_str(), "RECREATE"); */
|
|
|
- /* tree_obj->fChain->SetBranchStatus("*", false); */
|
|
|
- /* } */
|
|
|
-
|
|
|
- /* void print_branch_statuses(){ */
|
|
|
- /* TObjArray* obj_arr = tree_obj->fChain->GetListOfBranches(); */
|
|
|
- /* for(TIter iter=obj_arr->begin(); iter!= obj_arr->end(); ++iter){ */
|
|
|
- /* TBranch* br = (TBranch*)*iter; */
|
|
|
- /* std::cout << br->GetName() << ": " */
|
|
|
- /* << tree_obj->fChain->GetBranchStatus(br->GetName()) << std::endl; */
|
|
|
- /* } */
|
|
|
-
|
|
|
- /* } */
|
|
|
-
|
|
|
TreeDataSet(const std::string& output_filename, const std::vector<fv::util::DataFileDescriptor>& dfds,
|
|
|
const std::string& tree_name)
|
|
|
:DataSet(),
|