///////////////////////////////////////////////////////////////////////////
//
// Copyright (c) 2007, 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.
//
///////////////////////////////////////////////////////////////////////////


//-----------------------------------------------------------------------------
//
//	exr2aces -- a program that converts an
//	OpenEXR file to an ACES image file.
//
//-----------------------------------------------------------------------------


#include <ImfAcesFile.h>
#include <ImfArray.h>
#include <iostream>
#include <exception>
#include <string>
#include <string.h>
#include <stdlib.h>

#include <OpenEXRConfig.h>
using namespace OPENEXR_IMF_NAMESPACE;
using namespace IMATH_NAMESPACE;
using namespace std;

namespace {

void
usageMessage (const char argv0[], bool verbose = false)
{
    cerr << "usage: " << argv0 << " [options] infile outfile" << endl;

    if (verbose)
    {
	cerr << "\n"
		"Reads an OpenEXR file from infile and saves the contents\n"
		"in ACES image file outfile.\n"
		"\n"
		"The ACES image file format is a subset of the OpenEXR file\n"
		"format.  ACES image files are restricted as follows:\n"
		"\n"
		"* Images are stored as scanlines; tiles are not allowed.\n"
		"\n"
		"* Images contain three color channels, either\n"
		"      R, G, B (red, green, blue) or\n"
		"      Y, RY, BY (luminance, sub-sampled chroma)\n"
		"\n"
		"* Images may optionally contain an alpha channel.\n"
		"\n"
		"* Only three compression types are allowed:\n"
		"      NO_COMPRESSION (file is not compressed)\n"
		"      PIZ_COMPRESSION (lossless)\n"
		"      B44A_COMPRESSION (lossy)\n"
		"* The \"chromaticities\" header attribute must specify\n"
		"  the ACES RGB primaries and white point.\n"
		"\n"
		"Options:\n"
		"\n"
		"-v        verbose mode\n"
		"\n"
		"-h        prints this message\n";

	 cerr << endl;
    }

    exit (1);
}


void
exr2aces (const char inFileName[],
	  const char outFileName[],
	  bool verbose)
{
    Array2D<Rgba> p;
    Header h;
    RgbaChannels ch;
    Box2i dw;
    int width;
    int height;

    {
	if (verbose)
	    cout << "Reading file " << inFileName << endl;

	AcesInputFile in (inFileName);

	h = in.header();
	ch = in.channels();
	dw = h.dataWindow();

	width  = dw.max.x - dw.min.x + 1;
	height = dw.max.y - dw.min.y + 1;
	p.resizeErase (height, width);

	in.setFrameBuffer (&p[0][0] - dw.min.x - dw.min.y * width, 1, width);
	in.readPixels (dw.min.y, dw.max.y);
    }

    switch (h.compression())
    {
      case NO_COMPRESSION:
	break;

      case B44_COMPRESSION:
      case B44A_COMPRESSION:
	h.compression() = B44A_COMPRESSION;

      default:
	h.compression() = PIZ_COMPRESSION;
    }

    {
	if (verbose)
	    cout << "Writing file " << outFileName << endl;

	AcesOutputFile out (outFileName, h, ch);

	out.setFrameBuffer (&p[0][0] - dw.min.x - dw.min.y * width, 1, width);
	out.writePixels (height);
    }
}


} // namespace


int
main(int argc, char **argv)
{
    const char *inFile = 0;
    const char *outFile = 0;
    bool verbose = false;

    //
    // Parse the command line.
    //

    if (argc < 2)
	usageMessage (argv[0], true);

    int i = 1;

    while (i < argc)
    {
	if (!strcmp (argv[i], "-v"))
	{
	    //
	    // Verbose mode
	    //

	    verbose = true;
	    i += 1;
	}
	else if (!strcmp (argv[i], "-h"))
	{
	    //
	    // Print help message
	    //

	    usageMessage (argv[0], true);
	}
	else
	{
	    //
	    // Image file name
	    //

	    if (inFile == 0)
		inFile = argv[i];
	    else
		outFile = argv[i];

	    i += 1;
	}
    }

    if (inFile == 0 || outFile == 0)
	usageMessage (argv[0]);

    //
    // Load inFile, and save a tiled version in outFile.
    //

    int exitStatus = 0;

    try
    {
	exr2aces (inFile, outFile, verbose);
    }
    catch (const exception &e)
    {
	cerr << e.what() << endl;
	exitStatus = 1;
    }

    return exitStatus;
}
