#include "texception.h"
#include "tfilepath.h"
#include "tiio_bmp.h"
#include "filebmp.h"

namespace
{

string buildBMPExceptionString(int retCode)
{
	switch (retCode) {
	case UNSUPPORTED_BMP_FORMAT:
		return " : unsupported bmp format";
	case OUT_OF_MEMORY:
		return " : out of memory";
	case UNEXPECTED_EOF:
		return " : unexptected EOF";
	case CANT_OPEN_FILE:
		return " : can't open file";
	case WRITE_ERROR:
		return " : writing error";
	default:
		return " : I/O error";
	}
}

} //namespace
//------------------------------------------------------------

TImageP TImageReaderBmp::load()
{
	TImageP image;
	void *buff;
	int retCode = readbmp(getFilePath().getWideString().c_str(), &m_lx, &m_ly, &buff);
	if (retCode != OK) {
		throw TImageException(getFilePath(), buildBMPExceptionString(retCode));
	}

	TRaster32P raster;
	raster.create(m_lx, m_ly);
	raster->lock();
	memcpy(raster->pixels(), buff, m_lx * m_ly * sizeof(TPixel32));
	raster->unlock();
	TRasterImageP rasImage(raster);
	// image->setRaster(raster);
	delete[] buff;
	return TImageP(rasImage);
}

//------------------------------------------------------------

/*
void TImageReaderBmp::load(const TRasterP &ras, const TPoint &pos,
                              int shrinkX, int shrinkY)
{
void *buff;
int x0 = pos.x;
int y0 = pos.y;
int x1 = x0 + ras->getLx() * shrinkX - 1;
int y1 = y0 + ras->getLy() * shrinkY - 1;

int retCode = readbmpregion(m_path.getFullPath().c_str(), &buff, x0, y0, x1, y1, shrinkX);
if (retCode!= OK) 
  throw TImageException(getFilePath().getFullPath(), buildBMPExceptionString(retCode));
memcpy(ras->getRawData(), buff, ras->getLx() * ras->getLy() * 4);
delete [] buff;
}
*/

//------------------------------------------------------------

TDimension TImageReaderBmp::getSize() const
{
	TDimension d(0, 0);
	int retCode = readbmp_size(getFilePath().getWideString().c_str(), &(d.lx), &(d.ly));
	if (retCode != OK)
		throw TImageException(getFilePath(), buildBMPExceptionString(retCode));
	return d;
}

//------------------------------------------------------------

TRect TImageReaderBmp::getBBox() const
{
	TRect r;
	int retCode = readbmp_bbox(getFilePath().getWideString().c_str(), &(r.x0), &(r.y0), &(r.x1), &(r.y1));
	if (retCode != OK)
		throw TImageException(getFilePath(), buildBMPExceptionString(retCode));
	return r;
}

//============================================================

void TImageWriterBmp::save(const TImageP &image)
{
	TRasterImageP rasImage(image);

	TRaster32P ras32;
	ras32 = rasImage->getRaster();

	TRasterGR8P ras8;
	ras8 = rasImage->getRaster();
	void *buff = 0;
	int pixSize;

	if (!ras32 && !ras8))
    throw TImageException(m_path, buildBMPExceptionString(UNSUPPORTED_BMP_FORMAT));

	rasImage->getRaster()->lock();
	if (ras32) {
		buff = ras32->getRawData();
		m_lx = ras32->getLx();
		m_ly = ras32->getLy();
		pixSize = 32;
	} else {
		buff = ras8->getRawData();
		m_lx = ras8->getLx();
		m_ly = ras8->getLy();
		pixSize = 8;
	}

	int retCode = writebmp(getFilePath().getWideString().c_str(), m_lx, m_ly, buff, pixSize);
	rasImage->getRaster()->unlock();
	if (retCode != OK) {
		throw TImageException(m_path, buildBMPExceptionString(retCode));
	}
}