#include <ppmreader.h>
#include <iostream>
#include <fstream>
#include <cstring>

using namespace std;

PPMReader::PPMReader(char * filename) {
	loadPPMFile(filename);
}

void PPMReader::loadPPMFile(char * filename) {

	const int readBufferSize = 20;
	ifstream ifs;
	char buffer[readBufferSize];
	ifs.open(filename, ifstream::in);

	ifs.getline(buffer, readBufferSize);
	ifs.getline(buffer, readBufferSize, ' ');
	width = atoi(buffer);
	ifs.getline(buffer, readBufferSize);
	height = atoi(buffer);
	ifs.getline(buffer, readBufferSize);
	unsigned int bitDepth = atoi(buffer);
	bytesPerPix = (bitDepth == 255) ? 3 : 6;

	unsigned int imagePixs = width * height;
	unsigned int imageBytes = imagePixs * bytesPerPix;
	image = new char[imageBytes];
	unsigned int bytesPerLn = width * bytesPerPix;
	for (int i = 0; i < int(height); i++) {	
		char* tmp = new char[bytesPerLn];
		ifs.getline(tmp, bytesPerLn);
		strcat(image, tmp);
	}
	
	ifs.close();

}

unsigned int PPMReader::getBytesPerPix() {
	return bytesPerPix;
}

#ifndef PPMREADER_H
#define PPMREADER_H

#include <fstream>

using namespace std;

class PPMReader {

private:

	char * image;
	unsigned int width;
	unsigned int height;
	unsigned int bytesPerPix;

public:

	PPMReader(char * filename);

	inline void getImage(char * &dest) { dest = image; };

	inline unsigned int getWidth() { return width; };

	inline unsigned int getHeight() { return height; };

	void loadPPMFile(char * filename);

	unsigned int getBytesPerPix();

};

#endif