count_events.cc File Reference

#include "canvas/Persistency/Provenance/BranchType.h"
#include "canvas/Persistency/Provenance/rootNames.h"
#include "boost/program_options.hpp"
#include <algorithm>
#include <iostream>
#include <string>
#include <vector>
#include "TBranch.h"
#include "TError.h"
#include "TFile.h"
#include "TTree.h"

Go to the source code of this file.

Functions
int	main (int argc, char **argv)

Function Documentation

int main	(	int	argc,
		char **	argv
	)

Definition at line 93 of file count_events.cc.

{
  using stringvec = std::vector<std::string>;
  int result = 1;
  std::ostringstream descstr;
  descstr << argv[0] << "Usage: count_events [<options>] <filename>+\n";
  bpo::options_description desc(descstr.str());
  desc.add_options()("hr", "Human-readable output")(
    "help,h", "this help message.")("source,s",
                                    bpo::value<stringvec>()->composing(),
                                    "source data file (multiple OK).");
  bpo::options_description all_opts("All Options.");
  all_opts.add(desc);
  // Each non-option argument is interpreted as the name of a file to be
  // processed. Any number of filenames is allowed.
  bpo::positional_options_description pd;
  pd.add("source", -1);
  // The variables_map contains the actual program options.
  bpo::variables_map vm;
  try {
    bpo::store(bpo::command_line_parser(argc, argv)
                 .options(all_opts)
                 .positional(pd)
                 .run(),
               vm);
    bpo::notify(vm);
  }
  catch (bpo::error const& e) {
    std::cerr << "Exception from command line processing in " << argv[0] << ": "
              << e.what() << "\n";
    return 2;
  }
  if (vm.count("help")) {
    std::cerr << desc << std::endl;
    return 1;
  } else if (vm.count("hr")) {
    want_hr = true;
  }
  if (vm.count("source") == 0) {
    std::cerr << "Require at least one source file.\n";
    std::cerr << desc << "\n";
    return 1;
  }
  auto const& sources = vm["source"].as<stringvec>();
  auto const expected = sources.size();
  auto succeeded = std::count_if(sources.cbegin(),
                                 sources.cend(),
                                 std::bind(&count_events,
                                           std::placeholders::_1,
                                           std::ref(std::cout),
                                           std::ref(std::cerr)));
  if (expected == static_cast<size_t>(succeeded)) {
    std::cout << "Counted events successfully for " << expected
              << " specified files." << std::endl;
    result = 0;
  } else {
    result = 1;
    std::cout << "Failed to count events for " << expected - succeeded << " of "
              << expected << " specified files." << std::endl;
  }
  return result & 0xff;
}