D:/Home/Programs/iSpike/src/Reader/FileVisualReader.cpp

Go to the documentation of this file.

//iSpike includes
#include <iSpike/Reader/FileVisualReader.hpp>
#include <iSpike/Bitmap.hpp>
#include <iSpike/ISpikeException.hpp>
#include <iSpike/Log/Log.hpp>
using namespace ispike;

//Other includes
#include <iostream>
#include <fstream>
using namespace std;

//Enable/disable debug output
//#define DEBUG

FileVisualReader::FileVisualReader()
{
        // Define the properties of this reader
        addProperty(Property("imageIn.ppm", "File Name", "The file where the image will be read from", true));

        //Create description
        readerDescription = Description("File Visual Reader", "This is a file visual reader", "Visual Reader");
}


FileVisualReader::~FileVisualReader(){
}


/*--------------------------------------------------------------------*/
/*---------                 PUBLIC METHODS                     -------*/
/*--------------------------------------------------------------------*/

Bitmap& FileVisualReader::getBitmap(){
        return bitmap;
}


//Inherited from Reader
void FileVisualReader::initialize(map<string, Property>& properties){
        setProperties(properties);
        setInitialized(true);
}


//Inherited from PropertyHolder
void FileVisualReader::setProperties(map<string, Property>& properties){
        string fileName = updateStringProperty(properties["File Name"]);
        readPPMImage(fileName);
}


/*--------------------------------------------------------------------*/
/*---------                PRIVATE METHODS                     -------*/
/*--------------------------------------------------------------------*/

void FileVisualReader::readPPMImage(string& fname){
        #ifdef DEBUG
                LOG(LOG_INFO) << "FileVisualReader: Reading image from: " << fname;
        #endif//DEBUG

        ifstream ifp;

        ifp.open(fname.c_str(), ios::in | ios::binary);

        if (!ifp) {
                ostringstream messageStream;
                messageStream << "Can't read image: " << fname;
                string message(messageStream.str());
                throw ISpikeException(message);
        }

        // read header
        char header [100];
        ifp.getline(header,100,'\n');

        if ( (header[0]!=80) ||    /* 'P' */
                 (header[1]!=54) ) {   /* '6' */
                ostringstream messageStream;
                messageStream << "Image " << fname << " is not PPM";
                string message(messageStream.str());
                throw ISpikeException(message);
        }

        ifp.getline(header,100,'\n');
        while(header[0]=='#') {
                ifp.getline(header,100,'\n');
        }

        char* ptr;
        int M=strtol(header, &ptr, 0);
        int N=atoi(ptr);

        #ifdef DEBUG
                LOG(LOG_INFO) << "FileVisualReader: Retrieved image with Width: " << M << " Height: " << N;
        #endif//DEBUG

        ifp.getline(header,100,'\n');

        bitmap.reset(M, N, 3);
        unsigned char *charImage = bitmap.getContents();

        //Read image into bitmap's contents
        ifp.read((char*)charImage, 3*M*N);
        if (ifp.fail()) {
                ostringstream messageStream;
                messageStream << "Image " << fname << " has wrong size";
                string message(messageStream.str());
                throw ISpikeException(message);
        }

        //Clean up
        ifp.close();

        #ifdef DEBUG
                LOG(LOG_DEBUG)<<"File visual reader bitmap size: "<< bitmap.size();
        #endif//DEBUG

        //Increment image id
        ++imageID;
}