//
// SPDX-License-Identifier: BSD-3-Clause
// Copyright (c) Contributors to the OpenEXR Project.
//

#ifdef NDEBUG
#    undef NDEBUG
#endif

#include "ImfNamespace.h"
#include "OpenEXRConfigInternal.h"

#include "testAttributes.h"
#include "testB44ExpLogTable.h"
#include "testBackwardCompatibility.h"
#include "testBadTypeAttributes.h"
#include "testChannels.h"
#include "testCompositeDeepScanLine.h"
#include "testCompression.h"
#include "testConversion.h"
#include "testCopyDeepScanLine.h"
#include "testCopyDeepTiled.h"
#include "testCopyMultiPartFile.h"
#include "testCopyPixels.h"
#include "testCpuId.h"
#include "testCustomAttributes.h"
#include "testDeepScanLineBasic.h"
#include "testDeepScanLineHuge.h"
#include "testDeepScanLineMultipleRead.h"
#include "testDeepTiledBasic.h"
#include "testDwaCompressorSimd.h"
#include "testDwaLookups.h"
#include "testExistingStreams.h"
#include "testFutureProofing.h"
#include "testHeader.h"
#include "testHuf.h"
#include "testIDManifest.h"
#include "testInputPart.h"
#include "testIsComplete.h"
#include "testLargeDataWindowOffsets.h"
#include "testLineOrder.h"
#include "testLut.h"
#include "testMagic.h"
#include "testMultiPartApi.h"
#include "testMultiPartFileMixingBasic.h"
#include "testMultiPartSharedAttributes.h"
#include "testMultiPartThreading.h"
#include "testMultiScanlinePartThreading.h"
#include "testMultiTiledPartThreading.h"
#include "testMultiView.h"
#include "testNativeFormat.h"
#include "testOptimized.h"
#include "testOptimizedInterleavePatterns.h"
#include "testPartHelper.h"
#include "testPreviewImage.h"
#include "testRgba.h"
#include "testRgbaThreading.h"
#include "testRle.h"
#include "testSampleImages.h"
#include "testScanLineApi.h"
#include "testSharedFrameBuffer.h"
#include "testStandardAttributes.h"
#include "testTiledCompression.h"
#include "testTiledCopyPixels.h"
#include "testTiledLineOrder.h"
#include "testTiledRgba.h"
#include "testTiledYa.h"
#include "testWav.h"
#include "testXdr.h"
#include "testYca.h"

#include "ImathRandom.h"
#include "tmpDir.h"

// system includes

#include <errno.h>
#include <iostream>
#include <stdlib.h>
#include <string.h>
#include <time.h>

#include <set>

#ifdef _WIN32
#    include <windows.h>
#else
#    include <unistd.h>
#endif
#include <sstream>

using namespace std;

#define TEST_STRING(x) #x
#define TEST(x, y)                                                             \
    if (helpMode)                                                              \
    {                                                                          \
        tests.insert (string (TEST_STRING (x)));                               \
        suites.insert (string (y));                                            \
    }                                                                          \
    else if (                                                                  \
        argc < 2 ||                                                            \
        (!strcmp (argv[1], TEST_STRING (x)) || !strcmp (argv[1], y)))          \
    {                                                                          \
        cout << "\n=======\nRunning " << TEST_STRING (x) << endl;              \
        x (tempDir);                                                           \
    }

string
makeTempDir ()
{
    string tempDir;

    IMATH_NAMESPACE::Rand48 rand48 (time ((time_t*) 0));
    while (true)
    {
#ifdef _WIN32
        char  tmpbuf[4096];
        DWORD len = GetTempPathA (4096, tmpbuf);
        if (len == 0 || len > 4095)
        {
            cerr << "Cannot retrieve temporary directory" << endl;
            exit (1);
        }
        tempDir = tmpbuf;
        // windows does this automatically
        // tempDir += IMF_PATH_SEPARATOR;
        tempDir += "OpenEXRTest_";
#else
        tempDir = IMF_TMP_DIR "OpenEXRTest_";
#endif
        for (int i = 0; i < 8; ++i)
            tempDir += ('A' + rand48.nexti () % 26);

        std::cout << "tempDir = " << tempDir << std::endl;

        int status = mkdir (tempDir.c_str (), 0777);

        if (status == 0)
        {
            tempDir += IMF_PATH_SEPARATOR;
            break; // success
        }

        if (errno != EEXIST)
        {
            std::cerr << "ERROR -- mkdir(" << tempDir
                      << ") failed: "
                         "errno = "
                      << errno << std::endl;
            exit (1);
        }
    }

    return tempDir;
}

int
main (int argc, char* argv[])
{
    // Create temporary files in a uniquely named private temporary
    // subdirectory of IMF_TMP_DIR to avoid colliding with other
    // running instances of this program.

    std::string tempDir;

    bool helpMode = false;
    if (argc == 2 &&
        (strcmp (argv[1], "--help") == 0 || strcmp (argv[1], "-h") == 0))
    {
        helpMode = true;
    }
    set<string> tests;
    set<string> suites;

    if (!helpMode) { tempDir = makeTempDir (); }

    // NB: If you add a test here, make sure to enumerate it in the
    // CMakeLists.txt so it runs as part of the test suite
    TEST (testMagic, "core");
    TEST (testXdr, "core");
    TEST (testHuf, "core");
    TEST (testWav, "core");
    TEST (testRgba, "basic");
    TEST (testLargeDataWindowOffsets, "basic");
    TEST (testSharedFrameBuffer, "basic");
    TEST (testRgbaThreading, "basic");
    TEST (testChannels, "basic");
    TEST (testAttributes, "core");
    TEST (testCustomAttributes, "core");
    TEST (testLineOrder, "basic");
    TEST (testCompression, "basic");
    TEST (testCopyPixels, "basic");
    TEST (testLut, "basic");
    TEST (testSampleImages, "basic");
    TEST (testPreviewImage, "basic");
    TEST (testConversion, "basic");
    TEST (testTiledRgba, "basic");
    TEST (testTiledCopyPixels, "basic");
    TEST (testTiledCompression, "basic");
    TEST (testTiledLineOrder, "basic");
    TEST (testScanLineApi, "basic");
    TEST (testExistingStreams, "core");
    TEST (testStandardAttributes, "core");
    TEST (testOptimized, "basic");
    TEST (testOptimizedInterleavePatterns, "basic");
    TEST (testYca, "basic");
    TEST (testTiledYa, "basic");
    TEST (testNativeFormat, "basic");
    TEST (testMultiView, "basic");
    TEST (testIsComplete, "basic");
    TEST (testDeepScanLineBasic, "deep");
    TEST (testCopyDeepScanLine, "deep");
    TEST (testDeepScanLineMultipleRead, "deep");
    TEST (testDeepTiledBasic, "deep");
    TEST (testCopyDeepTiled, "deep");
    TEST (testCompositeDeepScanLine, "deep");
    TEST (testMultiPartFileMixingBasic, "multi");
    TEST (testInputPart, "multi");
    TEST (testPartHelper, "multi");
    TEST (testBadTypeAttributes, "multi");
    TEST (testMultiScanlinePartThreading, "multi");
    TEST (testMultiTiledPartThreading, "multi");
    TEST (testMultiPartThreading, "multi");
    TEST (testMultiPartApi, "multi");
    TEST (testMultiPartSharedAttributes, "multi");
    TEST (testCopyMultiPartFile, "multi");
    TEST (testBackwardCompatibility, "core");
    TEST (testFutureProofing, "core");
    TEST (testDwaCompressorSimd, "basic");
    TEST (testRle, "core");
    TEST (testB44ExpLogTable, "core");
    TEST (testDwaLookups, "core");
    TEST (testIDManifest, "core");
    TEST (testCpuId, "core");
    TEST (testHeader, "basic");

    // NB: If you add a test here, make sure to enumerate it in the
    // CMakeLists.txt so it runs as part of the test suite

    //#ifdef ENABLE_IMFHUGETEST
    // defined via configure with --enable-imfhugetest=yes/no
#if 0
        TEST (testDeepScanLineHuge, "deep");
#endif

    if (helpMode)
    {
        cout << "OpenEXRTest runs a series of tests to confirm\n"
                "correct behavior of the OpenEXR library.\n"
                "If all is correct, OpenEXRTest will complete without\n"
                "crashing or leaking memory.\n";
        cout << "\n";
        cout
            << "If a test fails, an individual test can be re-run, avoiding\n"
               "the wait for previous tests to complete. This allows easier debugging\n"
               "of the failure.\n";
        cout << "\n";
        cout
            << "A 'suite' of tests can also be run, to allow a subset of\n"
            << "tests to run. This is useful as an initial confirmation\n"
            << "that a modification to the library has not introduced an error.\n"
            << "Suites can be run in parallel for speed. Every test is in one suite.\n";
        cout << "\n";
        cout
            << "usage:\n"
            << "OpenEXRTest           : with no arguments, run all tests\n"
            << "OpenEXRTest TEST      : run only specific test, then quit\n"
            << "OpenEXRTest SUITE     : run all the tests in the given SUITE\n";
        cout << "\n";
        cout << "available TESTs:\n";
        for (auto i = tests.begin (); i != tests.end (); ++i)
        {
            cout << ' ' << *i << endl;
        }
        cout << "\n";
        cout << "available SUITEs:\n";
        for (auto i = suites.begin (); i != suites.end (); ++i)
        {
            cout << ' ' << *i << endl;
        }
    }
    else
    {
        cout << "removing temp dir " << tempDir << endl;
        rmdir (tempDir.c_str ());

#ifdef OPENEXR_IMF_HAVE_LINUX_PROCFS

        //
        // Allow the user to check for file descriptor leaks
        //

        cout << "open file descriptors:" << endl;

        stringstream ss;
        ss << "ls -lG /proc/" << getpid () << "/fd";

        if (system (ss.str ().c_str ()) == -1) { cout << "failed to run ls\n"; }

        cout << endl;

#endif
    }
    return 0;
}
