ShaderDX.h

/*
 *  Copyright (C) 2017-2020 Team Kodi
 *  This file is part of Kodi - https://kodi.tv
 *
 *  SPDX-License-Identifier: GPL-2.0-or-later
 *  See LICENSES/README.md for more information.
 */

#pragma once

#include "ShaderTextureDX.h"
#include "cores/RetroPlayer/shaders/IShader.h"
#include "cores/VideoPlayer/VideoRenderers/VideoShaders/WinVideoFilter.h"
#include "guilib/D3DResource.h"

#include <stdint.h>

namespace KODI
{
namespace RETRO
{
class CRenderContext;
}

namespace SHADER
{

// TODO: make renderer independent
// libretro's "Common shaders"
// Spec here: https://github.com/libretro/common-shaders/blob/master/docs/README
class CShaderDX : public CWinShader, public IShader
{
public:
  CShaderDX(RETRO::CRenderContext& context);
  ~CShaderDX() override;

  // implementation of IShader
  bool Create(const std::string& shaderSource,
              const std::string& shaderPath,
              ShaderParameterMap shaderParameters,
              IShaderSampler* sampler,
              ShaderLutVec luts,
              float2 viewPortSize,
              unsigned frameCountMod = 0) override;
  void Render(IShaderTexture* source, IShaderTexture* target) override;
  void SetSizes(const float2& prevSize, const float2& nextSize) override;
  void PrepareParameters(CPoint dest[4], bool isLastPass, uint64_t frameCount) override;
  CD3DEffect& GetEffect();
  void UpdateMVP() override;
  bool CreateInputBuffer() override;
  void UpdateInputBuffer(uint64_t frameCount);

  // expose these from CWinShader
  bool CreateVertexBuffer(unsigned vertCount, unsigned vertSize) override;

  bool CreateInputLayout(D3D11_INPUT_ELEMENT_DESC* layout, unsigned numElements);

protected:
  void SetShaderParameters(CD3DTexture& sourceTexture);

private:
  struct cbInput
  {
    XMFLOAT2 video_size;
    XMFLOAT2 texture_size;
    XMFLOAT2 output_size;
    float frame_count;
    float frame_direction;
  };

  // Currently loaded shader's source code
  std::string m_shaderSource;

  // Currently loaded shader's relative path
  std::string m_shaderPath;

  // Array of shader parameters
  ShaderParameterMap m_shaderParameters;

  // Sampler state
  ID3D11SamplerState* m_pSampler = nullptr;

  // Look-up textures that the shader uses
  ShaderLutVec m_luts; // todo: back to DX maybe

  // Resolution of the input of the shader
  float2 m_inputSize;

  // Resolution of the output of the shader
  float2 m_outputSize;

  // Resolution of the viewport/window
  float2 m_viewportSize;

  // Resolution of the texture that holds the input
  // float2 m_textureSize;

  // Holds the data bount to the input cbuffer (cbInput here)
  ID3D11Buffer* m_pInputBuffer = nullptr;

  // Projection matrix
  XMFLOAT4X4 m_MVP;

  // Value to modulo (%) frame count with
  // Unused if 0
  unsigned m_frameCountMod = 0;

private:
  cbInput GetInputData(uint64_t frameCount = 0);

  // Construction parameters
  RETRO::CRenderContext& m_context;
};

} // namespace SHADER
} // namespace KODI