check_benchmark_outputs.cpp File Reference

#include "TFile.h"
#include "TTree.h"
#include <iostream>
#include <string>
#include <vector>

Go to the source code of this file.

Typedefs
using	StringVec = std::vector< std::string >

Functions
bool	verifyTree (TTree *tree, int expectedEntries, const StringVec &expectedBranches)
void	verifyBMFile (const char *fileName, const StringVec &setupBranches, const StringVec &eventBranches, int expectedEvents=nExpectedEvents)
int	main (int, char *argv[])

Variables
constexpr int	nExpectedEvents = 2000

Typedef Documentation

StringVec

using StringVec = std::vector<std::string>

Definition at line 9 of file check_benchmark_outputs.cpp.

Function Documentation

main()

int main	(	int	,
		char *	argv[]
	)

We can't really make any checks on the actual execution times, but we can at least verify that the expected timing points are here.

Definition at line 68 of file check_benchmark_outputs.cpp.

                            {
  const StringVec writeBMSetupBranches = {"constructor", "finish", "register_for_write"};
  const StringVec writeBMEventBranches = {"write_event"};
  verifyBMFile(argv[1], writeBMSetupBranches, writeBMEventBranches);
 
  const StringVec readBMSetupBranches = {"constructor", "open_file", "close_file", "get_entries",
                                         "read_collection_ids"};
  const StringVec readBMEventBranches = {"read_collections", "read_ev_md",   "read_run_md",
                                         "read_coll_md",     "end_of_event", "read_event"};
  verifyBMFile(argv[2], readBMSetupBranches, readBMEventBranches);
 
  return 0;
}

verifyBMFile()

void verifyBMFile	(	const char *	fileName,
		const StringVec &	setupBranches,
		const StringVec &	eventBranches,
		int	expectedEvents = nExpectedEvents
	)

Definition at line 43 of file check_benchmark_outputs.cpp.

                                                        {
  TFile* bmFile = TFile::Open(fileName);
  if (!bmFile) {
    std::cerr << "Benchmark file \'" << fileName << "\' does not exist!" << std::endl;
    std::exit(1);
  }
  if (!verifyTree(static_cast<TTree*>(bmFile->Get("setup_times")), 1, setupBranches)) {
    std::cerr << "In file \'" << fileName << "\' setup_times Tree does not have the expected entries" << std::endl;
    bmFile->Close();
    std::exit(1);
  }
  if (!verifyTree(static_cast<TTree*>(bmFile->Get("event_times")), expectedEvents, eventBranches)) {
    std::cerr << "In file \'" << fileName << "\' event_times Tree does not have the expected entries" << std::endl;
    bmFile->Close();
    std::exit(1);
  }
 
  bmFile->Close();
}

Referenced by main().

verifyTree()

bool verifyTree	(	TTree *	tree,
		int	expectedEntries,
		const StringVec &	expectedBranches
	)

Definition at line 11 of file check_benchmark_outputs.cpp.

                                                                                     {
  const std::string treeName = tree->GetName();
  if (tree->GetEntries() != expectedEntries) {
    std::cerr << "Tree \'" << treeName << "\' should have " << expectedEntries << " but has " << tree->GetEntries()
              << std::endl;
    return false;
  }
 
  const auto* branches = tree->GetListOfBranches();
  for (const auto& branch : expectedBranches) {
    bool found = false;
    for (int i = 0; i < branches->GetEntries(); ++i) {
      if (branch == branches->At(i)->GetName()) {
        found = true;
        break;
      }
    }
    if (!found) {
      std::cerr << "Branch \'" << branch << "\' was expected to be in Tree \'" << treeName
                << "\' but could not be found" << std::endl;
      return false;
    }
  }
 
  if ((unsigned)branches->GetEntries() != expectedBranches.size()) {
    std::cerr << "Tree \'" << treeName << "\' has additional, unexpected branches" << std::endl;
    return false;
  }
 
  return true;
}

Referenced by verifyBMFile().

Variable Documentation

nExpectedEvents

constexpr int nExpectedEvents = 2000

constexpr

Definition at line 8 of file check_benchmark_outputs.cpp.