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

//

// Copyright (c) 2004-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 <ImfStandardAttributes.h>

#include <ImfFramesPerSecond.h>

#include <ImfArray.h>

#include "ImathRandom.h"

#include <fstream>

#include <iomanip>

#include <stdio.h>

#include <assert.h>

using namespace OPENEXR_IMF_NAMESPACE;

using namespace std;

using namespace IMATH_NAMESPACE;

namespace {

void

convertRGBtoXYZ ()

{

    cout << "conversion from RGB to XYZ" << endl;

    Chromaticities c;

    float Y = 100;

    M44f M1 = RGBtoXYZ (c, Y);

    V3f R1 = V3f (1, 0, 0) * M1;

    V3f G1 = V3f (0, 1, 0) * M1;

    V3f B1 = V3f (0, 0, 1) * M1;

    V3f W1 = V3f (1, 1, 1) * M1;

    cout << "red   XYZ = " << R1 << endl;

    cout << "green XYZ = " << G1 << endl;

    cout << "blue  XYZ = " << B1 << endl;

    cout << "white XYZ = " << W1 << endl;

    V2f r1 (R1.x / (R1.x + R1.y + R1.z), R1.y / (R1.x + R1.y + R1.z));

    V2f g1 (G1.x / (G1.x + G1.y + G1.z), G1.y / (G1.x + G1.y + G1.z));

    V2f b1 (B1.x / (B1.x + B1.y + B1.z), B1.y / (B1.x + B1.y + B1.z));

    V2f w1 (W1.x / (W1.x + W1.y + W1.z), W1.y / (W1.x + W1.y + W1.z));

    cout << "red   xy = " << r1 << endl;

    cout << "green xy = " << g1 << endl;

    cout << "blue  xy = " << b1 << endl;

    cout << "white xy = " << w1 << endl;

    assert (equalWithRelError (W1.y, Y, 1e-5F));

    assert (r1.equalWithAbsError (c.red, 1e-5F));

    assert (g1.equalWithAbsError (c.green, 1e-5F));

    assert (b1.equalWithAbsError (c.blue, 1e-5F));

    assert (w1.equalWithAbsError (c.white, 1e-5F));

    cout << "conversion from XYZ to RGB" << endl;

    M44f M2 = XYZtoRGB (c, Y);

    V3f R2 = R1 * M2;

    V3f G2 = G1 * M2;

    V3f B2 = B1 * M2;

    V3f W2 = W1 * M2;

    cout << "red   RGB = " << R2 << endl;

    cout << "green RGB = " << G2 << endl;

    cout << "blue  RGB = " << B2 << endl;

    cout << "white RGB = " << W2 << endl;

    assert (R2.equalWithAbsError (V3f (1, 0, 0), 1e-3F));

    assert (G2.equalWithAbsError (V3f (0, 1, 0), 1e-3F));

    assert (B2.equalWithAbsError (V3f (0, 0, 1), 1e-3F));

    assert (W2.equalWithAbsError (V3f (1, 1, 1), 1e-3F));

}

void

writeReadChromaticities (const char fileName[])

{

    cout << "chromaticities attribute" << endl;

    cout << "writing, ";

    Chromaticities c1 (V2f (1, 2), V2f (3, 4), V2f (5, 6), V2f (7, 8));

    static const int W = 100;

    static const int H = 100;

    Header header (W, H);

    assert (hasChromaticities (header) == false);

    addChromaticities (header, c1);

    assert (hasChromaticities (header) == true);

    {

	RgbaOutputFile out (fileName, header);

	Rgba pixels[W];

	for (int i = 0; i < W; ++i)

	{

	    pixels[i].r = 1;

	    pixels[i].g = 1;

	    pixels[i].b = 1;

	    pixels[i].a = 1;

	}

	out.setFrameBuffer (pixels, 1, 0);

	out.writePixels (H);

    }

    cout << "reading, comparing" << endl;

    {

	RgbaInputFile in (fileName);

	const Chromaticities &c2 = chromaticities (in.header());

	assert (hasChromaticities (in.header()) == true);

	assert (c1.red == c2.red);

	assert (c1.green == c2.green);

	assert (c1.blue == c2.blue);

	assert (c1.white == c2.white);

    }

    remove (fileName);

}

void

latLongMap (const char fileName1[], const char fileName2[])

{

    cout << "latitude-longitude environment map" << endl;

    const int W = 360;

    const int H = 180;

    Header header (W, H);

    addEnvmap (header, ENVMAP_LATLONG);

    V2f pos;

    pos = LatLongMap::latLong (V3f (0, 1, 0));

    assert (equalWithAbsError (pos.x, float (M_PI/2), 1e-6f));

    pos = LatLongMap::latLong (V3f (0, -1, 0));

    assert (equalWithAbsError (pos.x, float (-M_PI/2), 1e-6f));

    pos = LatLongMap::latLong (V3f (0, 0, 1));

    assert (pos.equalWithAbsError (V2f (0, 0), 1e-6f));

    pos = LatLongMap::latLong (V3f (1, 0, 0));

    assert (pos.equalWithAbsError (V2f (0, M_PI/2), 1e-6f));

    pos = LatLongMap::latLong (V3f (-1, 0, 0));

    assert (pos.equalWithAbsError (V2f (0, -M_PI/2), 1e-6f));

    pos = LatLongMap::latLong (V3f (0, 1, 1));

    assert (pos.equalWithAbsError (V2f (M_PI/4, 0), 1e-6f));

    pos = LatLongMap::latLong (V3f (0, -1, 1));

    assert (pos.equalWithAbsError (V2f (-M_PI/4, 0), 1e-6f));

    pos = LatLongMap::pixelPosition (header.dataWindow(), V2f (M_PI/2, M_PI));

    assert (pos.equalWithAbsError (V2f (0, 0), 1e-6f * W));

    pos = LatLongMap::pixelPosition(header.dataWindow(), V2f (-M_PI/2, -M_PI));

    assert (pos.equalWithAbsError (V2f (header.dataWindow().max), 1e-6f * W));

    Array2D<Rgba> pixels (H, W);

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

    {

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

	{

	    Rgba &p = pixels[y][x];

	    V3f dir = LatLongMap::direction (header.dataWindow(), V2f (x, y));

	    p.r = dir.x + 1;

	    p.g = dir.y + 1;

	    p.b = dir.z + 1;

	}

    }

    {

	RgbaOutputFile out (fileName1, header, WRITE_RGB);

	out.setFrameBuffer (&pixels[0][0], 1, W);

	out.writePixels (H);

    }

    Rand48 rand (0);

    for (int i = 0; i < W * H * 3; ++i)

    {

	V3f dir = hollowSphereRand<V3f> (rand);

	V2f pos = LatLongMap::pixelPosition (header.dataWindow(), dir);

	Rgba &p = pixels[int (pos.y + 0.5)][int (pos.x + 0.5)];

	p.r = (dir.x + 1) * 0.8;

	p.g = (dir.y + 1) * 0.8;

	p.b = (dir.z + 1) * 0.8;

	V3f dir1 = LatLongMap::direction (header.dataWindow(), pos);

	assert (dir.equalWithAbsError (dir1.normalized(), 1e-5f));

    }

    {

	RgbaOutputFile out (fileName2, header, WRITE_RGB);

	out.setFrameBuffer (&pixels[0][0], 1, W);

	out.writePixels (H);

    }

    remove (fileName1);

    remove (fileName2);

}

void

cubeMap (const char fileName1[], const char fileName2[])

{

    cout << "cube environment map" << endl;

    const int N = 128;

    const int W = N;

    const int H = N * 6;

    Header header (W, H);

    addEnvmap (header, ENVMAP_CUBE);

    int sof = CubeMap::sizeOfFace (header.dataWindow());

    assert (sof == N);

    for (int face1 = 0; face1 < 6; ++face1)

    {

	Box2i dw1 = CubeMap::dataWindowForFace (CubeMapFace (face1),

						header.dataWindow());

	assert (dw1.max.x - dw1.min.x == sof - 1);

	assert (dw1.max.y - dw1.min.y == sof - 1);

	assert (header.dataWindow().intersects (dw1.min));

	assert (header.dataWindow().intersects (dw1.max));

	for (int face2 = face1 + 1; face2 < 6; ++face2)

	{

	    Box2i dw2 = CubeMap::dataWindowForFace (CubeMapFace (face2),

						    header.dataWindow());

	    assert (!dw1.intersects (dw2));

	}

    }

    CubeMapFace face;

    V2f pos;

    CubeMap::faceAndPixelPosition (V3f (1, 0, 0),

				   header.dataWindow(),

				   face, pos);

    assert (face == CUBEFACE_POS_X);

    assert (pos.equalWithAbsError (V2f ((sof - 1), (sof - 1)) / 2, 1e-6 * W));

    CubeMap::faceAndPixelPosition (V3f (-1, 0, 0),

	                           header.dataWindow(),

				   face, pos);

    assert (face == CUBEFACE_NEG_X);

    assert (pos.equalWithAbsError (V2f ((sof - 1), (sof - 1)) / 2, 1e-6 * W));

    CubeMap::faceAndPixelPosition (V3f (0, 1, 0),

	    			   header.dataWindow(),

				   face, pos);

    assert (face == CUBEFACE_POS_Y);

    assert (pos.equalWithAbsError (V2f ((sof - 1), (sof - 1)) / 2, 1e-6 * W));

    CubeMap::faceAndPixelPosition (V3f (0, -1, 0),

	    			   header.dataWindow(),

				   face, pos);

    assert (face == CUBEFACE_NEG_Y);

    assert (pos.equalWithAbsError (V2f ((sof - 1), (sof - 1)) / 2, 1e-6 * W));

    CubeMap::faceAndPixelPosition (V3f (0, 0, 1),

	    			   header.dataWindow(),

				   face, pos);

    assert (face == CUBEFACE_POS_Z);

    assert (pos.equalWithAbsError (V2f ((sof - 1), (sof - 1)) / 2, 1e-6 * W));

    CubeMap::faceAndPixelPosition (V3f (0, 0, -1),

	    			   header.dataWindow(),

				   face, pos);

    assert (face == CUBEFACE_NEG_Z);

    assert (pos.equalWithAbsError (V2f ((sof - 1), (sof - 1)) / 2, 1e-6 * W));

    Array2D<Rgba> pixels (H, W);

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

    {

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

	{

	    Rgba &p = pixels[y][x];

	    p.r = p.g = p.b = 0;

	}

    }

    for (int face = 0; face < 6; ++face)

    {

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

	{

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

	    {

		V2f px = CubeMap::pixelPosition (CubeMapFace (face),

						 header.dataWindow(),

						 V2f (x, y));

		Rgba &p = pixels[int (px.y + 0.5)][int (px.x + 0.5)];

		V3f dir = CubeMap::direction (CubeMapFace (face),

			                      header.dataWindow(),

					      V2f (x, y));

		dir.normalize();

		p.r = dir.x + 1;

		p.g = dir.y + 1;

		p.b = dir.z + 1;

	    }

	}

    }

    {

	RgbaOutputFile out (fileName1, header, WRITE_RGB);

	out.setFrameBuffer (&pixels[0][0], 1, W);

	out.writePixels (H);

    }

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

    {

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

	{

	    Rgba &p = pixels[y][x];

	    assert (p.r != 0 || p.g != 0 || p.b != 0);

	}

    }

    Rand48 rand (0);

    for (int i = 0; i < W * H * 3; ++i)

    {

	V3f dir = hollowSphereRand<V3f> (rand);

	CubeMapFace face;

	V2f pif;

	CubeMap::faceAndPixelPosition (dir, header.dataWindow(), face, pif);

	V2f pos = CubeMap::pixelPosition (face, header.dataWindow(), pif);

	Rgba &p = pixels[int (pos.y + 0.5)][int (pos.x + 0.5)];

	p.r = (dir.x + 1) * 0.8;

	p.g = (dir.y + 1) * 0.8;

	p.b = (dir.z + 1) * 0.8;

	V3f dir1 = CubeMap::direction (face, header.dataWindow(), pif);

	assert (dir.equalWithAbsError (dir1.normalized(), 1e-6f));

    }

    {

	RgbaOutputFile out (fileName2, header, WRITE_RGB);

	out.setFrameBuffer (&pixels[0][0], 1, W);

	out.writePixels (H);

    }

    remove (fileName1);

    remove (fileName2);

}

void

writeReadKeyCode (const char fileName[])

{

    cout << "key code attribute" << endl;

    cout << "writing, ";

    KeyCode k1 (12,	// filmMfcCode

	        34,	// filmType

		123456,	// prefix

		1234,	// count

		45,	// perfOffset

		3,	// perfsPerFrame

		80);	// perfsPerCount

    assert (k1.filmMfcCode() == 12);

    assert (k1.filmType() == 34);

    assert (k1.prefix() == 123456);

    assert (k1.count() == 1234);

    assert (k1.perfOffset() == 45);

    assert (k1.perfsPerFrame() == 3);

    assert (k1.perfsPerCount() == 80);

    static const int W = 100;

    static const int H = 100;

    Header header (W, H);

    assert (hasKeyCode (header) == false);

    addKeyCode (header, k1);

    assert (hasKeyCode (header) == true);

    {

	RgbaOutputFile out (fileName, header);

	Rgba pixels[W];

	for (int i = 0; i < W; ++i)

	{

	    pixels[i].r = 1;

	    pixels[i].g = 1;

	    pixels[i].b = 1;

	    pixels[i].a = 1;

	}

	out.setFrameBuffer (pixels, 1, 0);

	out.writePixels (H);

    }

    cout << "reading, comparing" << endl;

    {

	RgbaInputFile in (fileName);

	const KeyCode &k2 = keyCode (in.header());

	assert (hasKeyCode (in.header()) == true);

	assert (k1.filmMfcCode() == k2.filmMfcCode());

	assert (k1.filmType() == k2.filmType());

	assert (k1.prefix() == k2.prefix());

	assert (k1.count() == k2.count());

	assert (k1.perfOffset() == k2.perfOffset());

	assert (k1.perfsPerFrame() == k2.perfsPerFrame());

	assert (k1.perfsPerCount() == k2.perfsPerCount());

    }

    remove (fileName);

}

void

timeCodeMethods ()

{

    cout << "time code methods" << endl;

    TimeCode t;

    assert (t.timeAndFlags() == 0);

    assert (t.userData() == 0);

    // Frames

    t.setTimeAndFlags (0x00000000);

    t.setFrame (29);

    assert (t.frame() == 29);

    assert (t.timeAndFlags() == 0x00000029);

    t.setTimeAndFlags (0xffffffff);

    t.setFrame (0);

    assert (t.frame() == 0);

    assert (t.timeAndFlags() == 0xffffffc0);

    // Seconds

    t.setTimeAndFlags (0x00000000);

    t.setSeconds (59);

    assert (t.seconds() == 59);

    assert (t.timeAndFlags() == 0x00005900);

    t.setTimeAndFlags (0xffffffff);

    t.setSeconds (0);

    assert (t.seconds() == 0);

    assert (t.timeAndFlags() == 0xffff80ff);

    // Minutes

    t.setTimeAndFlags (0x00000000);

    t.setMinutes (59);

    assert (t.minutes() == 59);

    assert (t.timeAndFlags() == 0x00590000);

    t.setTimeAndFlags (0xffffffff);

    t.setMinutes (0);

    assert (t.minutes() == 0);

    assert (t.timeAndFlags() == 0xff80ffff);

    // Hours

    t.setTimeAndFlags (0x00000000);

    t.setHours (23);

    assert (t.hours() == 23);

    assert (t.timeAndFlags() == 0x23000000);

    t.setTimeAndFlags (0xffffffff);

    t.setHours (0);

    assert (t.hours() == 0);

    assert (t.timeAndFlags() == 0xc0ffffff);

    // Drop frame flag

    t.setTimeAndFlags (0x00000000);

    t.setDropFrame (true);

    assert (t.dropFrame() == true);

    assert (t.timeAndFlags() == 0x00000040);

    t.setTimeAndFlags (0xffffffff);

    t.setDropFrame (false);

    assert (t.dropFrame() == false);

    assert (t.timeAndFlags() == 0xffffffbf);

    // Color frame flag

    t.setTimeAndFlags (0x00000000);

    t.setColorFrame (true);

    assert (t.colorFrame() == true);

    assert (t.timeAndFlags() == 0x00000080);

    t.setTimeAndFlags (0xffffffff);

    t.setColorFrame (false);

    assert (t.colorFrame() == false);

    assert (t.timeAndFlags() == 0xffffff7f);

    // Field/phase flag

    t.setTimeAndFlags (0x00000000);

    t.setFieldPhase (true);

    assert (t.fieldPhase() == true);

    assert (t.timeAndFlags (TimeCode::TV60_PACKING)   == 0x00008000);

    assert (t.timeAndFlags (TimeCode::TV50_PACKING)   == 0x80000000);

    assert (t.timeAndFlags (TimeCode::FILM24_PACKING) == 0x00008000);

    t.setTimeAndFlags (0xffffffff);

    t.setFieldPhase (false);

    assert (t.fieldPhase() == false);

    assert (t.timeAndFlags (TimeCode::TV60_PACKING)   == 0xffff7fff);

    assert (t.timeAndFlags (TimeCode::TV50_PACKING)   == 0x7fffffbf);

    assert (t.timeAndFlags (TimeCode::FILM24_PACKING) == 0xffff7f3f);

    t.setTimeAndFlags (0x23595929 | 0x00008000, TimeCode::TV60_PACKING);

    assert (t.timeAndFlags() == (0x23595929 | 0x00008000));

    t.setTimeAndFlags (0x23595929 | 0x80000000, TimeCode::TV50_PACKING);

    assert (t.timeAndFlags() == (0x23595929 | 0x00008000));

    t.setTimeAndFlags (0x23595929 | 0x00008000, TimeCode::FILM24_PACKING);

    assert (t.timeAndFlags() == (0x23595929 | 0x00008000));

    // bgf0

    t.setTimeAndFlags (0x00000000);

    t.setBgf0 (true);

    assert (t.bgf0() == true);

    assert (t.timeAndFlags (TimeCode::TV60_PACKING)   == 0x00800000);

    assert (t.timeAndFlags (TimeCode::TV50_PACKING)   == 0x00008000);

    assert (t.timeAndFlags (TimeCode::FILM24_PACKING) == 0x00800000);

    t.setTimeAndFlags (0xffffffff);

    t.setBgf0 (false);

    assert (t.bgf0() == false);

    assert (t.timeAndFlags (TimeCode::TV60_PACKING)   == 0xff7fffff);

    assert (t.timeAndFlags (TimeCode::TV50_PACKING)   == 0xffff7fbf);

    assert (t.timeAndFlags (TimeCode::FILM24_PACKING) == 0xff7fff3f);

    t.setTimeAndFlags (0x23595929 | 0x00800000, TimeCode::TV60_PACKING);

    assert (t.timeAndFlags() == (0x23595929 | 0x00800000));

    t.setTimeAndFlags (0x23595929 | 0x00008000, TimeCode::TV50_PACKING);

    assert (t.timeAndFlags() == (0x23595929 | 0x00800000));

    t.setTimeAndFlags (0x23595929 | 0x00800000, TimeCode::FILM24_PACKING);

    assert (t.timeAndFlags() == (0x23595929 | 0x00800000));

    // bgf1

    t.setTimeAndFlags (0x00000000);

    t.setBgf1 (true);

    assert (t.bgf1() == true);

    assert (t.timeAndFlags (TimeCode::TV60_PACKING)   == 0x40000000);

    assert (t.timeAndFlags (TimeCode::TV50_PACKING)   == 0x40000000);

    assert (t.timeAndFlags (TimeCode::FILM24_PACKING) == 0x40000000);

    t.setTimeAndFlags (0xffffffff);

    t.setBgf1 (false);

    assert (t.bgf1() == false);

    assert (t.timeAndFlags (TimeCode::TV60_PACKING)   == 0xbfffffff);

    assert (t.timeAndFlags (TimeCode::TV50_PACKING)   == 0xbfffffbf);

    assert (t.timeAndFlags (TimeCode::FILM24_PACKING) == 0xbfffff3f);

    t.setTimeAndFlags (0x23595929 | 0x40000000, TimeCode::TV60_PACKING);

    assert (t.timeAndFlags() == (0x23595929 | 0x40000000));

    t.setTimeAndFlags (0x23595929 | 0x40000000, TimeCode::TV50_PACKING);

    assert (t.timeAndFlags() == (0x23595929 | 0x40000000));

    t.setTimeAndFlags (0x23595929 | 0x40000000, TimeCode::FILM24_PACKING);

    assert (t.timeAndFlags() == (0x23595929 | 0x40000000));

    // bgf2

    t.setTimeAndFlags (0x00000000);

    t.setBgf2 (true);

    assert (t.bgf2() == true);

    assert (t.timeAndFlags (TimeCode::TV60_PACKING)   == 0x80000000);

    assert (t.timeAndFlags (TimeCode::TV50_PACKING)   == 0x00800000);

    assert (t.timeAndFlags (TimeCode::FILM24_PACKING) == 0x80000000);

    t.setTimeAndFlags (0xffffffff);

    t.setBgf2 (false);

    assert (t.bgf2() == false);

    assert (t.timeAndFlags (TimeCode::TV60_PACKING)   == 0x7fffffff);

    assert (t.timeAndFlags (TimeCode::TV50_PACKING)   == 0xff7fffbf);

    assert (t.timeAndFlags (TimeCode::FILM24_PACKING) == 0x7fffff3f);

    t.setTimeAndFlags (0x23595929 | 0x80000000, TimeCode::TV60_PACKING);

    assert (t.timeAndFlags() == (0x23595929 | 0x80000000));

    t.setTimeAndFlags (0x23595929 | 0x00800000, TimeCode::TV50_PACKING);

    assert (t.timeAndFlags() == (0x23595929 | 0x80000000));

    t.setTimeAndFlags (0x23595929 | 0x80000000, TimeCode::FILM24_PACKING);

    assert (t.timeAndFlags() == (0x23595929 | 0x80000000));

    // User-defined data

    t.setUserData (0x87654321);

    assert (t.userData() == 0x87654321);

    assert (t.binaryGroup (1) == 1);

    assert (t.binaryGroup (2) == 2);

    assert (t.binaryGroup (3) == 3);

    assert (t.binaryGroup (4) == 4);

    assert (t.binaryGroup (5) == 5);

    assert (t.binaryGroup (6) == 6);

    assert (t.binaryGroup (7) == 7);

    t.setBinaryGroup (1, 2);

    t.setBinaryGroup (2, 3);

    t.setBinaryGroup (3, 4);

    t.setBinaryGroup (4, 5);

    t.setBinaryGroup (5, 6);

    t.setBinaryGroup (6, 7);

    t.setBinaryGroup (7, 8);

    t.setBinaryGroup (8, 9);

    assert (t.userData() == 0x98765432);

    // Assignment

    TimeCode t1 (12, 17, 57, 14,	   // hours, minutes, seconds, frame

		 false, false, false,	   // dropFrame, colorFrame, fieldPhase

		 false, false, false,	   // bgf0, bgf1, bgf2

		 1, 2, 3, 4, 5, 6, 7, 8);  // binary groups 1 to 8

    t = t1;

    assert (t.timeAndFlags() == 0x12175714);

    assert (t.userData() == 0x87654321);

}

void

writeReadTimeCode (const char fileName[])

{

    cout << "time code attribute" << endl;

    cout << "writing, ";

    TimeCode t1 (0x23595829, 0x12345678, TimeCode::FILM24_PACKING);

    assert (t1.timeAndFlags (TimeCode::FILM24_PACKING) == 0x23595829);

    assert (t1.userData() == 0x12345678);

    static const int W = 100;

    static const int H = 100;

    Header header (W, H);

    assert (hasTimeCode (header) == false);

    addTimeCode (header, t1);

    assert (hasTimeCode (header) == true);

    {

	RgbaOutputFile out (fileName, header);

	Rgba pixels[W];

	for (int i = 0; i < W; ++i)

	{

	    pixels[i].r = 1;

	    pixels[i].g = 1;

	    pixels[i].b = 1;

	    pixels[i].a = 1;

	}

	out.setFrameBuffer (pixels, 1, 0);

	out.writePixels (H);

    }

    cout << "reading, comparing" << endl;

    {

	RgbaInputFile in (fileName);

	const TimeCode &t2 = timeCode (in.header());

	assert (hasTimeCode (in.header()) == true);

	assert (t1.timeAndFlags() == t2.timeAndFlags());

	assert (t1.userData() == t2.userData());

    }

    remove (fileName);

}

bool

equal (const Rational &a, const Rational &b)

{

    return a.n == b.n && a.d == b.d;

}

void

rationalMethods ()

{

    cout << "rational methods" << endl;

    Rational r0 (0, 1);

    assert (r0 == 0);

    Rational r1 (1, 1);

    assert (r1 == 1);

    Rational r2 (1, 4);

    assert (r2 == 0.25);

    Rational r3 (-8, 2);

    assert (r3 == -4);

    Rational r4 (0.0);

    assert (r4 == 0);

    Rational r5 (1e-50);

    assert (r5 == 0);

    Rational r6 (1.0);

    assert (r6 == 1.0 && r6.n == 1 && r6.d == 1);

    Rational r7 (-10.0);

    assert (r7 == -10.0 && r7.n == -10 && r7.d == 1);

    Rational r8 (0.03125);

    assert (r8 == 0.03125 && r8.n == 1 && r8.d == 32);

    Rational r9 (0.53125);

    assert (r9 == 0.53125 && r9.n == 17 && r9.d == 32);

    Rational r10 (10.1);

    assert (equalWithAbsError (double (r10), 10.1, 1e-8));

    Rational r11 (double ((1U << 30) - 1));

    assert (r11 == double ((1U << 30) - 1));

    assert (equal (guessExactFps (23.976), fps_23_976()));

    assert (equal (guessExactFps (24.000), fps_24()));

    assert (equal (guessExactFps (25.000), fps_25()));

    assert (equal (guessExactFps (29.970), fps_29_97()));

    assert (equal (guessExactFps (30.000), fps_30()));

    assert (equal (guessExactFps (47.952), fps_47_952()));

    assert (equal (guessExactFps (48.000), fps_48()));

    assert (equal (guessExactFps (50.000), fps_50()));

    assert (equal (guessExactFps (59.940), fps_59_94()));

    assert (equal (guessExactFps (60.000), fps_60()));

    assert (equal (guessExactFps (70.500), Rational (141, 2)));

}

void

writeReadRational (const char fileName[])

{

    cout << "rational attribute" << endl;

    cout << "writing, ";

    Rational r1 (12, 17);

    Rational r2 (-12, 3);

    static const int W = 100;

    static const int H = 100;

    Header header (W, H);

    header.insert ("r1", RationalAttribute (r1));

    header.insert ("r2", RationalAttribute (r2));

    {

	RgbaOutputFile out (fileName, header);

	Rgba pixels[W];

	for (int i = 0; i < W; ++i)

	{

	    pixels[i].r = 1;

	    pixels[i].g = 1;

	    pixels[i].b = 1;

	    pixels[i].a = 1;

	}

	out.setFrameBuffer (pixels, 1, 0);

	out.writePixels (H);

    }

    cout << "reading, comparing" << endl;

    {

	RgbaInputFile in (fileName);

	const Rational &r3 =

	    in.header().typedAttribute<RationalAttribute>("r1").value();

	const Rational &r4 =

	    in.header().typedAttribute<RationalAttribute>("r2").value();

	assert (equal (r1, r3));

	assert (equal (r2, r4));

    }

    remove (fileName);

}

void

generatedFunctions ()

{

    //

    // Most optional standard attributes are of type string, float,

    // etc.  The attribute types are already being tested elsewhere

    // (testAttributes.C), and the convenience functions to access

    // the standard attributes are all generated via macros.  Here

    // we just verify that all the convenience functions exist

    // (that is, ImfStandardAttributes.C and ImfStandardAttributes.h

    // contain the right macro invocations).  If any functions are

    // missing, we should get an error during compiling or linking.

    //

    cout << "automatically generated functions" << endl;

    Header header;