// Copyright (C) 2006  Davis E. King (davis@dlib.net)
// License: Boost Software License   See LICENSE.txt for the full license.


#include <sstream>
#include <string>
#include <cstdlib>
#include <ctime>
#include <dlib/misc_api.h>
#include <dlib/threads.h>

#include "tester.h"

namespace  
{
    using namespace test;
    using namespace dlib;
    using namespace std;

    logger dlog("test.threads");

    void test_async()
    {
#if __cplusplus >= 201103
        print_spinner();
        auto v1 = dlib::async([]() { dlib::sleep(500); return 1; }).share();
        auto v2 = dlib::async([v1]() { dlib::sleep(400); return v1.get()+1; }).share();
        auto v3 = dlib::async([v2](int a) { dlib::sleep(300); return v2.get()+a; },2).share();
        auto v4 = dlib::async([v3]() { dlib::sleep(200); return v3.get()+1; });

        DLIB_TEST(v4.get() == 5);

        print_spinner();
        auto except = dlib::async([](){ dlib::sleep(300); throw error("oops"); });
        bool got_exception = false;
        try
        {
            except.get();
        }
        catch (error&e)
        {
            got_exception = true;
            DLIB_TEST(e.what() == string("oops"));
        }
        DLIB_TEST(got_exception);
#endif
    }

    class threads_tester : public tester
    {
    public:
        threads_tester (
        ) :
            tester ("test_threads",
                    "Runs tests on the threads component."),
            sm(cm)
        {}

        thread_specific_data<int> tsd;
        rmutex cm;
        rsignaler sm;
        int count;
        bool failure;

        void perform_test (
        )
        {
            failure = false;
            print_spinner();


            count = 10;
            if (!create_new_thread<threads_tester,&threads_tester::thread1>(*this)) failure = true;
            if (!create_new_thread<threads_tester,&threads_tester::thread2>(*this)) failure = true;
            if (!create_new_thread<threads_tester,&threads_tester::thread3>(*this)) failure = true;
            if (!create_new_thread<threads_tester,&threads_tester::thread4>(*this)) failure = true;
            if (!create_new_thread<threads_tester,&threads_tester::thread5>(*this)) failure = true;
            if (!create_new_thread<threads_tester,&threads_tester::thread6>(*this)) failure = true;
            if (!create_new_thread<threads_tester,&threads_tester::thread7>(*this)) failure = true;
            if (!create_new_thread<threads_tester,&threads_tester::thread8>(*this)) failure = true;
            if (!create_new_thread<threads_tester,&threads_tester::thread9>(*this)) failure = true;
            if (!create_new_thread<threads_tester,&threads_tester::thread10>(*this)) failure = true;

            thread(66);

            // this should happen in the main program thread
            if (is_dlib_thread())
                failure = true;

            auto_mutex M(cm);
            while (count > 0 && !failure)
                sm.wait();


            DLIB_TEST(!failure);

            test_async();
        }

        void thread_end_handler (
        )
        {
            auto_mutex M(cm);
            --count;
            if (count == 0)
                sm.signal();
        }

        void thread1() { thread(1); }
        void thread2() 
        { 
            thread(2); 
            if (is_dlib_thread() == false)
                failure = true;
        }
        void thread3() { thread(3); }
        void thread4() { thread(4); }
        void thread5() { thread(5); }
        void thread6() { thread(6); }
        void thread7() { thread(7); }
        void thread8() { thread(8); }
        void thread9() { thread(9); }
        void thread10() { thread(10); }

        void thread (
            int num
        )
        {
            dlog << LTRACE << "starting thread num " << num;
            if (is_dlib_thread())
                register_thread_end_handler(*this,&threads_tester::thread_end_handler);
            tsd.data() = num;
            for (int i = 0; i < 0x3FFFF; ++i)
            {
                if ((i&0xFFF) == 0)
                {
                    print_spinner();
                    dlib::sleep(10);
                }
                // if this isn't equal to num then there is a problem with the thread specific data stuff
                if (tsd.data() != num)
                {
                    auto_mutex M(cm);
                    failure = true;
                    sm.signal();
                }
            }
            dlog << LTRACE << "ending of thread num " << num;


        }
    } a;


}