module Render.DepthOnly.Pipeline
  ( Pipeline
  , allocate

  , Config
  , config

  , stageCode
  , stageSpirv
  ) where

import RIO

import Control.Monad.Trans.Resource (ResourceT)
import Data.Tagged (Tagged(..))
import Vulkan.Core10 qualified as Vk

import Geomancy (Transform)
import Engine.Vulkan.Pipeline.Graphics qualified as Graphics
import Engine.Vulkan.Types (HasVulkan, HasRenderPass(..), DsBindings)
import Render.Code (compileVert)
import Render.DepthOnly.Code qualified as Code
import Render.DescSets.Set0 (Scene, vertexPos, instanceTransform)

type Pipeline = Graphics.Pipeline '[Scene] () Transform
type Config = Graphics.Configure Pipeline
type instance Graphics.Specialization Pipeline = ()

allocate
  :: ( HasVulkan env
     , HasRenderPass renderpass
     )
  => Vk.SampleCountFlagBits
  -> Tagged Scene DsBindings
  -> renderpass
  -> ResourceT (RIO env) Pipeline
allocate :: forall env renderpass.
(HasVulkan env, HasRenderPass renderpass) =>
SampleCountFlagBits
-> Tagged Scene DsBindings
-> renderpass
-> ResourceT (RIO env) Pipeline
allocate SampleCountFlagBits
multisample Tagged Scene DsBindings
tset0 renderpass
rp = do
  (ReleaseKey
_, Pipeline
p) <- Maybe Extent2D
-> SampleCountFlagBits
-> Config '[Scene] () Transform ()
-> renderpass
-> ResourceT (RIO env) (ReleaseKey, Pipeline)
forall config pipeline (dsl :: [*]) vertices instances spec env
       (m :: * -> *) renderpass.
(config ~ Configure pipeline,
 pipeline ~ Pipeline dsl vertices instances,
 spec ~ Specialization pipeline, Specialization spec, HasCallStack,
 MonadVulkan env m, MonadResource m, HasRenderPass renderpass) =>
Maybe Extent2D
-> SampleCountFlagBits
-> Config dsl vertices instances spec
-> renderpass
-> m (ReleaseKey, pipeline)
Graphics.allocate
    Maybe Extent2D
forall a. Maybe a
Nothing
    SampleCountFlagBits
multisample
    (Tagged Scene DsBindings -> Configure Pipeline
config Tagged Scene DsBindings
tset0)
    renderpass
rp
  Pipeline -> ResourceT (RIO env) Pipeline
forall (f :: * -> *) a. Applicative f => a -> f a
pure Pipeline
p

config :: Tagged Scene DsBindings -> Config
config :: Tagged Scene DsBindings -> Configure Pipeline
config (Tagged DsBindings
set0) = Config '[] Any Any ()
forall vertices instances. Config '[] vertices instances ()
Graphics.baseConfig
  { $sel:cDescLayouts:Config :: Tagged '[Scene] [DsBindings]
Graphics.cDescLayouts  = forall {s :: [*]} {b}. b -> Tagged s b
forall {k} (s :: k) b. b -> Tagged s b
Tagged @'[Scene] [DsBindings
set0]
  , $sel:cStages:Config :: StageSpirv
Graphics.cStages       = StageSpirv
stageSpirv
  , $sel:cVertexInput:Config :: SomeStruct PipelineVertexInputStateCreateInfo
Graphics.cVertexInput  = SomeStruct PipelineVertexInputStateCreateInfo
vertexInput
  }
  where
    vertexInput :: SomeStruct PipelineVertexInputStateCreateInfo
vertexInput = [(VertexInputRate, [Format])]
-> SomeStruct PipelineVertexInputStateCreateInfo
Graphics.vertexInput
      [ (VertexInputRate, [Format])
vertexPos
      , (VertexInputRate, [Format])
instanceTransform
      ]

stageCode  :: Graphics.StageCode
stageCode :: StageCode
stageCode = ("vert" ::: Code) -> StageCode
forall a. ("vert" ::: a) -> Stages (Maybe ("vert" ::: a))
Graphics.vertexOnly "vert" ::: Code
Code.vert

stageSpirv :: Graphics.StageSpirv
stageSpirv :: StageSpirv
stageSpirv = ("vert" ::: ByteString) -> StageSpirv
forall a. ("vert" ::: a) -> Stages (Maybe ("vert" ::: a))
Graphics.vertexOnly "vert" ::: ByteString
vertSpirv

vertSpirv :: ByteString
vertSpirv :: "vert" ::: ByteString
vertSpirv = $(compileVert Code.vert)