thread.h

Go to the documentation of this file.

/*
 * thread.h
 *
 * Executable thread encapsulation class (pre-emptive if OS allows).
 *
 * Portable Tools Library
 *
 * Copyright (c) 1993-1998 Equivalence Pty. Ltd.
 *
 * The contents of this file are subject to the Mozilla Public License
 * Version 1.0 (the "License"); you may not use this file except in
 * compliance with the License. You may obtain a copy of the License at
 * http://www.mozilla.org/MPL/
 *
 * Software distributed under the License is distributed on an "AS IS"
 * basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See
 * the License for the specific language governing rights and limitations
 * under the License.
 *
 * The Original Code is Portable Windows Library.
 *
 * The Initial Developer of the Original Code is Equivalence Pty. Ltd.
 *
 * Portions are Copyright (C) 1993 Free Software Foundation, Inc.
 * All Rights Reserved.
 *
 * Contributor(s): ______________________________________.
 *
 * $Revision: 24177 $
 * $Author: rjongbloed $
 * $Date: 2010-04-05 06:52:04 -0500 (Mon, 05 Apr 2010) $
 */

#ifndef PTLIB_THREAD_H
#define PTLIB_THREAD_H

#ifdef P_USE_PRAGMA
#pragma interface
#endif

#ifdef Priority
#undef Priority
#endif

#include <ptlib/mutex.h>

class PSemaphore;

#define PThreadIdentifer PThreadIdentifier

typedef P_THREADIDENTIFIER PThreadIdentifier;

// PThread

class PThread : public PObject
{
  PCLASSINFO(PThread, PObject);

  public:

    enum Priority {
      LowestPriority,  

      LowPriority,     

      NormalPriority,  

      HighPriority,    

      HighestPriority, 

      NumPriorities
    };

    enum AutoDeleteFlag {
      AutoDeleteThread,   

      NoAutoDeleteThread  
    };

    PThread(
      PINDEX,                 
      AutoDeleteFlag deletion = AutoDeleteThread,
      Priority priorityLevel = NormalPriority,  
      const PString & threadName = PString::Empty() 
    );

    ~PThread();

    void PrintOn(
      ostream & strm    
    ) const;

    virtual void Restart();

    virtual void Terminate();

    virtual PBoolean IsTerminated() const;

    void WaitForTermination() const;

    PBoolean WaitForTermination(
      const PTimeInterval & maxWait  
    ) const;

    virtual void Suspend(
      PBoolean susp = true    
    );

    virtual void Resume();

    virtual PBoolean IsSuspended() const;

    static void Sleep(
      const PTimeInterval & delay   
    );

    virtual void SetPriority(
      Priority priorityLevel    
    );

    virtual Priority GetPriority() const;

    virtual void SetAutoDelete(
      AutoDeleteFlag deletion = AutoDeleteThread  
    );

    void SetNoAutoDelete() { SetAutoDelete(NoAutoDeleteThread); }

    virtual PString GetThreadName() const;

    virtual void SetThreadName(
      const PString & name        
    );

    virtual PThreadIdentifier GetThreadId() const;
    static PThreadIdentifier GetCurrentThreadId();

    struct Times
    {
      PTimeInterval m_real;     
      PTimeInterval m_kernel;   
      PTimeInterval m_user;     
      friend ostream & operator<<(ostream & strm, const Times & times);
    };

    bool GetTimes(
      Times & times   
    );

    virtual void Main() = 0;

    static PThread * Current();

    static void Yield();

    static PThread * Create(
      const PNotifier & notifier,     
      INT parameter = 0,              
      AutoDeleteFlag deletion = AutoDeleteThread,
      Priority priorityLevel = NormalPriority,  
      const PString & threadName = PString::Empty(), 
      PINDEX stackSize = 65536         
    );
    static PThread * Create(
      const PNotifier & notifier,     
      const PString & threadName      
    ) { return Create(notifier, 0, NoAutoDeleteThread, NormalPriority, threadName); }
  
    bool IsAutoDelete() const { return autoDelete; }

  private:
    PThread();
    // Create a new thread instance as part of a <code>PProcess</code> class.

    friend class PProcess;
    friend class PExternalThread;
    friend class PHouseKeepingThread;
    // So a PProcess can get at PThread() constructor but nothing else.

    PThread(const PThread &) : PObject () { }
    // Empty constructor to prevent copying of thread instances.

    PThread & operator=(const PThread &) { return *this; }
    // Empty assignment operator to prevent copying of thread instances.

    PBoolean autoDelete;
    // Automatically delete the thread on completion.

    // Give the thread a name for debugging purposes.
    PString threadName;
    PMutex threadNameMutex;

#if PTRACING
  public:
    struct TraceInfo {
      TraceInfo()
      { traceBlockIndentLevel = 0; }

      PStack<PStringStream> traceStreams;
      unsigned traceLevel;
      unsigned traceBlockIndentLevel;
    };

#ifndef P_HAS_THREADLOCAL_STORAGE
  private:
    friend class PTrace;
    TraceInfo traceInfo;
#endif // P_HAS_THREADLOCAL_STORAGE
#endif // PTRACING

// Include platform dependent part of class
#ifdef _WIN32
#include "msos/ptlib/thread.h"
#else
#include "unix/ptlib/thread.h"
#endif
};

// Include definition of platform dependent thread ID format
#if defined(_WIN32) && !defined(_WIN32_WCE)
  #define PTHREAD_ID_FMT ":%u"
#else
  #define PTHREAD_ID_FMT ":0x%x"
#endif

#ifdef _MSC_VER
#pragma warning(disable:4355)
#endif

/*
   This class automates calling a global function with no arguments within it's own thread.
   It is used as follows:

   void GlobalFunction()
   {
   }

   ...
   PString arg;
   new PThreadMain(&GlobalFunction)
 */
class PThreadMain : public PThread
{
  PCLASSINFO(PThreadMain, PThread);
  public:
    typedef void (*FnType)(); 
    PThreadMain(FnType function, bool autoDel = false)
      : PThread(10000, autoDel ? PThread::AutoDeleteThread : PThread::NoAutoDeleteThread)
      , m_function(function)
      { PThread::Resume(); }
    PThreadMain(const char * file, int line, FnType function, bool autoDel = false)
      : PThread(10000, autoDel ? PThread::AutoDeleteThread : PThread::NoAutoDeleteThread,  NormalPriority,
                psprintf("%s:%08x-%s:%i", GetClass(), (void *)this, file, line))
      , m_function(function)
      { PThread::Resume(); }
    virtual void Main()
      { (*m_function)(); }

  protected:
    FnType m_function;
};

/*
   This template automates calling a global function with one argument within it's own thread.
   It is used as follows:

   void GlobalFunction(PString arg)
   {
   }

   ...
   PString arg;
   new PThread1Arg<PString>(arg, &GlobalFunction)
 */
template<typename Arg1Type>
class PThread1Arg : public PThread
{
  PCLASSINFO(PThread1Arg, PThread);
  public:
    typedef void (*FnType)(Arg1Type arg1);

    PThread1Arg(Arg1Type arg1, FnType function, bool autoDel = false)
      : PThread(10000, autoDel ? PThread::AutoDeleteThread : PThread::NoAutoDeleteThread)
      , m_function(function)
      , m_arg1(arg1)
    { PThread::Resume(); }
    PThread1Arg(const char * file, int line, Arg1Type arg1, FnType function, bool autoDel = false)
      : PThread(10000, autoDel ? PThread::AutoDeleteThread : PThread::NoAutoDeleteThread, NormalPriority,
                psprintf("%s:%08x-%s:%i", GetClass(), (void *)this, file, line))
      , m_function(function)
      , m_arg1(arg1)
      { PThread::Resume(); }
    virtual void Main()
      { (*m_function)(m_arg1); }

  protected:
    FnType   m_function;
    Arg1Type m_arg1;
};


/*
   This template automates calling a global function with two arguments within it's own thread.
   It is used as follows:

   void GlobalFunction(PString arg1, int arg2)
   {
   }

   ...
   PString arg;
   new PThread2Arg<PString, int>(arg1, arg2, &GlobalFunction)
 */
template<typename Arg1Type, typename Arg2Type>
class PThread2Arg : public PThread
{
  PCLASSINFO(PThread2Arg, PThread);
  public:
    typedef void (*FnType)(Arg1Type arg1, Arg2Type arg2); 
    PThread2Arg(Arg1Type arg1, Arg2Type arg2, FnType function, bool autoDel = false)
      : PThread(10000, autoDel ? PThread::AutoDeleteThread : PThread::NoAutoDeleteThread)
      , m_function(function)
      , m_arg1(arg1)
      , m_arg2(arg2)
      { PThread::Resume(); }
    PThread2Arg(const char * file, int line, Arg1Type arg1, Arg2Type arg2, FnType function, bool autoDel = false)
      : PThread(10000, autoDel ? PThread::AutoDeleteThread : PThread::NoAutoDeleteThread, NormalPriority,
                psprintf("%s:%08x-%s:%i", GetClass(), (void *)this, file, line))
      , m_function(function)
      , m_arg1(arg1)
      , m_arg2(arg2)
      { PThread::Resume(); }
    virtual void Main()
      { (*m_function)(m_arg1, m_arg2); }

  protected:
    FnType   m_function;
    Arg1Type m_arg1;
    Arg2Type m_arg2;
};

/*
   This template automates calling a global function with three arguments within it's own thread.
   It is used as follows:

   void GlobalFunction(PString arg1, int arg2, int arg3)
   {
   }

   ...
   PString arg;
   new PThread3Arg<PString, int, int>(arg1, arg2, arg3, &GlobalFunction)
 */
template<typename Arg1Type, typename Arg2Type, typename Arg3Type>
class PThread3Arg : public PThread
{
  PCLASSINFO(PThread3Arg, PThread);
  public:
    typedef void (*FnType)(Arg1Type arg1, Arg2Type arg2, Arg3Type arg3); 
    PThread3Arg(Arg1Type arg1, Arg2Type arg2, Arg3Type arg3, FnType function, bool autoDel = false)
      : PThread(10000, autoDel ? PThread::AutoDeleteThread : PThread::NoAutoDeleteThread)
      , m_function(function)
      , m_arg1(arg1)
      , m_arg2(arg2)
      , m_arg3(arg3)
      { PThread::Resume(); }
    PThread3Arg(const char * file, int line, Arg1Type arg1, Arg2Type arg2, Arg3Type arg3, FnType function, bool autoDel = false)
      : PThread(10000, autoDel ? PThread::AutoDeleteThread : PThread::NoAutoDeleteThread, NormalPriority,
                psprintf("%s:%08x-%s:%i", GetClass(), (void *)this, file, line))
      , m_function(function)
      , m_arg1(arg1)
      , m_arg2(arg2)
      , m_arg3(arg3)
      { PThread::Resume(); }
    virtual void Main()
      { (*m_function)(m_arg1, m_arg2, m_arg3); }

  protected:
    FnType   m_function;
    Arg1Type m_arg1;
    Arg2Type m_arg2;
    Arg2Type m_arg3;
};

/*
   This template automates calling a member function with no arguments within it's own thread.
   It is used as follows:

   class Example {
     public:
      void Function()
      {
      }
   };

   ...
   Example ex;
   new PThreadObj<Example>(ex, &Example::Function)
 */

template <typename ObjType>
class PThreadObj : public PThread
{
    PCLASSINFO(PThreadObj, PThread);
  public:
    typedef void (ObjType::*ObjTypeFn)(); 

    PThreadObj(
      ObjType & obj,
      ObjTypeFn function,
      bool autoDel = false,
      const char * name = NULL,
      PThread::Priority priority = PThread::NormalPriority
    ) : PThread(10000,
                autoDel ? PThread::AutoDeleteThread : PThread::NoAutoDeleteThread,
                priority,
                name)
      , m_object(obj)
      , m_function(function)
    {
      PThread::Resume();
    }

    void Main()
    {
      (m_object.*m_function)();
    }

  protected:
    ObjType & m_object;
    ObjTypeFn m_function;
};


/*
   This template automates calling a member function with one argument within it's own thread.
   It is used as follows:

   class Example {
     public:
      void Function(PString arg)
      {
      }
   };

   ...
   Example ex;
   PString str;
   new PThreadObj1Arg<Example>(ex, str, &Example::Function)
 */
template <class ObjType, typename Arg1Type>
class PThreadObj1Arg : public PThread
{
    PCLASSINFO(PThreadObj1Arg, PThread);
  public:
    typedef void (ObjType::*ObjTypeFn)(Arg1Type); 

    PThreadObj1Arg(
      ObjType & obj,
      Arg1Type arg1,
      ObjTypeFn function,
      bool autoDel = false,
      const char * name = NULL,
      PThread::Priority priority = PThread::NormalPriority
    ) : PThread(10000,
                autoDel ? PThread::AutoDeleteThread : PThread::NoAutoDeleteThread,
                priority,
                name)
      , m_object(obj)
      , m_function(function)
      , m_arg1(arg1)
    {
      PThread::Resume();
    }

    void Main()
    {
      (m_object.*m_function)(m_arg1);
    }

  protected:
    ObjType & m_object;
    ObjTypeFn m_function;
    Arg1Type  m_arg1;
};

template <class ObjType, typename Arg1Type, typename Arg2Type>
class PThreadObj2Arg : public PThread
{
    PCLASSINFO(PThreadObj2Arg, PThread);
  public:
    typedef void (ObjType::*ObjTypeFn)(Arg1Type, Arg2Type);

    PThreadObj2Arg(
      ObjType & obj,
      Arg1Type arg1,
      Arg2Type arg2,
      ObjTypeFn function,
      bool autoDel = false,
      const char * name = NULL,
      PThread::Priority priority = PThread::NormalPriority
    ) : PThread(10000,
                autoDel ? PThread::AutoDeleteThread : PThread::NoAutoDeleteThread,
                priority,
                name)
      , m_object(obj)
      , m_function(function)
      , m_arg1(arg1)
      , m_arg2(arg2)
    {
      PThread::Resume();
    }

    void Main()
    {
      (m_object.*m_function)(m_arg1, m_arg2);
    }

  protected:
    ObjType & m_object;
    ObjTypeFn m_function;
    Arg1Type  m_arg1;
    Arg2Type  m_arg2;
};


//
// PThreadLocalStorage
//

#ifdef _WIN32

#define P_HAS_THREADLOCAL_STORAGE 1

template <class Storage_T>
class PThreadLocalStorage
{
  public:
    typedef DWORD Key_T;
    typedef Storage_T value_type;

    PThreadLocalStorage()
    { key = TlsAlloc(); }

    ~PThreadLocalStorage()
    { TlsFree(key);  }

    Key_T GetKey() const
    { return key; }

    value_type * Get()
    { return (value_type *) TlsGetValue(key); }

    void Set(value_type * v)
    { TlsSetValue(key, (LPVOID)v); }

  protected:
    DWORD key;
};

#elif defined(P_PTHREADS)

#include <pthread.h>

#define P_HAS_THREADLOCAL_STORAGE 1

template <class Storage_T>
class PThreadLocalStorage
{
  public:
    typedef pthread_key_t Key_T;
    typedef Storage_T value_type;

    PThreadLocalStorage()
    { pthread_key_create(&key, NULL); }

    ~PThreadLocalStorage()
    { pthread_key_delete(key); }

    Key_T GetKey() const
    { return key; }

    value_type * Get()
    { return (value_type *)pthread_getspecific(key); }

    void Set(value_type * v)
    { pthread_setspecific(key, v); }

  private:
    Key_T key;
};

#else

#undef P_HAS_THREADLOCAL_STORAGE 1
#warning("Thread local storage not supported");

#endif


#ifdef _MSC_VER
#pragma warning(default:4355)
#endif

#endif // PTLIB_THREAD_H

// End Of File ///////////////////////////////////////////////////////////////

---