///////////////////////////////////////////////////////////////////////////
//
// 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.
//
///////////////////////////////////////////////////////////////////////////
#include "testDeepScanLineBasic.h"
#include "ImfDeepScanLineInputFile.h"
#include "ImfDeepScanLineOutputFile.h"
#include "ImfDeepFrameBuffer.h"
#include "ImfPartType.h"
#include "ImfChannelList.h"
#include "ImfArray.h"
#include "IlmThreadPool.h"
#include "tmpDir.h"
#include <assert.h>
#include <string.h>
#include <stdlib.h>
#include <stdio.h>
#include <vector>
namespace IMF = OPENEXR_IMF_NAMESPACE;
using namespace IMF;
using namespace std;
using namespace IMATH_NAMESPACE;
using namespace ILMTHREAD_NAMESPACE;
namespace
{
const int width = 8193;
const int height = 1;
const int minX = 0;
const int minY = 0;
const long numGib = 1; // number of GiB to allocate for huge test
const Box2i dataWindow(V2i(minX, minY), V2i(minX + width - 1, minY + height - 1));
const Box2i displayWindow(V2i(0, 0), V2i(minX + width * 2, minY + height * 2));
vector<int> channelTypes;
Array2D<unsigned int> sampleCount;
vector<unsigned char> storage; // actual pixel storage for entire image (effectively)
Header header;
void
generateRandomFile (int channelCount,
Compression compression,
bool random_channel_data,
const std::string & fn)
{
cout << "generating ... " << flush;
header = Header(displayWindow, dataWindow,
1,
IMATH_NAMESPACE::V2f (0, 0),
1,
INCREASING_Y,
compression);
cout << "compression " << compression << " " << flush;
//
// Add channels.
//
channelTypes.clear();
for (int i = 0; i < channelCount; i++)
{
int type = rand() % 3;
stringstream ss;
ss << i;
string str = ss.str();
if (type == 0)
header.channels().insert(str, Channel(IMF::UINT));
if (type == 1)
header.channels().insert(str, Channel(IMF::HALF));
if (type == 2)
header.channels().insert(str, Channel(IMF::FLOAT));
channelTypes.push_back(type);
}
header.setType(DEEPSCANLINE);
Array<Array2D< void* > > data(channelCount);
for (int i = 0; i < channelCount; i++)
data[i].resizeErase(height, width);
sampleCount.resizeErase(height, width);
remove (fn.c_str());
DeepScanLineOutputFile file (fn.c_str(), header, 8);
DeepFrameBuffer frameBuffer;
frameBuffer.insertSampleCountSlice (Slice (IMF::UINT, // type // 7
(char *) (&sampleCount[0][0]
- dataWindow.min.x
- dataWindow.min.y * width), // base // 8
sizeof (unsigned int) * 1, // xStride// 9
sizeof (unsigned int) * width)); // yStride// 10
// count total size of all pixels
Int64 bytes_per_sample = 0;
for (int i = 0; i < channelCount; i++)
{
PixelType type;
if (channelTypes[i] == 0)
type = IMF::UINT;
if (channelTypes[i] == 1)
type = IMF::HALF;
if (channelTypes[i] == 2)
type = IMF::FLOAT;
stringstream ss;
ss << i;
string str = ss.str();
int sampleSize;
if (channelTypes[i] == 0) sampleSize = sizeof (unsigned int);
if (channelTypes[i] == 1) sampleSize = sizeof (half);
if (channelTypes[i] == 2) sampleSize = sizeof (float);
int pointerSize = sizeof(char *);
bytes_per_sample+=sampleSize;
frameBuffer.insert (str, // name // 6
DeepSlice (type, // type // 7
(char *) (&data[i][0][0]
- dataWindow.min.x
- dataWindow.min.y * width), // base // 8
pointerSize * 1, // xStride// 9
pointerSize * width, // yStride// 10
sampleSize)); // sampleStride
}
file.setFrameBuffer(frameBuffer);
cout << "writing file " << endl;
Int64 total_number_of_samples = 0;
// compute ideal number of samples per pixel assuming we want abotut 5GiB of data
// int samples_per_pixel = int(5l*1024l*1024l*1024l/Int64(width*height)) / bytes_per_sample;
// compute ideal number of samples per pixel assuming we want abotut 15GiB of data
int samples_per_pixel = int(numGib*1024l*1024l*1024l/Int64(width*height)) / bytes_per_sample;
cout << " generating approx. " << samples_per_pixel << " samples per pixel\n";
for (int i = 0; i < height; i++)
{
for (int j = 0; j < width; j++)
{
sampleCount[i][j] = (rand() % 4000) + (samples_per_pixel-2000);
total_number_of_samples += sampleCount[i][j];
}
}
cout << " total number of samples: " << total_number_of_samples << std::endl;
cout << " storage required: " << total_number_of_samples*bytes_per_sample << " bytes (" <<
((total_number_of_samples*bytes_per_sample)>>30) << "GiB)" << std::endl;
//
// storage layout scheme:
// [Pixel1: [Channel1: [Sample1 Sample2 Sample...] ] [Channel2: [Sample1 Sample2...] ] [Channnel...] ]
// [Pixel2: [Channel1: [Sample1 Sample2 Sample...] ] [Channel2: [Sample1 Sample2...] ] [Channnel...] ]
// [Pixel...]
//
storage.resize(total_number_of_samples*bytes_per_sample);
Int64 write_pointer=0;
for (int i = 0; i < height; i++)
{
//
// Fill in data at the last minute.
//
for (int j = 0; j < width; j++)
{
for (int k = 0; k < channelCount; k++)
{
data[k][i][j]=&storage[write_pointer];
if (channelTypes[k] == 0)
write_pointer+=sizeof(int)*sampleCount[i][j];
if (channelTypes[k] == 1)
write_pointer+=sizeof(half)*sampleCount[i][j];
if (channelTypes[k] == 2)
write_pointer+=sizeof(float)*sampleCount[i][j];
if(random_channel_data)
{
for (int l = 0; l < sampleCount[i][j]; l++)
{
if (channelTypes[k] == 0)
((unsigned int*)data[k][i][j])[l] = rand();
if (channelTypes[k] == 1)
((half*)data[k][i][j])[l] = rand()/RAND_MAX;
if (channelTypes[k] == 2)
((float*)data[k][i][j])[l] = rand()/RAND_MAX;
}
}
else
{
for (int l = 0; l < sampleCount[i][j]; l++)
{
if (channelTypes[k] == 0)
((unsigned int*)data[k][i][j])[l] = (i * width + j) % 2049;
if (channelTypes[k] == 1)
((half*)data[k][i][j])[l] = (i * width + j) % 2049;
if (channelTypes[k] == 2)
((float*)data[k][i][j])[l] = (i * width + j) % 2049;
}
}
}
}
}
cout << " data prepared, writing ...";
file.writePixels(height);
cout << " data written\n";
}
void
readFile (int channelCount, bool bulkRead, const std::string & fn)
{
cout << "reading \n" << flush;
DeepScanLineInputFile file (fn.c_str(), 8);
const Header& fileHeader = file.header();
assert (fileHeader.displayWindow() == header.displayWindow());
assert (fileHeader.dataWindow() == header.dataWindow());
assert (fileHeader.pixelAspectRatio() == header.pixelAspectRatio());
assert (fileHeader.screenWindowCenter() == header.screenWindowCenter());
assert (fileHeader.screenWindowWidth() == header.screenWindowWidth());
assert (fileHeader.lineOrder() == header.lineOrder());
assert (fileHeader.compression() == header.compression());
assert (fileHeader.channels() == header.channels());
assert (fileHeader.type() == header.type());
Array2D<unsigned int> localSampleCount;
localSampleCount.resizeErase(height, width);
Array<Array2D< void* > > data(channelCount);
for (int i = 0; i < channelCount; i++)
data[i].resizeErase(height, width);
DeepFrameBuffer frameBuffer;
frameBuffer.insertSampleCountSlice (Slice (IMF::UINT, // type // 7
(char *) (&localSampleCount[0][0]
- dataWindow.min.x
- dataWindow.min.y * width), // base // 8)
sizeof (unsigned int) * 1, // xStride// 9
sizeof (unsigned int) * width)); // yStride// 10
Int64 bytes_per_sample=0;
for (int i = 0; i < channelCount; i++)
{
PixelType type;
if (channelTypes[i] == 0)
type = IMF::UINT;
if (channelTypes[i] == 1)
type = IMF::HALF;
if (channelTypes[i] == 2)
type = IMF::FLOAT;
stringstream ss;
ss << i;
string str = ss.str();
int sampleSize;
if (channelTypes[i] == 0) sampleSize = sizeof (unsigned int);
if (channelTypes[i] == 1) sampleSize = sizeof (half);
if (channelTypes[i] == 2) sampleSize = sizeof (float);
int pointerSize = sizeof (char *);
bytes_per_sample+=sampleSize;
frameBuffer.insert (str, // name // 6
DeepSlice (type, // type // 7
(char *) (&data[i][0][0]
- dataWindow.min.x
- dataWindow.min.y * width), // base // 8)
pointerSize * 1, // xStride// 9
pointerSize * width, // yStride// 10
sampleSize)); // sampleStride
}
file.setFrameBuffer(frameBuffer);
file.readPixelSampleCounts(dataWindow.min.y, dataWindow.max.y);
Int64 total_pixel_count = 0;
for (int i = 0; i < dataWindow.max.y - dataWindow.min.y + 1; i++)
{
int y = i + dataWindow.min.y;
for (int j = 0; j < width; j++)
{
assert(localSampleCount[i][j] == sampleCount[i][j]);
total_pixel_count += localSampleCount[i][j];
}
}
vector<char> localstorage(total_pixel_count*bytes_per_sample);
Int64 write_pointer=0;
for (int i = 0; i < height; i++)
{
//
// Fill in data at the last minute.
//
for (int j = 0; j < width; j++)
{
for (int k = 0; k < channelCount; k++)
{
data[k][i][j]=&localstorage[write_pointer];
if (channelTypes[k] == 0)
write_pointer+=sizeof(int)*sampleCount[i][j];
if (channelTypes[k] == 1)
write_pointer+=sizeof(half)*sampleCount[i][j];
if (channelTypes[k] == 2)
write_pointer+=sizeof(float)*sampleCount[i][j];
}
}
}
cout << "reading image data ... " << flush;
file.readPixels(dataWindow.min.y, dataWindow.max.y);
cout << " image read \n" << flush;
}
void
readWriteTest (int channelCount,
int testTimes,
bool random_channel_data,
const std::string & fn)
{
cout << "Testing files with " << channelCount << " channels " << testTimes << " times."
<< endl << flush;
for (int i = 0; i < testTimes; i++)
{
int compressionIndex = i % 3;
Compression compression;
switch (compressionIndex)
{
case 0:
compression = NO_COMPRESSION;
break;
case 1:
compression = RLE_COMPRESSION;
break;
case 2:
compression = ZIPS_COMPRESSION;
break;
}
generateRandomFile (channelCount, compression, random_channel_data, fn);
readFile (channelCount, false, fn);
remove (fn.c_str());
}
}
}; // namespace
void testDeepScanLineHuge (const std::string & tempDir)
{
try
{
cout << "\n\nTesting the DeepScanLineInput/OutputFile for huge scanlines:\n" << endl;
srand(1);
std::string fn = tempDir + "imf_test_deep_scanline_huge.exr";
readWriteTest (10, 5 , false, fn);
readWriteTest (10, 5 , true, fn);
cout << "ok\n" << endl;
}
catch (const std::exception &e)
{
cerr << "ERROR -- caught exception: " << e.what() << endl;
assert (false);
}
}