mirror of
https://github.com/etlegacy/EasyGen.git
synced 2024-11-10 06:42:16 +00:00
1863 lines
58 KiB
C++
1863 lines
58 KiB
C++
///////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// 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.
|
|
//
|
|
///////////////////////////////////////////////////////////////////////////
|
|
|
|
|
|
|
|
//-----------------------------------------------------------------------------
|
|
//
|
|
// Miscellaneous helper functions for OpenEXR image file I/O
|
|
//
|
|
//-----------------------------------------------------------------------------
|
|
|
|
#include <ImfMisc.h>
|
|
#include <ImfHeader.h>
|
|
#include <ImfAttribute.h>
|
|
#include <ImfCompressor.h>
|
|
#include <ImfChannelList.h>
|
|
#include <ImfXdr.h>
|
|
#include <ImathFun.h>
|
|
#include <Iex.h>
|
|
#include <ImfStdIO.h>
|
|
#include <ImfConvert.h>
|
|
#include <ImfPartType.h>
|
|
#include <ImfTileDescription.h>
|
|
#include "ImfNamespace.h"
|
|
|
|
OPENEXR_IMF_INTERNAL_NAMESPACE_SOURCE_ENTER
|
|
|
|
using IMATH_NAMESPACE::Box2i;
|
|
using IMATH_NAMESPACE::divp;
|
|
using IMATH_NAMESPACE::modp;
|
|
using std::vector;
|
|
|
|
int
|
|
pixelTypeSize (PixelType type)
|
|
{
|
|
int size;
|
|
|
|
switch (type)
|
|
{
|
|
case UINT:
|
|
|
|
size = Xdr::size <unsigned int> ();
|
|
break;
|
|
|
|
case HALF:
|
|
|
|
size = Xdr::size <half> ();
|
|
break;
|
|
|
|
case FLOAT:
|
|
|
|
size = Xdr::size <float> ();
|
|
break;
|
|
|
|
default:
|
|
|
|
throw IEX_NAMESPACE::ArgExc ("Unknown pixel type.");
|
|
}
|
|
|
|
return size;
|
|
}
|
|
|
|
|
|
int
|
|
numSamples (int s, int a, int b)
|
|
{
|
|
int a1 = divp (a, s);
|
|
int b1 = divp (b, s);
|
|
return b1 - a1 + ((a1 * s < a)? 0: 1);
|
|
}
|
|
|
|
|
|
size_t
|
|
bytesPerLineTable (const Header &header,
|
|
vector<size_t> &bytesPerLine)
|
|
{
|
|
const Box2i &dataWindow = header.dataWindow();
|
|
const ChannelList &channels = header.channels();
|
|
|
|
bytesPerLine.resize (dataWindow.max.y - dataWindow.min.y + 1);
|
|
|
|
for (ChannelList::ConstIterator c = channels.begin();
|
|
c != channels.end();
|
|
++c)
|
|
{
|
|
int nBytes = pixelTypeSize (c.channel().type) *
|
|
(dataWindow.max.x - dataWindow.min.x + 1) /
|
|
c.channel().xSampling;
|
|
|
|
for (int y = dataWindow.min.y, i = 0; y <= dataWindow.max.y; ++y, ++i)
|
|
if (modp (y, c.channel().ySampling) == 0)
|
|
bytesPerLine[i] += nBytes;
|
|
}
|
|
|
|
size_t maxBytesPerLine = 0;
|
|
|
|
for (int y = dataWindow.min.y, i = 0; y <= dataWindow.max.y; ++y, ++i)
|
|
if (maxBytesPerLine < bytesPerLine[i])
|
|
maxBytesPerLine = bytesPerLine[i];
|
|
|
|
return maxBytesPerLine;
|
|
}
|
|
|
|
|
|
const int&
|
|
sampleCount(const char* base, int xStride, int yStride, int x, int y)
|
|
{
|
|
const char* ptr = base + y * yStride + x * xStride;
|
|
int* intPtr = (int*) ptr;
|
|
|
|
return *intPtr;
|
|
}
|
|
|
|
int&
|
|
sampleCount(char* base, int xStride, int yStride, int x, int y)
|
|
{
|
|
char* ptr = base + y * yStride + x * xStride;
|
|
int* intPtr = (int*) ptr;
|
|
|
|
return *intPtr;
|
|
}
|
|
|
|
|
|
size_t
|
|
bytesPerDeepLineTable (const Header &header,
|
|
int minY, int maxY,
|
|
const char* base,
|
|
int xStride,
|
|
int yStride,
|
|
vector<size_t> &bytesPerLine)
|
|
{
|
|
const Box2i &dataWindow = header.dataWindow();
|
|
const ChannelList &channels = header.channels();
|
|
|
|
for (ChannelList::ConstIterator c = channels.begin();
|
|
c != channels.end();
|
|
++c)
|
|
{
|
|
for (int y = minY; y <= maxY; ++y)
|
|
if (modp (y, c.channel().ySampling) == 0)
|
|
{
|
|
int nBytes = 0;
|
|
for (int x = dataWindow.min.x; x <= dataWindow.max.x; x++)
|
|
{
|
|
if (modp (x, c.channel().xSampling) == 0)
|
|
nBytes += pixelTypeSize (c.channel().type) *
|
|
sampleCount(base, xStride, yStride, x, y);
|
|
}
|
|
bytesPerLine[y - dataWindow.min.y] += nBytes;
|
|
}
|
|
}
|
|
|
|
size_t maxBytesPerLine = 0;
|
|
|
|
for (int y = minY; y <= maxY; ++y)
|
|
if (maxBytesPerLine < bytesPerLine[y - dataWindow.min.y])
|
|
maxBytesPerLine = bytesPerLine[y - dataWindow.min.y];
|
|
|
|
return maxBytesPerLine;
|
|
}
|
|
|
|
|
|
size_t
|
|
bytesPerDeepLineTable (const Header &header,
|
|
char* base,
|
|
int xStride,
|
|
int yStride,
|
|
vector<size_t> &bytesPerLine)
|
|
{
|
|
return bytesPerDeepLineTable(header,
|
|
header.dataWindow().min.y,
|
|
header.dataWindow().max.y,
|
|
base,
|
|
xStride,
|
|
yStride,
|
|
bytesPerLine);
|
|
}
|
|
|
|
|
|
void
|
|
offsetInLineBufferTable (const vector<size_t> &bytesPerLine,
|
|
int scanline1, int scanline2,
|
|
int linesInLineBuffer,
|
|
vector<size_t> &offsetInLineBuffer)
|
|
{
|
|
offsetInLineBuffer.resize (bytesPerLine.size());
|
|
|
|
size_t offset = 0;
|
|
|
|
for (int i = scanline1; i <= scanline2; ++i)
|
|
{
|
|
if (i % linesInLineBuffer == 0)
|
|
offset = 0;
|
|
|
|
offsetInLineBuffer[i] = offset;
|
|
offset += bytesPerLine[i];
|
|
}
|
|
}
|
|
|
|
|
|
void
|
|
offsetInLineBufferTable (const vector<size_t> &bytesPerLine,
|
|
int linesInLineBuffer,
|
|
vector<size_t> &offsetInLineBuffer)
|
|
{
|
|
offsetInLineBufferTable (bytesPerLine,
|
|
0, bytesPerLine.size() - 1,
|
|
linesInLineBuffer,
|
|
offsetInLineBuffer);
|
|
}
|
|
|
|
|
|
int
|
|
lineBufferMinY (int y, int minY, int linesInLineBuffer)
|
|
{
|
|
return ((y - minY) / linesInLineBuffer) * linesInLineBuffer + minY;
|
|
}
|
|
|
|
|
|
int
|
|
lineBufferMaxY (int y, int minY, int linesInLineBuffer)
|
|
{
|
|
return lineBufferMinY (y, minY, linesInLineBuffer) + linesInLineBuffer - 1;
|
|
}
|
|
|
|
|
|
Compressor::Format
|
|
defaultFormat (Compressor * compressor)
|
|
{
|
|
return compressor? compressor->format(): Compressor::XDR;
|
|
}
|
|
|
|
|
|
int
|
|
numLinesInBuffer (Compressor * compressor)
|
|
{
|
|
return compressor? compressor->numScanLines(): 1;
|
|
}
|
|
|
|
|
|
void
|
|
copyIntoFrameBuffer (const char *& readPtr,
|
|
char * writePtr,
|
|
char * endPtr,
|
|
size_t xStride,
|
|
bool fill,
|
|
double fillValue,
|
|
Compressor::Format format,
|
|
PixelType typeInFrameBuffer,
|
|
PixelType typeInFile)
|
|
{
|
|
//
|
|
// Copy a horizontal row of pixels from an input
|
|
// file's line or tile buffer to a frame buffer.
|
|
//
|
|
|
|
if (fill)
|
|
{
|
|
//
|
|
// The file contains no data for this channel.
|
|
// Store a default value in the frame buffer.
|
|
//
|
|
|
|
switch (typeInFrameBuffer)
|
|
{
|
|
case UINT:
|
|
|
|
{
|
|
unsigned int fillVal = (unsigned int) (fillValue);
|
|
|
|
while (writePtr <= endPtr)
|
|
{
|
|
*(unsigned int *) writePtr = fillVal;
|
|
writePtr += xStride;
|
|
}
|
|
}
|
|
break;
|
|
|
|
case HALF:
|
|
|
|
{
|
|
half fillVal = half (fillValue);
|
|
|
|
while (writePtr <= endPtr)
|
|
{
|
|
*(half *) writePtr = fillVal;
|
|
writePtr += xStride;
|
|
}
|
|
}
|
|
break;
|
|
|
|
case FLOAT:
|
|
|
|
{
|
|
float fillVal = float (fillValue);
|
|
|
|
while (writePtr <= endPtr)
|
|
{
|
|
*(float *) writePtr = fillVal;
|
|
writePtr += xStride;
|
|
}
|
|
}
|
|
break;
|
|
|
|
default:
|
|
|
|
throw IEX_NAMESPACE::ArgExc ("Unknown pixel data type.");
|
|
}
|
|
}
|
|
else if (format == Compressor::XDR)
|
|
{
|
|
//
|
|
// The the line or tile buffer is in XDR format.
|
|
//
|
|
// Convert the pixels from the file's machine-
|
|
// independent representation, and store the
|
|
// results in the frame buffer.
|
|
//
|
|
|
|
switch (typeInFrameBuffer)
|
|
{
|
|
case UINT:
|
|
|
|
switch (typeInFile)
|
|
{
|
|
case UINT:
|
|
|
|
while (writePtr <= endPtr)
|
|
{
|
|
Xdr::read <CharPtrIO> (readPtr, *(unsigned int *) writePtr);
|
|
writePtr += xStride;
|
|
}
|
|
break;
|
|
|
|
case HALF:
|
|
|
|
while (writePtr <= endPtr)
|
|
{
|
|
half h;
|
|
Xdr::read <CharPtrIO> (readPtr, h);
|
|
*(unsigned int *) writePtr = halfToUint (h);
|
|
writePtr += xStride;
|
|
}
|
|
break;
|
|
|
|
case FLOAT:
|
|
|
|
while (writePtr <= endPtr)
|
|
{
|
|
float f;
|
|
Xdr::read <CharPtrIO> (readPtr, f);
|
|
*(unsigned int *)writePtr = floatToUint (f);
|
|
writePtr += xStride;
|
|
}
|
|
break;
|
|
|
|
default:
|
|
throw IEX_NAMESPACE::ArgExc ("Unknown pixel data type.");
|
|
}
|
|
break;
|
|
|
|
case HALF:
|
|
|
|
switch (typeInFile)
|
|
{
|
|
case UINT:
|
|
|
|
while (writePtr <= endPtr)
|
|
{
|
|
unsigned int ui;
|
|
Xdr::read <CharPtrIO> (readPtr, ui);
|
|
*(half *) writePtr = uintToHalf (ui);
|
|
writePtr += xStride;
|
|
}
|
|
break;
|
|
|
|
case HALF:
|
|
|
|
while (writePtr <= endPtr)
|
|
{
|
|
Xdr::read <CharPtrIO> (readPtr, *(half *) writePtr);
|
|
writePtr += xStride;
|
|
}
|
|
break;
|
|
|
|
case FLOAT:
|
|
|
|
while (writePtr <= endPtr)
|
|
{
|
|
float f;
|
|
Xdr::read <CharPtrIO> (readPtr, f);
|
|
*(half *) writePtr = floatToHalf (f);
|
|
writePtr += xStride;
|
|
}
|
|
break;
|
|
default:
|
|
|
|
throw IEX_NAMESPACE::ArgExc ("Unknown pixel data type.");
|
|
}
|
|
break;
|
|
|
|
case FLOAT:
|
|
|
|
switch (typeInFile)
|
|
{
|
|
case UINT:
|
|
|
|
while (writePtr <= endPtr)
|
|
{
|
|
unsigned int ui;
|
|
Xdr::read <CharPtrIO> (readPtr, ui);
|
|
*(float *) writePtr = float (ui);
|
|
writePtr += xStride;
|
|
}
|
|
break;
|
|
|
|
case HALF:
|
|
|
|
while (writePtr <= endPtr)
|
|
{
|
|
half h;
|
|
Xdr::read <CharPtrIO> (readPtr, h);
|
|
*(float *) writePtr = float (h);
|
|
writePtr += xStride;
|
|
}
|
|
break;
|
|
|
|
case FLOAT:
|
|
|
|
while (writePtr <= endPtr)
|
|
{
|
|
Xdr::read <CharPtrIO> (readPtr, *(float *) writePtr);
|
|
writePtr += xStride;
|
|
}
|
|
break;
|
|
default:
|
|
|
|
throw IEX_NAMESPACE::ArgExc ("Unknown pixel data type.");
|
|
}
|
|
break;
|
|
|
|
default:
|
|
|
|
throw IEX_NAMESPACE::ArgExc ("Unknown pixel data type.");
|
|
}
|
|
}
|
|
else
|
|
{
|
|
//
|
|
// The the line or tile buffer is in NATIVE format.
|
|
// Copy the results into the frame buffer.
|
|
//
|
|
|
|
switch (typeInFrameBuffer)
|
|
{
|
|
case UINT:
|
|
|
|
switch (typeInFile)
|
|
{
|
|
case UINT:
|
|
|
|
while (writePtr <= endPtr)
|
|
{
|
|
for (size_t i = 0; i < sizeof (unsigned int); ++i)
|
|
writePtr[i] = readPtr[i];
|
|
|
|
readPtr += sizeof (unsigned int);
|
|
writePtr += xStride;
|
|
}
|
|
break;
|
|
|
|
case HALF:
|
|
|
|
while (writePtr <= endPtr)
|
|
{
|
|
half h = *(half *) readPtr;
|
|
*(unsigned int *) writePtr = halfToUint (h);
|
|
readPtr += sizeof (half);
|
|
writePtr += xStride;
|
|
}
|
|
break;
|
|
|
|
case FLOAT:
|
|
|
|
while (writePtr <= endPtr)
|
|
{
|
|
float f;
|
|
|
|
for (size_t i = 0; i < sizeof (float); ++i)
|
|
((char *)&f)[i] = readPtr[i];
|
|
|
|
*(unsigned int *)writePtr = floatToUint (f);
|
|
readPtr += sizeof (float);
|
|
writePtr += xStride;
|
|
}
|
|
break;
|
|
|
|
default:
|
|
|
|
throw IEX_NAMESPACE::ArgExc ("Unknown pixel data type.");
|
|
}
|
|
break;
|
|
|
|
case HALF:
|
|
|
|
switch (typeInFile)
|
|
{
|
|
case UINT:
|
|
|
|
while (writePtr <= endPtr)
|
|
{
|
|
unsigned int ui;
|
|
|
|
for (size_t i = 0; i < sizeof (unsigned int); ++i)
|
|
((char *)&ui)[i] = readPtr[i];
|
|
|
|
*(half *) writePtr = uintToHalf (ui);
|
|
readPtr += sizeof (unsigned int);
|
|
writePtr += xStride;
|
|
}
|
|
break;
|
|
|
|
case HALF:
|
|
|
|
while (writePtr <= endPtr)
|
|
{
|
|
*(half *) writePtr = *(half *)readPtr;
|
|
readPtr += sizeof (half);
|
|
writePtr += xStride;
|
|
}
|
|
break;
|
|
|
|
case FLOAT:
|
|
|
|
while (writePtr <= endPtr)
|
|
{
|
|
float f;
|
|
|
|
for (size_t i = 0; i < sizeof (float); ++i)
|
|
((char *)&f)[i] = readPtr[i];
|
|
|
|
*(half *) writePtr = floatToHalf (f);
|
|
readPtr += sizeof (float);
|
|
writePtr += xStride;
|
|
}
|
|
break;
|
|
default:
|
|
|
|
throw IEX_NAMESPACE::ArgExc ("Unknown pixel data type.");
|
|
}
|
|
break;
|
|
|
|
case FLOAT:
|
|
|
|
switch (typeInFile)
|
|
{
|
|
case UINT:
|
|
|
|
while (writePtr <= endPtr)
|
|
{
|
|
unsigned int ui;
|
|
|
|
for (size_t i = 0; i < sizeof (unsigned int); ++i)
|
|
((char *)&ui)[i] = readPtr[i];
|
|
|
|
*(float *) writePtr = float (ui);
|
|
readPtr += sizeof (unsigned int);
|
|
writePtr += xStride;
|
|
}
|
|
break;
|
|
|
|
case HALF:
|
|
|
|
while (writePtr <= endPtr)
|
|
{
|
|
half h = *(half *) readPtr;
|
|
*(float *) writePtr = float (h);
|
|
readPtr += sizeof (half);
|
|
writePtr += xStride;
|
|
}
|
|
break;
|
|
|
|
case FLOAT:
|
|
|
|
while (writePtr <= endPtr)
|
|
{
|
|
for (size_t i = 0; i < sizeof (float); ++i)
|
|
writePtr[i] = readPtr[i];
|
|
|
|
readPtr += sizeof (float);
|
|
writePtr += xStride;
|
|
}
|
|
break;
|
|
default:
|
|
|
|
throw IEX_NAMESPACE::ArgExc ("Unknown pixel data type.");
|
|
}
|
|
break;
|
|
|
|
default:
|
|
|
|
throw IEX_NAMESPACE::ArgExc ("Unknown pixel data type.");
|
|
}
|
|
}
|
|
}
|
|
|
|
void
|
|
copyIntoDeepFrameBuffer (const char *& readPtr,
|
|
char * base,
|
|
const char* sampleCountBase,
|
|
ptrdiff_t sampleCountXStride,
|
|
ptrdiff_t sampleCountYStride,
|
|
int y, int minX, int maxX,
|
|
int xOffsetForSampleCount,
|
|
int yOffsetForSampleCount,
|
|
int xOffsetForData,
|
|
int yOffsetForData,
|
|
ptrdiff_t sampleStride,
|
|
ptrdiff_t xPointerStride,
|
|
ptrdiff_t yPointerStride,
|
|
bool fill,
|
|
double fillValue,
|
|
Compressor::Format format,
|
|
PixelType typeInFrameBuffer,
|
|
PixelType typeInFile)
|
|
{
|
|
//
|
|
// Copy a horizontal row of pixels from an input
|
|
// file's line or tile buffer to a frame buffer.
|
|
//
|
|
|
|
if (fill)
|
|
{
|
|
//
|
|
// The file contains no data for this channel.
|
|
// Store a default value in the frame buffer.
|
|
//
|
|
|
|
switch (typeInFrameBuffer)
|
|
{
|
|
case UINT:
|
|
|
|
{
|
|
unsigned int fillVal = (unsigned int) (fillValue);
|
|
|
|
for (int x = minX; x <= maxX; x++)
|
|
{
|
|
char* writePtr = *(char **)(base+(y-yOffsetForData)*yPointerStride + (x-xOffsetForData)*xPointerStride);
|
|
if(writePtr)
|
|
{
|
|
int count = sampleCount(sampleCountBase,
|
|
sampleCountXStride,
|
|
sampleCountYStride,
|
|
x - xOffsetForSampleCount,
|
|
y - yOffsetForSampleCount);
|
|
for (int i = 0; i < count; i++)
|
|
{
|
|
*(unsigned int *) writePtr = fillVal;
|
|
writePtr += sampleStride;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
break;
|
|
|
|
case HALF:
|
|
|
|
{
|
|
half fillVal = half (fillValue);
|
|
|
|
for (int x = minX; x <= maxX; x++)
|
|
{
|
|
char* writePtr = *(char **)(base+(y-yOffsetForData)*yPointerStride + (x-xOffsetForData)*xPointerStride);
|
|
|
|
if(writePtr)
|
|
{
|
|
int count = sampleCount(sampleCountBase,
|
|
sampleCountXStride,
|
|
sampleCountYStride,
|
|
x - xOffsetForSampleCount,
|
|
y - yOffsetForSampleCount);
|
|
for (int i = 0; i < count; i++)
|
|
{
|
|
*(half *) writePtr = fillVal;
|
|
writePtr += sampleStride;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
break;
|
|
|
|
case FLOAT:
|
|
|
|
{
|
|
float fillVal = float (fillValue);
|
|
|
|
for (int x = minX; x <= maxX; x++)
|
|
{
|
|
char* writePtr = *(char **)(base+(y-yOffsetForData)*yPointerStride + (x-xOffsetForData)*xPointerStride);
|
|
|
|
if(writePtr)
|
|
{
|
|
int count = sampleCount(sampleCountBase,
|
|
sampleCountXStride,
|
|
sampleCountYStride,
|
|
x - xOffsetForSampleCount,
|
|
y - yOffsetForSampleCount);
|
|
for (int i = 0; i < count; i++)
|
|
{
|
|
*(float *) writePtr = fillVal;
|
|
writePtr += sampleStride;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
break;
|
|
|
|
default:
|
|
|
|
throw IEX_NAMESPACE::ArgExc ("Unknown pixel data type.");
|
|
}
|
|
}
|
|
else if (format == Compressor::XDR)
|
|
{
|
|
//
|
|
// The the line or tile buffer is in XDR format.
|
|
//
|
|
// Convert the pixels from the file's machine-
|
|
// independent representation, and store the
|
|
// results in the frame buffer.
|
|
//
|
|
|
|
switch (typeInFrameBuffer)
|
|
{
|
|
case UINT:
|
|
|
|
switch (typeInFile)
|
|
{
|
|
case UINT:
|
|
|
|
for (int x = minX; x <= maxX; x++)
|
|
{
|
|
char* writePtr = *(char **)(base+(y-yOffsetForData)*yPointerStride + (x-xOffsetForData)*xPointerStride);
|
|
|
|
int count = sampleCount(sampleCountBase,
|
|
sampleCountXStride,
|
|
sampleCountYStride,
|
|
x - xOffsetForSampleCount,
|
|
y - yOffsetForSampleCount);
|
|
if(writePtr)
|
|
{
|
|
|
|
for (int i = 0; i < count; i++)
|
|
{
|
|
Xdr::read <CharPtrIO> (readPtr, *(unsigned int *) writePtr);
|
|
writePtr += sampleStride;
|
|
}
|
|
}else{
|
|
Xdr::skip <CharPtrIO> (readPtr,count*Xdr::size<unsigned int>());
|
|
}
|
|
}
|
|
break;
|
|
|
|
case HALF:
|
|
|
|
for (int x = minX; x <= maxX; x++)
|
|
{
|
|
char* writePtr = *(char **)(base+(y-yOffsetForData)*yPointerStride + (x-xOffsetForData)*xPointerStride);
|
|
|
|
int count = sampleCount(sampleCountBase,
|
|
sampleCountXStride,
|
|
sampleCountYStride,
|
|
x - xOffsetForSampleCount,
|
|
y - yOffsetForSampleCount);
|
|
if(writePtr)
|
|
{
|
|
|
|
for (int i = 0; i < count; i++)
|
|
{
|
|
half h;
|
|
Xdr::read <CharPtrIO> (readPtr, h);
|
|
*(unsigned int *) writePtr = halfToUint (h);
|
|
writePtr += sampleStride;
|
|
}
|
|
}else{
|
|
Xdr::skip <CharPtrIO> (readPtr,count*Xdr::size<half>());
|
|
}
|
|
}
|
|
break;
|
|
|
|
case FLOAT:
|
|
|
|
for (int x = minX; x <= maxX; x++)
|
|
{
|
|
char* writePtr = *(char **)(base+(y-yOffsetForData)*yPointerStride + (x-xOffsetForData)*xPointerStride);
|
|
|
|
int count = sampleCount(sampleCountBase,
|
|
sampleCountXStride,
|
|
sampleCountYStride,
|
|
x - xOffsetForSampleCount,
|
|
y - yOffsetForSampleCount);
|
|
|
|
if(writePtr)
|
|
{
|
|
for (int i = 0; i < count; i++)
|
|
{
|
|
float f;
|
|
Xdr::read <CharPtrIO> (readPtr, f);
|
|
*(unsigned int *)writePtr = floatToUint (f);
|
|
writePtr += sampleStride;
|
|
}
|
|
}else{
|
|
Xdr::skip <CharPtrIO> (readPtr,count*Xdr::size<float>());
|
|
}
|
|
|
|
}
|
|
break;
|
|
default:
|
|
throw IEX_NAMESPACE::ArgExc ("Unknown pixel data type.");
|
|
}
|
|
break;
|
|
|
|
case HALF:
|
|
|
|
switch (typeInFile)
|
|
{
|
|
case UINT:
|
|
|
|
for (int x = minX; x <= maxX; x++)
|
|
{
|
|
char* writePtr = *(char **)(base+(y-yOffsetForData)*yPointerStride + (x-xOffsetForData)*xPointerStride);
|
|
|
|
int count = sampleCount(sampleCountBase,
|
|
sampleCountXStride,
|
|
sampleCountYStride,
|
|
x - xOffsetForSampleCount,
|
|
y - yOffsetForSampleCount);
|
|
if(writePtr)
|
|
{
|
|
|
|
for (int i = 0; i < count; i++)
|
|
{
|
|
unsigned int ui;
|
|
Xdr::read <CharPtrIO> (readPtr, ui);
|
|
*(half *) writePtr = uintToHalf (ui);
|
|
writePtr += sampleStride;
|
|
}
|
|
}else{
|
|
Xdr::skip <CharPtrIO> (readPtr,count*Xdr::size<unsigned int>());
|
|
}
|
|
}
|
|
break;
|
|
|
|
case HALF:
|
|
|
|
for (int x = minX; x <= maxX; x++)
|
|
{
|
|
char* writePtr = *(char **)(base+(y-yOffsetForData)*yPointerStride + (x-xOffsetForData)*xPointerStride);
|
|
|
|
int count = sampleCount(sampleCountBase,
|
|
sampleCountXStride,
|
|
sampleCountYStride,
|
|
x - xOffsetForSampleCount,
|
|
y - yOffsetForSampleCount);
|
|
if(writePtr)
|
|
{
|
|
|
|
for (int i = 0; i < count; i++)
|
|
{
|
|
Xdr::read <CharPtrIO> (readPtr, *(half *) writePtr);
|
|
writePtr += sampleStride;
|
|
}
|
|
}else{
|
|
Xdr::skip <CharPtrIO> (readPtr,count*Xdr::size<half>());
|
|
}
|
|
}
|
|
break;
|
|
|
|
case FLOAT:
|
|
|
|
for (int x = minX; x <= maxX; x++)
|
|
{
|
|
char* writePtr = *(char **) (base+(y-yOffsetForData)*yPointerStride + (x-xOffsetForData)*xPointerStride);
|
|
|
|
int count = sampleCount(sampleCountBase,
|
|
sampleCountXStride,
|
|
sampleCountYStride,
|
|
x - xOffsetForSampleCount,
|
|
y - yOffsetForSampleCount);
|
|
if(writePtr)
|
|
{
|
|
for (int i = 0; i < count; i++)
|
|
{
|
|
float f;
|
|
Xdr::read <CharPtrIO> (readPtr, f);
|
|
*(half *) writePtr = floatToHalf (f);
|
|
writePtr += sampleStride;
|
|
}
|
|
}else{
|
|
Xdr::skip <CharPtrIO> (readPtr,count*Xdr::size<float>());
|
|
}
|
|
}
|
|
break;
|
|
default:
|
|
|
|
throw IEX_NAMESPACE::ArgExc ("Unknown pixel data type.");
|
|
}
|
|
break;
|
|
|
|
case FLOAT:
|
|
|
|
switch (typeInFile)
|
|
{
|
|
case UINT:
|
|
|
|
for (int x = minX; x <= maxX; x++)
|
|
{
|
|
char* writePtr = *(char **)(base+(y-yOffsetForData)*yPointerStride + (x-xOffsetForData)*xPointerStride);
|
|
|
|
int count = sampleCount(sampleCountBase,
|
|
sampleCountXStride,
|
|
sampleCountYStride,
|
|
x - xOffsetForSampleCount,
|
|
y - yOffsetForSampleCount);
|
|
if(writePtr)
|
|
{
|
|
for (int i = 0; i < count; i++)
|
|
{
|
|
unsigned int ui;
|
|
Xdr::read <CharPtrIO> (readPtr, ui);
|
|
*(float *) writePtr = float (ui);
|
|
writePtr += sampleStride;
|
|
}
|
|
}else{
|
|
Xdr::skip <CharPtrIO> (readPtr,count*Xdr::size<unsigned int>());
|
|
}
|
|
}
|
|
break;
|
|
|
|
case HALF:
|
|
|
|
for (int x = minX; x <= maxX; x++)
|
|
{
|
|
char* writePtr = *(char **)(base+(y-yOffsetForData)*yPointerStride + (x-xOffsetForData)*xPointerStride);
|
|
|
|
int count = sampleCount(sampleCountBase,
|
|
sampleCountXStride,
|
|
sampleCountYStride,
|
|
x - xOffsetForSampleCount,
|
|
y - yOffsetForSampleCount);
|
|
if(writePtr)
|
|
{
|
|
|
|
for (int i = 0; i < count; i++)
|
|
{
|
|
half h;
|
|
Xdr::read <CharPtrIO> (readPtr, h);
|
|
*(float *) writePtr = float (h);
|
|
writePtr += sampleStride;
|
|
}
|
|
|
|
}else{
|
|
Xdr::skip <CharPtrIO> (readPtr,count*Xdr::size<half>());
|
|
}
|
|
}
|
|
break;
|
|
|
|
case FLOAT:
|
|
|
|
for (int x = minX; x <= maxX; x++)
|
|
{
|
|
char* writePtr = *(char **)(base+(y-yOffsetForData)*yPointerStride + (x-xOffsetForData)*xPointerStride);
|
|
|
|
int count = sampleCount(sampleCountBase,
|
|
sampleCountXStride,
|
|
sampleCountYStride,
|
|
x - xOffsetForSampleCount,
|
|
y - yOffsetForSampleCount);
|
|
if(writePtr)
|
|
{
|
|
|
|
for (int i = 0; i < count; i++)
|
|
{
|
|
Xdr::read <CharPtrIO> (readPtr, *(float *) writePtr);
|
|
writePtr += sampleStride;
|
|
}
|
|
} else{
|
|
Xdr::skip <CharPtrIO> (readPtr,count*Xdr::size<float>());
|
|
}
|
|
|
|
}
|
|
break;
|
|
default:
|
|
|
|
throw IEX_NAMESPACE::ArgExc ("Unknown pixel data type.");
|
|
}
|
|
break;
|
|
|
|
default:
|
|
|
|
throw IEX_NAMESPACE::ArgExc ("Unknown pixel data type.");
|
|
}
|
|
}
|
|
else
|
|
{
|
|
//
|
|
// The the line or tile buffer is in NATIVE format.
|
|
// Copy the results into the frame buffer.
|
|
//
|
|
|
|
switch (typeInFrameBuffer)
|
|
{
|
|
case UINT:
|
|
|
|
switch (typeInFile)
|
|
{
|
|
case UINT:
|
|
|
|
for (int x = minX; x <= maxX; x++)
|
|
{
|
|
char* writePtr = *(char **)(base+(y-yOffsetForData)*yPointerStride + (x-xOffsetForData)*xPointerStride);
|
|
|
|
int count = sampleCount(sampleCountBase,
|
|
sampleCountXStride,
|
|
sampleCountYStride,
|
|
x - xOffsetForSampleCount,
|
|
y - yOffsetForSampleCount);
|
|
|
|
if(writePtr)
|
|
{
|
|
for (int i = 0; i < count; i++)
|
|
{
|
|
for (size_t i = 0; i < sizeof (unsigned int); ++i)
|
|
writePtr[i] = readPtr[i];
|
|
|
|
readPtr += sizeof (unsigned int);
|
|
writePtr += sampleStride;
|
|
}
|
|
}else{
|
|
readPtr+=sizeof(unsigned int)*count;
|
|
}
|
|
}
|
|
break;
|
|
|
|
case HALF:
|
|
|
|
for (int x = minX; x <= maxX; x++)
|
|
{
|
|
char* writePtr = *(char **)(base+(y-yOffsetForData)*yPointerStride + (x-xOffsetForData)*xPointerStride);
|
|
|
|
int count = sampleCount(sampleCountBase,
|
|
sampleCountXStride,
|
|
sampleCountYStride,
|
|
x - xOffsetForSampleCount,
|
|
y - yOffsetForSampleCount);
|
|
|
|
if(writePtr)
|
|
{
|
|
for (int i = 0; i < count; i++)
|
|
{
|
|
half h = *(half *) readPtr;
|
|
*(unsigned int *) writePtr = halfToUint (h);
|
|
readPtr += sizeof (half);
|
|
writePtr += sampleStride;
|
|
}
|
|
}else{
|
|
readPtr+=sizeof(half)*count;
|
|
}
|
|
}
|
|
break;
|
|
|
|
case FLOAT:
|
|
|
|
for (int x = minX; x <= maxX; x++)
|
|
{
|
|
char* writePtr = *(char **)(base+(y-yOffsetForData)*yPointerStride + (x-xOffsetForData)*xPointerStride);
|
|
|
|
int count = sampleCount(sampleCountBase,
|
|
sampleCountXStride,
|
|
sampleCountYStride,
|
|
x - xOffsetForSampleCount,
|
|
y - yOffsetForSampleCount);
|
|
|
|
if(writePtr)
|
|
{
|
|
|
|
for (int i = 0; i < count; i++)
|
|
{
|
|
float f;
|
|
|
|
for (size_t i = 0; i < sizeof (float); ++i)
|
|
((char *)&f)[i] = readPtr[i];
|
|
|
|
*(unsigned int *)writePtr = floatToUint (f);
|
|
readPtr += sizeof (float);
|
|
writePtr += sampleStride;
|
|
}
|
|
}else{
|
|
readPtr+=sizeof(float)*count;
|
|
}
|
|
}
|
|
break;
|
|
default:
|
|
|
|
throw IEX_NAMESPACE::ArgExc ("Unknown pixel data type.");
|
|
}
|
|
break;
|
|
|
|
case HALF:
|
|
|
|
switch (typeInFile)
|
|
{
|
|
case UINT:
|
|
|
|
for (int x = minX; x <= maxX; x++)
|
|
{
|
|
char* writePtr = *(char **)(base+(y-yOffsetForData)*yPointerStride + (x-xOffsetForData)*xPointerStride);
|
|
|
|
int count = sampleCount(sampleCountBase,
|
|
sampleCountXStride,
|
|
sampleCountYStride,
|
|
x - xOffsetForSampleCount,
|
|
y - yOffsetForSampleCount);
|
|
|
|
if(writePtr)
|
|
{
|
|
for (int i = 0; i < count; i++)
|
|
{
|
|
unsigned int ui;
|
|
|
|
for (size_t i = 0; i < sizeof (unsigned int); ++i)
|
|
((char *)&ui)[i] = readPtr[i];
|
|
|
|
*(half *) writePtr = uintToHalf (ui);
|
|
readPtr += sizeof (unsigned int);
|
|
writePtr += sampleStride;
|
|
}
|
|
}else{
|
|
readPtr+=sizeof(unsigned int)*count;
|
|
}
|
|
}
|
|
break;
|
|
|
|
case HALF:
|
|
|
|
for (int x = minX; x <= maxX; x++)
|
|
{
|
|
char* writePtr = *(char **)(base+(y-yOffsetForData)*yPointerStride + (x-xOffsetForData)*xPointerStride);
|
|
|
|
int count = sampleCount(sampleCountBase,
|
|
sampleCountXStride,
|
|
sampleCountYStride,
|
|
x - xOffsetForSampleCount,
|
|
y - yOffsetForSampleCount);
|
|
|
|
if(writePtr)
|
|
{
|
|
for (int i = 0; i < count; i++)
|
|
{
|
|
*(half *) writePtr = *(half *)readPtr;
|
|
readPtr += sizeof (half);
|
|
writePtr += sampleStride;
|
|
}
|
|
}else{
|
|
readPtr+=sizeof(half)*count;
|
|
}
|
|
}
|
|
break;
|
|
|
|
case FLOAT:
|
|
|
|
for (int x = minX; x <= maxX; x++)
|
|
{
|
|
char* writePtr = *(char **)(base+(y-yOffsetForData)*yPointerStride + (x-xOffsetForData)*xPointerStride);
|
|
|
|
int count = sampleCount(sampleCountBase,
|
|
sampleCountXStride,
|
|
sampleCountYStride,
|
|
x - xOffsetForSampleCount,
|
|
y - yOffsetForSampleCount);
|
|
|
|
if(writePtr)
|
|
{
|
|
for (int i = 0; i < count; i++)
|
|
{
|
|
float f;
|
|
|
|
for (size_t i = 0; i < sizeof (float); ++i)
|
|
((char *)&f)[i] = readPtr[i];
|
|
|
|
*(half *) writePtr = floatToHalf (f);
|
|
readPtr += sizeof (float);
|
|
writePtr += sampleStride;
|
|
}
|
|
}else{
|
|
readPtr+=sizeof(float)*count;
|
|
}
|
|
}
|
|
break;
|
|
default:
|
|
|
|
throw IEX_NAMESPACE::ArgExc ("Unknown pixel data type.");
|
|
}
|
|
break;
|
|
|
|
case FLOAT:
|
|
|
|
switch (typeInFile)
|
|
{
|
|
case UINT:
|
|
|
|
for (int x = minX; x <= maxX; x++)
|
|
{
|
|
char* writePtr = *(char **)(base+(y-yOffsetForData)*yPointerStride + (x-xOffsetForData)*xPointerStride);
|
|
|
|
int count = sampleCount(sampleCountBase,
|
|
sampleCountXStride,
|
|
sampleCountYStride,
|
|
x - xOffsetForSampleCount,
|
|
y - yOffsetForSampleCount);
|
|
|
|
if(writePtr)
|
|
{
|
|
for (int i = 0; i < count; i++)
|
|
{
|
|
unsigned int ui;
|
|
|
|
for (size_t i = 0; i < sizeof (unsigned int); ++i)
|
|
((char *)&ui)[i] = readPtr[i];
|
|
|
|
*(float *) writePtr = float (ui);
|
|
readPtr += sizeof (unsigned int);
|
|
writePtr += sampleStride;
|
|
}
|
|
}else{
|
|
readPtr+=sizeof(unsigned int)*count;
|
|
}
|
|
}
|
|
break;
|
|
|
|
case HALF:
|
|
|
|
for (int x = minX; x <= maxX; x++)
|
|
{
|
|
char* writePtr = *(char **)(base+(y-yOffsetForData)*yPointerStride + (x-xOffsetForData)*xPointerStride);
|
|
|
|
int count = sampleCount(sampleCountBase,
|
|
sampleCountXStride,
|
|
sampleCountYStride,
|
|
x - xOffsetForSampleCount,
|
|
y - yOffsetForSampleCount);
|
|
|
|
if(writePtr)
|
|
{
|
|
for (int i = 0; i < count; i++)
|
|
{
|
|
half h = *(half *) readPtr;
|
|
*(float *) writePtr = float (h);
|
|
readPtr += sizeof (half);
|
|
writePtr += sampleStride;
|
|
}
|
|
}else{
|
|
readPtr+=sizeof(half)*count;
|
|
}
|
|
}
|
|
break;
|
|
|
|
case FLOAT:
|
|
|
|
for (int x = minX; x <= maxX; x++)
|
|
{
|
|
char* writePtr = *(char **)(base+(y-yOffsetForData)*yPointerStride + (x-xOffsetForData)*xPointerStride);
|
|
|
|
int count = sampleCount(sampleCountBase,
|
|
sampleCountXStride,
|
|
sampleCountYStride,
|
|
x - xOffsetForSampleCount,
|
|
y - yOffsetForSampleCount);
|
|
|
|
if(writePtr)
|
|
{
|
|
for (int i = 0; i < count; i++)
|
|
{
|
|
for (size_t i = 0; i < sizeof (float); ++i)
|
|
writePtr[i] = readPtr[i];
|
|
|
|
readPtr += sizeof (float);
|
|
writePtr += sampleStride;
|
|
}
|
|
}else{
|
|
readPtr+=sizeof(float)*count;
|
|
}
|
|
}
|
|
break;
|
|
default:
|
|
|
|
throw IEX_NAMESPACE::ArgExc ("Unknown pixel data type.");
|
|
}
|
|
break;
|
|
|
|
default:
|
|
|
|
throw IEX_NAMESPACE::ArgExc ("Unknown pixel data type.");
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
void
|
|
skipChannel (const char *& readPtr,
|
|
PixelType typeInFile,
|
|
size_t xSize)
|
|
{
|
|
switch (typeInFile)
|
|
{
|
|
case UINT:
|
|
|
|
Xdr::skip <CharPtrIO> (readPtr, Xdr::size <unsigned int> () * xSize);
|
|
break;
|
|
|
|
case HALF:
|
|
|
|
Xdr::skip <CharPtrIO> (readPtr, Xdr::size <half> () * xSize);
|
|
break;
|
|
|
|
case FLOAT:
|
|
|
|
Xdr::skip <CharPtrIO> (readPtr, Xdr::size <float> () * xSize);
|
|
break;
|
|
|
|
default:
|
|
|
|
throw IEX_NAMESPACE::ArgExc ("Unknown pixel data type.");
|
|
}
|
|
}
|
|
|
|
|
|
void
|
|
convertInPlace (char *& writePtr,
|
|
const char *& readPtr,
|
|
PixelType type,
|
|
size_t numPixels)
|
|
{
|
|
switch (type)
|
|
{
|
|
case UINT:
|
|
|
|
for (size_t j = 0; j < numPixels; ++j)
|
|
{
|
|
Xdr::write <CharPtrIO> (writePtr, *(const unsigned int *) readPtr);
|
|
readPtr += sizeof(unsigned int);
|
|
}
|
|
break;
|
|
|
|
case HALF:
|
|
|
|
for (size_t j = 0; j < numPixels; ++j)
|
|
{
|
|
Xdr::write <CharPtrIO> (writePtr, *(const half *) readPtr);
|
|
readPtr += sizeof(half);
|
|
}
|
|
break;
|
|
|
|
case FLOAT:
|
|
|
|
for (size_t j = 0; j < numPixels; ++j)
|
|
{
|
|
Xdr::write <CharPtrIO> (writePtr, *(const float *) readPtr);
|
|
readPtr += sizeof(float);
|
|
}
|
|
break;
|
|
|
|
default:
|
|
|
|
throw IEX_NAMESPACE::ArgExc ("Unknown pixel data type.");
|
|
}
|
|
}
|
|
|
|
|
|
void
|
|
copyFromFrameBuffer (char *& writePtr,
|
|
const char *& readPtr,
|
|
const char * endPtr,
|
|
size_t xStride,
|
|
Compressor::Format format,
|
|
PixelType type)
|
|
{
|
|
//
|
|
// Copy a horizontal row of pixels from a frame
|
|
// buffer to an output file's line or tile buffer.
|
|
//
|
|
|
|
if (format == Compressor::XDR)
|
|
{
|
|
//
|
|
// The the line or tile buffer is in XDR format.
|
|
//
|
|
|
|
switch (type)
|
|
{
|
|
case UINT:
|
|
|
|
while (readPtr <= endPtr)
|
|
{
|
|
Xdr::write <CharPtrIO> (writePtr,
|
|
*(const unsigned int *) readPtr);
|
|
readPtr += xStride;
|
|
}
|
|
break;
|
|
|
|
case HALF:
|
|
|
|
while (readPtr <= endPtr)
|
|
{
|
|
Xdr::write <CharPtrIO> (writePtr, *(const half *) readPtr);
|
|
readPtr += xStride;
|
|
}
|
|
break;
|
|
|
|
case FLOAT:
|
|
|
|
while (readPtr <= endPtr)
|
|
{
|
|
Xdr::write <CharPtrIO> (writePtr, *(const float *) readPtr);
|
|
readPtr += xStride;
|
|
}
|
|
break;
|
|
|
|
default:
|
|
|
|
throw IEX_NAMESPACE::ArgExc ("Unknown pixel data type.");
|
|
}
|
|
}
|
|
else
|
|
{
|
|
//
|
|
// The the line or tile buffer is in NATIVE format.
|
|
//
|
|
|
|
switch (type)
|
|
{
|
|
case UINT:
|
|
|
|
while (readPtr <= endPtr)
|
|
{
|
|
for (size_t i = 0; i < sizeof (unsigned int); ++i)
|
|
*writePtr++ = readPtr[i];
|
|
|
|
readPtr += xStride;
|
|
}
|
|
break;
|
|
|
|
case HALF:
|
|
|
|
while (readPtr <= endPtr)
|
|
{
|
|
*(half *) writePtr = *(const half *) readPtr;
|
|
writePtr += sizeof (half);
|
|
readPtr += xStride;
|
|
}
|
|
break;
|
|
|
|
case FLOAT:
|
|
|
|
while (readPtr <= endPtr)
|
|
{
|
|
for (size_t i = 0; i < sizeof (float); ++i)
|
|
*writePtr++ = readPtr[i];
|
|
|
|
readPtr += xStride;
|
|
}
|
|
break;
|
|
|
|
default:
|
|
|
|
throw IEX_NAMESPACE::ArgExc ("Unknown pixel data type.");
|
|
}
|
|
}
|
|
}
|
|
|
|
void
|
|
copyFromDeepFrameBuffer (char *& writePtr,
|
|
const char * base,
|
|
char* sampleCountBase,
|
|
ptrdiff_t sampleCountXStride,
|
|
ptrdiff_t sampleCountYStride,
|
|
int y, int xMin, int xMax,
|
|
int xOffsetForSampleCount,
|
|
int yOffsetForSampleCount,
|
|
int xOffsetForData,
|
|
int yOffsetForData,
|
|
ptrdiff_t sampleStride,
|
|
ptrdiff_t dataXStride,
|
|
ptrdiff_t dataYStride,
|
|
Compressor::Format format,
|
|
PixelType type)
|
|
{
|
|
//
|
|
// Copy a horizontal row of pixels from a frame
|
|
// buffer to an output file's line or tile buffer.
|
|
//
|
|
|
|
if (format == Compressor::XDR)
|
|
{
|
|
//
|
|
// The the line or tile buffer is in XDR format.
|
|
//
|
|
|
|
switch (type)
|
|
{
|
|
case UINT:
|
|
|
|
for (int x = xMin; x <= xMax; x++)
|
|
{
|
|
unsigned int count =
|
|
sampleCount(sampleCountBase,
|
|
sampleCountXStride,
|
|
sampleCountYStride,
|
|
x - xOffsetForSampleCount,
|
|
y - yOffsetForSampleCount);
|
|
const char* ptr = base + (y-yOffsetForData) * dataYStride + (x-xOffsetForData) * dataXStride;
|
|
const char* readPtr = ((const char**) ptr)[0];
|
|
for (unsigned int i = 0; i < count; i++)
|
|
{
|
|
Xdr::write <CharPtrIO> (writePtr,
|
|
*(const unsigned int *) readPtr);
|
|
readPtr += sampleStride;
|
|
}
|
|
}
|
|
break;
|
|
|
|
case HALF:
|
|
|
|
for (int x = xMin; x <= xMax; x++)
|
|
{
|
|
unsigned int count =
|
|
sampleCount(sampleCountBase,
|
|
sampleCountXStride,
|
|
sampleCountYStride,
|
|
x - xOffsetForSampleCount,
|
|
y - yOffsetForSampleCount);
|
|
const char* ptr = base + (y-yOffsetForData) * dataYStride + (x-xOffsetForData) * dataXStride;
|
|
const char* readPtr = ((const char**) ptr)[0];
|
|
for (unsigned int i = 0; i < count; i++)
|
|
{
|
|
Xdr::write <CharPtrIO> (writePtr, *(const half *) readPtr);
|
|
readPtr += sampleStride;
|
|
}
|
|
}
|
|
break;
|
|
|
|
case FLOAT:
|
|
|
|
for (int x = xMin; x <= xMax; x++)
|
|
{
|
|
unsigned int count =
|
|
sampleCount(sampleCountBase,
|
|
sampleCountXStride,
|
|
sampleCountYStride,
|
|
x - xOffsetForSampleCount,
|
|
y - yOffsetForSampleCount);
|
|
const char* ptr = base + (y-yOffsetForData) * dataYStride + (x-xOffsetForData) * dataXStride;
|
|
|
|
const char* readPtr = ((const char**) ptr)[0];
|
|
for (unsigned int i = 0; i < count; i++)
|
|
{
|
|
Xdr::write <CharPtrIO> (writePtr, *(const float *) readPtr);
|
|
readPtr += sampleStride;
|
|
}
|
|
}
|
|
break;
|
|
|
|
default:
|
|
|
|
throw IEX_NAMESPACE::ArgExc ("Unknown pixel data type.");
|
|
}
|
|
}
|
|
else
|
|
{
|
|
//
|
|
// The the line or tile buffer is in NATIVE format.
|
|
//
|
|
|
|
switch (type)
|
|
{
|
|
case UINT:
|
|
|
|
for (int x = xMin; x <= xMax; x++)
|
|
{
|
|
unsigned int count =
|
|
sampleCount(sampleCountBase,
|
|
sampleCountXStride,
|
|
sampleCountYStride,
|
|
x - xOffsetForSampleCount,
|
|
y - yOffsetForSampleCount);
|
|
|
|
const char* ptr = base + (y-yOffsetForData) * dataYStride + (x-xOffsetForData) * dataXStride;
|
|
const char* readPtr = ((const char**) ptr)[0];
|
|
for (unsigned int i = 0; i < count; i++)
|
|
{
|
|
for (size_t j = 0; j < sizeof (unsigned int); ++j)
|
|
*writePtr++ = readPtr[j];
|
|
|
|
readPtr += sampleStride;
|
|
}
|
|
}
|
|
break;
|
|
|
|
case HALF:
|
|
|
|
for (int x = xMin; x <= xMax; x++)
|
|
{
|
|
unsigned int count =
|
|
sampleCount(sampleCountBase,
|
|
sampleCountXStride,
|
|
sampleCountYStride,
|
|
x - xOffsetForSampleCount,
|
|
y - yOffsetForSampleCount);
|
|
const char* ptr = base + (y-yOffsetForData) * dataYStride + (x-xOffsetForData) * dataXStride;
|
|
const char* readPtr = ((const char**) ptr)[0];
|
|
for (unsigned int i = 0; i < count; i++)
|
|
{
|
|
*(half *) writePtr = *(const half *) readPtr;
|
|
writePtr += sizeof (half);
|
|
readPtr += sampleStride;
|
|
}
|
|
}
|
|
break;
|
|
|
|
case FLOAT:
|
|
|
|
for (int x = xMin; x <= xMax; x++)
|
|
{
|
|
unsigned int count =
|
|
sampleCount(sampleCountBase,
|
|
sampleCountXStride,
|
|
sampleCountYStride,
|
|
x - xOffsetForSampleCount,
|
|
y - yOffsetForSampleCount);
|
|
|
|
const char* ptr = base + (y-yOffsetForData) * dataYStride + (x-xOffsetForData) * dataXStride;
|
|
const char* readPtr = ((const char**) ptr)[0];
|
|
for (unsigned int i = 0; i < count; i++)
|
|
{
|
|
for (size_t j = 0; j < sizeof (float); ++j)
|
|
*writePtr++ = readPtr[j];
|
|
|
|
readPtr += sampleStride;
|
|
}
|
|
}
|
|
break;
|
|
|
|
default:
|
|
|
|
throw IEX_NAMESPACE::ArgExc ("Unknown pixel data type.");
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
void
|
|
fillChannelWithZeroes (char *& writePtr,
|
|
Compressor::Format format,
|
|
PixelType type,
|
|
size_t xSize)
|
|
{
|
|
if (format == Compressor::XDR)
|
|
{
|
|
//
|
|
// Fill with data in XDR format.
|
|
//
|
|
|
|
switch (type)
|
|
{
|
|
case UINT:
|
|
|
|
for (size_t j = 0; j < xSize; ++j)
|
|
Xdr::write <CharPtrIO> (writePtr, (unsigned int) 0);
|
|
|
|
break;
|
|
|
|
case HALF:
|
|
|
|
for (size_t j = 0; j < xSize; ++j)
|
|
Xdr::write <CharPtrIO> (writePtr, (half) 0);
|
|
|
|
break;
|
|
|
|
case FLOAT:
|
|
|
|
for (size_t j = 0; j < xSize; ++j)
|
|
Xdr::write <CharPtrIO> (writePtr, (float) 0);
|
|
|
|
break;
|
|
|
|
default:
|
|
|
|
throw IEX_NAMESPACE::ArgExc ("Unknown pixel data type.");
|
|
}
|
|
}
|
|
else
|
|
{
|
|
//
|
|
// Fill with data in NATIVE format.
|
|
//
|
|
|
|
switch (type)
|
|
{
|
|
case UINT:
|
|
|
|
for (size_t j = 0; j < xSize; ++j)
|
|
{
|
|
static const unsigned int ui = 0;
|
|
|
|
for (size_t i = 0; i < sizeof (ui); ++i)
|
|
*writePtr++ = ((char *) &ui)[i];
|
|
}
|
|
break;
|
|
|
|
case HALF:
|
|
|
|
for (size_t j = 0; j < xSize; ++j)
|
|
{
|
|
*(half *) writePtr = half (0);
|
|
writePtr += sizeof (half);
|
|
}
|
|
break;
|
|
|
|
case FLOAT:
|
|
|
|
for (size_t j = 0; j < xSize; ++j)
|
|
{
|
|
static const float f = 0;
|
|
|
|
for (size_t i = 0; i < sizeof (f); ++i)
|
|
*writePtr++ = ((char *) &f)[i];
|
|
}
|
|
break;
|
|
|
|
default:
|
|
|
|
throw IEX_NAMESPACE::ArgExc ("Unknown pixel data type.");
|
|
}
|
|
}
|
|
}
|
|
|
|
bool
|
|
usesLongNames (const Header &header)
|
|
{
|
|
//
|
|
// If an OpenEXR file contains any attribute names, attribute type names
|
|
// or channel names longer than 31 characters, then the file cannot be
|
|
// read by older versions of the IlmImf library (up to OpenEXR 1.6.1).
|
|
// Before writing the file header, we check if the header contains
|
|
// any names longer than 31 characters; if it does, then we set the
|
|
// LONG_NAMES_FLAG in the file version number. Older versions of the
|
|
// IlmImf library will refuse to read files that have the LONG_NAMES_FLAG
|
|
// set. Without the flag, older versions of the library would mis-
|
|
// interpret the file as broken.
|
|
//
|
|
|
|
for (Header::ConstIterator i = header.begin();
|
|
i != header.end();
|
|
++i)
|
|
{
|
|
if (strlen (i.name()) >= 32 || strlen (i.attribute().typeName()) >= 32)
|
|
return true;
|
|
}
|
|
|
|
const ChannelList &channels = header.channels();
|
|
|
|
for (ChannelList::ConstIterator i = channels.begin();
|
|
i != channels.end();
|
|
++i)
|
|
{
|
|
if (strlen (i.name()) >= 32)
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
int
|
|
getScanlineChunkOffsetTableSize(const Header& header)
|
|
{
|
|
const Box2i &dataWindow = header.dataWindow();
|
|
|
|
vector<size_t> bytesPerLine;
|
|
size_t maxBytesPerLine = bytesPerLineTable (header,
|
|
bytesPerLine);
|
|
|
|
Compressor* compressor = newCompressor(header.compression(),
|
|
maxBytesPerLine,
|
|
header);
|
|
|
|
int linesInBuffer = numLinesInBuffer (compressor);
|
|
|
|
int lineOffsetSize = (dataWindow.max.y - dataWindow.min.y +
|
|
linesInBuffer) / linesInBuffer;
|
|
|
|
delete compressor;
|
|
|
|
return lineOffsetSize;
|
|
}
|
|
|
|
//
|
|
// Located in ImfTiledMisc.cpp
|
|
//
|
|
int
|
|
getTiledChunkOffsetTableSize(const Header& header);
|
|
|
|
int
|
|
getChunkOffsetTableSize(const Header& header,bool ignore_attribute)
|
|
{
|
|
if(!ignore_attribute && header.hasChunkCount())
|
|
{
|
|
return header.chunkCount();
|
|
}
|
|
|
|
if(header.hasType() && !isSupportedType(header.type()))
|
|
{
|
|
throw IEX_NAMESPACE::ArgExc ("unsupported header type to "
|
|
"get chunk offset table size");
|
|
}
|
|
if (isTiled(header.type()) == false)
|
|
return getScanlineChunkOffsetTableSize(header);
|
|
else
|
|
return getTiledChunkOffsetTableSize(header);
|
|
|
|
}
|
|
|
|
OPENEXR_IMF_INTERNAL_NAMESPACE_SOURCE_EXIT
|