PBM.hpp

#include <fstream>
#include <string>
#include <vector>


struct PBM {
    size_t width, height; 
    std::vector<int> bits; 

    PBM(size_t width, size_t height) {
        this->width = width;
        this->height = height;
        this->bits = std::vector<int>(width * height); 
    }
}; 


PBM read_pbm(const std::string& filename) {
    std::ifstream file(filename);
    if (!file.is_open()) throw std::runtime_error("Error: Could not open file '" + filename + "'.");
    
    std::string magic;
    file >> magic;
    if (magic != "P1") throw std::runtime_error("Error: Not a PBM file (P1 format)"); 

    std::string line;
    while (file.peek() == '#') std::getline(file, line);

    size_t width, height; 
    file >> width >> height;

    PBM pbm(width, height); 
    for (size_t y = 0; y < height; ++y) 
        for (size_t x = 0; x < width; ++x) 
            file >> pbm.bits[y * width + x];
        
    file.close();
    return pbm;
}