#include "FloatPlane.h"

#include <cstring>

extern "C" {

#include <stdio.h>

#include <pam.h>

}

FloatPlane::FloatPlane()

{

    setMaxMinValue();

    ready = false;

}

void FloatPlane::setMaxMinValue(unsigned int max, unsigned int min)

{

	MAXVALUE=max;

	MINVALUE=min;

}

FloatPlane::FloatPlane(float *_ptr)

{

	ptr = _ptr;

    setMaxMinValue();

    ready = true;

}

FloatPlane::FloatPlane(const unsigned int _width, const unsigned int _height)

{

    allocate(_width, _height);

    setMaxMinValue();

    ready = true;

}

void FloatPlane::allocate(const unsigned int _width, const unsigned int _height)

{

    width = _width;

    height = _height;

	ptr = new float[width*height];

    ready = true;

}

void FloatPlane::writePGM(const char * filename)

{

    struct pam outpam;

    unsigned int row;

    tuple *tuplerow;

    FILE* file;

    if (!ready)

        return;

    file = fopen(filename, "wb");

    if (file == NULL)

        return;

    outpam.size = sizeof(outpam);

    outpam.len = outpam.size;

    outpam.file = file;

    outpam.format = PGM_FORMAT;

    outpam.plainformat = 0; /* false */

    outpam.width = width;

    outpam.height = height;

    outpam.depth = 1; /* Grayscale - 1 sample per tuple */

    outpam.allocation_depth = 0; /* Same as depth. */

    outpam.maxval = MAXVALUE;

    std::strcpy(outpam.tuple_type, "GRAYSCALE"); /* PGM non transparent */

    /* needs size,len,file,format,height,width,depth,maxval, tuple_type */

    pnm_writepaminit(&outpam);

    tuplerow = pnm_allocpamrow(&outpam);

    for (row = 0; row < height; row++) {

        unsigned int column;

        for (column = 0; column < width; ++column) {

            /* Only one plane */

            /* float to int! */

            tuplerow[column][0] = ptr[column+width*row];

        }

        pnm_writepamrow(&outpam, tuplerow);

    }

    pnm_freepamrow(tuplerow);

    fclose(file);

}

unsigned int FloatPlane::getMaxValue()

{

    return MAXVALUE;

}

unsigned int FloatPlane::getWidth()

{

    return width;

}

unsigned int FloatPlane::getHeight()

{

    return height;

}

float * FloatPlane::getPtr()

{

    return ptr;

}

void FloatPlane::free()

{

    delete[] ptr;

}