/3rdparty/openexr/IlmImf/ImfInputFile.cpp
C++ | 648 lines | 420 code | 126 blank | 102 comment | 44 complexity | 230410b935a5fa1445906123e9be8caf MD5 | raw file
- ///////////////////////////////////////////////////////////////////////////
- //
- // Copyright (c) 2004, Industrial Light & Magic, a division of Lucas
- // Digital Ltd. LLC
- //
- // All rights reserved.
- //
- // Redistribution and use in source and binary forms, with or without
- // modification, are permitted provided that the following conditions are
- // met:
- // * Redistributions of source code must retain the above copyright
- // notice, this list of conditions and the following disclaimer.
- // * Redistributions in binary form must reproduce the above
- // copyright notice, this list of conditions and the following disclaimer
- // in the documentation and/or other materials provided with the
- // distribution.
- // * Neither the name of Industrial Light & Magic nor the names of
- // its contributors may be used to endorse or promote products derived
- // from this software without specific prior written permission.
- //
- // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
- // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
- // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
- // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
- // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
- // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
- // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
- // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
- // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
- // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
- // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
- //
- ///////////////////////////////////////////////////////////////////////////
- //-----------------------------------------------------------------------------
- //
- // class InputFile
- //
- //-----------------------------------------------------------------------------
- #include <ImfInputFile.h>
- #include <ImfScanLineInputFile.h>
- #include <ImfTiledInputFile.h>
- #include <ImfChannelList.h>
- #include <ImfMisc.h>
- #include <ImfStdIO.h>
- #include <ImfVersion.h>
- #include "ImathFun.h"
- #include "IlmThreadMutex.h"
- #include "Iex.h"
- #include "half.h"
- #include <fstream>
- #include <algorithm>
- namespace Imf {
- using Imath::Box2i;
- using Imath::divp;
- using Imath::modp;
- using IlmThread::Mutex;
- using IlmThread::Lock;
- //
- // Struct InputFile::Data stores things that will be
- // needed between calls to readPixels
- //
- struct InputFile::Data: public Mutex
- {
- Header header;
- int version;
- IStream * is;
- bool deleteStream;
- TiledInputFile * tFile;
- ScanLineInputFile * sFile;
- LineOrder lineOrder; // the file's lineorder
- int minY; // data window's min y coord
- int maxY; // data window's max x coord
- FrameBuffer tFileBuffer;
- FrameBuffer * cachedBuffer;
- int cachedTileY;
- int offset;
- int numThreads;
- Data (bool del, int numThreads);
- ~Data ();
- void deleteCachedBuffer();
- };
- InputFile::Data::Data (bool del, int numThreads):
- is (0),
- deleteStream (del),
- tFile (0),
- sFile (0),
- cachedBuffer (0),
- cachedTileY (-1),
- numThreads (numThreads)
- {
- // empty
- }
- InputFile::Data::~Data ()
- {
- delete tFile;
- delete sFile;
- if (deleteStream)
- delete is;
- deleteCachedBuffer();
- }
- void
- InputFile::Data::deleteCachedBuffer()
- {
- //
- // Delete the cached frame buffer, and all memory
- // allocated for the slices in the cached frameBuffer.
- //
- if (cachedBuffer)
- {
- for (FrameBuffer::Iterator k = cachedBuffer->begin();
- k != cachedBuffer->end();
- ++k)
- {
- Slice &s = k.slice();
- switch (s.type)
- {
- case UINT:
- delete [] (((unsigned int *)s.base) + offset);
- break;
- case HALF:
- delete [] ((half *)s.base + offset);
- break;
- case FLOAT:
- delete [] (((float *)s.base) + offset);
- break;
- }
- }
- //
- // delete the cached frame buffer
- //
- delete cachedBuffer;
- cachedBuffer = 0;
- }
- }
- namespace {
- void
- bufferedReadPixels (InputFile::Data* ifd, int scanLine1, int scanLine2)
- {
- //
- // bufferedReadPixels reads each row of tiles that intersect the
- // scan-line range (scanLine1 to scanLine2). The previous row of
- // tiles is cached in order to prevent redundent tile reads when
- // accessing scanlines sequentially.
- //
- int minY = std::min (scanLine1, scanLine2);
- int maxY = std::max (scanLine1, scanLine2);
- if (minY < ifd->minY || maxY > ifd->maxY)
- {
- throw Iex::ArgExc ("Tried to read scan line outside "
- "the image file's data window.");
- }
- //
- // The minimum and maximum y tile coordinates that intersect this
- // scanline range
- //
- int minDy = (minY - ifd->minY) / ifd->tFile->tileYSize();
- int maxDy = (maxY - ifd->minY) / ifd->tFile->tileYSize();
- //
- // Figure out which one is first in the file so we can read without seeking
- //
- int yStart, yEnd, yStep;
- if (ifd->lineOrder == DECREASING_Y)
- {
- yStart = maxDy;
- yEnd = minDy - 1;
- yStep = -1;
- }
- else
- {
- yStart = minDy;
- yEnd = maxDy + 1;
- yStep = 1;
- }
- //
- // the number of pixels in a row of tiles
- //
- Box2i levelRange = ifd->tFile->dataWindowForLevel(0);
- //
- // Read the tiles into our temporary framebuffer and copy them into
- // the user's buffer
- //
- for (int j = yStart; j != yEnd; j += yStep)
- {
- Box2i tileRange = ifd->tFile->dataWindowForTile (0, j, 0);
- int minYThisRow = std::max (minY, tileRange.min.y);
- int maxYThisRow = std::min (maxY, tileRange.max.y);
- if (j != ifd->cachedTileY)
- {
- //
- // We don't have any valid buffered info, so we need to read in
- // from the file.
- //
- ifd->tFile->readTiles (0, ifd->tFile->numXTiles (0) - 1, j, j);
- ifd->cachedTileY = j;
- }
- //
- // Copy the data from our cached framebuffer into the user's
- // framebuffer.
- //
- for (FrameBuffer::ConstIterator k = ifd->cachedBuffer->begin();
- k != ifd->cachedBuffer->end();
- ++k)
- {
- Slice fromSlice = k.slice(); // slice to write from
- Slice toSlice = ifd->tFileBuffer[k.name()]; // slice to write to
- char *fromPtr, *toPtr;
- int size = pixelTypeSize (toSlice.type);
- int xStart = levelRange.min.x;
- int yStart = minYThisRow;
- while (modp (xStart, toSlice.xSampling) != 0)
- ++xStart;
- while (modp (yStart, toSlice.ySampling) != 0)
- ++yStart;
- for (int y = yStart;
- y <= maxYThisRow;
- y += toSlice.ySampling)
- {
- //
- // Set the pointers to the start of the y scanline in
- // this row of tiles
- //
- fromPtr = fromSlice.base +
- (y - tileRange.min.y) * fromSlice.yStride +
- xStart * fromSlice.xStride;
- toPtr = toSlice.base +
- divp (y, toSlice.ySampling) * toSlice.yStride +
- divp (xStart, toSlice.xSampling) * toSlice.xStride;
- //
- // Copy all pixels for the scanline in this row of tiles
- //
- for (int x = xStart;
- x <= levelRange.max.x;
- x += toSlice.xSampling)
- {
- for (size_t i = 0; i < size; ++i)
- toPtr[i] = fromPtr[i];
- fromPtr += fromSlice.xStride * toSlice.xSampling;
- toPtr += toSlice.xStride;
- }
- }
- }
- }
- }
- } // namespace
- InputFile::InputFile (const char fileName[], int numThreads):
- _data (new Data (true, numThreads))
- {
- try
- {
- _data->is = new StdIFStream (fileName);
- initialize();
- }
- catch (Iex::BaseExc &e)
- {
- delete _data;
- REPLACE_EXC (e, "Cannot read image file "
- "\"" << fileName << "\". " << e);
- throw;
- }
- catch (...)
- {
- delete _data;
- throw;
- }
- }
- InputFile::InputFile (IStream &is, int numThreads):
- _data (new Data (false, numThreads))
- {
- try
- {
- _data->is = &is;
- initialize();
- }
- catch (Iex::BaseExc &e)
- {
- delete _data;
- REPLACE_EXC (e, "Cannot read image file "
- "\"" << is.fileName() << "\". " << e);
- throw;
- }
- catch (...)
- {
- delete _data;
- throw;
- }
- }
- void
- InputFile::initialize ()
- {
- _data->header.readFrom (*_data->is, _data->version);
- _data->header.sanityCheck (isTiled (_data->version));
- if (isTiled (_data->version))
- {
- _data->lineOrder = _data->header.lineOrder();
- //
- // Save the dataWindow information
- //
- const Box2i &dataWindow = _data->header.dataWindow();
- _data->minY = dataWindow.min.y;
- _data->maxY = dataWindow.max.y;
- _data->tFile = new TiledInputFile (_data->header,
- _data->is,
- _data->version,
- _data->numThreads);
- }
- else
- {
- _data->sFile = new ScanLineInputFile (_data->header,
- _data->is,
- _data->numThreads);
- }
- }
- InputFile::~InputFile ()
- {
- delete _data;
- }
- const char *
- InputFile::fileName () const
- {
- return _data->is->fileName();
- }
- const Header &
- InputFile::header () const
- {
- return _data->header;
- }
- int
- InputFile::version () const
- {
- return _data->version;
- }
- void
- InputFile::setFrameBuffer (const FrameBuffer &frameBuffer)
- {
- if (isTiled (_data->version))
- {
- Lock lock (*_data);
- //
- // We must invalidate the cached buffer if the new frame
- // buffer has a different set of channels than the old
- // frame buffer, or if the type of a channel has changed.
- //
- const FrameBuffer &oldFrameBuffer = _data->tFileBuffer;
- FrameBuffer::ConstIterator i = oldFrameBuffer.begin();
- FrameBuffer::ConstIterator j = frameBuffer.begin();
- while (i != oldFrameBuffer.end() && j != frameBuffer.end())
- {
- if (strcmp (i.name(), j.name()) || i.slice().type != j.slice().type)
- break;
- ++i;
- ++j;
- }
- if (i != oldFrameBuffer.end() || j != frameBuffer.end())
- {
- //
- // Invalidate the cached buffer.
- //
- _data->deleteCachedBuffer ();
- _data->cachedTileY = -1;
- //
- // Create new a cached frame buffer. It can hold a single
- // row of tiles. The cached buffer can be reused for each
- // row of tiles because we set the yTileCoords parameter of
- // each Slice to true.
- //
- const Box2i &dataWindow = _data->header.dataWindow();
- _data->cachedBuffer = new FrameBuffer();
- _data->offset = dataWindow.min.x;
- int tileRowSize = (dataWindow.max.x - dataWindow.min.x + 1) *
- _data->tFile->tileYSize();
- for (FrameBuffer::ConstIterator k = frameBuffer.begin();
- k != frameBuffer.end();
- ++k)
- {
- Slice s = k.slice();
- switch (s.type)
- {
- case UINT:
- _data->cachedBuffer->insert
- (k.name(),
- Slice (UINT,
- (char *)(new unsigned int[tileRowSize] -
- _data->offset),
- sizeof (unsigned int),
- sizeof (unsigned int) *
- _data->tFile->levelWidth(0),
- 1, 1,
- s.fillValue,
- false, true));
- break;
- case HALF:
- _data->cachedBuffer->insert
- (k.name(),
- Slice (HALF,
- (char *)(new half[tileRowSize] -
- _data->offset),
- sizeof (half),
- sizeof (half) *
- _data->tFile->levelWidth(0),
- 1, 1,
- s.fillValue,
- false, true));
- break;
- case FLOAT:
- _data->cachedBuffer->insert
- (k.name(),
- Slice (FLOAT,
- (char *)(new float[tileRowSize] -
- _data->offset),
- sizeof(float),
- sizeof(float) *
- _data->tFile->levelWidth(0),
- 1, 1,
- s.fillValue,
- false, true));
- break;
- default:
- throw Iex::ArgExc ("Unknown pixel data type.");
- }
- }
- _data->tFile->setFrameBuffer (*_data->cachedBuffer);
- }
- _data->tFileBuffer = frameBuffer;
- }
- else
- {
- _data->sFile->setFrameBuffer (frameBuffer);
- }
- }
- const FrameBuffer &
- InputFile::frameBuffer () const
- {
- if (isTiled (_data->version))
- {
- Lock lock (*_data);
- return _data->tFileBuffer;
- }
- else
- {
- return _data->sFile->frameBuffer();
- }
- }
- bool
- InputFile::isComplete () const
- {
- if (isTiled (_data->version))
- return _data->tFile->isComplete();
- else
- return _data->sFile->isComplete();
- }
- void
- InputFile::readPixels (int scanLine1, int scanLine2)
- {
- if (isTiled (_data->version))
- {
- Lock lock (*_data);
- bufferedReadPixels (_data, scanLine1, scanLine2);
- }
- else
- {
- _data->sFile->readPixels (scanLine1, scanLine2);
- }
- }
- void
- InputFile::readPixels (int scanLine)
- {
- readPixels (scanLine, scanLine);
- }
- void
- InputFile::rawPixelData (int firstScanLine,
- const char *&pixelData,
- int &pixelDataSize)
- {
- try
- {
- if (isTiled (_data->version))
- {
- throw Iex::ArgExc ("Tried to read a raw scanline "
- "from a tiled image.");
- }
- _data->sFile->rawPixelData (firstScanLine, pixelData, pixelDataSize);
- }
- catch (Iex::BaseExc &e)
- {
- REPLACE_EXC (e, "Error reading pixel data from image "
- "file \"" << fileName() << "\". " << e);
- throw;
- }
- }
- void
- InputFile::rawTileData (int &dx, int &dy,
- int &lx, int &ly,
- const char *&pixelData,
- int &pixelDataSize)
- {
- try
- {
- if (!isTiled (_data->version))
- {
- throw Iex::ArgExc ("Tried to read a raw tile "
- "from a scanline-based image.");
- }
- _data->tFile->rawTileData (dx, dy, lx, ly, pixelData, pixelDataSize);
- }
- catch (Iex::BaseExc &e)
- {
- REPLACE_EXC (e, "Error reading tile data from image "
- "file \"" << fileName() << "\". " << e);
- throw;
- }
- }
- TiledInputFile*
- InputFile::tFile()
- {
- if (!isTiled (_data->version))
- {
- throw Iex::ArgExc ("Cannot get a TiledInputFile pointer "
- "from an InputFile that is not tiled.");
- }
- return _data->tFile;
- }
- } // namespace Imf