Logo Search packages:      
Sourcecode: qtpfsgui version File versions

pfstiff.cpp

/**
 * @brief Tiff facilities
 * 
 * This file is a part of Qtpfsgui package.
 * ----------------------------------------------------------------------
 * Copyright (C) 2003,2004 Rafal Mantiuk and Grzegorz Krawczyk
 * Copyright (C) 2006 Giuseppe Rota
 * 
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software
 *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 * ----------------------------------------------------------------------
 *
 * @author Grzegorz Krawczyk, <krawczyk@mpi-sb.mpg.de>
 * slightly modified by Giuseppe Rota <grota@sourceforge.net> for qtpfsgui
 */

#include <cmath>
#include <QSysInfo>

#include "pfstiff.h"

00033 TiffReader::TiffReader( const char* filename ) {
  // read header containing width and height from file
  tif = TIFFOpen(filename, "r");
  if( !tif )
    throw pfs::Exception("TIFF: could not open file for reading.");

  //--- image size
  TIFFGetField(tif, TIFFTAG_IMAGEWIDTH, &width);
  TIFFGetField(tif, TIFFTAG_IMAGELENGTH, &height);

  if( width*height<=0 )
  {
    TIFFClose(tif);
    throw pfs::Exception("TIFF: illegal image size");
  }

  //--- image parameters
  if(!TIFFGetField(tif, TIFFTAG_COMPRESSION, &comp)) // compression type
    comp = COMPRESSION_NONE;
  qDebug("compression=%d",comp);

  // type of photometric data
  if(!TIFFGetFieldDefaulted(tif, TIFFTAG_PHOTOMETRIC, &phot))
    throw pfs::Exception("TIFF: unspecified photometric type");

  uint16 * extra_sample_types=0;
  uint16 extra_samples_per_pixel=0;
  switch(phot)
  {
    case PHOTOMETRIC_LOGLUV:
      qDebug("Photometric data: LogLuv");
      if (comp != COMPRESSION_SGILOG && comp != COMPRESSION_SGILOG24) {
      TIFFClose(tif);
      throw pfs::Exception("TIFF: only support SGILOG compressed LogLuv data");
      }
      TIFFGetField(tif, TIFFTAG_SAMPLESPERPIXEL, &nSamples);
      TIFFSetField(tif, TIFFTAG_SGILOGDATAFMT, SGILOGDATAFMT_FLOAT);
      TypeOfData = FLOATLOGLUV;
      break;
    case PHOTOMETRIC_RGB:
      qDebug("Photometric data: RGB");
     // read extra samples (# of alpha channels)
      if (TIFFGetField( tif, TIFFTAG_EXTRASAMPLES,
                        &extra_samples_per_pixel, &extra_sample_types )!=1)
      {
          extra_samples_per_pixel=0;
      }
      TIFFGetField(tif, TIFFTAG_SAMPLESPERPIXEL, &nSamples);
      bps = nSamples - extra_samples_per_pixel;
      if (bps!=3)
      {
      TIFFClose(tif);
      throw pfs::Exception("TIFF: unsupported samples per pixel for RGB");
      }
      if (!TIFFGetField(tif, TIFFTAG_BITSPERSAMPLE, &bps) || (bps!=8 && bps!=16 && bps!=32))
      {
      TIFFClose(tif);
      throw pfs::Exception("TIFF: unsupported bits per sample for RGB");
      }

      if( bps==8 )
      {
      TypeOfData = BYTE;
      qDebug("8bit per channel");
      }
      else if( bps==16 )
      {
      TypeOfData = WORD;
      qDebug("16bit per channel");
      }
      else
      {
      TypeOfData = FLOAT;
      qDebug("32bit float per channel");
      }
      break;
    default:
      qFatal("Unsupported photometric type: %d",phot);
      TIFFClose(tif);
      throw pfs::Exception("TIFF: unsupported photometric type");
  }

  if (!TIFFGetField(tif, TIFFTAG_STONITS, &stonits))
    stonits = 1.;
}

pfs::Frame* TiffReader::readIntoPfsFrame() {
  //--- scanline buffer with pointers to different data types
  union {
    float* fp;
    uint16* wp;
    uint8* bp;
    void* vp;
  } buf;

  pfs::DOMIO pfsio;
  pfs::Frame *frame = pfsio.createFrame( width, height );
  pfs::Channel *X, *Y, *Z;
  frame->createRGBChannels( X, Y, Z );

  //--- image length
  uint32 imagelength;
  TIFFGetField(tif, TIFFTAG_IMAGELENGTH, &imagelength);

  //--- image scanline size
  uint32 scanlinesize = TIFFScanlineSize(tif);
  buf.vp = _TIFFmalloc(scanlinesize);

  //--- read scan lines
  const int image_width = X->getCols();
  for(uint32 row = 0; row < imagelength; row++)
  {
    switch(TypeOfData)
    {
      case FLOAT:
      case FLOATLOGLUV:
      TIFFReadScanline(tif, buf.fp, row);
      for( int i=0; i < image_width; i++ )
      {
        (*X)(i,row) = buf.fp[i*nSamples];
        (*Y)(i,row) = buf.fp[i*nSamples+1];
        (*Z)(i,row) = buf.fp[i*nSamples+2];
      }
      break;
      case WORD:
      TIFFReadScanline(tif, buf.wp, row);
      for( int i=0; i<image_width; i++ )
      {
        (*X)(i,row) = buf.wp[i*nSamples];
        (*Y)(i,row) = buf.wp[i*nSamples+1];
        (*Z)(i,row) = buf.wp[i*nSamples+2];
      }
      break;
      case BYTE:
      TIFFReadScanline(tif, buf.bp, row);
      for( int i=0; i<image_width; i++ )
      {
        (*X)(i,row) = pow( buf.bp[i*nSamples]/255.0, 2.2 );
        (*Y)(i,row) = pow( buf.bp[i*nSamples+1]/255.0, 2.2 );
        (*Z)(i,row) = pow( buf.bp[i*nSamples+2]/255.0, 2.2 );
      }
      break;
    }
  }

  //--- free buffers and close files
  _TIFFfree(buf.vp);
  TIFFClose(tif);
  if (TypeOfData==FLOATLOGLUV)
    pfs::transformColorSpace( pfs::CS_XYZ, X,Y,Z, pfs::CS_RGB, X,Y,Z );
  return frame;
}

//given for granted that users of this function call it only after checking that TypeOfData==BYTE
QImage* TiffReader::readIntoQImage() {
      uchar *data=new uchar[width*height*4]; //this will contain the image data: data must be 32-bit aligned, in Format: 0xffRRGGBB
//    qDebug("pfstiff, data=%ld",data);
      assert(TypeOfData==BYTE);

      //--- image length
      uint32 imagelength;
      TIFFGetField(tif, TIFFTAG_IMAGELENGTH, &imagelength);

      //--- image scanline size
      uint32 scanlinesize = TIFFScanlineSize(tif);
      uint8* bp = (uint8 *)_TIFFmalloc(scanlinesize);

      //--- read scan lines
      for(uint y = 0; y < height; y++) {
            TIFFReadScanline(tif, bp, y);
            for( uint x=0; x<width; x++ ) {
                  if (QSysInfo::ByteOrder==QSysInfo::LittleEndian) {
                        *(data + 0 + (y*width+x)*4) = bp[x*nSamples+2] ;
                        *(data + 1 + (y*width+x)*4) = bp[x*nSamples+1] ;
                        *(data + 2 + (y*width+x)*4) = bp[x*nSamples] ;
                        *(data + 3 + (y*width+x)*4) = bp[x*nSamples+3];
                  } else {
                        *(data + 3 + (y*width+x)*4) = bp[x*nSamples+2];
                        *(data + 2 + (y*width+x)*4) = bp[x*nSamples+1];
                        *(data + 1 + (y*width+x)*4) = bp[x*nSamples];
                        *(data + 0 + (y*width+x)*4) = bp[x*nSamples+3];
                  }
            }
      }
      //--- free buffers and close files
      _TIFFfree(bp);
      TIFFClose(tif);

//special treament for qt 4.2.1... removing "const" doesn't seem to work.
#if QT_VERSION == 0x040201
      QImage *toreturn=new QImage(const_cast<const uchar *>(data), width, height, QImage::Format_ARGB32);
#elif QT_VERSION <= 0x040200
      QImage *toreturn=new QImage(data,width,height,QImage::Format_ARGB32);
#else
      QImage *toreturn=new QImage(const_cast<uchar *>(data),width,height,QImage::Format_ARGB32);
#endif
      return toreturn;
}

TiffWriter::TiffWriter( const char* filename, pfs::Frame *f ) : tif((TIFF *)NULL) {
      f->getRGBChannels(R,G,B);
      width=R->getWidth();
      height=R->getHeight();
//    qDebug("width=%d, heigh=%d",width,height);
      tif = TIFFOpen(filename, "w");
      if( !tif )
            throw pfs::Exception("TIFF: could not open file for reading.");

      TIFFSetField (tif, TIFFTAG_IMAGEWIDTH, width);
      TIFFSetField (tif, TIFFTAG_IMAGELENGTH, height);
      TIFFSetField (tif, TIFFTAG_PLANARCONFIG, PLANARCONFIG_CONTIG);
      TIFFSetField (tif, TIFFTAG_SAMPLESPERPIXEL, 3);
      TIFFSetField (tif, TIFFTAG_ROWSPERSTRIP, 1);
}

int TiffWriter::writeFloatTiff() { //write 32 bit float Tiff from pfs::Frame
      TIFFSetField (tif, TIFFTAG_COMPRESSION, COMPRESSION_DEFLATE); // TODO what about others?
      TIFFSetField (tif, TIFFTAG_PHOTOMETRIC, PHOTOMETRIC_RGB);
      TIFFSetField (tif, TIFFTAG_BITSPERSAMPLE, 32);

      tsize_t strip_size = TIFFStripSize (tif);
      tstrip_t strips_num = TIFFNumberOfStrips (tif);
      float* strip_buf=(float*)_TIFFmalloc(strip_size); //enough space for a strip
      if (!strip_buf)
            throw pfs::Exception("TIFF: error allocating buffer.");

      for (unsigned int s=0; s<strips_num; s++) {
            for (unsigned int col=0; col<width; col++) {
                  strip_buf[3*col+0]=(*R)(col,s);
                  strip_buf[3*col+1]=(*G)(col,s);
                  strip_buf[3*col+2]=(*B)(col,s);
            }
            if (TIFFWriteEncodedStrip (tif, s, strip_buf, strip_size) == 0) {
                  qDebug("error writing strip");
                  return -1;
            }
      }
      _TIFFfree(strip_buf);
      TIFFClose(tif);
      return 0;
}

int TiffWriter::writeLogLuvTiff() { //write LogLUv Tiff from pfs::Frame
      TIFFSetField(tif, TIFFTAG_COMPRESSION, COMPRESSION_SGILOG);
      TIFFSetField(tif, TIFFTAG_PHOTOMETRIC, PHOTOMETRIC_LOGLUV);
      TIFFSetField(tif, TIFFTAG_SGILOGDATAFMT, SGILOGDATAFMT_FLOAT);
      TIFFSetField(tif, TIFFTAG_STONITS, 1.);   /* not known */

      tsize_t strip_size = TIFFStripSize (tif);
      tstrip_t strips_num = TIFFNumberOfStrips (tif);
      float* strip_buf=(float*)_TIFFmalloc(strip_size); //enough space for a strip
      if (!strip_buf)
            throw pfs::Exception("TIFF: error allocating buffer.");

      pfs::transformColorSpace( pfs::CS_RGB, R,G,B, pfs::CS_XYZ, R,G,B );
      for (unsigned int s=0; s<strips_num; s++) {
            for (unsigned int col=0; col<width; col++) {
                  strip_buf[3*col+0]=(*R)(col,s);
                  strip_buf[3*col+1]=(*G)(col,s);
                  strip_buf[3*col+2]=(*B)(col,s);
            }
            if (TIFFWriteEncodedStrip (tif, s, strip_buf, strip_size) == 0) {
                  qDebug("error writing strip");
                  return -1;
            }
      }
      pfs::transformColorSpace( pfs::CS_XYZ, R,G,B, pfs::CS_RGB, R,G,B );

      _TIFFfree(strip_buf);
      TIFFClose(tif);
      return 0;
}

Generated by  Doxygen 1.6.0   Back to index