/* Implementation of SpriteLoader by Mohammed Khoory */

 #include "sprload.h"

 SpriteLoader::SpriteLoader()

 {

 	resetHeader();

 }

 SpriteLoader::~SpriteLoader()

 {

 	if(rawBitmapFile.is_open())

 		rawBitmapFile.close();

 	if(paletteFile.is_open())

 		paletteFile.close();

 }

 void SpriteLoader::resetHeader()

 {

 	width = 0;

 	height = 0;

 	size = 0;

 }

 // string filename must not have an extension!

 void SpriteLoader::loadFile(const string& filename) throw(int, ios_base::failure)

 {

 	if(rawBitmapFile.is_open() || paletteFile.is_open())

 		this->unloadFile();

 	// open rawBitmapFile and paletteFile (don't do anything)

 	string rawFileName = filename + ".spr";

 	string paletteFileName = filename + ".pal";

 	rawBitmapFile.open(rawFileName.c_str(), ifstream::in);

 	if(!rawBitmapFile.is_open())

 		throw BadFileNameException;

 	paletteFile.open(paletteFileName.c_str(), ifstream::in);

 	if(!paletteFile.is_open())

 		throw BadFileNameException;

 	rawBitmapFile.exceptions(ifstream::eofbit | ifstream::failbit | ifstream::badbit);

 	paletteFile.exceptions(ifstream::eofbit | ifstream::failbit | ifstream::badbit);

 	// open and read the header 

 	ifstream headerFile;

 	string headerFileName = filename + ".hdr";

 	headerFile.open(headerFileName.c_str(), ifstream::in );

 	if(!headerFile.is_open())

 		throw BadFileNameException;

 	headerFile.exceptions(ifstream::eofbit | ifstream::failbit | ifstream::badbit);

 	try

 	{

 		string widthString, heightString, sizeString;

 		getline(headerFile, widthString);

 		getline(headerFile, heightString);

 		getline(headerFile, sizeString);

 		width = atol(widthString.c_str());

 		height = atol(heightString.c_str());

 		size = atol(sizeString.c_str());

 		headerFile.close();

 	}

 	catch(ios_base::failure& e)

 	{

 		resetHeader();

 		bool eof = headerFile.eof();

 		headerFile.close();

 		rawBitmapFile.close();

 		paletteFile.close();

 		if(eof)

 			throw BadFileException;

 		else

 			throw e;

 	}

 }

 void SpriteLoader::unloadFile()

 {

 	if(paletteFile.is_open())

 		paletteFile.close();

 	if(rawBitmapFile.is_open())

 		rawBitmapFile.close();

 	resetHeader();

 }

 unsigned char * SpriteLoader::getBitmap() throw(int, bad_alloc, ios_base::failure)

 {

 	if(!rawBitmapFile.is_open() || !paletteFile.is_open())

 		throw FileNotLoadedException;

 	unsigned char * bitmapData = new unsigned char[size];

 	try

 	{

 		rawBitmapFile.read((char *)bitmapData, size);

 	}

 	catch(ios_base::failure e)

 	{

 		delete[] bitmapData;

 		if(rawBitmapFile.eof())

 			throw BadFileException;

 		else

 			throw e;

 	}	

 	return bitmapData;

 }

 struct sprite * SpriteLoader::getSprite(int x1, int y1, int x2, int y2) throw(int, bad_alloc, ios_base::failure)

 {

 	if(!paletteFile.is_open() || !rawBitmapFile.is_open())

 		throw FileNotLoadedException;

 	// if x1 or y1 is negative set to 0, if x2 or y2 is negative set to the last index in row/column

 	x1 = x1 < 0 ? 0 : x1;

 	y1 = y1 < 0 ? 0 : y1;

 	x2 = x2 < 0 ? width-1 : x2;

 	y2 = y2 < 0 ? height-1 : y2;

 	// check if anything is out of bounds, set it to largest size

 	x1 = x1 > width-1 ? width-1 : x1;

 	x2 = x2 > width-1 ? width-1 : x2;

 	y1 = y1 > height-1 ? height-1 : y1;

 	y2 = y2 > height-1 ? height-1 : y2;

 	// if x1 > x2, switch, if y1 > y2, switch

 	if( x1 > x2)

 	{

 		int temp = x1;

 		x1 = x2;

 		x2 = temp;

 	}

 	if( y1 > y2)

 	{

 		int temp = y1;

 		y1 = y2;

 		y2 = temp;

 	}

 	struct sprite * theSprite = new struct sprite;

 	theSprite->width = x2 - x1 +1;

 	theSprite->height = y2 - y1 +1;

 	theSprite->bitmap = new unsigned char[theSprite->width * theSprite->height];

 	// begin reading

 	try

 	{

 		// go to initial offset (at (x1,y1))

 		unsigned long int initialOffset = x1 + width*y1;

 		for(int i = 0; i < y2 - y1; i++)

 		{

 			// i is considered to be the current "row" we're reading

 			rawBitmapFile.seekg(initialOffset + (i *width));

 			rawBitmapFile.read((char*)theSprite->bitmap + (i*theSprite->width), theSprite->width);

 		}

 	}

 	catch (ios_base::failure& e)

 	{

 		if(theSprite != NULL)

 		{

 			if(theSprite->bitmap != NULL)

 				delete[] theSprite->bitmap;

 			delete theSprite;

 		}

 		if(rawBitmapFile.eof())

 			throw BadFileException;

 		else

 			throw e;

 	}

 	return theSprite;

 }

 void SpriteLoader::setPalette(GraphicsDevice& graphics) throw (int, bad_alloc, ios_base::failure)

 {

 	if(!paletteFile.is_open())

 		throw FileNotLoadedException;

 	char * palette = new char[768];

 	paletteFile.seekg(0);

 	paletteFile.read(palette, 768);

 	graphics.setPalette(palette, 768);

 	delete[] palette;

 }

 unsigned long int SpriteLoader::getWidth() throw(int)

 {

 	if(!paletteFile.is_open() && !rawBitmapFile.is_open())

 		throw FileNotLoadedException;

 	return width;

 }

 unsigned long int SpriteLoader::getHeight() throw(int)

 {

 	if(!paletteFile.is_open() && !rawBitmapFile.is_open())

 		throw FileNotLoadedException;	

 	return height;

 }

 unsigned long int SpriteLoader::getSize() throw(int)

 {

 	if(!paletteFile.is_open() && !rawBitmapFile.is_open())

 		throw FileNotLoadedException;	

 	return size;

 }

 bool SpriteLoader::isLoaded()

 {

 	return paletteFile.is_open() && rawBitmapFile.is_open();

 }