Configuration.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/

 
#include <ColumnarTestFixtures/Configuration.h>
 
#include <nlohmann/json.hpp>
 
#include <cstdlib>
#include <fstream>
#include <stdexcept>
 
namespace columnar
{
  namespace TestUtils
  {
    void UserConfiguration ::
    loadFromFile (const std::string& filename)
    {
      std::ifstream file (filename);
      if (!file.is_open())
        throw std::runtime_error ("failed to open configuration file: " + filename);
 
      auto json = nlohmann::json::parse (file);
 
      if (json.contains ("targetTime"))
        targetTime = std::chrono::seconds (json["targetTime"].get<int>());
      if (json.contains ("batchSize"))
        batchSize = json["batchSize"].get<unsigned int>();
      if (json.contains ("runToolTwice"))
        runToolTwice = json["runToolTwice"].get<bool>();
      if (json.contains ("skipShallowCopies"))
        skipShallowCopies = json["skipShallowCopies"].get<bool>();
      if (json.contains ("measureNonAccessForEmpty"))
        measureNonAccessForEmpty = json["measureNonAccessForEmpty"].get<bool>();
    }
 
    UserConfiguration UserConfiguration :: fromEnvironment ()
    {
      UserConfiguration config;
      const char *envPath = std::getenv ("COLUMNAR_TEST_CONFIG");
      if (envPath != nullptr && envPath[0] != '\0')
        config.loadFromFile (envPath);
      return config;
    }
  }
}