#include "Render/D3DX9Runtime.h"
#include "Render/DeviceOwnerDX9.h"
#include "Render/RendererDX9Operations.h"
#include "Render/UIRPData.h"

#include <bit>
#include <cstdint>

namespace {
  class ScreenshotOverlay : public UIRPData {
  public:
    const std::list<RenderPrimitive*>& Primitives() const {
      return this->m_primitives;
    }
  };
}

// OFFSET: 0x005680A0
// FUNCTION: fn_02236
bool RendererDX9Operations::SaveScreenshot(
  DeviceOwnerDX9& owner,
  CRenderSystemBase* renderSystem,
  const char* filename,
  int format,
  CTexture* overlay,
  const tagRECT* rectangle
) {
  if (overlay) {
    unsigned x = rectangle ? static_cast<unsigned>(rectangle->right) : owner.backBufferDescription.Width;
    unsigned y = rectangle ? static_cast<unsigned>(rectangle->bottom) : owner.backBufferDescription.Height;
    x -= overlay->Width();
    y -= overlay->Height();
    ScreenshotOverlay drawing;
    drawing.Draw(
      *this->context.primitives, std::bit_cast<std::int32_t>(x), std::bit_cast<std::int32_t>(y), overlay, 0xFFFFFFFFu,
      0, 0, false
    );

    if (owner.Begin(renderSystem)) {
      ScopedDeviceStateDX9 restore(this->state);
      this->state.SetRenderState(D3DRS_ZENABLE, 0);
      this->state.SetRenderState(D3DRS_FOGENABLE, 0);
      this->state.SetRenderState(D3DRS_ALPHAREF, 1);
      this->state.SetRenderState(D3DRS_ALPHATESTENABLE, 1);
      this->state.SetRenderState(D3DRS_ALPHABLENDENABLE, 1);

      for (RenderPrimitive* command : drawing.Primitives()) {
        this->state.SetTexture(
          0, command->texture ? static_cast<CTextureDX9*>(command->texture)->NativeTexture() : nullptr
        );
        this->DrawPrimitive(command);
      }

      owner.End();
    }
  }

  return SUCCEEDED(this->context.d3dx->saveSurfaceToFileA(filename, format, owner.backBuffer, nullptr, rectangle));
}