ttree2hdf5.cxx File Reference

#include "copyRootTree.h"
#include "getTree.h"
#include "treeCopyOpts.h"
#include "H5Cpp.h"
#include "TFile.h"
#include "TChain.h"
#include <iostream>
#include <memory>

Include dependency graph for ttree2hdf5.cxx:

Go to the source code of this file.

Functions
int	run (int argc, char *argv[])
int	main (int argc, char *argv[])

Function Documentation

main()

int main	(	int	argc,
		char *	argv[]
	)

Definition at line 19 of file ttree2hdf5.cxx.

                                 {
  try {
    return run(argc, argv);
  } catch (std::logic_error& e) {
    std::cerr << "ERROR: " << e.what() << ", quitting." << std::endl;
    return 1;
  }
}

run()

int run	(	int	argc,
		char *	argv[]
	)

Definition at line 28 of file ttree2hdf5.cxx.

                                {
  using namespace H5Utils;
  AppOpts opts = getTreeCopyOpts(argc, argv);
 
  if (opts.exit_code != 0) return opts.exit_code;
 
  // Read in the root tree. We pick whatever tree is on the top level
  // of the file. If there are two we throw an error.
  std::string tree_name = opts.file.tree;
  if (tree_name.size() == 0) tree_name = getTree(opts.file.in.at(0));
  if (opts.tree.verbose) std::cout << "tree: " << tree_name << std::endl;
  std::unique_ptr<TChain> chain(new TChain(tree_name.c_str()));
  for (const auto& file_name: opts.file.in) {
    if (opts.tree.verbose) std::cout << "adding " << file_name << std::endl;
    int ret_code = chain->Add(file_name.c_str(), -1);
    if (ret_code != 1) {
      std::cerr << "Tree '" << tree_name << "' is missing from "
                << file_name << std::endl;
      return 1;
    }
  }
 
  // make the output file
  H5::H5File out_file(opts.file.out, H5F_ACC_TRUNC);
 
  // All the magic appens here
  copyRootTree(*chain, out_file, opts.tree);
 
  return 0;
}