summaryrefslogtreecommitdiff
path: root/tools/asset_packer.cc
blob: 8b06829dea55a2e1e2d710aa6327aff178201584 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
#include <fstream>
#include <iostream>
#include <map>
#include <string>
#include <vector>

int main(int argc, char *argv[]) {
  if (argc != 4) {
    std::cerr << "Usage: " << argv[0]
              << " <assets.txt_path> <output_assets_h_path> "
                 "<output_assets_data_cc_path>\n";
    return 1;
  }

  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()) {
    std::cerr << "Error: Could not open assets.txt at " << assets_txt_path
              << "\n";
    return 1;
  }

  std::ofstream assets_h_file(output_assets_h_path);
  if (!assets_h_file.is_open()) {
    std::cerr << "Error: Could not open output assets.h at "
              << output_assets_h_path << "\n";
    return 1;
  }

  std::ofstream assets_data_cc_file(output_assets_data_cc_path);
  if (!assets_data_cc_file.is_open()) {
    std::cerr << "Error: Could not open output assets_data.cc at "
              << output_assets_data_cc_path << "\n";
    return 1;
  }

  // Generate assets.h
  assets_h_file << "#pragma once\n";
  assets_h_file << "#include <cstdint>\n\n";
  assets_h_file << "enum class AssetId : uint16_t {\n";

  // Generate assets_data.cc header
  assets_data_cc_file << "#include \"assets.h\"\n\n";
  assets_data_cc_file << "#include <vector>\n\n";

  std::string line;
  int asset_id_counter = 0;
  std::vector<std::string> asset_names;

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

    size_t first_comma = line.find(',');
    if (first_comma == std::string::npos) {
      std::cerr << "Warning: Skipping malformed line in assets.txt: " << line
                << "\n";
      continue;
    }

    std::string asset_name = line.substr(0, first_comma);
    asset_name.erase(0, asset_name.find_first_not_of(" \t\r\n"));
    asset_name.erase(asset_name.find_last_not_of(" \t\r\n") + 1);

    if (asset_name.empty()) {
      std::cerr << "Warning: Skipping line with empty asset name: " << line
                << "\n";
      continue;
    }

    asset_names.push_back(asset_name);

    // Add to assets.h enum
    assets_h_file << "  ASSET_" << asset_name << " = " << asset_id_counter
                  << ",\n";

    // Placeholder for assets_data.cc (empty data for now)
    assets_data_cc_file << "const uint8_t ASSET_DATA_" << asset_name
                        << "[] = {};\n";
    assets_data_cc_file << "const size_t ASSET_SIZE_" << asset_name
                        << " = 0;\n\n";

    asset_id_counter++;
  }

  assets_h_file << "};\n\n";

  // Generate GetAsset function declaration in assets.h
  assets_h_file
      << "const uint8_t *GetAsset(AssetId asset_id, size_t *out_size = nullptr);\n";
  assets_h_file << "void DropAsset(AssetId asset_id, const uint8_t *asset); // For lazy "
                   "decompression scaffolding\n";
  assets_h_file.close();

  // Generate GetAsset function implementation in assets_data.cc
  assets_data_cc_file << "const uint8_t *GetAsset(AssetId asset_id, size_t *out_size) {\n";
  assets_data_cc_file << "  if (out_size) *out_size = 0;\n"; // Default for now
  assets_data_cc_file << "  switch (asset_id) {\n";
  for (const std::string &name : asset_names) {
    assets_data_cc_file << "    case AssetId::ASSET_" << name << ":\n";
    assets_data_cc_file << "      if (out_size) *out_size = ASSET_SIZE_" << name << ";\n";
    assets_data_cc_file << "      return ASSET_DATA_" << name << ";\n";
  }
  assets_data_cc_file << "    default:\n";
  assets_data_cc_file << "      return nullptr;\n";
  assets_data_cc_file << "  }\n";
  assets_data_cc_file << "}\n\n";

  // Dummy DropAsset implementation
  assets_data_cc_file << "void DropAsset(AssetId asset_id, const uint8_t *asset) {\n";
  assets_data_cc_file << "  (void)asset_id;\n";
  assets_data_cc_file << "  (void)asset;\n";
  assets_data_cc_file << "  // No-op for now, actual implementation for lazy decompression goes here\n";
  assets_data_cc_file << "}\n";

  assets_data_cc_file.close();

  std::cout << "Asset packer successfully generated: " << output_assets_h_path
            << " and " << output_assets_data_cc_path << "\n";

  return 0;
}