include/DGImage.hpp

Go to the documentation of this file.

#ifndef DGIMAGE_H
#define DGIMAGE_H

#include <string>
#include <fstream>

#include "DGVGlobal.h"

using namespace std;

const int lineLength = 256;

template<typename type>
class DGV_TEMPLATEDLL DGImage
{
public:
    DGImage();
    virtual ~DGImage();

    bool isPNMBinaryFormat(string filename);

    bool readPNM(string filename, Array<type,2> &data, int &greyMax);
    bool writePNM(Array<type,2> &data, const int greyMax, string filename);
protected:

};

template<typename type>
DGImage<type>::DGImage()
{
    //ctor
}

template<typename type>
DGImage<type>::~DGImage()
{
    //dtor
}

template<typename type>
bool DGImage<type>::isPNMBinaryFormat(string filename)
{
    ifstream inFile(filename.c_str());
    char c;
    string magic;

    if(inFile.fail())
        return false;

    do
    {
        inFile >> c;
        if (c == '\0')
        {
            // Bad file.
            inFile.close();
            return false;
        }
    }
    while (c != 'P');
    magic += c;
    inFile >> c; 
    magic += c;

    //cerr << "PNM Magic Number: " << magic << endl;
    if ( magic == "P4" || magic == "P5" || magic == "P6" )
        return true;
    else
        return false;
}

template<typename type>
bool DGImage<type>::readPNM(string filename, Array<type,2> &data, int &greyMax)
{
    char portable = 0, fileType, comment[lineLength];
    int width = 0, height = 0;
    ifstream inFile;

    inFile.open(filename.c_str()); 

    if(inFile.fail())
        return false;

    while(portable != 'P')
        inFile >> portable;
        inFile >> fileType; 
        cerr << "File Type: " << portable << fileType << endl;
        while(portable != '#')
        inFile >> portable;
        inFile.getline(comment,lineLength); 
        cerr << "File Info: " << comment << endl;

    inFile >> width;
    inFile >> height;
    if(fileType != '1')
        inFile >> greyMax;
    else
        greyMax = 1;
        printf("File Size: %dx%d, Grey Scale Max: %d\n",width, height, greyMax);

    data.resizeAndPreserve(height,width);

    if((fileType == '2') || (fileType == '1'))
        {
                for(int j = 0; j < height; j ++)
            for(int k = 0; k < width; k ++)
                inFile >> data(j,k);
        }
    else if(fileType == '3')
        {
                for(int j = 0; j < height; j ++)
            for(int k = 0; k < width; k ++)
                        {
                inFile >> data(j,k); 
                inFile >> data(j,k);
                inFile >> data(j,k);
            }
        }

        inFile.close(); 

    return true;
}

template<typename type>
bool DGImage<type>::writePNM(Array<type,2> &data, const int greyMax, string filename)
{
    ofstream outFile;

    outFile.open(filename.c_str());
    if(outFile.fail())
        return false;

    outFile << "P2\n# Generated PGM by DGV.\n";
    outFile << data.cols() << " " << data.rows() << endl << greyMax << endl;

    for(int j = 0; j < data.rows(); j ++)
    {
        for(int k = 0; k < data.cols(); k ++)
            outFile << data(j,k) << " ";
        outFile << endl;
    }

    outFile.close();
    return true;
}

#endif // DGIMAGE_H