Io.hh

Go to the documentation of this file.

 
#ifndef IO_HH
#define IO_HH
 
#include <string>
 
namespace io
{
 
bool savePpm(const std::string& fileName,
        uint32_t           width,
        uint32_t           height,
        const void* dataP)
{
    std::ofstream outputStream(fileName.c_str(), std::ios::out | std::ios::binary);
 
    if (false == outputStream.good()) {
        std::cerr << "Failed to open \"" << fileName << "\"" << std::endl;
        return false;
    }
 
    const uint32_t imageSize = height * width * 3;
 
    outputStream << "P6\n"
        << width << " " << height << "\n"
        << 0xFF << "\n";
 
    outputStream.write(reinterpret_cast<const char*>(dataP), imageSize);
 
    outputStream.close();
    return true;
}
 
bool savePgm(const std::string& fileName,
             uint32_t           width,
             uint32_t           height,
             uint32_t           bitsPerPixel,
             const std::string& comment,
             const void         *dataP)
{
    std::ofstream outputStream(fileName.c_str(), std::ios::binary | std::ios::out);
 
    if (false == outputStream.good()) {
                std::cerr << "Failed to open \"" << fileName << "\"" << std::endl;
        return false;
    }
 
    const uint32_t imageSize = height * width;
 
    switch(bitsPerPixel) {
    case 8:
    {
 
        outputStream << "P5\n"
                     << "#" << comment << "\n"
                     << width << " " << height << "\n"
                     << 0xFF << "\n";
 
        outputStream.write(reinterpret_cast<const char*>(dataP), imageSize);
 
        break;
    }
    case 16:
    {
        outputStream << "P5\n"
                     << "#" << comment << "\n"
                     << width << " " << height << "\n"
                     << 0xFFFF << "\n";
 
        const uint16_t *imageP = reinterpret_cast<const uint16_t*>(dataP);
 
        for (uint32_t i=0; i<imageSize; ++i) {
            uint16_t o = htons(imageP[i]);
            outputStream.write(reinterpret_cast<const char*>(&o), sizeof(uint16_t));
        }
 
        break;
    }
    }
 
    outputStream.close();
    return true;
}
 
}
 
#endif //IO_HH