Add compute shader support

CMakeLists.txt

@@ -51,3 +51,4 @@ if (MSVC)
 else()
     target_compile_options(TheChef PRIVATE -Wall -Wextra -Wpedantic)
 endif()
+

ShaderCompiler.h

@@ -58,7 +58,7 @@ public:
     static bool IsEntryShader(const fs::path& p)
     {
         const auto ext = ToLower(p.extension().string());
-        return ext == ".vert" || ext == ".frag";
+        return ext == ".vert" || ext == ".frag" || ext == ".comp";
     }
 
     static bool IsIncludeOnlyShader(const fs::path& p)
@@ -114,6 +114,7 @@ private:
         const auto ext = ToLower(p.extension().string());
         if (ext == ".vert") return EShLangVertex;
         if (ext == ".frag") return EShLangFragment;
+        if (ext == ".comp") return EShLangCompute;
         throw std::runtime_error("Unknown entry shader stage extension: " + p.string());
     }
 

main.cpp

@@ -1,6 +1,3 @@
-// main.cpp (updated)
-
-// Standard
 #include <filesystem>
 #include <fstream>
 #include <string>
@@ -8,16 +5,12 @@
 #include <cctype>
 #include <chrono>
 #include <stdexcept>
-#include <cstdlib>
 
-// Manifest
 #include <nlohmann/json.hpp>
 
-// Logging
 #include <spdlog/spdlog.h>
 #include <spdlog/sinks/stdout_color_sinks.h>
 
-// Your shader compiler header
 #include "ShaderCompiler.h"
 
 namespace fs = std::filesystem;
@@ -122,6 +115,7 @@ int main(int argc, char** argv) {
         if (args.clean && fs::exists(args.output)) {
             spdlog::info("Cleaning output dir: {}", args.output.string());
             fs::remove_all(args.output);
+            return 0;
         }
         fs::create_directories(args.output);
 
@@ -130,6 +124,8 @@ int main(int argc, char** argv) {
         // ----------------------------
         SpirvShaderCompiler::Options scOpt;
         scOpt.debugInfo = false; // set true if you want debug info in SPIR-V
+        scOpt.defines = { "DEBUG=1" };
+
 
         // Common conventions (adjust to your folder layout):
         // - shaders live under <input>/shaders