iguana_ex_cpp_config_files.cc

Go to the documentation of this file.

#include <cassert>
#include <iguana/algorithms/clas12/ZVertexFilter/Algorithm.h>


int main(int argc, char** argv)
{
 
  // parse arguments
  std::string configDir;
  if(argc > 1)
    configDir = std::string(argv[1]);
  else
    configDir = iguana::ConfigFileReader::GetConfigInstallationPrefix() + "/examples";
  fmt::print("Using top-level configuration directory {}\n", configDir);
 
  // loop over multiple examples how to use configuration files and set options
  for(int example = 1; example <= 6; example++) {
    fmt::print("\n" + iguana::Logger::Header(fmt::format("CONFIG EXAMPLE {}", example)) + "\n");
 
    // start the algorithm; it will be destroyed at the end of each `example` iteration
    auto algo = std::make_unique<iguana::clas12::ZVertexFilter>();
    algo->SetLogLevel("debug");
 
    switch(example) {
 
    case 1:
      {
        // Use the default configuration, from `../src/iguana/algorithms/clas12/ZVertexFilter.yaml`
        algo->Start();
        auto key = algo->PrepareEvent(4800); // sets the run number and loads the cuts
        assert((algo->GetRunNum(key) == 4800)); // pass the key into the 'Get*' methods
        fmt::println("Z-vertex cuts: {} to {}", algo->GetElectronZcuts(key).at(0), algo->GetElectronZcuts(key).at(1));
        break;
      }
 
    case 2:
      {
        // Use `SetElectronZcuts` to set the cuts
        // - note that this will OVERRIDE any value used in any configuration file
        // - only use `SetElectronZcuts` if for any reason you want to hard-code a specific value; usage
        //   of configuration files is preferred in general
        algo->Start();
        iguana::concurrent_key_t const key = 0; // need the same key in `SetElectronZcuts` and `GetElectronZcuts`
        algo->SetElectronZcuts(-5.0, 3.0, key);
        assert((algo->GetElectronZcuts(key).at(0) == -5.0));
        assert((algo->GetElectronZcuts(key).at(1) == 3.0));
        break;
      }
 
    case 3:
      {
        // Use a specific configuration file
        algo->SetConfigFile(configDir + "/my_z_vertex_cuts.yaml");
        algo->Start();
        auto key = algo->PrepareEvent(5500);
        assert((algo->GetElectronZcuts(key).at(0) == -0.8));
        assert((algo->GetElectronZcuts(key).at(1) == 0.7));
        break;
      }
 
    case 4:
      {
        // Use the same specific configuration file, but don't set a run number;
        // note also the usage of `SetConfigDirectory`, as another example how to set a specific configuration file
        algo->SetConfigDirectory(configDir);
        algo->SetConfigFile("my_z_vertex_cuts.yaml");
        algo->Start();
        auto key = algo->PrepareEvent(0); // run number "0" means "no run number"
        assert((algo->GetElectronZcuts(key).at(0) == -1.5));
        assert((algo->GetElectronZcuts(key).at(1) == 1.3));
        break;
      }
 
    case 5:
      {
        // Use a custom directory of configuration files; if a configuration file within
        // has the same path and name as the default (`ZVertexFilter.yaml`), it will be used instead of the default.
        // This is designed such that if you copy the full installed configuration directory to a new location, you
        // may use that directory instead of the default, and modify any configuration file within.
        algo->SetConfigDirectory(configDir + "/my_config_directory");
        algo->Start();
        auto key = algo->PrepareEvent(0); // run number "0" means "no run number"
        assert((algo->GetElectronZcuts(key).at(0) == -15.0));
        assert((algo->GetElectronZcuts(key).at(1) == 15.0));
        break;
      }
 
    case 6:
      {
        // Use a single, combined configuration file; each algorithm's options are in a separate section
        algo->SetConfigDirectory(configDir);
        algo->SetConfigFile("my_combined_config_file.yaml");
        algo->Start();
        auto key = algo->PrepareEvent(0); // run number "0" means "no run number"
        assert((algo->GetElectronZcuts(key).at(0) == -33.0));
        assert((algo->GetElectronZcuts(key).at(1) == 11.0));
        break;
      }
 
    default:
      {
        fmt::print(stderr, "ERROR: unknown example number '{}'\n", example);
        return 1;
      }
    }
 
    algo->Stop();
  }
 
  return 0;
}