#include "concurrent/job_api.hpp"

#include <cassert>
#include <iostream>

static constexpr int k_DataSize = 100000;

static int s_ExampleData[k_DataSize] = {};

// Print the first 20 items for brevity
static void printFirst20Items()
{
  for (int i = 0; i < 20; ++i)
  {
    std::printf("data[%i] = %i\n", i, s_ExampleData[i]);
  }
}

int main()
{
   Job::Initialize(Job::MemRequirementsForConfig({}));

  // Initialize Dummy Data
  for (int i = 0; i < k_DataSize; ++i)
  {
    s_ExampleData[i] = i;
  }

  std::printf("Before:\n");
  printFirst20Items();

  auto* t = Job::ParallelFor(
   s_ExampleData, 
   k_DataSize, 
   Job::CountSplitter{6}, 
   [](int* data, std::size_t data_size) {
     for (std::size_t i = 0; i < data_size; ++i)
     {
       data[i] = data[i] * 5;
     }
   });

  Job::TaskSubmit(t);

  Job::WaitOnTask(t);

  std::printf("After:\n");
  printFirst20Items();

  // Check that the jobs finished working on all items.
  for (int i = 0; i < k_DataSize; ++i)
  {
    assert(s_ExampleData[i] == i * 5);
  }

  Job::Shutdown();

  return 0;
}