spv2h.cpp

#include <iostream>
#include <fstream>
#include <utility>

using namespace std;

int main(int argc, char **argv)
{
  auto loadSPIRV = [](const string &spirvPath) -> pair<uint8_t *, size_t> {
    ifstream file(spirvPath, ios::ate | ios::binary);
    
    if(!file.is_open())
      return { nullptr, 0 };
    
    size_t fileSize = (size_t)file.tellg();
    uint8_t *data = new uint8_t[fileSize];
    
    file.seekg(0);
    file.read((char *)data, fileSize);
    
    file.close();
    
    return { data, fileSize };
  };
  
  auto varName = [](string spirvPath) {
    string ret;
    
    for(auto c : spirvPath)
    {
      if(c == '.')
        ret += '_';
      else
        ret += c;
    }
    
    return ret;
  };
  
  if(argc != 3)
    return 1;
  
  auto [data, sz] = loadSPIRV(argv[1]);
  if(data)
  {
    ofstream out(argv[2]);
    
    if(!out.is_open())
      return 1;
    
    out << "/*spv2h for " << argv[1] << "*/" << endl << endl;
    out << "static const uint8_t " << varName(argv[1]) << "[] = {";
    
    for(size_t i = 0; i < sz; i++)
    {
      if(i % 40 == 0)
        out << endl;
      out << "\t" << hex << "0x" << (int)data[i] << ",";
    }
    out << endl;
    out << "};" << endl << endl;
    
    delete []data;
  }
  
  
  return 0;
}