TestValidator.h

#pragma once
// test an iguana algorithm validator
 
#include <filesystem>
#include <hipo4/reader.h>
#include <iguana/algorithms/Validator.h>
 
inline int TestValidator(
    std::string vdor_name,
    std::vector<std::string> bank_names,
    std::string data_file,
    int data_tag,
    int num_events,
    std::string output_dir,
    std::string log_level)
{
 
  // check arguments
  if(vdor_name == "" || bank_names.empty()) {
    fmt::print(stderr, "ERROR: need validator name and banks\n");
    return 1;
  }
  if(data_file == "") {
    fmt::print(stderr, "ERROR: need a data file for command 'validator'\n");
    return 1;
  }
 
  // open the HIPO file
  hipo::reader reader(data_file.c_str(), {data_tag});
  auto banks = reader.getBanks(bank_names);
 
  // make the output directory
  if(output_dir != "")
    std::filesystem::create_directories(output_dir);
 
  // define the validator
  auto vdor = iguana::AlgorithmFactory::Create(vdor_name);
  dynamic_cast<iguana::Validator*>(vdor.get())->SetOutputDirectory(output_dir);
  vdor->SetLogLevel(log_level);
 
  // event loop
  vdor->Start(banks);
  int it_ev = 0;
  while(reader.next(banks) && (num_events == 0 || it_ev++ < num_events)) {
    vdor->Run(banks);
  }
  vdor->Stop();
  return 0;
}