TiffExporter.cpp

#include "TiffExporter.h"

#include <QDebug>

#include "Buffer.h"
#include "FileName.h"
#include "IException.h"
#include "IString.h"

using namespace Isis;


namespace Isis {
  TiffExporter::TiffExporter() : StreamExporter() {
    m_image = NULL;
    m_raster = NULL;

    setExtension("tif");
  }


  TiffExporter::~TiffExporter() {
    if (m_image) {
      TIFFClose(m_image);
      m_image = NULL;
    }

    delete [] m_raster;
    m_raster = NULL;
  }


  void TiffExporter::createBuffer() {
    PixelType type = pixelType();
    int mult = (type == Isis::UnsignedByte) ? 1 : 2;
    int size = samples() * bands() * mult;

    try {
      m_raster = new unsigned char[size];
    }
    catch (...) {
      throw IException(IException::Unknown,
          "Could not allocate enough memory", _FILEINFO_);
    }
  }


  void TiffExporter::write(FileName outputName, int quality,
                           QString compression) {
    // Open the output image
    m_image = TIFFOpen(outputName.expanded().toAscii().data(), "w");

    if (m_image == NULL) {
      throw IException(IException::Programmer,
          "Could not open output image", _FILEINFO_);
    }

    TIFFSetField(m_image, TIFFTAG_IMAGEWIDTH, samples());
    TIFFSetField(m_image, TIFFTAG_IMAGELENGTH, lines());
    TIFFSetField(m_image, TIFFTAG_ROWSPERSTRIP, 1);
    if (compression == "packbits") {
      TIFFSetField(m_image, TIFFTAG_COMPRESSION, COMPRESSION_PACKBITS);
    }
    else if (compression == "lzw") {
      TIFFSetField(m_image, TIFFTAG_COMPRESSION, COMPRESSION_LZW);
    }
    else if (compression == "deflate") {
      TIFFSetField(m_image, TIFFTAG_COMPRESSION, COMPRESSION_ADOBE_DEFLATE);
    }
    else if (compression == "none") {
      TIFFSetField(m_image, TIFFTAG_COMPRESSION, COMPRESSION_NONE);
    }
    else {
      QString msg = "Invalid TIFF compression algorithm: " + compression;
      throw IException(IException::Programmer, msg, _FILEINFO_);
    }
    TIFFSetField(m_image, TIFFTAG_PLANARCONFIG, PLANARCONFIG_CONTIG);
    TIFFSetField(m_image, TIFFTAG_PHOTOMETRIC,
        bands() == 1 ? PHOTOMETRIC_MINISBLACK : PHOTOMETRIC_RGB);

    PixelType type = pixelType();
    int bps = (type == Isis::UnsignedByte) ? 8 : 16;
    TIFFSetField(m_image, TIFFTAG_BITSPERSAMPLE, bps);

    TIFFSetField(m_image, TIFFTAG_SAMPLESPERPIXEL, bands());

    ImageExporter::write(outputName, quality);
  }


  void TiffExporter::setBuffer(int s, int b, int dn) const {
    PixelType type = pixelType();
    int index = s * bands() + b;
    
    switch (type) {
      case UnsignedByte:
        m_raster[index] = (unsigned char) dn;
        break;
      case SignedWord:
        ((short int *) m_raster)[index] = (short int) dn;
        break;
      case UnsignedWord:
        ((short unsigned int *) m_raster)[index] = (short unsigned int) dn;
        break;
      default:
        throw IException(IException::Programmer,
            "Invalid pixel type for data [" + toString(type) + "]",
            _FILEINFO_);
    }
  }


  void TiffExporter::writeLine(int l) const {
    if (!TIFFWriteScanline(m_image, m_raster, l)) {
      throw IException(IException::Programmer,
          "Could not write image", _FILEINFO_);
    }
  }


  bool TiffExporter::canWriteFormat(QString format) {
    return format == "tiff";
  }
};