|
Packit Service |
6754ca |
///////////////////////////////////////////////////////////////////////////
|
|
Packit Service |
6754ca |
//
|
|
Packit Service |
6754ca |
// Copyright (c) 2007, Industrial Light & Magic, a division of Lucas
|
|
Packit Service |
6754ca |
// Digital Ltd. LLC
|
|
Packit Service |
6754ca |
//
|
|
Packit Service |
6754ca |
// All rights reserved.
|
|
Packit Service |
6754ca |
//
|
|
Packit Service |
6754ca |
// Redistribution and use in source and binary forms, with or without
|
|
Packit Service |
6754ca |
// modification, are permitted provided that the following conditions are
|
|
Packit Service |
6754ca |
// met:
|
|
Packit Service |
6754ca |
// * Redistributions of source code must retain the above copyright
|
|
Packit Service |
6754ca |
// notice, this list of conditions and the following disclaimer.
|
|
Packit Service |
6754ca |
// * Redistributions in binary form must reproduce the above
|
|
Packit Service |
6754ca |
// copyright notice, this list of conditions and the following disclaimer
|
|
Packit Service |
6754ca |
// in the documentation and/or other materials provided with the
|
|
Packit Service |
6754ca |
// distribution.
|
|
Packit Service |
6754ca |
// * Neither the name of Industrial Light & Magic nor the names of
|
|
Packit Service |
6754ca |
// its contributors may be used to endorse or promote products derived
|
|
Packit Service |
6754ca |
// from this software without specific prior written permission.
|
|
Packit Service |
6754ca |
//
|
|
Packit Service |
6754ca |
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
|
Packit Service |
6754ca |
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
|
Packit Service |
6754ca |
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
|
Packit Service |
6754ca |
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
|
Packit Service |
6754ca |
// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
|
Packit Service |
6754ca |
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
|
Packit Service |
6754ca |
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
|
Packit Service |
6754ca |
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
|
Packit Service |
6754ca |
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
Packit Service |
6754ca |
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
|
Packit Service |
6754ca |
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
Packit Service |
6754ca |
//
|
|
Packit Service |
6754ca |
///////////////////////////////////////////////////////////////////////////
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
#include <ImfMultiPartOutputFile.h>
|
|
Packit Service |
6754ca |
#include <ImfMultiPartInputFile.h>
|
|
Packit Service |
6754ca |
#include <ImfPartType.h>
|
|
Packit Service |
6754ca |
#include <ImfOutputPart.h>
|
|
Packit Service |
6754ca |
#include <ImfInputPart.h>
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
#include <ImfRgbaFile.h>
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
#include <ImfArray.h>
|
|
Packit Service |
6754ca |
#include <ImfThreading.h>
|
|
Packit Service |
6754ca |
#include <IlmThread.h>
|
|
Packit Service |
6754ca |
#include <Iex.h>
|
|
Packit Service |
6754ca |
#include <iostream>
|
|
Packit Service |
6754ca |
#include <cassert>
|
|
Packit Service |
6754ca |
#include <stdio.h>
|
|
Packit Service |
6754ca |
#include <vector>
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
#include "tmpDir.h"
|
|
Packit Service |
6754ca |
#include "fuzzFile.h"
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
// Handle the case when the custom namespace is not exposed
|
|
Packit Service |
6754ca |
#include <OpenEXRConfig.h>
|
|
Packit Service |
6754ca |
#include <ImfChannelList.h>
|
|
Packit Service |
6754ca |
using namespace OPENEXR_IMF_INTERNAL_NAMESPACE;
|
|
Packit Service |
6754ca |
using namespace std;
|
|
Packit Service |
6754ca |
using namespace IMATH_NAMESPACE;
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
namespace {
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
void
|
|
Packit Service |
6754ca |
fillPixels (Array2D<Rgba> &pixels, int w, int h)
|
|
Packit Service |
6754ca |
{
|
|
Packit Service |
6754ca |
for (int y = 0; y < h; ++y)
|
|
Packit Service |
6754ca |
{
|
|
Packit Service |
6754ca |
for (int x = 0; x < w; ++x)
|
|
Packit Service |
6754ca |
{
|
|
Packit Service |
6754ca |
Rgba &p = pixels[y][x];
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
p.r = 0.5 + 0.5 * sin (0.1 * x + 0.1 * y);
|
|
Packit Service |
6754ca |
p.g = 0.5 + 0.5 * sin (0.1 * x + 0.2 * y);
|
|
Packit Service |
6754ca |
p.b = 0.5 + 0.5 * sin (0.1 * x + 0.3 * y);
|
|
Packit Service |
6754ca |
p.a = (p.r + p.b + p.g) / 3.0;
|
|
Packit Service |
6754ca |
}
|
|
Packit Service |
6754ca |
}
|
|
Packit Service |
6754ca |
}
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
void
|
|
Packit Service |
6754ca |
writeImage (const char fileName[],
|
|
Packit Service |
6754ca |
int width,
|
|
Packit Service |
6754ca |
int height,
|
|
Packit Service |
6754ca |
const Array2D<Rgba> &pixels,
|
|
Packit Service |
6754ca |
int parts,
|
|
Packit Service |
6754ca |
Compression comp)
|
|
Packit Service |
6754ca |
{
|
|
Packit Service |
6754ca |
//
|
|
Packit Service |
6754ca |
// Save the image with the specified compression.
|
|
Packit Service |
6754ca |
//
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
cout << parts << " parts with compression: " << comp << endl;
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
Header header (width, height);
|
|
Packit Service |
6754ca |
header.compression() = comp;
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
if(parts==1)
|
|
Packit Service |
6754ca |
{
|
|
Packit Service |
6754ca |
RgbaOutputFile out (fileName, header, WRITE_RGBA);
|
|
Packit Service |
6754ca |
out.setFrameBuffer (&pixels[0][0], 1, width);
|
|
Packit Service |
6754ca |
out.writePixels (height);
|
|
Packit Service |
6754ca |
}else{
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
header.setType(SCANLINEIMAGE);
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
header.channels().insert("R",Channel(HALF));
|
|
Packit Service |
6754ca |
header.channels().insert("G",Channel(HALF));
|
|
Packit Service |
6754ca |
header.channels().insert("B",Channel(HALF));
|
|
Packit Service |
6754ca |
header.channels().insert("A",Channel(HALF));
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
vector<Header> headers(parts);
|
|
Packit Service |
6754ca |
for(int i=0;i
|
|
Packit Service |
6754ca |
{
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
headers[i]=header;
|
|
Packit Service |
6754ca |
ostringstream s;
|
|
Packit Service |
6754ca |
s << i;
|
|
Packit Service |
6754ca |
headers[i].setName(s.str());
|
|
Packit Service |
6754ca |
}
|
|
Packit Service |
6754ca |
MultiPartOutputFile out(fileName,&headers[0],parts);
|
|
Packit Service |
6754ca |
FrameBuffer f;
|
|
Packit Service |
6754ca |
f.insert("R",Slice(HALF,(char *) &(pixels[0][0].r),sizeof(Rgba),width*sizeof(Rgba)));
|
|
Packit Service |
6754ca |
f.insert("G",Slice(HALF,(char *) &(pixels[0][0].g),sizeof(Rgba),width*sizeof(Rgba)));
|
|
Packit Service |
6754ca |
f.insert("B",Slice(HALF,(char *) &(pixels[0][0].b),sizeof(Rgba),width*sizeof(Rgba)));
|
|
Packit Service |
6754ca |
f.insert("A",Slice(HALF,(char *) &(pixels[0][0].a),sizeof(Rgba),width*sizeof(Rgba)));
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
for(int i=0;i
|
|
Packit Service |
6754ca |
{
|
|
Packit Service |
6754ca |
OutputPart(out,i).setFrameBuffer(f);
|
|
Packit Service |
6754ca |
OutputPart(out,i).writePixels(height);
|
|
Packit Service |
6754ca |
}
|
|
Packit Service |
6754ca |
}
|
|
Packit Service |
6754ca |
}
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
void
|
|
Packit Service |
6754ca |
readImage (const char fileName[])
|
|
Packit Service |
6754ca |
{
|
|
Packit Service |
6754ca |
//
|
|
Packit Service |
6754ca |
// Try to read the specified file.
|
|
Packit Service |
6754ca |
// Reading should either succeed or throw an exception,
|
|
Packit Service |
6754ca |
// but it should not crash for any reason.
|
|
Packit Service |
6754ca |
//
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
try
|
|
Packit Service |
6754ca |
{
|
|
Packit Service |
6754ca |
// first , test Rgba interface
|
|
Packit Service |
6754ca |
RgbaInputFile in (fileName);
|
|
Packit Service |
6754ca |
const Box2i &dw = in.dataWindow();
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
int w = dw.max.x - dw.min.x + 1;
|
|
Packit Service |
6754ca |
int dx = dw.min.x;
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
if (w > (1 << 24))
|
|
Packit Service |
6754ca |
return;
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
Array<Rgba> pixels (w);
|
|
Packit Service |
6754ca |
in.setFrameBuffer (&pixels[-dx], 1, 0);
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
for (int y = dw.min.y; y <= dw.max.y; ++y)
|
|
Packit Service |
6754ca |
in.readPixels (y);
|
|
Packit Service |
6754ca |
}catch(...)
|
|
Packit Service |
6754ca |
{
|
|
Packit Service |
6754ca |
// expect exceptions
|
|
Packit Service |
6754ca |
}
|
|
Packit Service |
6754ca |
try{
|
|
Packit Service |
6754ca |
// now test Multipart interface (even for single part files)
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
MultiPartInputFile file(fileName);
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
for(int p=0;p
|
|
Packit Service |
6754ca |
{
|
|
Packit Service |
6754ca |
InputPart in(file,p);
|
|
Packit Service |
6754ca |
const Box2i &dw = in.header().dataWindow();
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
int w = dw.max.x - dw.min.x + 1;
|
|
Packit Service |
6754ca |
int dx = dw.min.x;
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
if (w > (1 << 24))
|
|
Packit Service |
6754ca |
return;
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
Array<Rgba> pixels (w);
|
|
Packit Service |
6754ca |
FrameBuffer i;
|
|
Packit Service |
6754ca |
i.insert("R",Slice(HALF,(char *)&(pixels[-dx].r),sizeof(Rgba),0));
|
|
Packit Service |
6754ca |
i.insert("G",Slice(HALF,(char *)&(pixels[-dx].g),sizeof(Rgba),0));
|
|
Packit Service |
6754ca |
i.insert("B",Slice(HALF,(char *)&(pixels[-dx].b),sizeof(Rgba),0));
|
|
Packit Service |
6754ca |
i.insert("A",Slice(HALF,(char *)&(pixels[-dx].a),sizeof(Rgba),0));
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
in.setFrameBuffer (i);
|
|
Packit Service |
6754ca |
for (int y = dw.min.y; y <= dw.max.y; ++y)
|
|
Packit Service |
6754ca |
in.readPixels (y);
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
}
|
|
Packit Service |
6754ca |
}
|
|
Packit Service |
6754ca |
catch (...)
|
|
Packit Service |
6754ca |
{
|
|
Packit Service |
6754ca |
// empty
|
|
Packit Service |
6754ca |
}
|
|
Packit Service |
6754ca |
}
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
void
|
|
Packit Service |
6754ca |
fuzzScanLines (int numThreads, Rand48 &random)
|
|
Packit Service |
6754ca |
{
|
|
Packit Service |
6754ca |
if (ILMTHREAD_NAMESPACE::supportsThreads())
|
|
Packit Service |
6754ca |
{
|
|
Packit Service |
6754ca |
setGlobalThreadCount (numThreads);
|
|
Packit Service |
6754ca |
cout << "\nnumber of threads: " << globalThreadCount() << endl;
|
|
Packit Service |
6754ca |
}
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
Header::setMaxImageSize (10000, 10000);
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
const int W = 217;
|
|
Packit Service |
6754ca |
const int H = 197;
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
Array2D<Rgba> pixels (H, W);
|
|
Packit Service |
6754ca |
fillPixels (pixels, W, H);
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
const char *goodFile = IMF_TMP_DIR "imf_test_scanline_file_fuzz_good.exr";
|
|
Packit Service |
6754ca |
const char *brokenFile = IMF_TMP_DIR "imf_test_scanline_file_fuzz_broken.exr";
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
// re-attempt to read broken file if it still remains on disk from previous aborted run
|
|
Packit Service |
6754ca |
readImage(brokenFile);
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
for (int parts = 1 ; parts<4 ; ++parts)
|
|
Packit Service |
6754ca |
{
|
|
Packit Service |
6754ca |
for (int comp = 0; comp < NUM_COMPRESSION_METHODS; ++comp)
|
|
Packit Service |
6754ca |
{
|
|
Packit Service |
6754ca |
writeImage (goodFile, W, H, pixels, parts,Compression (comp));
|
|
Packit Service |
6754ca |
fuzzFile (goodFile, brokenFile, readImage, 5000, 3000, random);
|
|
Packit Service |
6754ca |
}
|
|
Packit Service |
6754ca |
}
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
remove (goodFile);
|
|
Packit Service |
6754ca |
remove (brokenFile);
|
|
Packit Service |
6754ca |
}
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
} // namespace
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
void
|
|
Packit Service |
6754ca |
testFuzzScanLines ()
|
|
Packit Service |
6754ca |
{
|
|
Packit Service |
6754ca |
try
|
|
Packit Service |
6754ca |
{
|
|
Packit Service |
6754ca |
cout << "Testing scanline-based files "
|
|
Packit Service |
6754ca |
"with randomly inserted errors" << endl;
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
Rand48 random (1);
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
fuzzScanLines (0, random);
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
if (ILMTHREAD_NAMESPACE::supportsThreads())
|
|
Packit Service |
6754ca |
fuzzScanLines (2, random);
|
|
Packit Service |
6754ca |
|
|
Packit Service |
6754ca |
cout << "ok\n" << endl;
|
|
Packit Service |
6754ca |
}
|
|
Packit Service |
6754ca |
catch (const std::exception &e)
|
|
Packit Service |
6754ca |
{
|
|
Packit Service |
6754ca |
cerr << "ERROR -- caught exception: " << e.what() << endl;
|
|
Packit Service |
6754ca |
assert (false);
|
|
Packit Service |
6754ca |
}
|
|
Packit Service |
6754ca |
}
|