Seele/src/Engine/Graphics/Metal/Shader.mm

#include "Shader.h"
#include "Foundation/NSError.hpp"
#include "Foundation/NSString.hpp"
#include "Graphics.h"
#include "Graphics/Enums.h"
#include "Graphics/slang-compile.h"
#include "Metal/MTLDevice.hpp"
#include "Metal/MTLLibrary.hpp"
#include <fstream>
#include <iostream>
#include <slang.h>
#include <spirv_cross.hpp>
#include <spirv_cross/spirv_msl.hpp>

using namespace Seele;
using namespace Seele::Metal;

Shader::Shader(PGraphics graphics, Gfx::SeShaderStageFlags stage) : stage(stage), graphics(graphics) {}

Shader::~Shader() {
  if (function) {
    function->release();
    library->release();
  }
}

void Shader::create(const ShaderCreateInfo& createInfo) {
  Slang::ComPtr<slang::IBlob> kernelBlob = generateShader(createInfo, SLANG_SPIRV);
  hash = CRC::Calculate(kernelBlob->getBufferPointer(), kernelBlob->getBufferSize(), CRC::CRC_32());
  spirv_cross::CompilerMSL comp((const uint32*)kernelBlob->getBufferPointer(), kernelBlob->getBufferSize() / 4);
  auto options = comp.get_msl_options();
  options.argument_buffers = true;
  options.argument_buffers_tier = spirv_cross::CompilerMSL::Options::ArgumentBuffersTier::Tier2;
  options.set_msl_version(3);
  comp.set_msl_options(options);
  std::string metalCode = comp.compile();
  NS::Error* error = nullptr;
  MTL::CompileOptions* mtlOptions = MTL::CompileOptions::alloc()->init();

  library = graphics->getDevice()->newLibrary(NS::String::string(metalCode.c_str(), NS::ASCIIStringEncoding),
                                              mtlOptions, &error);
  if (error) {
    std::cout << error->debugDescription() << std::endl;
    assert(false);
  }
  function = library->newFunction(NS::String::string("main0", NS::ASCIIStringEncoding));
  if(!function)
  {
    std::ofstream shaderFile("error.metal");
    shaderFile << metalCode;
    shaderFile.close();
    assert(!function);
  }
  mtlOptions->release();
}

uint32 Shader::getShaderHash() const { return hash; }