Function runBenchmark

apps/oidnBenchmark.cpp:104–238

Runs a benchmark and returns the total runtime

// Runs a benchmark and returns the total runtime
double runBenchmark(DeviceRef& device, const Benchmark& bench)
{
  std::cout << bench.name << " ..." << std::flush;

  // Initialize the filter and the buffers
  FilterRef filter = device.newFilter(bench.filter.c_str());
  Random rng;

  std::shared_ptr<ImageBuffer> input;

  std::shared_ptr<ImageBuffer> albedo;
  if (bench.hasInput("alb") || bench.hasInput("calb"))
  {
    input = albedo = newImage(device, bench.width, bench.height);
    initImage(*albedo, rng, 0.f, 1.f);
    filter.setImage("albedo", albedo->getBuffer(), albedo->getFormat(), bench.width, bench.height);
  }

  std::shared_ptr<ImageBuffer> normal;
  if (bench.hasInput("nrm") || bench.hasInput("cnrm"))
  {
    input = normal = newImage(device, bench.width, bench.height);
    initImage(*normal, rng, -1.f, 1.f);
    filter.setImage("normal", normal->getBuffer(), normal->getFormat(), bench.width, bench.height);
  }

  std::shared_ptr<ImageBuffer> color;
  if (bench.hasInput("hdr"))
  {
    input = color = newImage(device, bench.width, bench.height);
    initImage(*color, rng, 0.f, 100.f);
    filter.setImage("color", color->getBuffer(), color->getFormat(), bench.width, bench.height);
    if (bench.filter != "RTLightmap")
      filter.set("hdr", true);
  }
  else if (bench.hasInput("ldr"))
  {
    input = color = newImage(device, bench.width, bench.height);
    initImage(*color, rng, 0.f, 1.f);
    filter.setImage("color", color->getBuffer(), color->getFormat(), bench.width, bench.height);
    filter.set("hdr", false);
  }

  if (bench.hasInput("calb") || bench.hasInput("cnrm"))
    filter.set("cleanAux", true);

  std::shared_ptr<ImageBuffer> output;
  if (inplace)
    output = input;
  else
    output = newImage(device, bench.width, bench.height);
  filter.setImage("output", output->getBuffer(), output->getFormat(), bench.width, bench.height);

  if (quality != Quality::Default)
    filter.set("quality", quality);

  if (maxMemoryMB >= 0)
    filter.set("maxMemoryMB", maxMemoryMB);