PPMIO.cpp

Go to the documentation of this file.

/*
 * PPMIO.cpp
 *
 *  Created on:  08.09.2011
 *      Author:  Kim Rinnewitz  ( krinnewitz@uos.de )
 *  Modified on: 11.12.2011
 *      Author:  Thomas Wiemann ( twiemann@uos.de )
 *  Modified on: 15.02.2011
 *      Author:  Denis Meyer    ( denmeyer@uos.de )
 */

#include "lvr2/io/PPMIO.hpp"

#include <iostream>
#include <fstream>
#include <string.h>

using namespace std;

namespace lvr2
{

PPMIO::PPMIO()
{
    m_pixels    = 0;
    m_width   = 0;
    m_height   = 0;
}

PPMIO::PPMIO( string filename ) : m_width(0), m_height(0), m_pixels(0)
{

    // Try to open file
    ifstream in(filename.c_str());

    // Parse file
    if(in.good())
    {
        // Line buffer
        char buffer[1024];

        // Read file tag
        readLine(in, buffer);

        // Check tag
        string tag(buffer);
        if(tag == "P3")
        {
            // Read width, height and color information
            stringstream ss;
            readLine(in, buffer);
            ss << buffer << " ";
            readLine(in, buffer);
            ss << buffer << " ";

            // Read formatted data
            ss >> m_width >> m_height;

            // Alloc data
            m_pixels = new unsigned char[m_width * m_height * 3];

            // Read pixels
            int p;
            for(int i = 0; i < m_width * m_height * 3; i++)
            {
                in >> p;
                m_pixels[i] = (unsigned char)p;
            }
        }
        else
        {
            in.close();
            in.open(filename.c_str(), ios::binary);
            //      readLine(in, buffer);
            //      char tmp[3];
            //      sscanf(buffer, "%s %d %d 255", tmp, &m_width, &m_height);

            string tag;
            in >> tag;


            if(tag == "P6") // TODO: hacked in for our output
            {
                int n_colors;
                in >> m_width >> m_height >> n_colors;
                in.getline(0,0);
                m_pixels = new unsigned char[m_width * m_height * 3];
                in.read((char *)m_pixels, m_width * m_height * 3);
            }
            else
            {
                cerr << "Unsupported tag, only P3 or P6 possible." << endl;
            }
        }
    }
    else
    {
        cout << "ReadPPM: Unable to open file " << filename << "." << endl;
    }
}

void PPMIO::write( string filename )
{
    ofstream out(filename.c_str());

    if(out.good())
    {
        out<<"P6"<<" "<<m_width<<" "<<m_height<<" "<<"255"<<endl;
        out.write((char*) m_pixels, m_width * m_height * 3);
    }

    out.close();

}

void PPMIO::setDataArray( unsigned char* array, int width, int height )
{
    m_pixels = array;
    m_width = width;
    m_height = height;
}

void PPMIO::readLine( ifstream & in, char* buffer )
{
    // Read lines until no comment line was found
    do
    {
      in.getline(buffer, 256);
    }
    while(buffer[0] == '#' && in.good() );
}

} // namespace lvr2