Revision e1c6ee354570561a7c9813c38e4a6d24d21ac430 authored by Andrew Adams on 10 October 2022, 22:44:15 UTC, committed by Andrew Adams on 10 October 2022, 22:44:15 UTC
1 parent 87853de
build_halide_h.cpp
#include <cstdlib>
#include <stdio.h>
#include <string.h>
#include <set>
#include <string>
#include <assert.h>
std::set<std::string> done;
void dump_header(std::string header) {
if (done.find(header) != done.end()) return;
done.insert(header);
if (header.find("runtime_internal") != std::string::npos) {
fprintf(stdout, "#error \"COMPILING_HALIDE_RUNTIME should never be defined for Halide.h\"\n");
return;
}
FILE *f = fopen(header.c_str(), "r");
if (f == nullptr) {
fprintf(stderr, "Could not open header %s.\n", header.c_str());
exit(1);
}
char line[1024];
const int line_len = sizeof(line);
const char include_str[] = "#include \"";
const int include_str_len = sizeof(include_str) - 1; // remove null terminator
while (fgets(line, line_len, f)) {
if (strncmp(line, include_str, include_str_len) == 0) {
char *sub_header = line + include_str_len;
for (int i = 0; i < line_len - include_str_len; i++) {
if (sub_header[i] == '"') sub_header[i] = 0;
}
size_t slash_pos = header.rfind('/');
std::string path;
if (slash_pos != std::string::npos)
path = header.substr(0, slash_pos + 1);
dump_header(path + sub_header);
} else {
fputs(line, stdout);
}
}
fclose(f);
}
int main(int argc, char **files) {
if (argc < 3) {
fprintf(stderr, "Usage: %s LICENSE.txt [headers...]\n", files[0]);
exit(1);
}
fprintf(stdout, "/* Halide.h -- interface for the 'Halide' library.\n\n");
{
FILE *f = fopen(files[1], "r");
if (f == nullptr) {
fprintf(stderr, "Could not open %s.\n", files[1]);
exit(1);
}
char line[1024];
while (fgets(line, sizeof(line), f)) {
fprintf(stdout, " %s", line);
}
fclose(f);
}
fprintf(stdout, "\n*/\n\n");
fprintf(stdout, "#ifndef HALIDE_H\n");
fprintf(stdout, "#define HALIDE_H\n\n");
// If we're building on visual studio and Halide_SHARED is defined, we'd better
// also define it for clients so that dllimport gets used.
#if defined(_MSC_VER) && defined(Halide_SHARED)
fprintf(stdout, "#define Halide_SHARED\n");
#endif
for (int i = 2; i < argc; i++) {
dump_header(files[i]);
}
fprintf(stdout, "\n");
fprintf(stdout,
"// Clean up macros used inside Halide headers\n"
"#undef user_assert\n"
"#undef user_error\n"
"#undef user_warning\n"
"#undef internal_error\n"
"#undef internal_assert\n"
"#undef halide_runtime_error\n"
"#undef HALIDE_EXPORT\n\n"
"#endif // HALIDE_H\n"
);
return 0;
}
Computing file changes ...