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

//

// Copyright (c) 2011, Industrial Light & Magic, a division of Lucas

// Digital Ltd. LLC

//

// Portions (c) 2012, Weta Digital Ltd

//

// 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 "testCopyDeepScanLine.h"

#include <assert.h>

#include <string.h>

#include <ImfDeepScanLineInputFile.h>

#include <ImfDeepScanLineOutputFile.h>

#include <ImfPartType.h>

#include <ImfChannelList.h>

#include <ImfDeepFrameBuffer.h>

#include <ImfArray.h>

#include <IlmThreadPool.h>

#include <stdlib.h>

#include <stdio.h>

#include <vector>

#include "tmpDir.h"

namespace IMF = OPENEXR_IMF_NAMESPACE;

using namespace IMF;

using namespace std;

using namespace IMATH_NAMESPACE;

using namespace ILMTHREAD_NAMESPACE;

namespace

{

const int width = 538;

const int height = 234;

const int minX = 42;

const int minY = 51;

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));

const char source_filename[] = IMF_TMP_DIR "imf_test_copy_deep_scanline_source.exr";

const char copy_filename[]  = IMF_TMP_DIR "imf_test_copy_deep_scanline_copy.exr";

vector<int> channelTypes;

Array2D<unsigned int> sampleCount;

Header header;

void generateRandomFile (const std::string & source_filename,

                         int channelCount,

                         Compression compression)

{

    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 (source_filename.c_str());

    DeepScanLineOutputFile file(source_filename.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

    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 *);

        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 " << flush;

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

    {

            //

            // Fill in data at the last minute.

            //

            for (int j = 0; j < width; j++)

            {

                sampleCount[i][j] = rand() % 10 + 1;

                for (int k = 0; k < channelCount; k++)

                {

                    if (channelTypes[k] == 0)

                        data[k][i][j] = new unsigned int[sampleCount[i][j]];

                    if (channelTypes[k] == 1)

                        data[k][i][j] = new half[sampleCount[i][j]];

                    if (channelTypes[k] == 2)

                        data[k][i][j] = new float[sampleCount[i][j]];

                    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;

                    }

                }

            }

        }

        file.writePixels(height);

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

        for (int j = 0; j < width; j++)

            for (int k = 0; k < channelCount; k++)

            {

                if (channelTypes[k] == 0)

                    delete[] (unsigned int*) data[k][i][j];

                if (channelTypes[k] == 1)

                    delete[] (half*) data[k][i][j];

                if (channelTypes[k] == 2)

                    delete[] (float*) data[k][i][j];

            }

}

void copyFile (const std::string & source_filename,

               const std::string & copy_filename)

{

    cout << "copying " ;

    cout.flush();

    {

       DeepScanLineInputFile in_file(source_filename.c_str(), 8);

       remove (copy_filename.c_str());

       DeepScanLineOutputFile out_file (copy_filename.c_str(),

                                        in_file.header(),

                                        8);

       out_file.copyPixels(in_file);

    }

    // remove the source file here to prevent accidentally reading it

  //  remove (source_filename);

}

void readFile (const std::string & copy_filename, int channelCount)

{

    cout << "reading " ;

    cout.flush();

    DeepScanLineInputFile file(copy_filename.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

    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 *);

            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);

        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]);

            for (int j = 0; j < width; j++)

            {

                for (int k = 0; k < channelCount; k++)

                {

                         if (channelTypes[k] == 0)

                              data[k][i][j] = new unsigned int[localSampleCount[i][j]];

                         if (channelTypes[k] == 1)

                              data[k][i][j] = new half[localSampleCount[i][j]];

                         if (channelTypes[k] == 2)

                             data[k][i][j] = new float[localSampleCount[i][j]];

                }

            }

        }

        file.readPixels(dataWindow.min.y, dataWindow.max.y);

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

        for (int j = 0; j < width; j++)

            for (int k = 0; k < channelCount; k++)

            {

                    for (int l = 0; l < sampleCount[i][j]; l++)

                    {

                        if (channelTypes[k] == 0)

                        {

                            unsigned int* value = (unsigned int*)(data[k][i][j]);

                            if (value[l] != (i * width + j) % 2049)

                                cout << j << ", " << i << " error, should be "

                                     << (i * width + j) % 2049 << ", is " << value[l]

                                     << endl << flush;

                            assert (value[l] == (i * width + j) % 2049);

                        }

                        if (channelTypes[k] == 1)

                        {

                            half* value = (half*)(data[k][i][j]);

                            if (value[l] != (i * width + j) % 2049)

                                cout << j << ", " << i << " error, should be "

                                     << (i * width + j) % 2049 << ", is " << value[l]

                                     << endl << flush;

                            assert (((half*)(data[k][i][j]))[l] == (i * width + j) % 2049);

                        }

                        if (channelTypes[k] == 2)

                        {

                            float* value = (float*)(data[k][i][j]);

                            if (value[l] != (i * width + j) % 2049)

                                cout << j << ", " << i << " error, should be "

                                     << (i * width + j) % 2049 << ", is " << value[l]

                                     << endl << flush;

                            assert (((float*)(data[k][i][j]))[l] == (i * width + j) % 2049);

                        }

                    }

            }

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

        for (int j = 0; j < width; j++)

            for (int k = 0; k < channelCount; k++)

            {

                    if (channelTypes[k] == 0)

                        delete[] (unsigned int*) data[k][i][j];

                    if (channelTypes[k] == 1)

                        delete[] (half*) data[k][i][j];

                    if (channelTypes[k] == 2)

                        delete[] (float*) data[k][i][j];

            }

}

void readCopyWriteTest (const std::string & tempDir,

                        int channelCount,

                        int testTimes)

{

    cout << "Testing files with " << channelCount << " channels " << testTimes << " times."

         << endl << flush;

    std::string source_filename = tempDir + "imf_test_copy_deep_scanline_source.exr";

    std::string copy_filename   = tempDir + "imf_test_copy_deep_scanline_copy.exr";

    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 (source_filename, channelCount, compression);

        copyFile (source_filename, copy_filename);

        readFile (copy_filename, channelCount );

        remove (source_filename.c_str());

        remove (copy_filename.c_str());

        cout << endl << flush;

    }

}

}; // namespace

void testCopyDeepScanLine (const std::string &tempDir)

{

    try

    {

        cout << "\n\nTesting raw data copy in DeepScanLineInput/OutputFile:\n" << endl;

        srand(1);

        int numThreads = ThreadPool::globalThreadPool().numThreads();

        ThreadPool::globalThreadPool().setNumThreads(4);

        readCopyWriteTest (tempDir,  1, 100);

        readCopyWriteTest (tempDir,  3,  50);

        readCopyWriteTest (tempDir, 10,  10);

        ThreadPool::globalThreadPool().setNumThreads(numThreads);

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

    }

    catch (const std::exception &e)

    {

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

        assert (false);

    }

}