///////////////////////////////////////////////////////////////////////////

//

// Copyright (c) 2003-2012, 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 <ImfRgbaFile.h>

#include <ImfArray.h>

#include <ImfPreviewImage.h>

#include <fstream>

#include <stdio.h>

#include <assert.h>

#ifndef ILM_IMF_TEST_IMAGEDIR

    #define ILM_IMF_TEST_IMAGEDIR

#endif

using namespace OPENEXR_IMF_NAMESPACE;

using namespace std;

using namespace IMATH_NAMESPACE;

namespace {

void

readWriteFiles (const char fileName1[],

	        const char fileName2[],

		const char fileName3[])

{

    //

    // Test if the preview image attribute works correctly:

    //

    // Read file1, which does not contain a preview image.

    //

    // Generate a preview image, and store both the pixels

    // from file 1 and the preview image in file 2.

    //

    // Read file 2, and verify that both the preview image, and

    // the main image are exactly what we stored in the file.

    //

    // Write file 3, with the same main image as file 2, but

    // initially leave the preview image blank.  Update the

    // preview image half way through writing the main image's

    // pixels.

    //

    // Compare file 2 and file 3 byte by byte, and verify that

    // the files are identical.

    //

    cout << "reading file " << fileName1 << endl;

    RgbaInputFile file1 (fileName1);

    assert (!file1.header().hasPreviewImage());

    const Box2i &dw = file1.dataWindow();

    int w = dw.max.x - dw.min.x + 1;

    int h = dw.max.y - dw.min.y + 1;

    int dx = dw.min.x;

    int dy = dw.min.y;

    Array<OPENEXR_IMF_NAMESPACE::Rgba> pixels1 (w * h);

    file1.setFrameBuffer (pixels1 - dx - dy * w, 1, w);

    file1.readPixels (dw.min.y, dw.max.y);

    cout << "generating preview image" << endl;

    const int PREVIEW_WIDTH  = 128;

    const int PREVIEW_HEIGHT = 64;

    PreviewImage preview1 (PREVIEW_WIDTH, PREVIEW_HEIGHT);

    for (int y = 0; y < PREVIEW_HEIGHT; ++y)

	for (int x = 0; x < PREVIEW_WIDTH; ++x)

	    preview1.pixel (x, y) = PreviewRgba (x*2, y*4, x+y, 128);

    cout << "writing file " << fileName2 << endl;

    {

	Header header (file1.header());

	header.setPreviewImage (preview1);

	RgbaOutputFile file2 (fileName2, header);

	file2.setFrameBuffer (pixels1 - dx - dy * w, 1, w);

	for (int y = dw.min.y; y <= dw.max.y; ++y)

	    file2.writePixels (1);

    }

    cout << "reading file " << fileName2 << endl;

    {

	RgbaInputFile file2 (fileName2);

	assert (file2.header().hasPreviewImage());

	const PreviewImage &preview2 = file2.header().previewImage();

	for (int i = 0; i < PREVIEW_WIDTH * PREVIEW_HEIGHT; ++i)

	{

	    assert (preview1.pixels()[i].r == preview2.pixels()[i].r);

	    assert (preview1.pixels()[i].g == preview2.pixels()[i].g);

	    assert (preview1.pixels()[i].b == preview2.pixels()[i].b);

	    assert (preview1.pixels()[i].a == preview2.pixels()[i].a);

	}

	assert (dw == file2.dataWindow());

	int w = dw.max.x - dw.min.x + 1;

	int h = dw.max.y - dw.min.y + 1;

	int dx = dw.min.x;

	int dy = dw.min.y;

	Array<OPENEXR_IMF_NAMESPACE::Rgba> pixels2 (w * h);

	file2.setFrameBuffer (pixels2 - dx - dy * w, 1, w);

	file2.readPixels (dw.min.y, dw.max.y);

	for (int i = 0; i < w * h; ++h)

	{

	    assert (pixels1[i].r == pixels2[i].r);

	    assert (pixels1[i].g == pixels2[i].g);

	    assert (pixels1[i].b == pixels2[i].b);

	    assert (pixels1[i].a == pixels2[i].a);

	}

    }

    cout << "writing file " << fileName3 << endl;

    {

	Header header (file1.header());

	header.setPreviewImage (PreviewImage (PREVIEW_WIDTH, PREVIEW_HEIGHT));

	RgbaOutputFile file3 (fileName3, header);

	file3.setFrameBuffer (pixels1 - dx - dy * w, 1, w);

	for (int y = dw.min.y; y <= dw.max.y; ++y)

	{

	    file3.writePixels (1);

	    if (y == (dw.min.y + dw.max.y) / 2)

		file3.updatePreviewImage (preview1.pixels());

	}

    }

    cout << "comparing files " << fileName2 << " and " << fileName3 << endl;

    {

	ifstream file2 (fileName2, std::ios_base::binary);

	ifstream file3 (fileName3, std::ios_base::binary);

	while (true)

	{

	    int c2 = file2.get();

	    int c3 = file3.get();

	    if (file2.eof())

		break;

	    assert (c2 == c3);

	    assert (!!file2 && !!file3);

	}

    }

    remove (fileName2);

    remove (fileName3);

}

} // namespace

void

testPreviewImage (const std::string &tempDir)

{

    std::string filename1 = tempDir + "imf_preview1.exr";

    std::string filename2 = tempDir + "imf_preview2.exr";

    try

    {

	cout << "Testing preview image attribute" << endl;

	readWriteFiles (ILM_IMF_TEST_IMAGEDIR "comp_piz.exr",

			filename1.c_str(),

			filename2.c_str());

	cout << "ok\n" << endl;

    }

    catch (const std::exception &e)

    {

	cerr << "ERROR -- caught exception: " << e.what() << endl;

	assert (false);

    }

}