mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2024-12-27 03:14:05 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
625eebad54
opencv/3rdparty/openexr/IlmImf/ImfDeepTiledOutputFile.cpp

2056 lines
60 KiB
C++
Raw Normal View History

Merge pull request #14725 from alalek:update_openexr_2.3.0 3rdparty: update OpenEXR 2.3.0 (#14725) * openexr 2.2.1 * openexr 2.3.0 * openexr: build fixes * openexr: build dwa tables on-demand
2019-06-11 01:04:23 +08:00
///////////////////////////////////////////////////////////////////////////
//
// Copyright (c) 2011, 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 DeepTiledOutputFile
//
//-----------------------------------------------------------------------------
#include "ImfDeepTiledOutputFile.h"
#include "ImfDeepTiledInputFile.h"
#include "ImfDeepTiledInputPart.h"
#include "ImfInputFile.h"
#include "ImfTileDescriptionAttribute.h"
#include "ImfPreviewImageAttribute.h"
#include "ImfChannelList.h"
#include "ImfMisc.h"
#include "ImfTiledMisc.h"
#include "ImfStdIO.h"
#include "ImfCompressor.h"
#include "ImfOutputStreamMutex.h"
#include "ImfOutputPartData.h"
#include "ImfArray.h"
#include "ImfXdr.h"
#include "ImfVersion.h"
#include "ImfTileOffsets.h"
#include "ImfThreading.h"
#include "ImfPartType.h"
#include "ImathBox.h"
#include "IlmThreadPool.h"
#include "IlmThreadSemaphore.h"
#include "IlmThreadMutex.h"
#include "Iex.h"
#include <string>
#include <vector>
#include <fstream>
#include <assert.h>
#include <map>
#include <algorithm>
#include "ImfNamespace.h"
OPENEXR_IMF_INTERNAL_NAMESPACE_SOURCE_ENTER
using IMATH_NAMESPACE::Box2i;
using IMATH_NAMESPACE::V2i;
using std::string;
using std::vector;
using std::ofstream;
using std::map;
using std::min;
using std::max;
using std::swap;
using ILMTHREAD_NAMESPACE::Mutex;
using ILMTHREAD_NAMESPACE::Lock;
using ILMTHREAD_NAMESPACE::Semaphore;
using ILMTHREAD_NAMESPACE::Task;
using ILMTHREAD_NAMESPACE::TaskGroup;
using ILMTHREAD_NAMESPACE::ThreadPool;
namespace {
struct TOutSliceInfo
{
PixelType type;
const char * base;
size_t sampleStride;
size_t xStride;
size_t yStride;
bool zero;
int xTileCoords;
int yTileCoords;
TOutSliceInfo (PixelType type = HALF,
size_t sampleStride = 0,
size_t xStride = 0,
size_t yStride = 0,
bool zero = false,
int xTileCoords = 0,
int yTileCoords = 0);
};
TOutSliceInfo::TOutSliceInfo (PixelType t,
size_t spst,
size_t xStride,
size_t yStride,
bool z,
int xtc,
int ytc)
:
type (t),
sampleStride (spst),
xStride(xStride),
yStride(yStride),
zero (z),
xTileCoords (xtc),
yTileCoords (ytc)
{
// empty
}
struct TileCoord
{
int dx;
int dy;
int lx;
int ly;
TileCoord (int xTile = 0, int yTile = 0,
int xLevel = 0, int yLevel = 0)
:
dx (xTile), dy (yTile),
lx (xLevel), ly (yLevel)
{
// empty
}
bool
operator < (const TileCoord &other) const
{
return (ly < other.ly) ||
(ly == other.ly && lx < other.lx) ||
((ly == other.ly && lx == other.lx) &&
((dy < other.dy) || (dy == other.dy && dx < other.dx)));
}
bool
operator == (const TileCoord &other) const
{
return lx == other.lx &&
ly == other.ly &&
dx == other.dx &&
dy == other.dy;
}
};
struct BufferedTile
{
char * pixelData;
Int64 pixelDataSize;
Int64 unpackedDataSize;
char * sampleCountTableData;
Int64 sampleCountTableSize;
BufferedTile (const char *data, int size, int unpackedSize,
const char *tableData, int tableSize):
pixelData (0),
pixelDataSize(size),
unpackedDataSize(unpackedSize),
sampleCountTableData(0),
sampleCountTableSize(tableSize)
{
pixelData = new char[pixelDataSize];
memcpy (pixelData, data, pixelDataSize);
sampleCountTableData = new char[tableSize];
memcpy (sampleCountTableData, tableData, tableSize);
}
~BufferedTile()
{
delete [] pixelData;
delete [] sampleCountTableData;
}
};
typedef map <TileCoord, BufferedTile *> TileMap;
struct TileBuffer
{
Array<char> buffer;
const char * dataPtr;
Int64 dataSize;
Int64 uncompressedSize;
Compressor * compressor;
Array<char> sampleCountTableBuffer;
const char * sampleCountTablePtr;
Int64 sampleCountTableSize;
Compressor* sampleCountTableCompressor;
TileCoord tileCoord;
bool hasException;
string exception;
TileBuffer ();
~TileBuffer ();
inline void wait () {_sem.wait();}
inline void post () {_sem.post();}
protected:
Semaphore _sem;
};
TileBuffer::TileBuffer ():
dataPtr (0),
dataSize (0),
compressor (0),
sampleCountTablePtr (0),
sampleCountTableCompressor (0),
hasException (false),
exception (),
_sem (1)
{
// empty
}
TileBuffer::~TileBuffer ()
{
if (compressor != 0)
delete compressor;
if (sampleCountTableCompressor != 0)
delete sampleCountTableCompressor;
}
} // namespace
struct DeepTiledOutputFile::Data
{
Header header; // the image header
int version; // file format version
bool multipart; // file is multipart
TileDescription tileDesc; // describes the tile layout
DeepFrameBuffer frameBuffer; // framebuffer to write into
Int64 previewPosition;
LineOrder lineOrder; // the file's lineorder
int minX; // data window's min x coord
int maxX; // data window's max x coord
int minY; // data window's min y coord
int maxY; // data window's max x coord
int numXLevels; // number of x levels
int numYLevels; // number of y levels
int * numXTiles; // number of x tiles at a level
int * numYTiles; // number of y tiles at a level
TileOffsets tileOffsets; // stores offsets in file for
// each tile
Compressor::Format format; // compressor's data format
vector<TOutSliceInfo*> slices; // info about channels in file
vector<TileBuffer*> tileBuffers;
Int64 tileOffsetsPosition; // position of the tile index
TileMap tileMap; // the map of buffered tiles
TileCoord nextTileToWrite;
int partNumber; // the output part number
char* sampleCountSliceBase; // the pointer to the number
// of samples in each pixel
int sampleCountXStride; // the x stride for sampleCountSliceBase
int sampleCountYStride; // the y stride for sampleCountSliceBase
int sampleCountXTileCoords; // using x coordinates relative to current tile
int sampleCountYTileCoords; // using y coordinates relative to current tile
Int64 maxSampleCountTableSize;// the max size in bytes for a pixel
// sample count table
OutputStreamMutex* _streamData;
bool _deleteStream;
Data (int numThreads);
~Data ();
inline TileBuffer * getTileBuffer (int number);
// hash function from tile
// buffer coords into our
// vector of tile buffers
int& getSampleCount(int x, int y);
// get the number of samples
// in each pixel
TileCoord nextTileCoord (const TileCoord &a);
};
DeepTiledOutputFile::Data::Data (int numThreads):
numXTiles(0),
numYTiles(0),
tileOffsetsPosition (0),
partNumber(-1),
_streamData(NULL),
_deleteStream(true)
{
//
// We need at least one tileBuffer, but if threading is used,
// to keep n threads busy we need 2*n tileBuffers
//
tileBuffers.resize (max (1, 2 * numThreads));
for (size_t i = 0; i < tileBuffers.size(); i++)
tileBuffers[i] = 0;
}
DeepTiledOutputFile::Data::~Data ()
{
delete [] numXTiles;
delete [] numYTiles;
//
// Delete all the tile buffers, if any still happen to exist
//
for (TileMap::iterator i = tileMap.begin(); i != tileMap.end(); ++i)
delete i->second;
for (size_t i = 0; i < tileBuffers.size(); i++)
if (tileBuffers[i] != 0)
delete tileBuffers[i];
for (size_t i = 0; i < slices.size(); i++)
delete slices[i];
}
int&
DeepTiledOutputFile::Data::getSampleCount(int x, int y)
{
return sampleCount(sampleCountSliceBase,
sampleCountXStride,
sampleCountYStride,
x, y);
}
TileBuffer*
DeepTiledOutputFile::Data::getTileBuffer (int number)
{
return tileBuffers[number % tileBuffers.size()];
}
TileCoord
DeepTiledOutputFile::Data::nextTileCoord (const TileCoord &a)
{
TileCoord b = a;
if (lineOrder == INCREASING_Y)
{
b.dx++;
if (b.dx >= numXTiles[b.lx])
{
b.dx = 0;
b.dy++;
if (b.dy >= numYTiles[b.ly])
{
//
// the next tile is in the next level
//
b.dy = 0;
switch (tileDesc.mode)
{
case ONE_LEVEL:
case MIPMAP_LEVELS:
b.lx++;
b.ly++;
break;
case RIPMAP_LEVELS:
b.lx++;
if (b.lx >= numXLevels)
{
b.lx = 0;
b.ly++;
#ifdef DEBUG
assert (b.ly <= numYLevels);
#endif
}
break;
case NUM_LEVELMODES :
throw IEX_NAMESPACE::LogicExc("unknown level mode computing nextTileCoord");
}
}
}
}
else if (lineOrder == DECREASING_Y)
{
b.dx++;
if (b.dx >= numXTiles[b.lx])
{
b.dx = 0;
b.dy--;
if (b.dy < 0)
{
//
// the next tile is in the next level
//
switch (tileDesc.mode)
{
case ONE_LEVEL:
case MIPMAP_LEVELS:
b.lx++;
b.ly++;
break;
case RIPMAP_LEVELS:
b.lx++;
if (b.lx >= numXLevels)
{
b.lx = 0;
b.ly++;
#ifdef DEBUG
assert (b.ly <= numYLevels);
#endif
}
break;
case NUM_LEVELMODES :
throw IEX_NAMESPACE::LogicExc("unknown level mode computing nextTileCoord");
}
if (b.ly < numYLevels)
b.dy = numYTiles[b.ly] - 1;
}
}
}else if(lineOrder==RANDOM_Y)
{
THROW (IEX_NAMESPACE::ArgExc,
"can't compute next tile from randomly ordered image: use getTilesInOrder instead");
}
return b;
}
namespace {
void
writeTileData (DeepTiledOutputFile::Data *ofd,
int dx, int dy,
int lx, int ly,
const char pixelData[],
Int64 pixelDataSize,
Int64 unpackedDataSize,
const char sampleCountTableData[],
Int64 sampleCountTableSize)
{
//
// Store a block of pixel data in the output file, and try
// to keep track of the current writing position the file,
// without calling tellp() (tellp() can be fairly expensive).
//
Int64 currentPosition = ofd->_streamData->currentPosition;
ofd->_streamData->currentPosition = 0;
if (currentPosition == 0)
currentPosition = ofd->_streamData->os->tellp();
ofd->tileOffsets (dx, dy, lx, ly) = currentPosition;
#ifdef DEBUG
assert (ofd->_streamData->os->tellp() == currentPosition);
#endif
//
// Write the tile header.
//
if (ofd->multipart)
{
Xdr::write <StreamIO> (*ofd->_streamData->os, ofd->partNumber);
}
Xdr::write <StreamIO> (*ofd->_streamData->os, dx);
Xdr::write <StreamIO> (*ofd->_streamData->os, dy);
Xdr::write <StreamIO> (*ofd->_streamData->os, lx);
Xdr::write <StreamIO> (*ofd->_streamData->os, ly);
//
// Write the packed size of the pixel sample count table (64 bits)
//
Xdr::write <StreamIO> (*ofd->_streamData->os, sampleCountTableSize);
//
// Write the packed and unpacked data size (64 bits each)
//
Xdr::write <StreamIO> (*ofd->_streamData->os, pixelDataSize);
Xdr::write <StreamIO> (*ofd->_streamData->os, unpackedDataSize);
//
// Write the compressed pixel sample count table.
//
ofd->_streamData->os->write (sampleCountTableData, sampleCountTableSize);
//
// Write the compressed data.
//
ofd->_streamData->os->write (pixelData, pixelDataSize);
//
// Keep current position in the file so that we can avoid
// redundant seekg() operations (seekg() can be fairly expensive).
//
ofd->_streamData->currentPosition = currentPosition +
4 * Xdr::size<int>() + // dx, dy, lx, ly,
3 * Xdr::size<Int64>() + // sampleCountTableSize,
// pixelDataSize,
// unpackedDataSize
sampleCountTableSize +
pixelDataSize;
if (ofd->multipart)
{
ofd->_streamData->currentPosition += Xdr::size<int>();
}
}
void
bufferedTileWrite (
DeepTiledOutputFile::Data *ofd,
int dx, int dy,
int lx, int ly,
const char pixelData[],
Int64 pixelDataSize,
Int64 unpackedDataSize,
const char sampleCountTableData[],
Int64 sampleCountTableSize)
{
//
// Check if a tile with coordinates (dx,dy,lx,ly) has already been written.
//
if (ofd->tileOffsets (dx, dy, lx, ly))
{
THROW (IEX_NAMESPACE::ArgExc,
"Attempt to write tile "
"(" << dx << ", " << dy << ", " << lx << ", " << ly << ") "
"more than once.");
}
//
// If tiles can be written in random order, then don't buffer anything.
//
if (ofd->lineOrder == RANDOM_Y)
{
writeTileData (ofd, dx, dy, lx, ly,
pixelData, pixelDataSize, unpackedDataSize,
sampleCountTableData, sampleCountTableSize);
return;
}
//
// If the tiles cannot be written in random order, then check if a
// tile with coordinates (dx,dy,lx,ly) has already been buffered.
//
TileCoord currentTile = TileCoord(dx, dy, lx, ly);
if (ofd->tileMap.find (currentTile) != ofd->tileMap.end())
{
THROW (IEX_NAMESPACE::ArgExc,
"Attempt to write tile "
"(" << dx << ", " << dy << ", " << lx << ", " << ly << ") "
"more than once.");
}
//
// If all the tiles before this one have already been written to the file,
// then write this tile immediately and check if we have buffered tiles
// that can be written after this tile.
//
// Otherwise, buffer the tile so it can be written to file later.
//
if (ofd->nextTileToWrite == currentTile)
{
writeTileData (ofd, dx, dy, lx, ly,
pixelData, pixelDataSize, unpackedDataSize,
sampleCountTableData, sampleCountTableSize);
ofd->nextTileToWrite = ofd->nextTileCoord (ofd->nextTileToWrite);
TileMap::iterator i = ofd->tileMap.find (ofd->nextTileToWrite);
//
// Step through the tiles and write all successive buffered tiles after
// the current one.
//
while(i != ofd->tileMap.end())
{
//
// Write the tile, and then delete the tile's buffered data
//
writeTileData (ofd,
i->first.dx, i->first.dy,
i->first.lx, i->first.ly,
i->second->pixelData,
i->second->pixelDataSize,
i->second->unpackedDataSize,
i->second->sampleCountTableData,
i->second->sampleCountTableSize);
delete i->second;
ofd->tileMap.erase (i);
//
// Proceed to the next tile
//
ofd->nextTileToWrite = ofd->nextTileCoord (ofd->nextTileToWrite);
i = ofd->tileMap.find (ofd->nextTileToWrite);
}
}
else
{
//
// Create a new BufferedTile, copy the pixelData into it, and
// insert it into the tileMap.
//
ofd->tileMap[currentTile] =
new BufferedTile ((const char *)pixelData, pixelDataSize, unpackedDataSize,
sampleCountTableData, sampleCountTableSize);
}
}
void
convertToXdr (DeepTiledOutputFile::Data *ofd,
Array<char>& tileBuffer,
int numScanLines,
vector<Int64>& bytesPerLine)
{
//
// Convert the contents of a TiledOutputFile's tileBuffer from the
// machine's native representation to Xdr format. This function is called
// by writeTile(), below, if the compressor wanted its input pixel data
// in the machine's native format, but then failed to compress the data
// (most compressors will expand rather than compress random input data).
//
// Note that this routine assumes that the machine's native representation
// of the pixel data has the same size as the Xdr representation. This
// makes it possible to convert the pixel data in place, without an
// intermediate temporary buffer.
//
//
// Set these to point to the start of the tile.
// We will write to toPtr, and read from fromPtr.
//
char *writePtr = tileBuffer;
const char *readPtr = writePtr;
//
// Iterate over all scan lines in the tile.
//
for (int y = 0; y < numScanLines; ++y)
{
//
// Iterate over all slices in the file.
//
for (unsigned int i = 0; i < ofd->slices.size(); ++i)
{
const TOutSliceInfo &slice = *ofd->slices[i];
//
// Convert the samples in place.
//
Int64 numPixelsPerScanLine = bytesPerLine[y];
convertInPlace (writePtr, readPtr, slice.type,
numPixelsPerScanLine);
}
}
#ifdef DEBUG
assert (writePtr == readPtr);
#endif
}
//
// A TileBufferTask encapsulates the task of copying a tile from
// the user's framebuffer into a LineBuffer and compressing the data
// if necessary.
//
class TileBufferTask: public Task
{
public:
TileBufferTask (TaskGroup *group,
DeepTiledOutputFile::Data *ofd,
int number,
int dx, int dy,
int lx, int ly);
virtual ~TileBufferTask ();
virtual void execute ();
private:
DeepTiledOutputFile::Data * _ofd;
TileBuffer * _tileBuffer;
};
TileBufferTask::TileBufferTask
(TaskGroup *group,
DeepTiledOutputFile::Data *ofd,
int number,
int dx, int dy,
int lx, int ly)
:
Task (group),
_ofd (ofd),
_tileBuffer (_ofd->getTileBuffer (number))
{
//
// Wait for the tileBuffer to become available
//
_tileBuffer->wait ();
_tileBuffer->tileCoord = TileCoord (dx, dy, lx, ly);
}
TileBufferTask::~TileBufferTask ()
{
//
// Signal that the tile buffer is now free
//
_tileBuffer->post ();
}
void
TileBufferTask::execute ()
{
try
{
//
// First copy the pixel data from the frame buffer
// into the tile buffer
//
// Convert one tile's worth of pixel data to
// a machine-independent representation, and store
// the result in _tileBuffer->buffer.
//
Box2i tileRange = OPENEXR_IMF_INTERNAL_NAMESPACE::dataWindowForTile (
_ofd->tileDesc,
_ofd->minX, _ofd->maxX,
_ofd->minY, _ofd->maxY,
_tileBuffer->tileCoord.dx,
_tileBuffer->tileCoord.dy,
_tileBuffer->tileCoord.lx,
_tileBuffer->tileCoord.ly);
int numScanLines = tileRange.max.y - tileRange.min.y + 1;
// int numPixelsPerScanLine = tileRange.max.x - tileRange.min.x + 1;
//
// Get the bytes for each line.
//
vector<Int64> bytesPerLine(_ofd->tileDesc.ySize);
vector<int> xOffsets(_ofd->slices.size());
vector<int> yOffsets(_ofd->slices.size());
for (size_t i = 0; i < _ofd->slices.size(); i++)
{
const TOutSliceInfo &slice = *_ofd->slices[i];
xOffsets[i] = slice.xTileCoords * tileRange.min.x;
yOffsets[i] = slice.yTileCoords * tileRange.min.y;
}
calculateBytesPerLine(_ofd->header,
_ofd->sampleCountSliceBase,
_ofd->sampleCountXStride,
_ofd->sampleCountYStride,
tileRange.min.x, tileRange.max.x,
tileRange.min.y, tileRange.max.y,
xOffsets, yOffsets,
bytesPerLine);
//
// Allocate the memory for internal buffer.
// (TODO) more efficient memory management?
//
Int64 totalBytes = 0;
Int64 maxBytesPerTileLine = 0;
for (size_t i = 0; i < bytesPerLine.size(); i++)
{
totalBytes += bytesPerLine[i];
if (bytesPerLine[i] > maxBytesPerTileLine)
maxBytesPerTileLine = bytesPerLine[i];
}
_tileBuffer->buffer.resizeErase(totalBytes);
char *writePtr = _tileBuffer->buffer;
//
// Iterate over the scan lines in the tile.
//
int xOffsetForSampleCount =
(_ofd->sampleCountXTileCoords == 0) ? 0 : tileRange.min.x;
int yOffsetForSampleCount =
(_ofd->sampleCountYTileCoords == 0) ? 0 : tileRange.min.y;
for (int y = tileRange.min.y; y <= tileRange.max.y; ++y)
{
//
// Iterate over all image channels.
//
for (unsigned int i = 0; i < _ofd->slices.size(); ++i)
{
const TOutSliceInfo &slice = *_ofd->slices[i];
//
// Fill the tile buffer with pixel data.
//
if (slice.zero)
{
//
// The frame buffer contains no data for this channel.
// Store zeroes in _data->tileBuffer.
//
fillChannelWithZeroes (writePtr, _ofd->format, slice.type,
bytesPerLine[y - tileRange.min.y]);
}
else
{
//
// The frame buffer contains data for this channel.
//
int xOffsetForData = slice.xTileCoords ? tileRange.min.x : 0;
int yOffsetForData = slice.yTileCoords ? tileRange.min.y : 0;
// (TOOD) treat sample count offsets differently.
copyFromDeepFrameBuffer (writePtr,
slice.base,
_ofd->sampleCountSliceBase,
_ofd->sampleCountXStride,
_ofd->sampleCountYStride,
y,
tileRange.min.x,
tileRange.max.x,
xOffsetForSampleCount,
yOffsetForSampleCount,
xOffsetForData,
yOffsetForData,
slice.sampleStride,
slice.xStride,
slice.yStride,
_ofd->format,
slice.type);
#if defined(DEBUG)
assert(writePtr-_tileBuffer->buffer<=totalBytes);
#endif
}
}
}
//
// Compress the pixel sample count table.
//
char* ptr = _tileBuffer->sampleCountTableBuffer;
Int64 tableDataSize = 0;
for (int i = tileRange.min.y; i <= tileRange.max.y; i++)
{
int count = 0;
for (int j = tileRange.min.x; j <= tileRange.max.x; j++)
{
count += _ofd->getSampleCount(j - xOffsetForSampleCount,
i - yOffsetForSampleCount);
Xdr::write <CharPtrIO> (ptr, count);
tableDataSize += sizeof (int);
}
}
if(_tileBuffer->sampleCountTableCompressor)
{
_tileBuffer->sampleCountTableSize =
_tileBuffer->sampleCountTableCompressor->compress (
_tileBuffer->sampleCountTableBuffer,
tableDataSize,
tileRange.min.y,
_tileBuffer->sampleCountTablePtr);
}
//
// If we can't make data shrink (or compression was disabled), then just use the raw data.
//
if ( ! _tileBuffer->sampleCountTableCompressor ||
_tileBuffer->sampleCountTableSize >= _ofd->maxSampleCountTableSize)
{
_tileBuffer->sampleCountTableSize = _ofd->maxSampleCountTableSize;
_tileBuffer->sampleCountTablePtr = _tileBuffer->sampleCountTableBuffer;
}
//
// Compress the contents of the tileBuffer,
// and store the compressed data in the output file.
//
_tileBuffer->dataSize = writePtr - _tileBuffer->buffer;
_tileBuffer->uncompressedSize = _tileBuffer->dataSize;
_tileBuffer->dataPtr = _tileBuffer->buffer;
// (TODO) don't do this all the time.
if (_tileBuffer->compressor != 0)
delete _tileBuffer->compressor;
_tileBuffer->compressor = newTileCompressor
(_ofd->header.compression(),
maxBytesPerTileLine,
_ofd->tileDesc.ySize,
_ofd->header);
if (_tileBuffer->compressor)
{
const char *compPtr;
Int64 compSize = _tileBuffer->compressor->compressTile
(_tileBuffer->dataPtr,
_tileBuffer->dataSize,
tileRange, compPtr);
if (compSize < _tileBuffer->dataSize)
{
_tileBuffer->dataSize = compSize;
_tileBuffer->dataPtr = compPtr;
}
else if (_ofd->format == Compressor::NATIVE)
{
//
// The data did not shrink during compression, but
// we cannot write to the file using native format,
// so we need to convert the lineBuffer to Xdr.
//
convertToXdr (_ofd, _tileBuffer->buffer, numScanLines,
bytesPerLine);
}
}
}
catch (std::exception &e)
{
if (!_tileBuffer->hasException)
{
_tileBuffer->exception = e.what ();
_tileBuffer->hasException = true;
}
}
catch (...)
{
if (!_tileBuffer->hasException)
{
_tileBuffer->exception = "unrecognized exception";
_tileBuffer->hasException = true;
}
}
}
} // namespace
DeepTiledOutputFile::DeepTiledOutputFile
(const char fileName[],
const Header &header,
int numThreads)
:
_data (new Data (numThreads))
{
_data->_streamData=new OutputStreamMutex();
_data->_deleteStream =true;
try
{
header.sanityCheck (true);
_data->_streamData->os = new StdOFStream (fileName);
initialize (header);
_data->_streamData->currentPosition = _data->_streamData->os->tellp();
// Write header and empty offset table to the file.
writeMagicNumberAndVersionField(*_data->_streamData->os, _data->header);
_data->previewPosition = _data->header.writeTo (*_data->_streamData->os, true);
_data->tileOffsetsPosition = _data->tileOffsets.writeTo (*_data->_streamData->os);
_data->multipart = false;
}
catch (IEX_NAMESPACE::BaseExc &e)
{
if (_data && _data->_streamData && _data->_streamData->os) delete _data->_streamData->os;
if (_data && _data->_streamData) delete _data->_streamData;
if (_data) delete _data;
REPLACE_EXC (e, "Cannot open image file "
"\"" << fileName << "\". " << e.what());
throw;
}
catch (...)
{
if (_data && _data->_streamData && _data->_streamData->os) delete _data->_streamData->os;
if (_data->_streamData) delete _data->_streamData;
if (_data) delete _data;
throw;
}
}
DeepTiledOutputFile::DeepTiledOutputFile
(OPENEXR_IMF_INTERNAL_NAMESPACE::OStream &os,
const Header &header,
int numThreads)
:
_data (new Data (numThreads))
{
_data->_streamData=new OutputStreamMutex();
_data->_deleteStream=false;
try
{
header.sanityCheck(true);
_data->_streamData->os = &os;
initialize (header);
_data->_streamData->currentPosition = _data->_streamData->os->tellp();
// Write header and empty offset table to the file.
writeMagicNumberAndVersionField(*_data->_streamData->os, _data->header);
_data->previewPosition = _data->header.writeTo (*_data->_streamData->os, true);
_data->tileOffsetsPosition = _data->tileOffsets.writeTo (*_data->_streamData->os);
_data->multipart = false;
}
catch (IEX_NAMESPACE::BaseExc &e)
{
if (_data && _data->_streamData) delete _data->_streamData;
if (_data) delete _data;
REPLACE_EXC (e, "Cannot open image file "
"\"" << os.fileName() << "\". " << e.what());
throw;
}
catch (...)
{
if (_data && _data->_streamData) delete _data->_streamData;
if (_data) delete _data;
throw;
}
}
DeepTiledOutputFile::DeepTiledOutputFile(const OutputPartData* part)
{
try
{
if (part->header.type() != DEEPTILE)
throw IEX_NAMESPACE::ArgExc("Can't build a DeepTiledOutputFile from "
"a type-mismatched part.");
_data = new Data (part->numThreads);
_data->_streamData=part->mutex;
_data->_deleteStream=false;
initialize(part->header);
_data->partNumber = part->partNumber;
_data->tileOffsetsPosition = part->chunkOffsetTablePosition;
_data->previewPosition = part->previewPosition;
_data->multipart = part->multipart;
}
catch (IEX_NAMESPACE::BaseExc &e)
{
if (_data) delete _data;
REPLACE_EXC (e, "Cannot initialize output part "
"\"" << part->partNumber << "\". " << e.what());
throw;
}
catch (...)
{
if (_data) delete _data;
throw;
}
}
void
DeepTiledOutputFile::initialize (const Header &header)
{
_data->header = header;
_data->header.setType(DEEPTILE);
_data->lineOrder = _data->header.lineOrder();
//
// Check that the file is indeed tiled
//
_data->tileDesc = _data->header.tileDescription();
//
// Save the dataWindow information
//
const Box2i &dataWindow = _data->header.dataWindow();
_data->minX = dataWindow.min.x;
_data->maxX = dataWindow.max.x;
_data->minY = dataWindow.min.y;
_data->maxY = dataWindow.max.y;
//
// Precompute level and tile information to speed up utility functions
//
precalculateTileInfo (_data->tileDesc,
_data->minX, _data->maxX,
_data->minY, _data->maxY,
_data->numXTiles, _data->numYTiles,
_data->numXLevels, _data->numYLevels);
//
// Determine the first tile coordinate that we will be writing
// if the file is not RANDOM_Y.
//
_data->nextTileToWrite = (_data->lineOrder == INCREASING_Y)?
TileCoord (0, 0, 0, 0):
TileCoord (0, _data->numYTiles[0] - 1, 0, 0);
Compressor* compressor = newTileCompressor
(_data->header.compression(),
0,
_data->tileDesc.ySize,
_data->header);
_data->format = defaultFormat (compressor);
if (compressor != 0)
delete compressor;
_data->tileOffsets = TileOffsets (_data->tileDesc.mode,
_data->numXLevels,
_data->numYLevels,
_data->numXTiles,
_data->numYTiles);
//ignore the existing value of chunkCount - correct it if it's wrong
_data->header.setChunkCount(getChunkOffsetTableSize(_data->header,true));
_data->maxSampleCountTableSize = _data->tileDesc.ySize *
_data->tileDesc.xSize *
sizeof(int);
for (size_t i = 0; i < _data->tileBuffers.size(); i++)
{
_data->tileBuffers[i] = new TileBuffer ();
_data->tileBuffers[i]->sampleCountTableBuffer.
resizeErase(_data->maxSampleCountTableSize);
char * p = &(_data->tileBuffers[i]->sampleCountTableBuffer[0]);
memset (p, 0, _data->maxSampleCountTableSize);
_data->tileBuffers[i]->sampleCountTableCompressor =
newCompressor (_data->header.compression(),
_data->maxSampleCountTableSize,
_data->header);
}
}
DeepTiledOutputFile::~DeepTiledOutputFile ()
{
if (_data)
{
{
Lock lock(*_data->_streamData);
Int64 originalPosition = _data->_streamData->os->tellp();
if (_data->tileOffsetsPosition > 0)
{
try
{
_data->_streamData->os->seekp (_data->tileOffsetsPosition);
_data->tileOffsets.writeTo (*_data->_streamData->os);
//
// Restore the original position.
//
_data->_streamData->os->seekp (originalPosition);
}
catch (...)
{
//
// We cannot safely throw any exceptions from here.
// This destructor may have been called because the
// stack is currently being unwound for another
// exception.
//
}
}
}
if (_data->_deleteStream && _data->_streamData)
delete _data->_streamData->os;
//
// (TODO) we should have a way to tell if the stream data is owned by
// this file or by a parent multipart file.
//
if (_data->partNumber == -1 && _data->_streamData)
delete _data->_streamData;
delete _data;
}
}
const char *
DeepTiledOutputFile::fileName () const
{
return _data->_streamData->os->fileName();
}
const Header &
DeepTiledOutputFile::header () const
{
return _data->header;
}
void
DeepTiledOutputFile::setFrameBuffer (const DeepFrameBuffer &frameBuffer)
{
Lock lock (*_data->_streamData);
//
// Check if the new frame buffer descriptor
// is compatible with the image file header.
//
const ChannelList &channels = _data->header.channels();
for (ChannelList::ConstIterator i = channels.begin();
i != channels.end();
++i)
{
DeepFrameBuffer::ConstIterator j = frameBuffer.find (i.name());
if (j == frameBuffer.end())
continue;
if (i.channel().type != j.slice().type)
THROW (IEX_NAMESPACE::ArgExc, "Pixel type of \"" << i.name() << "\" channel "
"of output file \"" << fileName() << "\" is "
"not compatible with the frame buffer's "
"pixel type.");
if (j.slice().xSampling != 1 || j.slice().ySampling != 1)
THROW (IEX_NAMESPACE::ArgExc, "All channels in a tiled file must have"
"sampling (1,1).");
}
//
// Store the pixel sample count table.
//
const Slice& sampleCountSlice = frameBuffer.getSampleCountSlice();
if (sampleCountSlice.base == 0)
{
throw IEX_NAMESPACE::ArgExc ("Invalid base pointer, please set a proper sample count slice.");
}
else
{
_data->sampleCountSliceBase = sampleCountSlice.base;
_data->sampleCountXStride = sampleCountSlice.xStride;
_data->sampleCountYStride = sampleCountSlice.yStride;
_data->sampleCountXTileCoords = sampleCountSlice.xTileCoords;
_data->sampleCountYTileCoords = sampleCountSlice.yTileCoords;
}
//
// Initialize slice table for writePixels().
// Pixel sample count slice is not presented in the header,
// so it wouldn't be added here.
// Store the pixel base pointer table.
//
vector<TOutSliceInfo*> slices;
for (ChannelList::ConstIterator i = channels.begin();
i != channels.end();
++i)
{
DeepFrameBuffer::ConstIterator j = frameBuffer.find (i.name());
if (j == frameBuffer.end())
{
//
// Channel i is not present in the frame buffer.
// In the file, channel i will contain only zeroes.
//
slices.push_back (new TOutSliceInfo (i.channel().type,
0, // sampleStride,
0, // xStride
0, // yStride
true)); // zero
}
else
{
//
// Channel i is present in the frame buffer.
//
slices.push_back (new TOutSliceInfo (j.slice().type,
j.slice().sampleStride,
j.slice().xStride,
j.slice().yStride,
false, // zero
(j.slice().xTileCoords)? 1: 0,
(j.slice().yTileCoords)? 1: 0));
TOutSliceInfo* slice = slices.back();
slice->base = j.slice().base;
}
}
//
// Store the new frame buffer.
//
_data->frameBuffer = frameBuffer;
for (size_t i = 0; i < _data->slices.size(); i++)
delete _data->slices[i];
_data->slices = slices;
}
const DeepFrameBuffer &
DeepTiledOutputFile::frameBuffer () const
{
Lock lock (*_data->_streamData);
return _data->frameBuffer;
}
void
DeepTiledOutputFile::writeTiles (int dx1, int dx2, int dy1, int dy2,
int lx, int ly)
{
try
{
Lock lock (*_data->_streamData);
if (_data->slices.size() == 0)
throw IEX_NAMESPACE::ArgExc ("No frame buffer specified "
"as pixel data source.");
if (!isValidTile (dx1, dy1, lx, ly) || !isValidTile (dx2, dy2, lx, ly))
throw IEX_NAMESPACE::ArgExc ("Tile coordinates are invalid.");
if (!isValidLevel (lx, ly))
THROW (IEX_NAMESPACE::ArgExc,
"Level coordinate "
"(" << lx << ", " << ly << ") "
"is invalid.");
//
// Determine the first and last tile coordinates in both dimensions
// based on the file's lineOrder
//
if (dx1 > dx2)
swap (dx1, dx2);
if (dy1 > dy2)
swap (dy1, dy2);
int dyStart = dy1;
int dyStop = dy2 + 1;
int dY = 1;
if (_data->lineOrder == DECREASING_Y)
{
dyStart = dy2;
dyStop = dy1 - 1;
dY = -1;
}
int numTiles = (dx2 - dx1 + 1) * (dy2 - dy1 + 1);
int numTasks = min ((int)_data->tileBuffers.size(), numTiles);
//
// Create a task group for all tile buffer tasks. When the
// task group goes out of scope, the destructor waits until
// all tasks are complete.
//
{
TaskGroup taskGroup;
//
// Add in the initial compression tasks to the thread pool
//
int nextCompBuffer = 0;
int dxComp = dx1;
int dyComp = dyStart;
while (nextCompBuffer < numTasks)
{
ThreadPool::addGlobalTask (new TileBufferTask (&taskGroup,
_data,
nextCompBuffer++,
dxComp, dyComp,
lx, ly));
dxComp++;
if (dxComp > dx2)
{
dxComp = dx1;
dyComp += dY;
}
}
//
// Write the compressed buffers and add in more compression
// tasks until done
//
int nextWriteBuffer = 0;
int dxWrite = dx1;
int dyWrite = dyStart;
while (nextWriteBuffer < numTiles)
{
//
// Wait until the nextWriteBuffer is ready to be written
//
TileBuffer* writeBuffer =
_data->getTileBuffer (nextWriteBuffer);
writeBuffer->wait();
//
// Write the tilebuffer
//
bufferedTileWrite ( _data, dxWrite, dyWrite, lx, ly,
writeBuffer->dataPtr,
writeBuffer->dataSize,
writeBuffer->uncompressedSize,
writeBuffer->sampleCountTablePtr,
writeBuffer->sampleCountTableSize);
//
// Release the lock on nextWriteBuffer
//
writeBuffer->post();
//
// If there are no more tileBuffers to compress, then
// only continue to write out remaining tileBuffers,
// otherwise keep adding compression tasks.
//
if (nextCompBuffer < numTiles)
{
//
// add nextCompBuffer as a compression Task
//
ThreadPool::addGlobalTask
(new TileBufferTask (&taskGroup,
_data,
nextCompBuffer,
dxComp, dyComp,
lx, ly));
}
nextWriteBuffer++;
dxWrite++;
if (dxWrite > dx2)
{
dxWrite = dx1;
dyWrite += dY;
}
nextCompBuffer++;
dxComp++;
if (dxComp > dx2)
{
dxComp = dx1;
dyComp += dY;
}
}
//
// finish all tasks
//
}
//
// Exeption handling:
//
// TileBufferTask::execute() may have encountered exceptions, but
// those exceptions occurred in another thread, not in the thread
// that is executing this call to TiledOutputFile::writeTiles().
// TileBufferTask::execute() has caught all exceptions and stored
// the exceptions' what() strings in the tile buffers.
// Now we check if any tile buffer contains a stored exception; if
// this is the case then we re-throw the exception in this thread.
// (It is possible that multiple tile buffers contain stored
// exceptions. We re-throw the first exception we find and
// ignore all others.)
//
const string *exception = 0;
for (size_t i = 0; i < _data->tileBuffers.size(); ++i)
{
TileBuffer *tileBuffer = _data->tileBuffers[i];
if (tileBuffer->hasException && !exception)
exception = &tileBuffer->exception;
tileBuffer->hasException = false;
}
if (exception)
throw IEX_NAMESPACE::IoExc (*exception);
}
catch (IEX_NAMESPACE::BaseExc &e)
{
REPLACE_EXC (e, "Failed to write pixel data to image "
"file \"" << fileName() << "\". " << e.what());
throw;
}
}
void
DeepTiledOutputFile::writeTiles (int dx1, int dxMax, int dyMin, int dyMax, int l)
{
writeTiles (dx1, dxMax, dyMin, dyMax, l, l);
}
void
DeepTiledOutputFile::writeTile (int dx, int dy, int lx, int ly)
{
writeTiles (dx, dx, dy, dy, lx, ly);
}
void
DeepTiledOutputFile::writeTile (int dx, int dy, int l)
{
writeTile(dx, dy, l, l);
}
void
DeepTiledOutputFile::copyPixels (DeepTiledInputFile &in)
{
//
// Check if this file's and and the InputFile's
// headers are compatible.
//
const Header &hdr = _data->header;
const Header &inHdr = in.header();
if (!(hdr.tileDescription() == inHdr.tileDescription()))
THROW (IEX_NAMESPACE::ArgExc, "Quick pixel copy from image "
"file \"" << in.fileName() << "\" to image "
"file \"" << fileName() << "\" failed. "
"The files have different tile descriptions.");
if (!(hdr.dataWindow() == inHdr.dataWindow()))
THROW (IEX_NAMESPACE::ArgExc, "Cannot copy pixels from image "
"file \"" << in.fileName() << "\" to image "
"file \"" << fileName() << "\". The "
"files have different data windows.");
if (!(hdr.lineOrder() == inHdr.lineOrder()))
THROW (IEX_NAMESPACE::ArgExc, "Quick pixel copy from image "
"file \"" << in.fileName() << "\" to image "
"file \"" << fileName() << "\" failed. "
"The files have different line orders.");
if (!(hdr.compression() == inHdr.compression()))
THROW (IEX_NAMESPACE::ArgExc, "Quick pixel copy from image "
"file \"" << in.fileName() << "\" to image "
"file \"" << fileName() << "\" failed. "
"The files use different compression methods.");
if (!(hdr.channels() == inHdr.channels()))
THROW (IEX_NAMESPACE::ArgExc, "Quick pixel copy from image "
"file \"" << in.fileName() << "\" to image "
"file \"" << fileName() << "\" "
"failed. The files have different channel "
"lists.");
// Verify that no pixel data have been written to this file yet.
//
if (!_data->tileOffsets.isEmpty())
THROW (IEX_NAMESPACE::LogicExc, "Quick pixel copy from image "
"file \"" << in.fileName() << "\" to image "
"file \"" << _data->_streamData->os->fileName() << "\" "
"failed. \"" << fileName() << "\" "
"already contains pixel data.");
int numAllTiles = in.totalTiles();
Lock lock (*_data->_streamData);
//
// special handling for random tiles
//
vector<int> dx_list(_data->lineOrder==RANDOM_Y ? numAllTiles : 1);
vector<int> dy_list(_data->lineOrder==RANDOM_Y ? numAllTiles : 1);
vector<int> lx_list(_data->lineOrder==RANDOM_Y ? numAllTiles : 1);
vector<int> ly_list(_data->lineOrder==RANDOM_Y ? numAllTiles : 1);
if(_data->lineOrder==RANDOM_Y)
{
in.getTileOrder(&dx_list[0],&dy_list[0],&lx_list[0],&ly_list[0]);
_data->nextTileToWrite.dx=dx_list[0];
_data->nextTileToWrite.dy=dy_list[0];
_data->nextTileToWrite.lx=lx_list[0];
_data->nextTileToWrite.ly=ly_list[0];
}
vector<char> data(4096);
for (int i = 0; i < numAllTiles; ++i)
{
int dx = _data->nextTileToWrite.dx;
int dy = _data->nextTileToWrite.dy;
int lx = _data->nextTileToWrite.lx;
int ly = _data->nextTileToWrite.ly;
Int64 dataSize = data.size();
in.rawTileData (dx, dy, lx, ly, &data[0], dataSize);
if(dataSize>data.size())
{
data.resize(dataSize);
in.rawTileData (dx, dy, lx, ly, &data[0], dataSize);
}
Int64 sampleCountTableSize = *(Int64 *)(&data[0] + 16);
Int64 pixelDataSize = *(Int64 *)(&data[0] + 24);
Int64 unpackedPixelDataSize = *(Int64 *)(&data[0] + 32);
char * sampleCountTable = &data[0]+40;
char * pixelData = sampleCountTable + sampleCountTableSize;
writeTileData (_data, dx, dy, lx, ly, pixelData, pixelDataSize,unpackedPixelDataSize,sampleCountTable,sampleCountTableSize);
if(_data->lineOrder==RANDOM_Y)
{
if(i<numAllTiles-1)
{
_data->nextTileToWrite.dx=dx_list[i+1];
_data->nextTileToWrite.dy=dy_list[i+1];
_data->nextTileToWrite.lx=lx_list[i+1];
_data->nextTileToWrite.ly=ly_list[i+1];
}
}else{
_data->nextTileToWrite = _data->nextTileCoord (_data->nextTileToWrite);
}
}
}
void
DeepTiledOutputFile::copyPixels (DeepTiledInputPart &in)
{
copyPixels(*in.file);
}
unsigned int
DeepTiledOutputFile::tileXSize () const
{
return _data->tileDesc.xSize;
}
unsigned int
DeepTiledOutputFile::tileYSize () const
{
return _data->tileDesc.ySize;
}
LevelMode
DeepTiledOutputFile::levelMode () const
{
return _data->tileDesc.mode;
}
LevelRoundingMode
DeepTiledOutputFile::levelRoundingMode () const
{
return _data->tileDesc.roundingMode;
}
int
DeepTiledOutputFile::numLevels () const
{
if (levelMode() == RIPMAP_LEVELS)
THROW (IEX_NAMESPACE::LogicExc, "Error calling numLevels() on image "
"file \"" << fileName() << "\" "
"(numLevels() is not defined for RIPMAPs).");
return _data->numXLevels;
}
int
DeepTiledOutputFile::numXLevels () const
{
return _data->numXLevels;
}
int
DeepTiledOutputFile::numYLevels () const
{
return _data->numYLevels;
}
bool
DeepTiledOutputFile::isValidLevel (int lx, int ly) const
{
if (lx < 0 || ly < 0)
return false;
if (levelMode() == MIPMAP_LEVELS && lx != ly)
return false;
if (lx >= numXLevels() || ly >= numYLevels())
return false;
return true;
}
int
DeepTiledOutputFile::levelWidth (int lx) const
{
try
{
int retVal = levelSize (_data->minX, _data->maxX, lx,
_data->tileDesc.roundingMode);
return retVal;
}
catch (IEX_NAMESPACE::BaseExc &e)
{
REPLACE_EXC (e, "Error calling levelWidth() on image "
"file \"" << fileName() << "\". " << e.what());
throw;
}
}
int
DeepTiledOutputFile::levelHeight (int ly) const
{
try
{
return levelSize (_data->minY, _data->maxY, ly,
_data->tileDesc.roundingMode);
}
catch (IEX_NAMESPACE::BaseExc &e)
{
REPLACE_EXC (e, "Error calling levelHeight() on image "
"file \"" << fileName() << "\". " << e.what());
throw;
}
}
int
DeepTiledOutputFile::numXTiles (int lx) const
{
if (lx < 0 || lx >= _data->numXLevels)
THROW (IEX_NAMESPACE::LogicExc, "Error calling numXTiles() on image "
"file \"" << _data->_streamData->os->fileName() << "\" "
"(Argument is not in valid range).");
return _data->numXTiles[lx];
}
int
DeepTiledOutputFile::numYTiles (int ly) const
{
if (ly < 0 || ly >= _data->numYLevels)
THROW (IEX_NAMESPACE::LogicExc, "Error calling numXTiles() on image "
"file \"" << _data->_streamData->os->fileName() << "\" "
"(Argument is not in valid range).");
return _data->numYTiles[ly];
}
Box2i
DeepTiledOutputFile::dataWindowForLevel (int l) const
{
return dataWindowForLevel (l, l);
}
Box2i
DeepTiledOutputFile::dataWindowForLevel (int lx, int ly) const
{
try
{
return OPENEXR_IMF_INTERNAL_NAMESPACE::dataWindowForLevel (
_data->tileDesc,
_data->minX, _data->maxX,
_data->minY, _data->maxY,
lx, ly);
}
catch (IEX_NAMESPACE::BaseExc &e)
{
REPLACE_EXC (e, "Error calling dataWindowForLevel() on image "
"file \"" << fileName() << "\". " << e.what());
throw;
}
}
Box2i
DeepTiledOutputFile::dataWindowForTile (int dx, int dy, int l) const
{
return dataWindowForTile (dx, dy, l, l);
}
Box2i
DeepTiledOutputFile::dataWindowForTile (int dx, int dy, int lx, int ly) const
{
try
{
if (!isValidTile (dx, dy, lx, ly))
throw IEX_NAMESPACE::ArgExc ("Arguments not in valid range.");
return OPENEXR_IMF_INTERNAL_NAMESPACE::dataWindowForTile (
_data->tileDesc,
_data->minX, _data->maxX,
_data->minY, _data->maxY,
dx, dy,
lx, ly);
}
catch (IEX_NAMESPACE::BaseExc &e)
{
REPLACE_EXC (e, "Error calling dataWindowForTile() on image "
"file \"" << fileName() << "\". " << e.what());
throw;
}
}
bool
DeepTiledOutputFile::isValidTile (int dx, int dy, int lx, int ly) const
{
return ((lx < _data->numXLevels && lx >= 0) &&
(ly < _data->numYLevels && ly >= 0) &&
(dx < _data->numXTiles[lx] && dx >= 0) &&
(dy < _data->numYTiles[ly] && dy >= 0));
}
void
DeepTiledOutputFile::updatePreviewImage (const PreviewRgba newPixels[])
{
Lock lock (*_data->_streamData);
if (_data->previewPosition <= 0)
THROW (IEX_NAMESPACE::LogicExc, "Cannot update preview image pixels. "
"File \"" << fileName() << "\" does not "
"contain a preview image.");
//
// Store the new pixels in the header's preview image attribute.
//
PreviewImageAttribute &pia =
_data->header.typedAttribute <PreviewImageAttribute> ("preview");
PreviewImage &pi = pia.value();
PreviewRgba *pixels = pi.pixels();
int numPixels = pi.width() * pi.height();
for (int i = 0; i < numPixels; ++i)
pixels[i] = newPixels[i];
//
// Save the current file position, jump to the position in
// the file where the preview image starts, store the new
// preview image, and jump back to the saved file position.
//
Int64 savedPosition = _data->_streamData->os->tellp();
try
{
_data->_streamData->os->seekp (_data->previewPosition);
pia.writeValueTo (*_data->_streamData->os, _data->version);
_data->_streamData->os->seekp (savedPosition);
}
catch (IEX_NAMESPACE::BaseExc &e)
{
REPLACE_EXC (e, "Cannot update preview image pixels for "
"file \"" << fileName() << "\". " << e.what());
throw;
}
}
void
DeepTiledOutputFile::breakTile
(int dx, int dy,
int lx, int ly,
int offset,
int length,
char c)
{
Lock lock (*_data->_streamData);
Int64 position = _data->tileOffsets (dx, dy, lx, ly);
if (!position)
THROW (IEX_NAMESPACE::ArgExc,
"Cannot overwrite tile "
"(" << dx << ", " << dy << ", " << lx << "," << ly << "). "
"The tile has not yet been stored in "
"file \"" << fileName() << "\".");
_data->_streamData->currentPosition = 0;
_data->_streamData->os->seekp (position + offset);
for (int i = 0; i < length; ++i)
_data->_streamData->os->write (&c, 1);
}
OPENEXR_IMF_INTERNAL_NAMESPACE_SOURCE_EXIT
Reference in New Issue Copy Permalink