#include "CWorldShadowShader.hpp"
#include "hecl/Pipeline.hpp"
#include "Graphics/CGraphics.hpp"

namespace urde {

static boo::ObjToken<boo::IShaderPipeline> s_Pipeline;
static boo::ObjToken<boo::IShaderPipeline> s_ZPipeline;

void CWorldShadowShader::Initialize() {
  s_Pipeline = hecl::conv->convert(Shader_CWorldShadowShader{});
  s_ZPipeline = hecl::conv->convert(Shader_CWorldShadowShaderZ{});
}

void CWorldShadowShader::Shutdown() {
  s_Pipeline.reset();
  s_ZPipeline.reset();
}

CWorldShadowShader::CWorldShadowShader(u32 w, u32 h) : m_w(w), m_h(h) {
  CGraphics::CommitResources([&](boo::IGraphicsDataFactory::Context& ctx) {
    m_vbo = ctx.newDynamicBuffer(boo::BufferUse::Vertex, 16, 4);
    m_uniBuf = ctx.newDynamicBuffer(boo::BufferUse::Uniform, sizeof(Uniform), 1);
    boo::ObjToken<boo::IGraphicsBuffer> bufs[] = {m_uniBuf.get()};
    boo::PipelineStage stages[] = {boo::PipelineStage::Vertex};
    m_dataBind = ctx.newShaderDataBinding(s_Pipeline, m_vbo.get(), nullptr, nullptr, 1, bufs, stages, nullptr, nullptr,
                                          0, nullptr, nullptr, nullptr);
    m_zDataBind = ctx.newShaderDataBinding(s_ZPipeline, m_vbo.get(), nullptr, nullptr, 1, bufs, stages, nullptr,
                                           nullptr, 0, nullptr, nullptr, nullptr);
    m_tex = ctx.newRenderTexture(m_w, m_h, boo::TextureClampMode::ClampToWhite, 1, 0);
    return true;
  } BooTrace);
}

void CWorldShadowShader::bindRenderTarget() { CGraphics::g_BooMainCommandQueue->setRenderTarget(m_tex); }

void CWorldShadowShader::drawBase(float extent) {
  zeus::CVector3f verts[] = {
      {-extent, 0.f, extent}, {extent, 0.f, extent}, {-extent, 0.f, -extent}, {extent, 0.f, -extent}};
  m_vbo->load(verts, sizeof(zeus::CVector3f) * 4);

  m_uniform.m_matrix = CGraphics::GetPerspectiveProjectionMatrix(true) * CGraphics::g_GXModelView.toMatrix4f();
  m_uniform.m_color = zeus::skWhite;
  m_uniBuf->load(&m_uniform, sizeof(m_uniform));

  CGraphics::SetShaderDataBinding(m_zDataBind);
  CGraphics::DrawArray(0, 4);
}

void CWorldShadowShader::lightenShadow() {
  m_uniform.m_color = zeus::CColor(1.f, 0.25f);
  m_uniBuf->load(&m_uniform, sizeof(m_uniform));

  CGraphics::SetShaderDataBinding(m_dataBind);
  CGraphics::DrawArray(0, 4);
}

void CWorldShadowShader::blendPreviousShadow() {
  if (!m_prevQuad)
    m_prevQuad.emplace(EFilterType::Blend, m_tex.get());
  zeus::CRectangle rect(0.f, 1.f, 1.f, -1.f);
  m_prevQuad->draw({1.f, 0.85f}, 1.f, rect);
}

void CWorldShadowShader::resolveTexture() {
  boo::SWindowRect rect = {0, 0, int(m_w), int(m_h)};
  CGraphics::g_BooMainCommandQueue->resolveBindTexture(m_tex, rect, false, 0, true, false, true);
}

} // namespace urde