// Copyright (C) 2004  Davis E. King (davisking@users.sourceforge.net)
// License: Boost Software License   See LICENSE.txt for the full license.


#include <sstream>
#include <string>
#include <ctime>
#include <cstdlib>

#include <dlib/sliding_buffer.h>
#include "tester.h"

namespace  
{

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

    logger dlog("test.sliding_buffer");

    template <
        typename buf
        >
    void sliding_buffer_kernel_test (
    )
    /*!
        requires
            - buf is an implementation of sliding_buffer/sliding_buffer_kernel_abstract.h 
            - buf is instantiated with T=unsigned char
        ensures
            - runs tests on buf for compliance with the specs 
    !*/
    {        

        print_spinner();

        buf test;

        DLIB_CASSERT(test.size() == 0,"");

        test.set_size(3);
        buf test2;

        DLIB_CASSERT(test.size() == 8,"");

        for (int g = 0; g < 2; ++g)
        {

            test.clear();

            DLIB_CASSERT(test.size() == 0,"");
            test.set_size(2);

            DLIB_CASSERT(test.size() == 4,"");



            test[0] = 'a';
            test[1] = 's';
            test[2] = 'd';
            test[3] = 'f';

            unsigned long id = test.get_element_id(2);
            DLIB_CASSERT(test[test.get_element_index(id)] == 'd',"");


            DLIB_CASSERT(test[0] == 'a',"");
            DLIB_CASSERT(test[1] == 's',"");
            DLIB_CASSERT(test[2] == 'd',"");
            DLIB_CASSERT(test[3] == 'f',"");

            DLIB_CASSERT(test2.size() == 0,"");
            swap(test,test2);
            DLIB_CASSERT(test2.size() == 4,"");

            DLIB_CASSERT(test2[0] == 'a',"");
            DLIB_CASSERT(test2[1] == 's',"");
            DLIB_CASSERT(test2[2] == 'd',"");
            DLIB_CASSERT(test2[3] == 'f',"");

            swap(test,test2);

            test.rotate_left(4);


            DLIB_CASSERT(test[test.get_element_index(id)] == 'd',"");

            DLIB_CASSERT(test[0] == 'a',"");
            DLIB_CASSERT(test[1] == 's',"");
            DLIB_CASSERT(test[2] == 'd',"");
            DLIB_CASSERT(test[3] == 'f',"");

            test.rotate_right(1);

            DLIB_CASSERT(test[test.get_element_index(id)] == 'd',"");

            DLIB_CASSERT(test[0] == 's',"");
            DLIB_CASSERT(test[1] == 'd',"");
            DLIB_CASSERT(test[2] == 'f',"");
            DLIB_CASSERT(test[3] == 'a',"");   


            test.rotate_left(1);

            DLIB_CASSERT(test[test.get_element_index(id)] == 'd',"");
            DLIB_CASSERT(test[0] == 'a',"");
            DLIB_CASSERT(test[1] == 's',"");
            DLIB_CASSERT(test[2] == 'd',"");
            DLIB_CASSERT(test[3] == 'f',"");


            test.rotate_left(16);

            DLIB_CASSERT(test[test.get_element_index(id)] == 'd',"");
            DLIB_CASSERT(test[0] == 'a',"");
            DLIB_CASSERT(test[1] == 's',"");
            DLIB_CASSERT(test[2] == 'd',"");
            DLIB_CASSERT(test[3] == 'f',"");


            test.rotate_left(2);

            DLIB_CASSERT(test[test.get_element_index(id)] == 'd',"");

            DLIB_CASSERT(test[0] == 'd',"");
            DLIB_CASSERT(test[1] == 'f',"");
            DLIB_CASSERT(test[2] == 'a',"");
            DLIB_CASSERT(test[3] == 's',"");

            test.rotate_left(1);

            DLIB_CASSERT(test[test.get_element_index(id)] == 'd',"");
            DLIB_CASSERT(test[0] == 's',"");
            DLIB_CASSERT(test[1] == 'd',"");
            DLIB_CASSERT(test[2] == 'f',"");
            DLIB_CASSERT(test[3] == 'a',"");

            test.rotate_left(1);

            DLIB_CASSERT(test[test.get_element_index(id)] == 'd',"");
            DLIB_CASSERT(test[0] == 'a',"");
            DLIB_CASSERT(test[1] == 's',"");
            DLIB_CASSERT(test[2] == 'd',"");
            DLIB_CASSERT(test[3] == 'f',"");

            DLIB_CASSERT(test.size() == 4,"");

            test[0] = 'x';

            DLIB_CASSERT(test[0] == 'x',"");
            DLIB_CASSERT(test[1] == 's',"");
            DLIB_CASSERT(test[2] == 'd',"");
            DLIB_CASSERT(test[3] == 'f',"");

            test.rotate_left(1);

            DLIB_CASSERT(test[0] == 'f',test[0]);
            DLIB_CASSERT(test[1] == 'x',"");
            DLIB_CASSERT(test[2] == 's',"");
            DLIB_CASSERT(test[3] == 'd',"");


            test[0] = 'x';

            DLIB_CASSERT(test[0] == 'x',"");
            DLIB_CASSERT(test[1] == 'x',"");
            DLIB_CASSERT(test[2] == 's',"");
            DLIB_CASSERT(test[3] == 'd',"");


            test.rotate_left(1);


            DLIB_CASSERT(test[0] == 'd',"");
            DLIB_CASSERT(test[1] == 'x',"");
            DLIB_CASSERT(test[2] == 'x',"");
            DLIB_CASSERT(test[3] == 's',"");



            DLIB_CASSERT(test.at_start() == true,"");
            DLIB_CASSERT(test.current_element_valid() == false,"");
            DLIB_CASSERT(test.move_next() == true,"");
            DLIB_CASSERT(test.at_start() == false,"");
            DLIB_CASSERT(test.current_element_valid() == true,"");

            test.clear();
            test2.clear();


            DLIB_CASSERT(test.size() == 0,"");
            DLIB_CASSERT(test.at_start() == true,"");
            DLIB_CASSERT(test.current_element_valid() == false,"");
            DLIB_CASSERT(test.move_next() == false,"");
            DLIB_CASSERT(test.at_start() == false,"");
            DLIB_CASSERT(test.current_element_valid() == false,"");

            swap(test,test2);

            DLIB_CASSERT(test2.at_start() == false,"");
            DLIB_CASSERT(test2.current_element_valid() == false,"");
            DLIB_CASSERT(test2.move_next() == false,"");
            DLIB_CASSERT(test2.at_start() == false,"");
            DLIB_CASSERT(test2.current_element_valid() == false,"");


            DLIB_CASSERT(test.at_start() == true,"");
            DLIB_CASSERT(test.current_element_valid() == false,"");
            DLIB_CASSERT(test.move_next() == false,"");
            DLIB_CASSERT(test.at_start() == false,"");

            test.set_size(3);
            DLIB_CASSERT(test.size() == 8,"");

            DLIB_CASSERT(test.at_start() == true,"");
            DLIB_CASSERT(test.current_element_valid() == false,"");
            DLIB_CASSERT(test.move_next() == true,"");
            DLIB_CASSERT(test.at_start() == false,"");
            DLIB_CASSERT(test.current_element_valid() == true,"");
            test.reset();
            DLIB_CASSERT(test.size() == 8,"");
            DLIB_CASSERT(test.at_start() == true,"");
            DLIB_CASSERT(test.current_element_valid() == false,"");
            DLIB_CASSERT(test.move_next() == true,"");
            DLIB_CASSERT(test.at_start() == false,"");
            DLIB_CASSERT(test.current_element_valid() == true,"");


            test.rotate_right(1);
            DLIB_CASSERT(test.size() == 8,"");
            DLIB_CASSERT(test.at_start() == true,"");
            DLIB_CASSERT(test.current_element_valid() == false,"");
            DLIB_CASSERT(test.move_next() == true,"");
            DLIB_CASSERT(test.at_start() == false,"");
            DLIB_CASSERT(test.current_element_valid() == true,"");

            test.rotate_left(1);
            DLIB_CASSERT(test.size() == 8,"");
            DLIB_CASSERT(test.at_start() == true,"");
            DLIB_CASSERT(test.current_element_valid() == false,"");
            DLIB_CASSERT(test.move_next() == true,"");
            DLIB_CASSERT(test.at_start() == false,"");
            DLIB_CASSERT(test.current_element_valid() == true,"");
            test.reset();


            for (unsigned long i = 0; i < test.size(); ++i)
            {
                test[i] = static_cast<unsigned char>(i);
            }

            unsigned long count = 0;
            while (test.move_next())
            {
                DLIB_CASSERT(test.element() == count,"");
                ++count;
            }

            DLIB_CASSERT(count == test.size(),"");


            test2.clear();
            ostringstream sout;
            istringstream sin;

            serialize(test,sout);
            sin.str(sout.str());
            deserialize(test2,sin);

            char ch;
            sin >> ch;
            DLIB_CASSERT( !sin, "");

            DLIB_CASSERT(test2.size() == test.size(),"");


            for (unsigned long i = 0; i < test.size(); ++i)
            {
                DLIB_CASSERT(test[i] == test2[i],
                             "\ni:        " << i <<
                             "\ntest[i]:  " << test[i] <<
                             "\ntest2[i]: " << test2[i]);
            }

            count = 0;
            while (test.move_next() && test2.move_next())
            {
                DLIB_CASSERT(test.element() == count,"");
                DLIB_CASSERT(test2.element() == count,"");
                ++count;
            }

            DLIB_CASSERT(test2.size() == count,"");
            DLIB_CASSERT(test.size() == count,"");

            test2.clear();


        } // for (int g = 0; g < 2; ++g)


    }






    class sliding_buffer_tester : public tester
    {
    public:
        sliding_buffer_tester (
        ) :
            tester ("test_sliding_buffer",
                    "Runs tests on the sliding_buffer component.")
        {}

        void perform_test (
        )
        {
            dlog << LINFO << "testing kernel_1a";
            sliding_buffer_kernel_test<sliding_buffer<unsigned char>::kernel_1a>  ();
            dlog << LINFO << "testing kernel_1a_c";
            sliding_buffer_kernel_test<sliding_buffer<unsigned char>::kernel_1a_c>();
        }
    } a;

}

