Blame IlmImfTest/testDeepTiledBasic.cpp

Packit 0d464f
///////////////////////////////////////////////////////////////////////////
Packit 0d464f
//
Packit 0d464f
// Copyright (c) 2011, Industrial Light & Magic, a division of Lucas
Packit 0d464f
// Digital Ltd. LLC
Packit 0d464f
//
Packit 0d464f
// All rights reserved.
Packit 0d464f
//
Packit 0d464f
// Redistribution and use in source and binary forms, with or without
Packit 0d464f
// modification, are permitted provided that the following conditions are
Packit 0d464f
// met:
Packit 0d464f
// *       Redistributions of source code must retain the above copyright
Packit 0d464f
// notice, this list of conditions and the following disclaimer.
Packit 0d464f
// *       Redistributions in binary form must reproduce the above
Packit 0d464f
// copyright notice, this list of conditions and the following disclaimer
Packit 0d464f
// in the documentation and/or other materials provided with the
Packit 0d464f
// distribution.
Packit 0d464f
// *       Neither the name of Industrial Light & Magic nor the names of
Packit 0d464f
// its contributors may be used to endorse or promote products derived
Packit 0d464f
// from this software without specific prior written permission.
Packit 0d464f
//
Packit 0d464f
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
Packit 0d464f
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
Packit 0d464f
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
Packit 0d464f
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
Packit 0d464f
// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
Packit 0d464f
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
Packit 0d464f
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
Packit 0d464f
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
Packit 0d464f
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
Packit 0d464f
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
Packit 0d464f
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
Packit 0d464f
//
Packit 0d464f
///////////////////////////////////////////////////////////////////////////
Packit 0d464f
Packit 0d464f
#include "testDeepTiledBasic.h"
Packit 0d464f
Packit 0d464f
#include <assert.h>
Packit 0d464f
#include <string.h>
Packit 0d464f
Packit 0d464f
#include <ImfDeepTiledInputFile.h>
Packit 0d464f
#include <ImfDeepTiledOutputFile.h>
Packit 0d464f
#include <ImfChannelList.h>
Packit 0d464f
#include <ImfArray.h>
Packit 0d464f
#include <ImfPartType.h>
Packit 0d464f
#include <IlmThreadPool.h>
Packit 0d464f
Packit 0d464f
#include <stdio.h>
Packit 0d464f
#include <stdlib.h>
Packit 0d464f
#include <vector>
Packit 0d464f
Packit 0d464f
namespace IMF = OPENEXR_IMF_NAMESPACE;
Packit 0d464f
using namespace IMF;
Packit 0d464f
using namespace IMATH_NAMESPACE;
Packit 0d464f
using namespace ILMTHREAD_NAMESPACE;
Packit 0d464f
using namespace std;
Packit 0d464f
Packit 0d464f
namespace
Packit 0d464f
{
Packit 0d464f
Packit 0d464f
const int width = 273;
Packit 0d464f
const int height = 169;
Packit 0d464f
const int minX = 10;
Packit 0d464f
const int minY = 11;
Packit 0d464f
const Box2i dataWindow(V2i(minX, minY), V2i(minX + width - 1, minY + height - 1));
Packit 0d464f
const Box2i displayWindow(V2i(0, 0), V2i(minX + width * 2, minY + height * 2));
Packit 0d464f
Packit 0d464f
vector<int> channelTypes;
Packit 0d464f
Array2D< Array2D<unsigned int> > sampleCountWhole;
Packit 0d464f
Header header;
Packit 0d464f
Packit 0d464f
void generateRandomFile (int channelCount,
Packit 0d464f
                         Compression compression,
Packit 0d464f
                         bool bulkWrite,
Packit 0d464f
                         bool relativeCoords,
Packit 0d464f
                         const std::string & filename)
Packit 0d464f
{
Packit 0d464f
    if (relativeCoords)
Packit 0d464f
        assert(bulkWrite == false);
Packit 0d464f
Packit 0d464f
    cout << "generating " << flush;
Packit 0d464f
    header = Header(displayWindow, dataWindow,
Packit 0d464f
                    1,
Packit 0d464f
                    IMATH_NAMESPACE::V2f (0, 0),
Packit 0d464f
                    1,
Packit 0d464f
                    INCREASING_Y,
Packit 0d464f
                    compression);
Packit 0d464f
    cout << "compression " << compression << " " << flush;
Packit 0d464f
Packit 0d464f
    //
Packit 0d464f
    // Add channels.
Packit 0d464f
    //
Packit 0d464f
Packit 0d464f
    channelTypes.clear();
Packit 0d464f
Packit 0d464f
    for (int i = 0; i < channelCount; i++)
Packit 0d464f
    {
Packit 0d464f
        int type = rand() % 3;
Packit 0d464f
        stringstream ss;
Packit 0d464f
        ss << i;
Packit 0d464f
        string str = ss.str();
Packit 0d464f
        if (type == 0)
Packit 0d464f
            header.channels().insert(str, Channel(IMF::UINT));
Packit 0d464f
        if (type == 1)
Packit 0d464f
            header.channels().insert(str, Channel(IMF::HALF));
Packit 0d464f
        if (type == 2)
Packit 0d464f
            header.channels().insert(str, Channel(IMF::FLOAT));
Packit 0d464f
        channelTypes.push_back(type);
Packit 0d464f
    }
Packit 0d464f
Packit 0d464f
    header.setType(DEEPTILE);
Packit 0d464f
    header.setTileDescription(
Packit 0d464f
        TileDescription(rand() % width + 1, rand() % height + 1, RIPMAP_LEVELS));
Packit 0d464f
Packit 0d464f
Packit 0d464f
    //
Packit 0d464f
    // Set up the output file
Packit 0d464f
    //
Packit 0d464f
    remove (filename.c_str());
Packit 0d464f
    DeepTiledOutputFile file(filename.c_str(), header, 8);
Packit 0d464f
Packit 0d464f
    DeepFrameBuffer frameBuffer;
Packit 0d464f
Packit 0d464f
    Array<Array2D< void* > > data(channelCount);
Packit 0d464f
    for (int i = 0; i < channelCount; i++)
Packit 0d464f
        data[i].resizeErase(height, width);
Packit 0d464f
Packit 0d464f
    Array2D<unsigned int> sampleCount;
Packit 0d464f
    sampleCount.resizeErase(height, width);
Packit 0d464f
Packit 0d464f
    cout << " tileSizeX " << file.tileXSize()
Packit 0d464f
         << " tileSizeY " << file.tileYSize() << " ";
Packit 0d464f
Packit 0d464f
    sampleCountWhole.resizeErase(file.numYLevels(), file.numXLevels());
Packit 0d464f
    for (int i = 0; i < sampleCountWhole.height(); i++)
Packit 0d464f
        for (int j = 0; j < sampleCountWhole.width(); j++)
Packit 0d464f
            sampleCountWhole[i][j].resizeErase(height, width);
Packit 0d464f
Packit 0d464f
    int memOffset;
Packit 0d464f
    if (relativeCoords)
Packit 0d464f
        memOffset = 0;
Packit 0d464f
    else
Packit 0d464f
        memOffset = dataWindow.min.x + dataWindow.min.y * width;
Packit 0d464f
Packit 0d464f
    frameBuffer.insertSampleCountSlice (Slice (IMF::UINT,
Packit 0d464f
                                        (char *) (&sampleCount[0][0] - memOffset),
Packit 0d464f
                                        sizeof (unsigned int) * 1,
Packit 0d464f
                                        sizeof (unsigned int) * width,
Packit 0d464f
                                        1, 1,
Packit 0d464f
                                        0,
Packit 0d464f
                                        relativeCoords,
Packit 0d464f
                                        relativeCoords));
Packit 0d464f
Packit 0d464f
    for (int i = 0; i < channelCount; i++)
Packit 0d464f
    {
Packit 0d464f
        PixelType type;
Packit 0d464f
        if (channelTypes[i] == 0)
Packit 0d464f
            type = IMF::UINT;
Packit 0d464f
        if (channelTypes[i] == 1)
Packit 0d464f
            type = IMF::HALF;
Packit 0d464f
        if (channelTypes[i] == 2)
Packit 0d464f
            type = IMF::FLOAT;
Packit 0d464f
Packit 0d464f
        stringstream ss;
Packit 0d464f
        ss << i;
Packit 0d464f
        string str = ss.str();
Packit 0d464f
Packit 0d464f
        int sampleSize;
Packit 0d464f
        if (channelTypes[i] == 0) sampleSize = sizeof (unsigned int);
Packit 0d464f
        if (channelTypes[i] == 1) sampleSize = sizeof (half);
Packit 0d464f
        if (channelTypes[i] == 2) sampleSize = sizeof (float);
Packit 0d464f
Packit 0d464f
        int pointerSize = sizeof (char *);
Packit 0d464f
Packit 0d464f
        frameBuffer.insert (str,
Packit 0d464f
                            DeepSlice (type,
Packit 0d464f
                            (char *) (&data[i][0][0] - memOffset),
Packit 0d464f
                            pointerSize * 1,
Packit 0d464f
                            pointerSize * width,
Packit 0d464f
                            sampleSize,
Packit 0d464f
                            1, 1,
Packit 0d464f
                            0,
Packit 0d464f
                            relativeCoords,
Packit 0d464f
                            relativeCoords));
Packit 0d464f
    }
Packit 0d464f
Packit 0d464f
    file.setFrameBuffer(frameBuffer);
Packit 0d464f
Packit 0d464f
    cout << "writing " << flush;
Packit 0d464f
Packit 0d464f
    if (bulkWrite)
Packit 0d464f
        cout << "bulk " << flush;
Packit 0d464f
    else
Packit 0d464f
    {
Packit 0d464f
        if (relativeCoords == false)
Packit 0d464f
            cout << "per-tile " << flush;
Packit 0d464f
        else
Packit 0d464f
            cout << "per-tile with relative coordinates " << flush;
Packit 0d464f
    }
Packit 0d464f
Packit 0d464f
    for (int ly = 0; ly < file.numYLevels(); ly++)
Packit 0d464f
        for (int lx = 0; lx < file.numXLevels(); lx++)
Packit 0d464f
        {
Packit 0d464f
            Box2i dataWindowL = file.dataWindowForLevel(lx, ly);
Packit 0d464f
Packit 0d464f
            if (bulkWrite)
Packit 0d464f
            {
Packit 0d464f
                //
Packit 0d464f
                // Bulk write (without relative coordinates).
Packit 0d464f
                //
Packit 0d464f
Packit 0d464f
                for (int j = 0; j < file.numYTiles(ly); j++)
Packit 0d464f
                {
Packit 0d464f
                    for (int i = 0; i < file.numXTiles(lx); i++)
Packit 0d464f
                    {
Packit 0d464f
                        Box2i box = file.dataWindowForTile(i, j, lx, ly);
Packit 0d464f
                        for (int y = box.min.y; y <= box.max.y; y++)
Packit 0d464f
                            for (int x = box.min.x; x <= box.max.x; x++)
Packit 0d464f
                            {
Packit 0d464f
                                int dwy = y - dataWindowL.min.y;
Packit 0d464f
                                int dwx = x - dataWindowL.min.x;
Packit 0d464f
                                sampleCount[dwy][dwx] = rand() % 10 + 1;
Packit 0d464f
                                sampleCountWhole[ly][lx][dwy][dwx] = sampleCount[dwy][dwx];
Packit 0d464f
                                for (int k = 0; k < channelCount; k++)
Packit 0d464f
                                {
Packit 0d464f
                                    if (channelTypes[k] == 0)
Packit 0d464f
                                        data[k][dwy][dwx] = new unsigned int[sampleCount[dwy][dwx]];
Packit 0d464f
                                    if (channelTypes[k] == 1)
Packit 0d464f
                                        data[k][dwy][dwx] = new half[sampleCount[dwy][dwx]];
Packit 0d464f
                                    if (channelTypes[k] == 2)
Packit 0d464f
                                        data[k][dwy][dwx] = new float[sampleCount[dwy][dwx]];
Packit 0d464f
                                    for (int l = 0; l < sampleCount[dwy][dwx]; l++)
Packit 0d464f
                                    {
Packit 0d464f
                                        if (channelTypes[k] == 0)
Packit 0d464f
                                            ((unsigned int*)data[k][dwy][dwx])[l] = (dwy * width + dwx) % 2049;
Packit 0d464f
                                        if (channelTypes[k] == 1)
Packit 0d464f
                                            ((half*)data[k][dwy][dwx])[l] = (dwy* width + dwx) % 2049;
Packit 0d464f
                                        if (channelTypes[k] == 2)
Packit 0d464f
                                            ((float*)data[k][dwy][dwx])[l] = (dwy * width + dwx) % 2049;
Packit 0d464f
                                    }
Packit 0d464f
                                }
Packit 0d464f
                            }
Packit 0d464f
                    }
Packit 0d464f
                }
Packit 0d464f
Packit 0d464f
                file.writeTiles(0, file.numXTiles(lx) - 1, 0, file.numYTiles(ly) - 1, lx, ly);
Packit 0d464f
            }
Packit 0d464f
            else if (bulkWrite == false)
Packit 0d464f
            {
Packit 0d464f
                if (relativeCoords == false)
Packit 0d464f
                {
Packit 0d464f
                    //
Packit 0d464f
                    // Per-tile write without relative coordinates.
Packit 0d464f
                    //
Packit 0d464f
Packit 0d464f
                    for (int j = 0; j < file.numYTiles(ly); j++)
Packit 0d464f
                    {
Packit 0d464f
                        for (int i = 0; i < file.numXTiles(lx); i++)
Packit 0d464f
                        {
Packit 0d464f
                            Box2i box = file.dataWindowForTile(i, j, lx, ly);
Packit 0d464f
                            for (int y = box.min.y; y <= box.max.y; y++)
Packit 0d464f
                                for (int x = box.min.x; x <= box.max.x; x++)
Packit 0d464f
                                {
Packit 0d464f
                                    int dwy = y - dataWindowL.min.y;
Packit 0d464f
                                    int dwx = x - dataWindowL.min.x;
Packit 0d464f
                                    sampleCount[dwy][dwx] = rand() % 10 + 1;
Packit 0d464f
                                    sampleCountWhole[ly][lx][dwy][dwx] = sampleCount[dwy][dwx];
Packit 0d464f
                                    for (int k = 0; k < channelCount; k++)
Packit 0d464f
                                    {
Packit 0d464f
                                        if (channelTypes[k] == 0)
Packit 0d464f
                                            data[k][dwy][dwx] = new unsigned int[sampleCount[dwy][dwx]];
Packit 0d464f
                                        if (channelTypes[k] == 1)
Packit 0d464f
                                            data[k][dwy][dwx] = new half[sampleCount[dwy][dwx]];
Packit 0d464f
                                        if (channelTypes[k] == 2)
Packit 0d464f
                                            data[k][dwy][dwx] = new float[sampleCount[dwy][dwx]];
Packit 0d464f
                                        for (int l = 0; l < sampleCount[dwy][dwx]; l++)
Packit 0d464f
                                        {
Packit 0d464f
                                            if (channelTypes[k] == 0)
Packit 0d464f
                                                ((unsigned int*)data[k][dwy][dwx])[l] = (dwy * width + dwx) % 2049;
Packit 0d464f
                                            if (channelTypes[k] == 1)
Packit 0d464f
                                                ((half*)data[k][dwy][dwx])[l] = (dwy* width + dwx) % 2049;
Packit 0d464f
                                            if (channelTypes[k] == 2)
Packit 0d464f
                                                ((float*)data[k][dwy][dwx])[l] = (dwy * width + dwx) % 2049;
Packit 0d464f
                                        }
Packit 0d464f
                                    }
Packit 0d464f
                                }
Packit 0d464f
                            file.writeTile(i, j, lx, ly);
Packit 0d464f
                        }
Packit 0d464f
                    }
Packit 0d464f
                }
Packit 0d464f
                else if (relativeCoords)
Packit 0d464f
                {
Packit 0d464f
                    //
Packit 0d464f
                    // Per-tile write with relative coordinates.
Packit 0d464f
                    //
Packit 0d464f
Packit 0d464f
                    for (int j = 0; j < file.numYTiles(ly); j++)
Packit 0d464f
                    {
Packit 0d464f
                        for (int i = 0; i < file.numXTiles(lx); i++)
Packit 0d464f
                        {
Packit 0d464f
                            Box2i box = file.dataWindowForTile(i, j, lx, ly);
Packit 0d464f
                            for (int y = box.min.y; y <= box.max.y; y++)
Packit 0d464f
                                for (int x = box.min.x; x <= box.max.x; x++)
Packit 0d464f
                                {
Packit 0d464f
                                    int dwy = y - dataWindowL.min.y;
Packit 0d464f
                                    int dwx = x - dataWindowL.min.x;
Packit 0d464f
                                    int ty = y - box.min.y;
Packit 0d464f
                                    int tx = x - box.min.x;
Packit 0d464f
                                    sampleCount[ty][tx] = rand() % 10 + 1;
Packit 0d464f
                                    sampleCountWhole[ly][lx][dwy][dwx] = sampleCount[ty][tx];
Packit 0d464f
                                    for (int k = 0; k < channelCount; k++)
Packit 0d464f
                                    {
Packit 0d464f
                                        if (channelTypes[k] == 0)
Packit 0d464f
                                            data[k][ty][tx] = new unsigned int[sampleCount[ty][tx]];
Packit 0d464f
                                        if (channelTypes[k] == 1)
Packit 0d464f
                                            data[k][ty][tx] = new half[sampleCount[ty][tx]];
Packit 0d464f
                                        if (channelTypes[k] == 2)
Packit 0d464f
                                            data[k][ty][tx] = new float[sampleCount[ty][tx]];
Packit 0d464f
                                        for (int l = 0; l < sampleCount[ty][tx]; l++)
Packit 0d464f
                                        {
Packit 0d464f
                                            if (channelTypes[k] == 0)
Packit 0d464f
                                                ((unsigned int*)data[k][ty][tx])[l] =
Packit 0d464f
                                                        (dwy * width + dwx) % 2049;
Packit 0d464f
                                            if (channelTypes[k] == 1)
Packit 0d464f
                                                ((half*)data[k][ty][tx])[l] =
Packit 0d464f
                                                        (dwy * width + dwx) % 2049;
Packit 0d464f
                                            if (channelTypes[k] == 2)
Packit 0d464f
                                                ((float*)data[k][ty][tx])[l] =
Packit 0d464f
                                                        (dwy * width + dwx) % 2049;
Packit 0d464f
                                        }
Packit 0d464f
                                    }
Packit 0d464f
                                }
Packit 0d464f
                            file.writeTile(i, j, lx, ly);
Packit 0d464f
Packit 0d464f
                            for (int y = box.min.y; y <= box.max.y; y++)
Packit 0d464f
                                for (int x = box.min.x; x <= box.max.x; x++)
Packit 0d464f
                                    for (int k = 0; k < channelCount; k++)
Packit 0d464f
                                    {
Packit 0d464f
                                        int ty = y - box.min.y;
Packit 0d464f
                                        int tx = x - box.min.x;
Packit 0d464f
                                        if (channelTypes[k] == 0)
Packit 0d464f
                                            delete[] (unsigned int*) data[k][ty][tx];
Packit 0d464f
                                        if (channelTypes[k] == 1)
Packit 0d464f
                                            delete[] (half*) data[k][ty][tx];
Packit 0d464f
                                        if (channelTypes[k] == 2)
Packit 0d464f
                                            delete[] (float*) data[k][ty][tx];
Packit 0d464f
                                    }
Packit 0d464f
                        }
Packit 0d464f
                    }
Packit 0d464f
                }
Packit 0d464f
            }
Packit 0d464f
Packit 0d464f
            if (relativeCoords == false)
Packit 0d464f
            {
Packit 0d464f
                for (int i = 0; i < file.levelHeight(ly); i++)
Packit 0d464f
                    for (int j = 0; j < file.levelWidth(lx); j++)
Packit 0d464f
                        for (int k = 0; k < channelCount; k++)
Packit 0d464f
                        {
Packit 0d464f
                            if (channelTypes[k] == 0)
Packit 0d464f
                                delete[] (unsigned int*) data[k][i][j];
Packit 0d464f
                            if (channelTypes[k] == 1)
Packit 0d464f
                                delete[] (half*) data[k][i][j];
Packit 0d464f
                            if (channelTypes[k] == 2)
Packit 0d464f
                                delete[] (float*) data[k][i][j];
Packit 0d464f
                        }
Packit 0d464f
            }
Packit 0d464f
        }
Packit 0d464f
}
Packit 0d464f
Packit 0d464f
void checkValue (void* sampleRawData,
Packit 0d464f
                 int sampleCount,
Packit 0d464f
                 int channelType,
Packit 0d464f
                 int dwx,
Packit 0d464f
                 int dwy)
Packit 0d464f
{
Packit 0d464f
    for (int l = 0; l < sampleCount; l++)
Packit 0d464f
    {
Packit 0d464f
        if (channelType == 0)
Packit 0d464f
        {
Packit 0d464f
            unsigned int* value = (unsigned int*)(sampleRawData);
Packit 0d464f
            if (value[l] != (dwy * width + dwx) % 2049)
Packit 0d464f
                cout << dwx << ", " << dwy << " error, should be "
Packit 0d464f
                     << (dwy * width + dwx) % 2049 << ", is " << value[l]
Packit 0d464f
                     << endl << flush;
Packit 0d464f
            assert (value[l] == (dwy * width + dwx) % 2049);
Packit 0d464f
        }
Packit 0d464f
        if (channelType == 1)
Packit 0d464f
        {
Packit 0d464f
            half* value = (half*)(sampleRawData);
Packit 0d464f
            if (value[l] != (dwy * width + dwx) % 2049)
Packit 0d464f
                cout << dwx << ", " << dwy << " error, should be "
Packit 0d464f
                     << (dwy * width + dwx) % 2049 << ", is " << value[l]
Packit 0d464f
                     << endl << flush;
Packit 0d464f
            assert (value[l] == (dwy * width + dwx) % 2049);
Packit 0d464f
        }
Packit 0d464f
        if (channelType == 2)
Packit 0d464f
        {
Packit 0d464f
            float* value = (float*)(sampleRawData);
Packit 0d464f
            if (value[l] != (dwy * width + dwx) % 2049)
Packit 0d464f
                cout << dwx << ", " << dwy << " error, should be "
Packit 0d464f
                     << (dwy * width + dwx) % 2049 << ", is " << value[l]
Packit 0d464f
                     << endl << flush;
Packit 0d464f
            assert (value[l] == (dwy * width + dwx) % 2049);
Packit 0d464f
        }
Packit 0d464f
    }
Packit 0d464f
}
Packit 0d464f
Packit 0d464f
void readFile (int channelCount,
Packit 0d464f
               bool bulkRead,
Packit 0d464f
               bool relativeCoords,
Packit 0d464f
               const std::string & filename)
Packit 0d464f
{
Packit 0d464f
    if (relativeCoords)
Packit 0d464f
        assert(bulkRead == false);
Packit 0d464f
Packit 0d464f
    cout << "reading " << flush;
Packit 0d464f
Packit 0d464f
    DeepTiledInputFile file (filename.c_str(), 4);
Packit 0d464f
Packit 0d464f
    const Header& fileHeader = file.header();
Packit 0d464f
    assert (fileHeader.displayWindow() == header.displayWindow());
Packit 0d464f
    assert (fileHeader.dataWindow() == header.dataWindow());
Packit 0d464f
    assert (fileHeader.pixelAspectRatio() == header.pixelAspectRatio());
Packit 0d464f
    assert (fileHeader.screenWindowCenter() == header.screenWindowCenter());
Packit 0d464f
    assert (fileHeader.screenWindowWidth() == header.screenWindowWidth());
Packit 0d464f
    assert (fileHeader.lineOrder() == header.lineOrder());
Packit 0d464f
    assert (fileHeader.compression() == header.compression());
Packit 0d464f
    assert (fileHeader.channels() == header.channels());
Packit 0d464f
    assert (fileHeader.type() == header.type());
Packit 0d464f
    assert (fileHeader.tileDescription() == header.tileDescription());
Packit 0d464f
Packit 0d464f
    Array2D<unsigned int> localSampleCount;
Packit 0d464f
    localSampleCount.resizeErase(height, width);
Packit 0d464f
    Array<Array2D< void* > > data(channelCount);
Packit 0d464f
    for (int i = 0; i < channelCount; i++)
Packit 0d464f
        data[i].resizeErase(height, width);
Packit 0d464f
Packit 0d464f
    DeepFrameBuffer frameBuffer;
Packit 0d464f
Packit 0d464f
    int memOffset;
Packit 0d464f
    if (relativeCoords)
Packit 0d464f
        memOffset = 0;
Packit 0d464f
    else
Packit 0d464f
        memOffset = dataWindow.min.x + dataWindow.min.y * width;
Packit 0d464f
    frameBuffer.insertSampleCountSlice (Slice (IMF::UINT,
Packit 0d464f
                                        (char *) (&localSampleCount[0][0] - memOffset),
Packit 0d464f
                                        sizeof (unsigned int) * 1,
Packit 0d464f
                                        sizeof (unsigned int) * width,
Packit 0d464f
                                        1, 1,
Packit 0d464f
                                        0,
Packit 0d464f
                                        relativeCoords,
Packit 0d464f
                                        relativeCoords));
Packit 0d464f
Packit 0d464f
    for (int i = 0; i < channelCount; i++)
Packit 0d464f
    {
Packit 0d464f
        PixelType type;
Packit 0d464f
        if (channelTypes[i] == 0)
Packit 0d464f
            type = IMF::UINT;
Packit 0d464f
        if (channelTypes[i] == 1)
Packit 0d464f
            type = IMF::HALF;
Packit 0d464f
        if (channelTypes[i] == 2)
Packit 0d464f
            type = IMF::FLOAT;
Packit 0d464f
Packit 0d464f
        stringstream ss;
Packit 0d464f
        ss << i;
Packit 0d464f
        string str = ss.str();
Packit 0d464f
Packit 0d464f
        int sampleSize;
Packit 0d464f
        if (channelTypes[i] == 0) sampleSize = sizeof (unsigned int);
Packit 0d464f
        if (channelTypes[i] == 1) sampleSize = sizeof (half);
Packit 0d464f
        if (channelTypes[i] == 2) sampleSize = sizeof (float);
Packit 0d464f
Packit 0d464f
        int pointerSize = sizeof (char *);
Packit 0d464f
Packit 0d464f
        frameBuffer.insert (str,
Packit 0d464f
                            DeepSlice (type,
Packit 0d464f
                            (char *) (&data[i][0][0] - memOffset),
Packit 0d464f
                            pointerSize * 1,
Packit 0d464f
                            pointerSize * width,
Packit 0d464f
                            sampleSize,
Packit 0d464f
                            1, 1,
Packit 0d464f
                            0,
Packit 0d464f
                            relativeCoords,
Packit 0d464f
                            relativeCoords));
Packit 0d464f
    }
Packit 0d464f
Packit 0d464f
    file.setFrameBuffer(frameBuffer);
Packit 0d464f
Packit 0d464f
    if (bulkRead)
Packit 0d464f
        cout << "bulk " << flush;
Packit 0d464f
    else
Packit 0d464f
    {
Packit 0d464f
        if (relativeCoords == false)
Packit 0d464f
            cout << "per-tile " << flush;
Packit 0d464f
        else
Packit 0d464f
            cout << "per-tile with relative coordinates " << flush;
Packit 0d464f
    }
Packit 0d464f
Packit 0d464f
    for (int ly = 0; ly < file.numYLevels(); ly++)
Packit 0d464f
        for (int lx = 0; lx < file.numXLevels(); lx++)
Packit 0d464f
        {
Packit 0d464f
            Box2i dataWindowL = file.dataWindowForLevel(lx, ly);
Packit 0d464f
Packit 0d464f
            if (bulkRead)
Packit 0d464f
            {
Packit 0d464f
                //
Packit 0d464f
                // Testing bulk read (without relative coordinates).
Packit 0d464f
                //
Packit 0d464f
Packit 0d464f
                file.readPixelSampleCounts(0, file.numXTiles(lx) - 1, 0, file.numYTiles(ly) - 1, lx, ly);
Packit 0d464f
Packit 0d464f
                for (int i = 0; i < file.numYTiles(ly); i++)
Packit 0d464f
                {
Packit 0d464f
                    for (int j = 0; j < file.numXTiles(lx); j++)
Packit 0d464f
                    {
Packit 0d464f
                        Box2i box = file.dataWindowForTile(j, i, lx, ly);
Packit 0d464f
                        for (int y = box.min.y; y <= box.max.y; y++)
Packit 0d464f
                            for (int x = box.min.x; x <= box.max.x; x++)
Packit 0d464f
                            {
Packit 0d464f
                                int dwy = y - dataWindowL.min.y;
Packit 0d464f
                                int dwx = x - dataWindowL.min.x;
Packit 0d464f
                                assert(localSampleCount[dwy][dwx] == sampleCountWhole[ly][lx][dwy][dwx]);
Packit 0d464f
Packit 0d464f
                                for (int k = 0; k < channelTypes.size(); k++)
Packit 0d464f
                                {
Packit 0d464f
                                    if (channelTypes[k] == 0)
Packit 0d464f
                                        data[k][dwy][dwx] = new unsigned int[localSampleCount[dwy][dwx]];
Packit 0d464f
                                    if (channelTypes[k] == 1)
Packit 0d464f
                                        data[k][dwy][dwx] = new half[localSampleCount[dwy][dwx]];
Packit 0d464f
                                    if (channelTypes[k] == 2)
Packit 0d464f
                                        data[k][dwy][dwx] = new float[localSampleCount[dwy][dwx]];
Packit 0d464f
                                }
Packit 0d464f
                            }
Packit 0d464f
                    }
Packit 0d464f
                }
Packit 0d464f
Packit 0d464f
                file.readTiles(0, file.numXTiles(lx) - 1, 0, file.numYTiles(ly) - 1, lx, ly);
Packit 0d464f
            }
Packit 0d464f
            else if (bulkRead == false)
Packit 0d464f
            {
Packit 0d464f
                if (relativeCoords == false)
Packit 0d464f
                {
Packit 0d464f
                    //
Packit 0d464f
                    // Testing per-tile read without relative coordinates.
Packit 0d464f
                    //
Packit 0d464f
Packit 0d464f
                    for (int i = 0; i < file.numYTiles(ly); i++)
Packit 0d464f
                    {
Packit 0d464f
                        for (int j = 0; j < file.numXTiles(lx); j++)
Packit 0d464f
                        {
Packit 0d464f
                            file.readPixelSampleCount(j, i, lx, ly);
Packit 0d464f
Packit 0d464f
                            Box2i box = file.dataWindowForTile(j, i, lx, ly);
Packit 0d464f
                            for (int y = box.min.y; y <= box.max.y; y++)
Packit 0d464f
                                for (int x = box.min.x; x <= box.max.x; x++)
Packit 0d464f
                                {
Packit 0d464f
                                    int dwy = y - dataWindowL.min.y;
Packit 0d464f
                                    int dwx = x - dataWindowL.min.x;
Packit 0d464f
                                    assert(localSampleCount[dwy][dwx] == sampleCountWhole[ly][lx][dwy][dwx]);
Packit 0d464f
Packit 0d464f
                                    for (int k = 0; k < channelTypes.size(); k++)
Packit 0d464f
                                    {
Packit 0d464f
                                        if (channelTypes[k] == 0)
Packit 0d464f
                                            data[k][dwy][dwx] = new unsigned int[localSampleCount[dwy][dwx]];
Packit 0d464f
                                        if (channelTypes[k] == 1)
Packit 0d464f
                                            data[k][dwy][dwx] = new half[localSampleCount[dwy][dwx]];
Packit 0d464f
                                        if (channelTypes[k] == 2)
Packit 0d464f
                                            data[k][dwy][dwx] = new float[localSampleCount[dwy][dwx]];
Packit 0d464f
                                    }
Packit 0d464f
                                }
Packit 0d464f
Packit 0d464f
                            file.readTile(j, i, lx, ly);
Packit 0d464f
                        }
Packit 0d464f
                    }
Packit 0d464f
                }
Packit 0d464f
                else if (relativeCoords)
Packit 0d464f
                {
Packit 0d464f
                    //
Packit 0d464f
                    // Testing per-tile read with relative coordinates.
Packit 0d464f
                    //
Packit 0d464f
Packit 0d464f
                    for (int i = 0; i < file.numYTiles(ly); i++)
Packit 0d464f
                    {
Packit 0d464f
                        for (int j = 0; j < file.numXTiles(lx); j++)
Packit 0d464f
                        {
Packit 0d464f
                            file.readPixelSampleCount(j, i, lx, ly);
Packit 0d464f
Packit 0d464f
                            Box2i box = file.dataWindowForTile(j, i, lx, ly);
Packit 0d464f
                            for (int y = box.min.y; y <= box.max.y; y++)
Packit 0d464f
                                for (int x = box.min.x; x <= box.max.x; x++)
Packit 0d464f
                                {
Packit 0d464f
                                    int dwy = y - dataWindowL.min.y;
Packit 0d464f
                                    int dwx = x - dataWindowL.min.x;
Packit 0d464f
                                    int ty = y - box.min.y;
Packit 0d464f
                                    int tx = x - box.min.x;
Packit 0d464f
                                    assert(localSampleCount[ty][tx] == sampleCountWhole[ly][lx][dwy][dwx]);
Packit 0d464f
Packit 0d464f
                                    for (int k = 0; k < channelTypes.size(); k++)
Packit 0d464f
                                    {
Packit 0d464f
                                        if (channelTypes[k] == 0)
Packit 0d464f
                                            data[k][ty][tx] = new unsigned int[localSampleCount[ty][tx]];
Packit 0d464f
                                        if (channelTypes[k] == 1)
Packit 0d464f
                                            data[k][ty][tx] = new half[localSampleCount[ty][tx]];
Packit 0d464f
                                        if (channelTypes[k] == 2)
Packit 0d464f
                                            data[k][ty][tx] = new float[localSampleCount[ty][tx]];
Packit 0d464f
                                    }
Packit 0d464f
                                }
Packit 0d464f
Packit 0d464f
                            file.readTile(j, i, lx, ly);
Packit 0d464f
Packit 0d464f
                            for (int y = box.min.y; y <= box.max.y; y++)
Packit 0d464f
                                for (int x = box.min.x; x <= box.max.x; x++)
Packit 0d464f
                                {
Packit 0d464f
                                    int dwy = y - dataWindowL.min.y;
Packit 0d464f
                                    int dwx = x - dataWindowL.min.x;
Packit 0d464f
                                    int ty = y - box.min.y;
Packit 0d464f
                                    int tx = x - box.min.x;
Packit 0d464f
Packit 0d464f
                                    for (int k = 0; k < channelTypes.size(); k++)
Packit 0d464f
                                    {
Packit 0d464f
                                        checkValue(data[k][ty][tx],
Packit 0d464f
                                                   localSampleCount[ty][tx],
Packit 0d464f
                                                   channelTypes[k],
Packit 0d464f
                                                   dwx, dwy);
Packit 0d464f
                                        if (channelTypes[k] == 0)
Packit 0d464f
                                            delete[] (unsigned int*) data[k][ty][tx];
Packit 0d464f
                                        if (channelTypes[k] == 1)
Packit 0d464f
                                            delete[] (half*) data[k][ty][tx];
Packit 0d464f
                                        if (channelTypes[k] == 2)
Packit 0d464f
                                            delete[] (float*) data[k][ty][tx];
Packit 0d464f
                                    }
Packit 0d464f
                                }
Packit 0d464f
                        }
Packit 0d464f
                    }
Packit 0d464f
                }
Packit 0d464f
            }
Packit 0d464f
Packit 0d464f
            if (relativeCoords == false)
Packit 0d464f
            {
Packit 0d464f
                for (int i = 0; i < file.levelHeight(ly); i++)
Packit 0d464f
                    for (int j = 0; j < file.levelWidth(lx); j++)
Packit 0d464f
                        for (int k = 0; k < channelCount; k++)
Packit 0d464f
                        {
Packit 0d464f
                            for (int l = 0; l < localSampleCount[i][j]; l++)
Packit 0d464f
                            {
Packit 0d464f
                                if (channelTypes[k] == 0)
Packit 0d464f
                                {
Packit 0d464f
                                    unsigned int* value = (unsigned int*)(data[k][i][j]);
Packit 0d464f
                                    if (value[l] != (i * width + j) % 2049)
Packit 0d464f
                                        cout << j << ", " << i << " error, should be "
Packit 0d464f
                                             << (i * width + j) % 2049 << ", is " << value[l]
Packit 0d464f
                                             << endl << flush;
Packit 0d464f
                                    assert (value[l] == (i * width + j) % 2049);
Packit 0d464f
                                }
Packit 0d464f
                                if (channelTypes[k] == 1)
Packit 0d464f
                                {
Packit 0d464f
                                    half* value = (half*)(data[k][i][j]);
Packit 0d464f
                                    if (value[l] != (i * width + j) % 2049)
Packit 0d464f
                                        cout << j << ", " << i << " error, should be "
Packit 0d464f
                                             << (i * width + j) % 2049 << ", is " << value[l]
Packit 0d464f
                                             << endl << flush;
Packit 0d464f
                                    assert (((half*)(data[k][i][j]))[l] == (i * width + j) % 2049);
Packit 0d464f
                                }
Packit 0d464f
                                if (channelTypes[k] == 2)
Packit 0d464f
                                {
Packit 0d464f
                                    float* value = (float*)(data[k][i][j]);
Packit 0d464f
                                    if (value[l] != (i * width + j) % 2049)
Packit 0d464f
                                        cout << j << ", " << i << " error, should be "
Packit 0d464f
                                             << (i * width + j) % 2049 << ", is " << value[l]
Packit 0d464f
                                             << endl << flush;
Packit 0d464f
                                    assert (((float*)(data[k][i][j]))[l] == (i * width + j) % 2049);
Packit 0d464f
                                }
Packit 0d464f
                            }
Packit 0d464f
                        }
Packit 0d464f
Packit 0d464f
                for (int i = 0; i < file.levelHeight(ly); i++)
Packit 0d464f
                    for (int j = 0; j < file.levelWidth(lx); j++)
Packit 0d464f
                        for (int k = 0; k < channelCount; k++)
Packit 0d464f
                        {
Packit 0d464f
                            if (channelTypes[k] == 0)
Packit 0d464f
                                delete[] (unsigned int*) data[k][i][j];
Packit 0d464f
                            if (channelTypes[k] == 1)
Packit 0d464f
                                delete[] (half*) data[k][i][j];
Packit 0d464f
                            if (channelTypes[k] == 2)
Packit 0d464f
                                delete[] (float*) data[k][i][j];
Packit 0d464f
                        }
Packit 0d464f
            }
Packit 0d464f
        }
Packit 0d464f
}
Packit 0d464f
Packit 0d464f
void readWriteTestWithAbsoluateCoordinates (int channelCount,
Packit 0d464f
                                            int testTimes,
Packit 0d464f
                                            const std::string & tempDir)
Packit 0d464f
{
Packit 0d464f
    cout << "Testing files with " << channelCount
Packit 0d464f
         << " channels, using absolute coordinates "
Packit 0d464f
         << testTimes << " times."
Packit 0d464f
         << endl << flush;
Packit 0d464f
Packit 0d464f
    std::string fn = tempDir + "imf_test_deep_tiled_basic.exr";
Packit 0d464f
Packit 0d464f
    for (int i = 0; i < testTimes; i++)
Packit 0d464f
    {
Packit 0d464f
        int compressionIndex = i % 3;
Packit 0d464f
        Compression compression;
Packit 0d464f
        switch (compressionIndex)
Packit 0d464f
        {
Packit 0d464f
            case 0:
Packit 0d464f
                compression = NO_COMPRESSION;
Packit 0d464f
                break;
Packit 0d464f
            case 1:
Packit 0d464f
                compression = RLE_COMPRESSION;
Packit 0d464f
                break;
Packit 0d464f
            case 2:
Packit 0d464f
                compression = ZIPS_COMPRESSION;
Packit 0d464f
                break;
Packit 0d464f
        }
Packit 0d464f
Packit 0d464f
        generateRandomFile (channelCount, compression, false, false, fn);
Packit 0d464f
        readFile (channelCount, false, false, fn);
Packit 0d464f
        remove (fn.c_str());
Packit 0d464f
        cout << endl << flush;
Packit 0d464f
Packit 0d464f
        generateRandomFile (channelCount, compression, true, false, fn);
Packit 0d464f
        readFile (channelCount, true, false, fn);
Packit 0d464f
        remove (fn.c_str());
Packit 0d464f
        cout << endl << flush;
Packit 0d464f
Packit 0d464f
        generateRandomFile (channelCount, compression, false, true, fn);
Packit 0d464f
        readFile (channelCount, false, true, fn);
Packit 0d464f
        remove (fn.c_str());
Packit 0d464f
        cout << endl << flush;
Packit 0d464f
    }
Packit 0d464f
}
Packit 0d464f
Packit 0d464f
} // namespace
Packit 0d464f
Packit 0d464f
void testDeepTiledBasic (const std::string & tempDir)
Packit 0d464f
{
Packit 0d464f
    try
Packit 0d464f
    {
Packit 0d464f
        cout << "Testing the DeepTiledInput/OutputFile for basic use" << endl;
Packit 0d464f
Packit 0d464f
        srand(1);
Packit 0d464f
Packit 0d464f
        int numThreads = ThreadPool::globalThreadPool().numThreads();
Packit 0d464f
        ThreadPool::globalThreadPool().setNumThreads(2);
Packit 0d464f
Packit 0d464f
        readWriteTestWithAbsoluateCoordinates ( 1, 2, tempDir);
Packit 0d464f
        readWriteTestWithAbsoluateCoordinates ( 3, 2, tempDir);
Packit 0d464f
        readWriteTestWithAbsoluateCoordinates (10, 2, tempDir);
Packit 0d464f
Packit 0d464f
        ThreadPool::globalThreadPool().setNumThreads(numThreads);
Packit 0d464f
Packit 0d464f
        cout << "ok\n" << endl;
Packit 0d464f
    }
Packit 0d464f
    catch (const std::exception &e)
Packit 0d464f
    {
Packit 0d464f
        cerr << "ERROR -- caught exception: " << e.what() << endl;
Packit 0d464f
        assert (false);
Packit 0d464f
    }
Packit 0d464f
}