package pipeline

import (
	"log"

	"zworld/engine/renderapi/descriptor"
	"zworld/engine/renderapi/device"
	"zworld/engine/util"

	"github.com/vkngwrapper/core/v2/core1_0"
)

type Layout interface {
	device.Resource[core1_0.PipelineLayout]
}

type layout struct {
	ptr    core1_0.PipelineLayout
	device device.T
}

func NewLayout(device device.T, descriptors []descriptor.SetLayout, constants []PushConstant) Layout {
	offset := 0
	info := core1_0.PipelineLayoutCreateInfo{

		SetLayouts: util.Map(descriptors, func(desc descriptor.SetLayout) core1_0.DescriptorSetLayout {
			return desc.Ptr()
		}),

		PushConstantRanges: util.Map(constants, func(push PushConstant) core1_0.PushConstantRange {
			size := push.Size()
			log.Printf("push: %d bytes", size)
			rng := core1_0.PushConstantRange{
				StageFlags: core1_0.ShaderStageFlags(push.Stages),
				Offset:     offset,
				Size:       size,
			}
			offset += size
			return rng
		}),
	}

	ptr, _, err := device.Ptr().CreatePipelineLayout(nil, info)
	if err != nil {
		panic(err)
	}

	return &layout{
		ptr:    ptr,
		device: device,
	}
}

func (l *layout) Ptr() core1_0.PipelineLayout {
	return l.ptr
}

func (l *layout) Destroy() {
	if l.ptr != nil {
		l.ptr.Destroy(nil)
		l.ptr = nil
	}
}