dlib C++ Library - multithreaded_object.cpp

// Copyright (C) 2007 Davis E. King (davis@dlib.net)
// License: Boost Software License See LICENSE.txt for the full license.
#include <string>
#include <sstream>
#include <dlib/threads.h>
#include "tester.h"
namespace 
{
 using namespace test;
 using namespace std;
 using namespace dlib;
 
 logger dlog("test.multithreaded_object");
 dlib::mutex cm;
 int count;
 class test1 : multithreaded_object
 {
 public:
 test1 ()
 {
 DLIB_TEST(number_of_threads_registered() == 0);
 DLIB_TEST(number_of_threads_alive() == 0);
 DLIB_TEST(is_running() == false);
 clear();
 DLIB_TEST(number_of_threads_registered() == 0);
 DLIB_TEST(number_of_threads_alive() == 0);
 DLIB_TEST(is_running() == false);
 }
 ~test1 ()
 {
 DLIB_TEST(number_of_threads_registered() == 0);
 DLIB_TEST(number_of_threads_alive() == 0);
 DLIB_TEST(is_running() == false);
 stop();
 DLIB_TEST(number_of_threads_registered() == 0);
 DLIB_TEST(number_of_threads_alive() == 0);
 DLIB_TEST(is_running() == false);
 wait();
 DLIB_TEST(number_of_threads_registered() == 0);
 DLIB_TEST(number_of_threads_alive() == 0);
 DLIB_TEST(is_running() == false);
 }
 private:
 };
 class test2 : private multithreaded_object
 {
 public:
 test2()
 {
 DLIB_TEST(number_of_threads_registered() == 0);
 DLIB_TEST(number_of_threads_alive() == 0);
 DLIB_TEST(is_running() == false);
 register_thread(*this,&test2::thread);
 DLIB_TEST(number_of_threads_registered() == 1);
 DLIB_TEST(number_of_threads_alive() == 0);
 DLIB_TEST(is_running() == false);
 clear();
 DLIB_TEST(number_of_threads_registered() == 0);
 DLIB_TEST(number_of_threads_alive() == 0);
 DLIB_TEST(is_running() == false);
 register_thread(*this,&test2::thread);
 DLIB_TEST(number_of_threads_registered() == 1);
 DLIB_TEST(number_of_threads_alive() == 0);
 DLIB_TEST(is_running() == false);
 }
 ~test2()
 {
 DLIB_TEST(number_of_threads_registered() == 1);
 DLIB_TEST(number_of_threads_alive() == 0);
 DLIB_TEST(is_running() == false);
 stop();
 DLIB_TEST(number_of_threads_registered() == 1);
 DLIB_TEST(number_of_threads_alive() == 0);
 DLIB_TEST(is_running() == false);
 wait();
 DLIB_TEST(number_of_threads_registered() == 1);
 DLIB_TEST(number_of_threads_alive() == 0);
 DLIB_TEST(is_running() == false);
 }
 private:
 void thread()
 {
 auto_mutex M(cm);
 ++count;
 }
 };
 class test3_c1 : private multithreaded_object
 {
 public:
 test3_c1()
 {
 DLIB_TEST(number_of_threads_registered() == 0);
 DLIB_TEST(number_of_threads_alive() == 0);
 DLIB_TEST(is_running() == false);
 register_thread(*this,&test3_c1::thread);
 DLIB_TEST(number_of_threads_registered() == 1);
 DLIB_TEST(number_of_threads_alive() == 0);
 DLIB_TEST(is_running() == false);
 start();
 DLIB_TEST(number_of_threads_registered() == 1);
 DLIB_TEST(is_running() == true);
 }
 ~test3_c1()
 {
 DLIB_TEST(number_of_threads_registered() == 1);
 stop();
 DLIB_TEST(is_running() == false);
 DLIB_TEST(number_of_threads_registered() == 1);
 wait();
 DLIB_TEST(number_of_threads_registered() == 1);
 DLIB_TEST(number_of_threads_alive() == 0);
 DLIB_TEST(is_running() == false);
 }
 private:
 void thread()
 {
 cm.lock();
 ++count;
 cm.unlock();
 // wait until we are supposed to stop
 while (!should_stop())
 dlib::sleep(1);
 }
 };
 class test4_c2 : private multithreaded_object
 {
 public:
 test4_c2()
 {
 DLIB_TEST(number_of_threads_registered() == 0);
 DLIB_TEST(number_of_threads_alive() == 0);
 DLIB_TEST(is_running() == false);
 register_thread(*this,&test4_c2::thread);
 DLIB_TEST(number_of_threads_registered() == 1);
 DLIB_TEST(number_of_threads_alive() == 0);
 DLIB_TEST(is_running() == false);
 start();
 DLIB_TEST(number_of_threads_registered() == 1);
 DLIB_TEST(number_of_threads_alive() == 1);
 DLIB_TEST(is_running() == true);
 register_thread(*this,&test4_c2::thread);
 DLIB_TEST(number_of_threads_registered() == 2);
 DLIB_TEST(number_of_threads_alive() == 2);
 DLIB_TEST(is_running() == true);
 start();
 DLIB_TEST(number_of_threads_registered() == 2);
 DLIB_TEST(number_of_threads_alive() == 2);
 DLIB_TEST(is_running() == true);
 start();
 DLIB_TEST(number_of_threads_registered() == 2);
 DLIB_TEST(number_of_threads_alive() == 2);
 DLIB_TEST(is_running() == true);
 start();
 DLIB_TEST(number_of_threads_registered() == 2);
 DLIB_TEST(number_of_threads_alive() == 2);
 DLIB_TEST(is_running() == true);
 start();
 DLIB_TEST(number_of_threads_registered() == 2);
 DLIB_TEST(number_of_threads_alive() == 2);
 DLIB_TEST(is_running() == true);
 pause();
 DLIB_TEST(number_of_threads_registered() == 2);
 DLIB_TEST(number_of_threads_alive() == 2);
 DLIB_TEST(is_running() == false);
 }
 ~test4_c2() 
 {
 try
 {
 DLIB_TEST(number_of_threads_registered() == 2);
 DLIB_TEST(number_of_threads_alive() == 2);
 DLIB_TEST_MSG(is_running() == false,"is_running(): " << is_running());
 stop();
 DLIB_TEST(number_of_threads_registered() == 2);
 DLIB_TEST(is_running() == false);
 wait();
 DLIB_TEST(number_of_threads_registered() == 2);
 DLIB_TEST(number_of_threads_alive() == 0);
 DLIB_TEST(is_running() == false);
 }
 catch(std::exception& e)
 {
 std::cerr << e.what() << std::endl;
 exit(1);
 }
 }
 private:
 void thread()
 {
 auto_mutex M(cm);
 ++count;
 while (!should_stop())
 dlib::sleep(10);
 }
 };
 class test5 : private multithreaded_object
 {
 public:
 test5()
 {
 register_thread(*this,&test5::thread1);
 register_thread(*this,&test5::thread2);
 register_thread(*this,&test5::thread3);
 register_thread(*this,&test5::thread3);
 start();
 }
 ~test5()
 {
 stop();
 wait();
 }
 private:
 void thread1()
 {
 while (!should_stop())
 dlib::sleep(10);
 }
 void thread2()
 {
 while (!should_stop())
 dlib::sleep(10);
 }
 void thread3()
 {
 while (!should_stop())
 dlib::sleep(10);
 }
 };
 void multithreaded_object_test (
 )
 /*!
 ensures
 - runs tests on dlib::multithreaded_object for compliance with the specs 
 !*/
 { 
 count = 0;
 for (int i = 0; i < 5; ++i)
 {
 {
 test1 a1;
 test2 a2;
 test3_c1 a3;
 test4_c2 a4;
 test5 a5;
 }
 DLIB_TEST(count == (i+1)*3);
 print_spinner();
 }
 count = 0;
 for (int i = 0; i < 5; ++i)
 {
 {
 test1 a1;
 test2 a2;
 test3_c1 a3;
 test4_c2 a4;
 test5 a5;
 dlib::sleep(50);
 }
 DLIB_TEST(count == (i+1)*3);
 print_spinner();
 }
 }
 class multithreaded_object_tester : public tester
 {
 public:
 multithreaded_object_tester (
 ) :
 tester ("test_multithreaded_object",
 "Runs tests on the multithreaded_object component.")
 {}
 void perform_test (
 )
 {
 multithreaded_object_test();
 }
 } a;
}