MeshLayout.h

Go to the documentation of this file.

#pragma once
 
#include <SDL3/SDL_gpu.h>
#include "math_types.h"
#include <vector>
 
inline bool operator==(const SDL_GPUVertexBufferDescription &lhs,
                       const SDL_GPUVertexBufferDescription &rhs) {
#define _c(field) lhs.field == rhs.field
  return _c(slot) && _c(pitch) && _c(input_rate) && _c(instance_step_rate);
#undef _c
}
 
inline bool operator==(const SDL_GPUVertexAttribute &lhs,
                       const SDL_GPUVertexAttribute &rhs) {
#define _c(field) lhs.field == rhs.field
  return _c(location) && _c(buffer_slot) && _c(format) && _c(offset);
#undef _c
}
 
namespace candlewick {
 
constexpr Uint64 vertexElementSize(SDL_GPUVertexElementFormat format) {
  switch (format) {
  case SDL_GPU_VERTEXELEMENTFORMAT_INVALID:
    return 0u;
  case SDL_GPU_VERTEXELEMENTFORMAT_INT:
    return sizeof(Sint32);
  case SDL_GPU_VERTEXELEMENTFORMAT_INT2:
    return sizeof(Sint32[2]);
  case SDL_GPU_VERTEXELEMENTFORMAT_INT3:
    return sizeof(Sint32[3]);
  case SDL_GPU_VERTEXELEMENTFORMAT_INT4:
    return sizeof(Sint32[4]);
  case SDL_GPU_VERTEXELEMENTFORMAT_UINT:
    return sizeof(Uint32);
  case SDL_GPU_VERTEXELEMENTFORMAT_UINT2:
    return sizeof(Uint32[2]);
  case SDL_GPU_VERTEXELEMENTFORMAT_UINT3:
    return sizeof(Uint32[3]);
  case SDL_GPU_VERTEXELEMENTFORMAT_UINT4:
    return sizeof(Uint32[4]);
  case SDL_GPU_VERTEXELEMENTFORMAT_FLOAT:
    return sizeof(float);
  case SDL_GPU_VERTEXELEMENTFORMAT_FLOAT2:
    return sizeof(float[2]);
  case SDL_GPU_VERTEXELEMENTFORMAT_FLOAT3:
    return sizeof(float[3]);
  case SDL_GPU_VERTEXELEMENTFORMAT_FLOAT4:
    return sizeof(float[4]);
  case SDL_GPU_VERTEXELEMENTFORMAT_BYTE2:
  case SDL_GPU_VERTEXELEMENTFORMAT_UBYTE2:
  case SDL_GPU_VERTEXELEMENTFORMAT_BYTE2_NORM:
  case SDL_GPU_VERTEXELEMENTFORMAT_UBYTE2_NORM:
    return 16u;
  case SDL_GPU_VERTEXELEMENTFORMAT_BYTE4:
  case SDL_GPU_VERTEXELEMENTFORMAT_UBYTE4:
  case SDL_GPU_VERTEXELEMENTFORMAT_BYTE4_NORM:
  case SDL_GPU_VERTEXELEMENTFORMAT_UBYTE4_NORM:
    return 32u;
  case SDL_GPU_VERTEXELEMENTFORMAT_SHORT2:
  case SDL_GPU_VERTEXELEMENTFORMAT_USHORT2:
  case SDL_GPU_VERTEXELEMENTFORMAT_SHORT2_NORM:
  case SDL_GPU_VERTEXELEMENTFORMAT_USHORT2_NORM:
  case SDL_GPU_VERTEXELEMENTFORMAT_HALF2:
    return 32ul;
  case SDL_GPU_VERTEXELEMENTFORMAT_SHORT4:
  case SDL_GPU_VERTEXELEMENTFORMAT_USHORT4:
  case SDL_GPU_VERTEXELEMENTFORMAT_SHORT4_NORM:
  case SDL_GPU_VERTEXELEMENTFORMAT_USHORT4_NORM:
  case SDL_GPU_VERTEXELEMENTFORMAT_HALF4:
    return 64ul;
  }
}

enum class VertexAttrib : Uint16 {
  Position,
  Normal,
  Tangent,
  Bitangent,
  Color0,
  Color1,
  TexCoord0,
  TexCoord1,
};

class MeshLayout {
public:
  MeshLayout() : m_bufferDescs{}, m_attrs{}, m_totalVertexSize(0) {}

  MeshLayout &addBinding(Uint32 slot, Uint32 pitch) {
    m_bufferDescs.emplace_back(slot, pitch, SDL_GPU_VERTEXINPUTRATE_VERTEX, 0u);
    return *this;
  }

  constexpr MeshLayout &addAttribute(VertexAttrib loc, Uint32 binding,
                                     SDL_GPUVertexElementFormat format,
                                     Uint32 offset) {
    const Uint16 _loc = static_cast<Uint16>(loc);
    m_attrs.emplace_back(_loc, binding, format, offset);
    const Uint32 attrSize =
        math::roundUpTo16(Uint32(vertexElementSize(format)));
    m_totalVertexSize = std::max(m_totalVertexSize, offset + attrSize);
    return *this;
  }
 
  const SDL_GPUVertexAttribute *getAttribute(VertexAttrib loc) const noexcept {
    for (Uint32 i = 0; i < numAttributes(); i++) {
      if (m_attrs[i].location == static_cast<Uint16>(loc)) {
        return &m_attrs[i];
      }
    }
    return nullptr;
  }

  SDL_GPUVertexInputState toVertexInputState() const noexcept {
    return {
        m_bufferDescs.data(),
        numBuffers(),
        m_attrs.data(),
        numAttributes(),
    };
  }
 
  operator SDL_GPUVertexInputState() const noexcept {
    return toVertexInputState();
  }
 
  bool operator==(const MeshLayout &other) const noexcept {
    if (numBuffers() != other.numBuffers() &&
        numAttributes() != other.numAttributes())
      return false;
 
    // for (Uint16 i = 0; i < numBuffers(); i++) {
    //   if (m_bufferDescs[i] != other.m_bufferDescs[i])
    //     return false;
    // }
    if (m_bufferDescs != other.m_bufferDescs)
      return false;
    // for (Uint16 i = 0; i < numAttributes(); i++) {
    //   if (m_attrs[i] != other.m_attrs[i])
    //     return false;
    // }
    if (m_attrs != other.m_attrs)
      return false;
    return true;
  }

  Uint32 numBuffers() const { return Uint32(m_bufferDescs.size()); }
  Uint32 numAttributes() const { return Uint32(m_attrs.size()); }
  Uint32 vertexSize() const { return m_totalVertexSize; }
  Uint32 indexSize() const { return sizeof(Uint32); }
 
  std::vector<SDL_GPUVertexBufferDescription> m_bufferDescs;
  std::vector<SDL_GPUVertexAttribute> m_attrs;
 
private:
  Uint32 m_totalVertexSize;
};

inline bool validateMeshLayout(const MeshLayout &layout) {
  return (layout.numBuffers() > 0) && (layout.numAttributes() > 0);
}

template <typename V>
concept IsVertexType = std::is_standard_layout_v<V> && (alignof(V) == 16);
 
template <IsVertexType V> struct VertexTraits;

template <IsVertexType V> MeshLayout meshLayoutFor() {
  return VertexTraits<V>::layout();
}
 
} // namespace candlewick