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

//

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

// Digital Ltd. LLC and 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 "fuzzFile.h"

#include <ImfDeepScanLineOutputFile.h>

#include <ImfDeepScanLineInputFile.h>

#include <ImfDeepFrameBuffer.h>

#include <ImfPartType.h>

#include <ImfArray.h>

#include <ImfThreading.h>

#include <IlmThread.h>

#include <Iex.h>

#include <iostream>

#include <cassert>

#include <stdio.h>

#include <vector>

#include "tmpDir.h"

// Handle the case when the custom namespace is not exposed

#include <ImfNamespace.h>

#include <ImfChannelList.h>

#include <ImfMultiPartOutputFile.h>

#include <ImfDeepScanLineOutputPart.h>

#include <ImfMultiPartInputFile.h>

#include <ImfDeepScanLineInputPart.h>

namespace IMF = OPENEXR_IMF_NAMESPACE;

using namespace IMF;

using namespace std;

using namespace IMATH_NAMESPACE;

namespace

{

const int width = 90;

const int height = 80;

const int minX = 10;

const int minY = 11;

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

Array2D<unsigned int> sampleCount;

void generateRandomFile(const char filename[], int channelCount,int parts , Compression compression)

{

    cout << "generating file with " << parts << " parts and compression " << compression << flush;

    vector<Header> headers(parts);

    headers[0] = Header(displayWindow, dataWindow,

                    1,

                    IMATH_NAMESPACE::V2f (0, 0),

                    1,

                    INCREASING_Y,

                    compression);

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

    {

        stringstream ss;

        ss << i;

        string str = ss.str();

        headers[0].channels().insert(str, Channel(IMF::FLOAT));

    }

     headers[0].setType(DEEPSCANLINE);

     headers[0].setName("bob");

     for(int p=1;p

     {

         headers[p]=headers[0];

         ostringstream s;

         s << p;

         headers[p].setName(s.str());

     }

     Array<Array2D< void* > > data(channelCount);

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

         data[i].resizeErase(height, width);

     sampleCount.resizeErase(height, width);

     remove (filename);

     MultiPartOutputFile file(filename,&headers[0],parts);

     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 = IMF::FLOAT;

         stringstream ss;

         ss << i;

         string str = ss.str();

         int 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

     }

    for(int p=0;p

    {

        DeepScanLineOutputPart pt(file,p);

        pt.setFrameBuffer(frameBuffer);

        cout << "writing " << p << 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() % 4 + 1;

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

                {

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

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

                    {

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

                    }

                }

            }

        }

        pt.writePixels(height);

    }

}

void readFile(const char filename[])

{

    //single part interface to read file

    try{

        DeepScanLineInputFile file(filename, 8);

        const Header& fileHeader = file.header();

        int channelCount=0;

        for(ChannelList::ConstIterator i=fileHeader.channels().begin();i!=fileHeader.channels().end();++i,++channelCount);

        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

        vector<int> read_channel(channelCount);

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

        {

            PixelType type = IMF::FLOAT;

            stringstream ss;

            ss << i;

            string str = ss.str();

            int sampleSize = sizeof (float);

            int pointerSize = sizeof (char *);

            frameBuffer.insert (str,                    

                                DeepSlice (type,        

                                           (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++)

            {

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

                {

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

                }

            }

        }

        try{

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

        }catch(...)

        {

            // if readPixels excepts we must clean up

        }

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

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

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

                {

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

                }

    }catch(std::exception & e)

    {

        /* ... yeah, that's likely to happen a lot ... */

    }

    try{

        MultiPartInputFile file(filename, 8);

        for(int p=0;p

        {

            DeepScanLineInputPart inpart(file,p);

            const Header& fileHeader = inpart.header();

            int channelCount=0;

            for(ChannelList::ConstIterator i=fileHeader.channels().begin();i!=fileHeader.channels().end();++i,++channelCount);

            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

            vector<int> read_channel(channelCount);

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

            {

                PixelType type = IMF::FLOAT;

                stringstream ss;

                ss << i;

                string str = ss.str();

                int sampleSize = sizeof (float);

                int pointerSize = sizeof (char *);

                frameBuffer.insert (str,                    

                                    DeepSlice (type,        

                                               (char *) (&data[i][0][0]

                                               - dataWindow.min.x

                                               - dataWindow.min.y * width), // base // 8)

                                               pointerSize * 1,             // xStride// 9

                                               pointerSize * width,         // yStride// 10

                                               sampleSize));                // sampleStride

            }

            inpart.setFrameBuffer(frameBuffer);

            inpart.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++)

                {

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

                    {

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

                    }

                }

            }

            try{

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

            }catch(...)

            {

            }

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

            {

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

                {

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

                    {

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

                    }

                }

            }

        }

    }catch(...)

    {

        // nothing

    }

}

void

fuzzDeepScanLines (int numThreads, Rand48 &random)

{

    if (ILMTHREAD_NAMESPACE::supportsThreads())

    {

	setGlobalThreadCount (numThreads);

	cout << "\nnumber of threads: " << globalThreadCount() << endl;

    }

    Header::setMaxImageSize (10000, 10000);

    const char *goodFile = IMF_TMP_DIR "imf_test_deep_scanline_file_fuzz_good.exr";

    const char *brokenFile = IMF_TMP_DIR "imf_test_deep_scanline_file_fuzz_broken.exr";

    // read file if it already exists: allows re-testing reading of broken file

    readFile(brokenFile);

    for(int parts=1 ; parts < 3 ; parts++)

    {

        for(int comp_method=0;comp_method<2;comp_method++)

        {

            generateRandomFile(goodFile,8,parts,comp_method==0 ? NO_COMPRESSION : ZIPS_COMPRESSION);

            fuzzFile (goodFile, brokenFile, readFile, 5000, 3000, random);

        }

    }

    remove (goodFile);

    remove (brokenFile);

}

} // namespace

void

testFuzzDeepScanLines ()

{

    try

    {

	cout << "Testing deep scanline-based files "

		"with randomly inserted errors" << endl;

	Rand48 random (1);

	fuzzDeepScanLines (0, random);

	if (ILMTHREAD_NAMESPACE::supportsThreads())

	    fuzzDeepScanLines (2, random);

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

    }

    catch (const std::exception &e)

    {

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

	assert (false);

    }

}