Function main

examples/shadertui/run.cpp:63–145

}

int main() {

  Context ctx = createContext();
  
  // static constexpr size_t kRows = 64;
  // static constexpr size_t kCols = 96;
  static constexpr size_t kRows = 64;
  static constexpr size_t kCols = 96;

  kDefLog.level = kError; // suppress screen logging
  LOG(kDefLog, kInfo, "Creating screen tensor");

  std::array<float, kRows * kCols> screenArr;
  // std::fill(begin(screenArr), end(screenArr), 0.0);
  auto gen = std::mt19937{std::random_device{}()};
  randint(screenArr, gen, 0, 1);
  Tensor screen = createTensor(ctx, {kRows, kCols}, kf32, screenArr.data());

  std::promise<void> promise;
  std::future<void> future = promise.get_future();

  std::string codeString;
  struct Params {
    float time;
    uint32_t screenWidth;
    uint32_t screenHeight;
  } params = {0.0, kCols, kRows};

  LOG(kDefLog, kInfo, "Loading shader code from shader.wgsl");

  loadKernelCode("shader.wgsl", codeString);
  KernelCode shader{codeString.c_str(), Shape{16, 16, 1}};
  Kernel renderKernel =
      createKernel(ctx, shader, Bindings{screen},
                   cdiv({kCols, kRows, 1}, shader.workgroupSize), params);

  LOG(kDefLog, kInfo, "Starting render loop");

  std::array<char, kRows *(kCols + 1)> raster;

  auto start = std::chrono::high_resolution_clock::now();
  std::chrono::duration<float> elapsed;
  size_t ticks = 0;
  printf("\033[2J\033[H");
  size_t framesPerLoad = 20;
  size_t frame = 0;
  while (true) {
    if (frame % framesPerLoad == 0) { 
      loadKernelCode("shader.wgsl", codeString);
      if (codeString != shader.data) {
        // TODO(avh): Use a better way to avoid write/read race conditions
        // and recover from partial write errors
        std::this_thread::sleep_for(std::chrono::milliseconds(20));
        loadKernelCode("shader.wgsl", codeString);
        shader = {codeString.c_str(), Shape{16, 16, 1}};
        renderKernel =
            createKernel(ctx, shader, Bindings{screen},
                         cdiv({kCols, kRows, 1}, shader.workgroupSize), params);