#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));

  }

}