source: Revenant/marslib/include/mars_debug.h@ 25c4774

#ifndef __MARSDEBUGTOOLS_H__
#define __MARSDEBUGTOOLS_H__

#include <boost/thread.hpp>
#include <map>
#include <set>
#include <string>

namespace mars
{
        enum ThreadingModel
        {
                ThrMdl_Main,
                ThrMdl_Single,
                ThrMdl_Any
        };

        extern class ThreadingModelManager
        {
        public:
                ThreadingModelManager();

                void registerMainThread ();
                size_t registerThread (const char * szFuncName, const void * pThis);
                size_t deregisterThread (const char * szFuncName, const void * pThis);
                bool checkMainThread (const char * szFuncName);

        private:
                typedef std::set< boost::thread::id > ThreadIDSet;
                typedef std::map< std::string, ThreadIDSet > ThreadConcurrencyMap;
                typedef std::map< std::pair< std::string, const void * >, size_t > ThreadReferenceCountMap;

                mutable boost::mutex _mtxMutex;
                ThreadConcurrencyMap _mapThreadTracker;
                ThreadReferenceCountMap _mapRefCounter;
                boost::thread::id _idMainThread;
                bool _bMainThreadSet;
        } gtmm_ThreadingModelManager;

        class ThreadingModelMonitor
        {
        public:
                ThreadingModelMonitor (const char * szFuncName, const void * oInst, const ThreadingModel entmThreadingModel);
                ~ThreadingModelMonitor();

        private:
                ThreadingModel _entmThreadingModel;
                boost::thread::id _idThread;
                const char * _szFuncName;
                const void * _pInst;
        };
}

#ifdef _DEBUG
        #define assert_threadmodel(entmThreadingModel) mars::ThreadingModelMonitor tmm(__FUNCTION__, this, entmThreadingModel)
        #define register_mainthread() mars::gtmm_ThreadingModelManager.registerMainThread()
#else
        #define assert_threadmodel(x)
        #define register_mainthread()
#endif

#endif