binviz.c

#include <math.h>
#include <stdint.h>

#define NOB_IMPLEMENTATION
#include "nob.h"

#define STB_IMAGE_WRITE_IMPLEMENTATION
#include "stb_image_write.h"

#define MAP_SIZE 256

static size_t map[MAP_SIZE][MAP_SIZE][MAP_SIZE] = {0};
static uint8_t pixels[MAP_SIZE * MAP_SIZE * MAP_SIZE] = {0};

int main(int argc, char **argv) {
    const char *program = nob_shift_args(&argc, &argv);

    if (argc <= 0) {
        nob_log(NOB_ERROR, "No input is provided");
        nob_log(NOB_INFO, "Usage: %s <input>", program);
        return 1;
    }

    Nob_String_Builder content = {0};
    for (int i = 0; i < argc; ++i) {
        const char *input_file_path = argv[i];
        content.count = 0;
        nob_log(NOB_INFO, "Reading %s", input_file_path);
        if (!nob_read_entire_file(input_file_path, &content))
            return 1;

        memset(map, 0, sizeof(map));
        if (content.count > 0) {
            for (size_t i = 0; i < content.count - 1; ++i) {
                uint8_t x = content.items[i];
                uint8_t y = content.items[i + 1];
                uint8_t z = content.items[i + 2];
                map[y][x][z] += 1;
            }
        }

        float max = 0;

        for (size_t y = 0; y < MAP_SIZE; ++y) {
            for (size_t x = 0; x < MAP_SIZE; ++x) {
                for (size_t z = 0; z < MAP_SIZE; ++z) {
                    float f = 0.0f;
                    if (map[y][x][z] > 0)
                        f = logf(map[y][x][z]);
                    if (f > max)
                        max = f;
                }
            }
        }

        for (size_t y = 0; y < MAP_SIZE; ++y) {
            for (size_t x = 0; x < MAP_SIZE; ++x) {
                for (size_t z = 0; z < MAP_SIZE; ++z) {
                    float t = logf(map[y][x][z]) / max;
                    uint8_t b = t * 255;
                    pixels[y * MAP_SIZE * MAP_SIZE + x * MAP_SIZE + z] = b;
                }
            }
        }

        const char* output_file_path = nob_temp_sprintf("%s.binviz3d", input_file_path);
        if (!nob_write_entire_file(output_file_path, pixels, MAP_SIZE * MAP_SIZE * MAP_SIZE * sizeof(uint8_t))) {
            nob_log(NOB_ERROR, "Could not save file %s", output_file_path);
            return 1;
        }

        nob_log(NOB_INFO, "Generated %s", output_file_path);

        nob_temp_reset();
    }

    return 0;
}