#include "Contents/Ground/GroundGeometry.h"
#include "Contents/Models/ModelActor.h"
#include "Client/Mode/CGameMode.h"
#include "Contents/Actors/Pc.h"
#include "Contents/World/World.h"
#include "Contents/Camera/Camera.h"
#include "Contents/World/WorldRayQuery.h"
#include "Contents/World/WorldSpatialTree.h"
#include "Render/Renderer.h"

#include <cstring>

// OFFSET: 0x00CD5AD0
// FUNCTION: fn_21286
void CGameMode::MarkWorldOcclusion() {
  if (!this->scene || !this->scene->player || !g_renderer) {
    return;
  }

  const int screenX = this->scene->player->render.screenX;
  const int screenY = static_cast<int>(static_cast<float>(this->scene->player->render.screenY) - g_renderer->viewport.uiScaleY * 23.0f);
  WorldRayQuery query;
  float ray[6]{};
  g_renderer->viewport.Ray(screenX, screenY, this->camera->frame.eye, this->camera->frame.inverse.values, ray);

  float actorRay[7];
  std::memcpy(actorRay, ray, sizeof(ray));
  actorRay[6] = (this->scene->player->render.position.y + this->camera->camera.distance) - 16.0f;
  query.Collect(ray, &this->scene->spatial);

  for (WorldSpatialTree* node : query.nodes) {
    for (C3dActor* model : node->objects) {
      model->MarkOccluded(actorRay);
    }
  }

  if (this->checkGroundAlpha) {
    for (WorldSpatialTree* node : query.nodes) {
      int row = node->groundCells[1];

      while (row <= node->groundCells[3]) {
        int column = node->groundCells[0];

        while (column <= node->groundCells[2]) {
          node->ground->MarkRay(ray, column, row, actorRay[6]);
          ++column;
        }

        ++row;
      }
    }
  }
}