// This file is part of the 64k demo project.
// It implements the asset packer tool for demoscene resource management.
// Converts external files into embedded C++ byte arrays and look-up records.

#include <algorithm> // For std::count
#include <cmath>
#include <cstdio>     // for simplicity, use fprintf() for output generation
#include <cstring>    // For std::memcpy
#include <filesystem> // For path normalization
#include <fstream>
#include <map>
#include <regex>     // For std::regex
#include <stdexcept> // For std::stof exceptions
#include <string>
#include <vector>

#define STB_IMAGE_IMPLEMENTATION
#define STBI_NO_LINEAR // Don't apply gamma correction, we want raw bytes
#define STBI_ONLY_PNG
#define STBI_ONLY_JPEG
#define STBI_ONLY_TGA
#define STBI_ONLY_BMP
#include "stb_image.h"

#include "procedural/generator.h" // For ProcGenFunc and procedural functions
#include "util/ans.h"             // ANS compression for WGSL assets
#include "util/asset_manager.h"   // For AssetRecord and AssetId

// Map of procedural function names to their pointers (used only internally by
// asset_packer here, not generated)
static const std::map<std::string, ProcGenFunc> kAssetPackerProcGenFuncMap = {
    {"gen_noise", procedural::gen_noise},
    {"gen_perlin", procedural::gen_perlin},
    {"gen_grid", procedural::gen_grid},
    {"make_periodic", procedural::make_periodic},
    {"gen_plasma", procedural::gen_plasma},
    {"gen_voronoi", procedural::gen_voronoi},
    {"gen_normalmap", procedural::gen_normalmap},
#if !defined(DEMO_STRIP_ALL)
    {"gen_fail", procedural::gen_fail},
#endif
};

// Forward declaration
struct AssetBuildInfo;

static bool ParseProceduralParams(const std::string& params_str,
                                  std::vector<float>* out_params,
                                  const std::string& asset_name) {
  size_t current_pos = 0;
  while (current_pos < params_str.length()) {
    size_t comma_pos = params_str.find(',', current_pos);
    std::string param_val_str =
        (comma_pos == std::string::npos)
            ? params_str.substr(current_pos)
            : params_str.substr(current_pos, comma_pos - current_pos);
    param_val_str.erase(0, param_val_str.find_first_not_of(" \t\r\n"));
    param_val_str.erase(param_val_str.find_last_not_of(" \t\r\n") + 1);
    try {
      out_params->push_back(std::stof(param_val_str));
    } catch (...) {
      fprintf(stderr, "Error: Invalid proc param for %s: %s\n",
              asset_name.c_str(), param_val_str.c_str());
      return false;
    }
    if (comma_pos == std::string::npos)
      break;
    current_pos = comma_pos + 1;
  }
  return true;
}

// Helper struct to hold all information about an asset during parsing
struct AssetBuildInfo {
  std::string name;
  std::string filename;           // Original filename for static assets
  std::string asset_type;         // "STATIC", "PROC", "PROC_GPU", "MP3"
  std::string proc_func_name;     // Function name string
  std::vector<float> proc_params; // Parameters for procedural function

  // For generated C++ code
  std::string data_array_name;    // ASSET_DATA_xxx for static
  std::string params_array_name;  // ASSET_PROC_PARAMS_xxx for procedural
  std::string func_name_str_name; // ASSET_PROC_FUNC_STR_xxx for procedural

  // Set during the per-asset emit step (only for embedded data, not
  // disk-load and not procedural).
  std::string compression = "NONE"; // "NONE" | "ANS_ASCII"
  size_t uncompressed_size = 0;     // 0 when 'compression' == "NONE"
};

static bool ParseProceduralFunction(const std::string& compression_type_str,
                                    AssetBuildInfo* info, bool is_gpu) {
  const char* prefix = is_gpu ? "PROC_GPU(" : "PROC(";
  size_t prefix_len = is_gpu ? 9 : 5;

  size_t open_paren = compression_type_str.find('(');
  size_t close_paren = compression_type_str.rfind(')');
  if (open_paren == std::string::npos || close_paren == std::string::npos) {
    fprintf(stderr, "Error: Invalid %s syntax for asset: %s, string: [%s]\n",
            prefix, info->name.c_str(), compression_type_str.c_str());
    return false;
  }

  std::string func_and_params_str =
      compression_type_str.substr(open_paren + 1, close_paren - open_paren - 1);
  size_t params_start = func_and_params_str.find(',');

  if (params_start != std::string::npos) {
    info->proc_func_name = func_and_params_str.substr(0, params_start);
    std::string params_str = func_and_params_str.substr(params_start + 1);
    if (!ParseProceduralParams(params_str, &info->proc_params, info->name)) {
      return false;
    }
  } else {
    info->proc_func_name = func_and_params_str;
  }

  if (is_gpu) {
    if (info->proc_func_name != "gen_noise" &&
        info->proc_func_name != "gen_perlin" &&
        info->proc_func_name != "gen_grid") {
      fprintf(stderr,
              "Error: PROC_GPU only supports gen_noise, gen_perlin, gen_grid, "
              "got: %s for asset: %s\n",
              info->proc_func_name.c_str(), info->name.c_str());
      return false;
    }
  } else {
    if (kAssetPackerProcGenFuncMap.find(info->proc_func_name) ==
        kAssetPackerProcGenFuncMap.end()) {
      fprintf(stderr,
              "Warning: Unknown procedural function: %s for asset: %s "
              "(Runtime error will occur)\n",
              info->proc_func_name.c_str(), info->name.c_str());
    }
  }

  return true;
}

struct Vec3 {
  float x, y, z;
  Vec3 operator+(const Vec3& o) const {
    return {x + o.x, y + o.y, z + o.z};
  }
  Vec3 operator+=(const Vec3& o) {
    x += o.x;
    y += o.y;
    z += o.z;
    return *this;
  }
  Vec3 operator-(const Vec3& o) const {
    return {x - o.x, y - o.y, z - o.z};
  }
  Vec3 operator*(float s) const {
    return {x * s, y * s, z * s};
  }
  static Vec3 cross(const Vec3& a, const Vec3& b) {
    return {a.y * b.z - a.z * b.y, a.z * b.x - a.x * b.z,
            a.x * b.y - a.y * b.x};
  }
  Vec3 normalize() const {
    float len = std::sqrt(x * x + y * y + z * z);
    return (len > 1e-6f) ? Vec3{x / len, y / len, z / len} : Vec3{0, 0, 0};
  }
};

struct Vertex {
  float p[3], n[3], u[2];
};

static bool ProcessMeshFile(const std::string& full_path,
                            std::vector<uint8_t>* buffer,
                            const std::string& asset_name) {
  std::ifstream obj_file(full_path);
  if (!obj_file.is_open()) {
    fprintf(stderr, "Error: Could not open mesh file: %s\n", full_path.c_str());
    return false;
  }

  std::vector<float> v_pos, v_norm, v_uv;
  struct RawFace {
    int v[3], vt[3], vn[3];
  };
  std::vector<RawFace> raw_faces;

  std::string obj_line;
  while (std::getline(obj_file, obj_line)) {
    if (obj_line.compare(0, 2, "v ") == 0) {
      float x, y, z;
      std::sscanf(obj_line.c_str(), "v %f %f %f", &x, &y, &z);
      v_pos.push_back(x);
      v_pos.push_back(y);
      v_pos.push_back(z);
    } else if (obj_line.compare(0, 3, "vn ") == 0) {
      float x, y, z;
      std::sscanf(obj_line.c_str(), "vn %f %f %f", &x, &y, &z);
      v_norm.push_back(x);
      v_norm.push_back(y);
      v_norm.push_back(z);
    } else if (obj_line.compare(0, 3, "vt ") == 0) {
      float u, v;
      std::sscanf(obj_line.c_str(), "vt %f %f", &u, &v);
      v_uv.push_back(u);
      v_uv.push_back(v);
    } else if (obj_line.compare(0, 2, "f ") == 0) {
      char s1[64], s2[64], s3[64];
      if (std::sscanf(obj_line.c_str(), "f %s %s %s", s1, s2, s3) == 3) {
        std::string parts[3] = {s1, s2, s3};
        RawFace face = {};
        for (int i = 0; i < 3; ++i) {
          int v_idx = 0, vt_idx = 0, vn_idx = 0;
          if (parts[i].find("//") != std::string::npos) {
            std::sscanf(parts[i].c_str(), "%d//%d", &v_idx, &vn_idx);
          } else if (std::count(parts[i].begin(), parts[i].end(), '/') == 2) {
            std::sscanf(parts[i].c_str(), "%d/%d/%d", &v_idx, &vt_idx, &vn_idx);
          } else if (std::count(parts[i].begin(), parts[i].end(), '/') == 1) {
            std::sscanf(parts[i].c_str(), "%d/%d", &v_idx, &vt_idx);
          } else {
            std::sscanf(parts[i].c_str(), "%d", &v_idx);
          }
          face.v[i] = v_idx;
          face.vt[i] = vt_idx;
          face.vn[i] = vn_idx;
        }
        raw_faces.push_back(face);
      }
    }
  }

  // Generate normals if missing
  if (v_norm.empty() && !v_pos.empty()) {
    printf("Generating normals for %s...\n", asset_name.c_str());
    std::vector<Vec3> temp_normals(v_pos.size() / 3, {0, 0, 0});
    for (auto& face : raw_faces) {
      int idx0 = face.v[0] - 1;
      int idx1 = face.v[1] - 1;
      int idx2 = face.v[2] - 1;

      if (idx0 >= 0 && idx1 >= 0 && idx2 >= 0) {
        Vec3 p0 = {v_pos[idx0 * 3], v_pos[idx0 * 3 + 1], v_pos[idx0 * 3 + 2]};
        Vec3 p1 = {v_pos[idx1 * 3], v_pos[idx1 * 3 + 1], v_pos[idx1 * 3 + 2]};
        Vec3 p2 = {v_pos[idx2 * 3], v_pos[idx2 * 3 + 1], v_pos[idx2 * 3 + 2]};

        Vec3 normal = Vec3::cross(p1 - p0, p2 - p0).normalize();
        temp_normals[idx0] += normal;
        temp_normals[idx1] += normal;
        temp_normals[idx2] += normal;
      }
    }

    for (const auto& n : temp_normals) {
      Vec3 normalized = n.normalize();
      v_norm.push_back(normalized.x);
      v_norm.push_back(normalized.y);
      v_norm.push_back(normalized.z);
    }

    for (auto& face : raw_faces) {
      face.vn[0] = face.v[0];
      face.vn[1] = face.v[1];
      face.vn[2] = face.v[2];
    }
  }

  // Build final vertices
  std::vector<Vertex> final_vertices;
  std::vector<uint32_t> final_indices;
  std::map<std::string, uint32_t> vertex_map;

  for (const auto& face : raw_faces) {
    for (int i = 0; i < 3; ++i) {
      char key_buf[128];
      std::snprintf(key_buf, sizeof(key_buf), "%d/%d/%d", face.v[i], face.vt[i],
                    face.vn[i]);
      std::string key = key_buf;

      if (vertex_map.find(key) == vertex_map.end()) {
        vertex_map[key] = (uint32_t)final_vertices.size();

        Vertex v = {};
        if (face.v[i] > 0) {
          v.p[0] = v_pos[(face.v[i] - 1) * 3];
          v.p[1] = v_pos[(face.v[i] - 1) * 3 + 1];
          v.p[2] = v_pos[(face.v[i] - 1) * 3 + 2];
        }
        if (face.vn[i] > 0) {
          v.n[0] = v_norm[(face.vn[i] - 1) * 3];
          v.n[1] = v_norm[(face.vn[i] - 1) * 3 + 1];
          v.n[2] = v_norm[(face.vn[i] - 1) * 3 + 2];
        }
        if (face.vt[i] > 0) {
          v.u[0] = v_uv[(face.vt[i] - 1) * 2];
          v.u[1] = v_uv[(face.vt[i] - 1) * 2 + 1];
        }
        final_vertices.push_back(v);
      }
      final_indices.push_back(vertex_map[key]);
    }
  }

  // Format: [num_vertices][Vertex*N][num_indices][uint32_t*N]
  buffer->resize(sizeof(uint32_t) + final_vertices.size() * sizeof(Vertex) +
                 sizeof(uint32_t) + final_indices.size() * sizeof(uint32_t));
  uint8_t* out_ptr = buffer->data();
  *(uint32_t*)(out_ptr) = (uint32_t)final_vertices.size();
  out_ptr += sizeof(uint32_t);
  std::memcpy(out_ptr, final_vertices.data(),
              final_vertices.size() * sizeof(Vertex));
  out_ptr += final_vertices.size() * sizeof(Vertex);
  *(uint32_t*)(out_ptr) = (uint32_t)final_indices.size();
  out_ptr += sizeof(uint32_t);
  std::memcpy(out_ptr, final_indices.data(),
              final_indices.size() * sizeof(uint32_t));

  printf("Processed mesh asset %s: %zu vertices, %zu indices\n",
         asset_name.c_str(), final_vertices.size(), final_indices.size());

  return true;
}

static bool ProcessImageFile(const std::string& full_path,
                             std::vector<uint8_t>* buffer,
                             const std::string& asset_name) {
  int w, h, channels;
  unsigned char* img_data =
      stbi_load(full_path.c_str(), &w, &h, &channels, 4); // Force RGBA
  if (!img_data) {
    fprintf(stderr, "Error: Could not load image file: %s (Reason: %s)\n",
            full_path.c_str(), stbi_failure_reason());
    return false;
  }

  // Format: [Width(4)][Height(4)][Pixels...]
  buffer->resize(sizeof(uint32_t) * 2 + w * h * 4);
  uint32_t* header = (uint32_t*)(buffer->data());
  header[0] = (uint32_t)w;
  header[1] = (uint32_t)h;
  std::memcpy(buffer->data() + sizeof(uint32_t) * 2, img_data, w * h * 4);

  stbi_image_free(img_data);
  printf("Processed image asset %s: %dx%d RGBA\n", asset_name.c_str(), w, h);

  return true;
}

// ANS-compress 'raw' with the seeded histogram and round-trip verify the
// payload. Returns true on success and writes the compressed bytes to '*out'.
// Returns false (without populating *out) if encoding fails, the compressed
// payload is not smaller, or the round-trip mismatches.
static bool TryAnsCompress(const std::vector<uint8_t>& raw,
                           const uint32_t* hist,
                           std::vector<uint8_t>* out) {
  if (raw.empty()) return false;
  std::vector<uint8_t> enc;
  if (!ans::Encode(raw.data(), raw.size(), &enc, hist)) return false;
  if (enc.size() >= raw.size()) return false;
  std::vector<uint8_t> verify(raw.size());
  size_t got = 0;
  if (!ans::Decode(enc.data(), enc.size(), verify.data(), verify.size(), &got,
                   hist) ||
      got != raw.size() ||
      std::memcmp(verify.data(), raw.data(), raw.size()) != 0) {
    return false;
  }
  *out = std::move(enc);
  return true;
}

// Emits a comma-separated list of values as a C array initializer, wrapping
// at 12 entries per line.
template <typename T, typename FormatFn>
static void EmitArrayInit(FILE* f, const T* data, size_t n, FormatFn fmt) {
  for (size_t i = 0; i < n; ++i) {
    if (i % 12 == 0) fprintf(f, "\n    ");
    fmt(f, data[i]);
    if (i + 1 != n) fprintf(f, ", ");
  }
  fprintf(f, "\n");
}

int main(int argc, char* argv[]) {
  if (argc < 4) {
    fprintf(stderr,
            "Usage: %s <assets.txt_path> <output_assets_h_path> "
            "<output_assets_data_cc_path> [--disk_load]\n",
            argv[0]);
    return 1;
  }

  bool disk_load_mode = false;
  if (argc > 4 && std::strcmp(argv[4], "--disk_load") == 0) {
    disk_load_mode = true;
    printf("Asset packer running in disk-load mode.\n");
  }

  std::string assets_txt_path = argv[1];
  std::string output_assets_h_path = argv[2];
  std::string output_assets_data_cc_path = argv[3];

  std::ifstream assets_txt_file(assets_txt_path);
  if (!assets_txt_file.is_open()) {
    fprintf(stderr, "Error: Could not open assets.txt at %s\n",
            assets_txt_path.c_str());
    return 1;
  }

  FILE* assets_h_file = std::fopen(output_assets_h_path.c_str(), "w");
  if (!assets_h_file) {
    fprintf(stderr, "Error: Could not open output assets.h at %s\n",
            output_assets_h_path.c_str());
    return 1;
  }

  FILE* assets_data_cc_file =
      std::fopen(output_assets_data_cc_path.c_str(), "w");
  if (!assets_data_cc_file) {
    fprintf(stderr, "Error: Could not open output assets_data.cc at %s\n",
            output_assets_data_cc_path.c_str());
    return 1;
  }

  // Generate assets.h header
  fprintf(
      assets_h_file,
      "// This file is auto-generated by asset_packer.cc. Do not edit.\n\n");
  fprintf(assets_h_file, "#pragma once\n");
  fprintf(assets_h_file, "#include <cstdint>\n\n");
  fprintf(assets_h_file, "enum class AssetId : uint16_t {\n");

  std::string generated_header_name =
      output_assets_h_path.substr(output_assets_h_path.find_last_of("/\\") + 1);

  // Generate assets_data.cc header
  fprintf(
      assets_data_cc_file,
      "// This file is auto-generated by asset_packer.cc. Do not edit.\n\n");
  fprintf(assets_data_cc_file, "#include <cstring>\n");
  fprintf(assets_data_cc_file, "#include \"util/asset_manager.h\"\n");
  fprintf(assets_data_cc_file, "#include \"%s\"\n",
          generated_header_name.c_str());

  // Forward declare procedural functions for AssetRecord initialization
  fprintf(assets_data_cc_file,
          "namespace procedural { void gen_noise(uint8_t*, int, int, const "
          "float*, int); }\n");
  fprintf(assets_data_cc_file,
          "namespace procedural { void gen_grid(uint8_t*, int, int, const "
          "float*, int); }\n");
  fprintf(assets_data_cc_file,
          "namespace procedural { void make_periodic(uint8_t*, int, int, const "
          "float*, int); }\n\n");

  std::vector<AssetBuildInfo> asset_build_infos;
  int asset_id_counter = 0;
  std::string line;
  // Updated regex pattern for new asset list format (Name, CompressionType,
  // Filename, Description)
  std::regex asset_line_regex(
      R"(^\s*([A-Z0-9_]+)\s*,\s*(PROC\([^)]*\)|[^,]+)\s*(?:,\s*([^,]*))?\s*(?:,\s*\"(.*)\")?\s*$)");

  while (std::getline(assets_txt_file, line)) {
    if (line.empty() || line[0] == '#')
      continue;

    std::smatch matches;
    if (std::regex_search(line, matches, asset_line_regex) &&
        matches.size() >= 3) {
      AssetBuildInfo info;
      info.name = matches[1].str();
      std::string compression_type_str = matches[2].str();
      // Filename is now matches[3]
      info.filename =
          (matches.size() >= 4 && matches[3].matched) ? matches[3].str() : "_";

      info.data_array_name = "ASSET_DATA_" + info.name;
      info.params_array_name = "ASSET_PROC_PARAMS_" + info.name;
      info.func_name_str_name = "ASSET_PROC_FUNC_STR_" + info.name;
      info.asset_type = compression_type_str;

      if (compression_type_str.rfind("PROC_GPU(", 0) == 0) {
        info.asset_type = "PROC_GPU";
        if (!ParseProceduralFunction(compression_type_str, &info, true)) {
          return 1;
        }
      } else if (compression_type_str.rfind("PROC(", 0) == 0) {
        info.asset_type = "PROC";
        if (!ParseProceduralFunction(compression_type_str, &info, false)) {
          return 1;
        }
      } else if (compression_type_str == "MP3") {
        info.asset_type = "MP3";
      } else if (compression_type_str != "WGSL" &&
                 compression_type_str != "SPEC" &&
                 compression_type_str != "TEXTURE" &&
                 compression_type_str != "MESH" &&
                 compression_type_str != "BINARY") {
        fprintf(stderr,
                "Warning: Unknown compression type '%s' for asset: %s\n",
                compression_type_str.c_str(), info.name.c_str());
      }

      asset_build_infos.push_back(info);
      fprintf(assets_h_file, "  ASSET_%s = %d,\n", info.name.c_str(),
              asset_id_counter++);
    } else {
      fprintf(stderr, "Warning: Skipping malformed line in assets.txt: %s\n",
              line.c_str());
    }
  }

  fprintf(assets_h_file, "  ASSET_LAST_ID = %d,\n", asset_id_counter);
  fprintf(assets_h_file, "};\n");

  fprintf(assets_h_file,
          "#include \"util/asset_manager.h\"\n"); // Include here AFTER enum
                                                  // definition
  fprintf(assets_h_file,
          "\n// Accessors to avoid static initialization order issues\n");
  fprintf(assets_h_file, "const struct AssetRecord* GetAssetRecordTable();\n");
  fprintf(assets_h_file, "size_t GetAssetCount();\n");

  std::fclose(assets_h_file);

  // ---------------------------------------------------------------------
  // Pre-pass: build a corpus-wide byte histogram from all WGSL assets to
  // seed the ANS coder. Skipped in disk-load mode (WGSL data is not
  // embedded then, so we never run the encoder).
  // ---------------------------------------------------------------------
  uint32_t ans_ascii_hist[256] = {};
  if (!disk_load_mode) {
    for (const auto& info : asset_build_infos) {
      if (info.asset_type != "WGSL") continue;
      std::string base_dir =
          assets_txt_path.substr(0, assets_txt_path.find_last_of("/\\") + 1);
      std::filesystem::path p = std::filesystem::absolute(base_dir) / info.filename;
      std::ifstream f(p.lexically_normal().string(), std::ios::binary);
      if (!f.is_open()) continue;
      std::vector<uint8_t> buf((std::istreambuf_iterator<char>(f)),
                               std::istreambuf_iterator<char>());
      ans::Histogram(buf.data(), buf.size(), ans_ascii_hist);
    }
  }

  fprintf(assets_data_cc_file,
          "// Per-corpus byte histogram, seed for ANS_ASCII decompression.\n");
  fprintf(assets_data_cc_file,
          "static const uint32_t kAnsAsciiHistogram[256] = {");
  EmitArrayInit(assets_data_cc_file, ans_ascii_hist, 256,
                [](FILE* f, uint32_t v) { fprintf(f, "%u", v); });
  fprintf(assets_data_cc_file, "};\n");
  fprintf(assets_data_cc_file,
          "const uint32_t* GetAnsAsciiHistogram() { return kAnsAsciiHistogram; }\n\n");

  for (auto& info : asset_build_infos) {
    if (info.asset_type != "PROC" && info.asset_type != "PROC_GPU") {
      std::string base_dir =
          assets_txt_path.substr(0, assets_txt_path.find_last_of("/\\") + 1);
      std::filesystem::path base_path = std::filesystem::absolute(base_dir);
      std::filesystem::path combined_path = base_path / info.filename;
      std::string full_path = combined_path.lexically_normal().string();

      if (disk_load_mode &&
          (info.asset_type == "SPEC" || info.asset_type == "MP3" ||
           info.asset_type == "WGSL")) {
        fprintf(assets_data_cc_file,
                "alignas(16) static const char %s[] = \"%s\";\n",
                info.data_array_name.c_str(), full_path.c_str());
        fprintf(assets_data_cc_file, "const size_t ASSET_SIZE_%s = %zu;\n",
                info.name.c_str(), full_path.length() + 1);
      } else {
        std::vector<uint8_t> buffer;
        if (info.asset_type == "TEXTURE") {
          if (!ProcessImageFile(full_path, &buffer, info.name)) {
            return 1;
          }
        } else if (info.asset_type == "MESH") {
          if (!ProcessMeshFile(full_path, &buffer, info.name)) {
            return 1;
          }
        } else {
          std::ifstream asset_file(full_path, std::ios::binary);
          if (!asset_file.is_open()) {
            fprintf(stderr,
                    "Warning: Asset file not found, skipping: %s (%s)\n",
                    info.name.c_str(), full_path.c_str());
            fprintf(assets_data_cc_file,
                    "const size_t ASSET_SIZE_%s = 0;\n",
                    info.name.c_str());
            fprintf(assets_data_cc_file,
                    "alignas(16) static const uint8_t %s[] = {0};\n",
                    info.data_array_name.c_str());
            continue;
          }
          buffer.assign((std::istreambuf_iterator<char>(asset_file)),
                        std::istreambuf_iterator<char>());
        }

        const size_t original_size = buffer.size();

        // ANS-compress WGSL (ASCII text) using the corpus histogram.
        // Compressed payload replaces the raw buffer; we don't null-terminate
        // compressed blobs since the runtime decoder writes NUL itself.
        std::vector<uint8_t> compressed;
        const bool use_ans =
            (info.asset_type == "WGSL") &&
            TryAnsCompress(buffer, ans_ascii_hist, &compressed);
        if (use_ans) {
          info.compression = "ANS_ASCII";
          info.uncompressed_size = original_size;
          printf("  ANS %-32s %7zu -> %7zu (%.2f x)\n", info.name.c_str(),
                 original_size, compressed.size(),
                 (double)compressed.size() / (double)original_size);
        } else {
          buffer.push_back(0);  // null-terminate raw assets
        }

        const std::vector<uint8_t>& payload = use_ans ? compressed : buffer;
        fprintf(assets_data_cc_file, "const size_t ASSET_SIZE_%s = %zu;\n",
                info.name.c_str(),
                use_ans ? payload.size() : original_size);
        fprintf(assets_data_cc_file,
                "alignas(16) static const uint8_t %s[] = {",
                info.data_array_name.c_str());
        EmitArrayInit(assets_data_cc_file, payload.data(), payload.size(),
                      [](FILE* f, uint8_t v) { fprintf(f, "0x%02x", v); });
        fprintf(assets_data_cc_file, "};\n");
      }
    } else {
      fprintf(assets_data_cc_file, "static const float %s[] = {",
              info.params_array_name.c_str());
      for (size_t i = 0; i < info.proc_params.size(); ++i) {
        if (i > 0)
          fprintf(assets_data_cc_file, ", ");
        fprintf(assets_data_cc_file, "%f", info.proc_params[i]);
      }
      fprintf(assets_data_cc_file, "};\n\n");
      fprintf(assets_data_cc_file, "static const char* %s = \"%s\";\n\n",
              info.func_name_str_name.c_str(), info.proc_func_name.c_str());
    }
  }

  fprintf(assets_data_cc_file, "const AssetRecord* GetAssetRecordTable() {\n");
  fprintf(assets_data_cc_file, "  static const AssetRecord assets[] = {\n");
  for (const auto& info : asset_build_infos) {
    fprintf(assets_data_cc_file, "    { ");
    if (info.asset_type == "PROC" || info.asset_type == "PROC_GPU") {
      // data, size, type, compression, uncompressed_size, proc_func, params, n
      fprintf(assets_data_cc_file,
              "nullptr, 0, AssetType::%s, AssetCompression::NONE, 0, "
              "%s, %s, %zu",
              info.asset_type.c_str(), info.func_name_str_name.c_str(),
              info.params_array_name.c_str(), info.proc_params.size());
    } else {
      fprintf(assets_data_cc_file,
              "(const uint8_t*)%s, ASSET_SIZE_%s, AssetType::%s, "
              "AssetCompression::%s, %zu, nullptr, nullptr, 0",
              info.data_array_name.c_str(), info.name.c_str(),
              info.asset_type.c_str(), info.compression.c_str(),
              info.uncompressed_size);
    }
    fprintf(assets_data_cc_file, " },\n");
  }
  fprintf(assets_data_cc_file, "  };\n");
  fprintf(assets_data_cc_file, "  return assets;\n");
  fprintf(assets_data_cc_file, "}\n\n");

  fprintf(assets_data_cc_file, "size_t GetAssetCount() {\n");
  fprintf(assets_data_cc_file, "  return %zu;\n", asset_build_infos.size());
  fprintf(assets_data_cc_file, "}\n\n");

  fprintf(assets_data_cc_file,
          "AssetId GetAssetIdByName(const char* name) {\n");
  for (const auto& info : asset_build_infos) {
    fprintf(assets_data_cc_file,
            "  if (std::strcmp(name, \"%s\") == 0) return AssetId::ASSET_%s;\n",
            info.name.c_str(), info.name.c_str());
  }
  fprintf(assets_data_cc_file, "  return AssetId::ASSET_LAST_ID;\n");
  fprintf(assets_data_cc_file, "}\n\n");

  std::fclose(assets_data_cc_file);

  printf("Asset packer successfully generated records for %zu assets.\n",
         asset_build_infos.size());

  return 0;
}