Main Page   Groups   Namespace List   Class Hierarchy   Alphabetical List   Compound List   File List   Namespace Members   Compound Members   File Members   Concepts

itkMultiThreader.h

Go to the documentation of this file.
00001 /*=========================================================================
00002 
00003   Program:   Insight Segmentation & Registration Toolkit
00004   Module:    $RCSfile: itkMultiThreader.h,v $
00005   Language:  C++
00006   Date:      $Date: 2004/05/19 02:07:53 $
00007   Version:   $Revision: 1.25 $
00008 
00009   Copyright (c) Insight Software Consortium. All rights reserved.
00010   See ITKCopyright.txt or http://www.itk.org/HTML/Copyright.htm for details.
00011 
00012   Portions of this code are covered under the VTK copyright.
00013   See VTKCopyright.txt or http://www.kitware.com/VTKCopyright.htm for details.
00014 
00015      This software is distributed WITHOUT ANY WARRANTY; without even 
00016      the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR 
00017      PURPOSE.  See the above copyright notices for more information.
00018 
00019 =========================================================================*/
00020 #ifndef __itkMultiThreader_h
00021 #define __itkMultiThreader_h
00022 
00023 #include "itkObject.h"
00024 #include "itkMutexLock.h"
00025 
00026 #ifdef ITK_USE_SPROC
00027 #include <sys/types.h>
00028 #include <sys/resource.h>
00029 #include <sys/types.h>
00030 #include <sys/prctl.h>
00031 #include <wait.h>
00032 #include <errno.h>
00033 #include <ulocks.h>
00034 #include <fcntl.h>
00035 #include <unistd.h>
00036 #include <stdlib.h>
00037 #include <signal.h>
00038 #include <sys/signal.h>
00039 #include <sys/sysmp.h>
00040 #include <sys/errno.h>
00041 #include <sys/syssgi.h>
00042 
00043 extern "C" {
00044 #include <sys/pmo.h>
00045 #include <fetchop.h>
00046 }
00047 #endif
00048 
00049 #ifdef ITK_USE_PTHREADS
00050 #include <pthread.h>
00051 #endif
00052 
00053 namespace itk
00054 {
00071 // The maximum number of threads allowed
00072 #ifdef ITK_USE_SPROC
00073 #define ITK_MAX_THREADS              128
00074 #endif
00075 
00076 #ifdef ITK_USE_PTHREADS
00077 #define ITK_MAX_THREADS              128
00078 #endif
00079 
00080 #ifdef ITK_USE_WIN32_THREADS
00081 #define ITK_MAX_THREADS              128
00082 #endif
00083 
00084 // cygwin threads are unreliable
00085 #ifdef __CYGWIN__
00086 #undef ITK_MAX_THREADS
00087 #define ITK_MAX_THREADS 128 
00088 #endif
00089 
00090 #ifndef ITK_MAX_THREADS
00091 #define ITK_MAX_THREADS 1
00092 #endif
00093 
00099 #ifdef ITK_USE_SPROC
00100 typedef int ThreadProcessIDType;
00101 #endif
00102 
00103 #ifdef ITK_USE_PTHREADS
00104 typedef void *(*ThreadFunctionType)(void *);
00105 typedef pthread_t ThreadProcessIDType;
00106 #define ITK_THREAD_RETURN_VALUE  NULL
00107 #define ITK_THREAD_RETURN_TYPE   void *
00108 #endif
00109 
00110 #ifdef ITK_USE_WIN32_THREADS
00111 typedef LPTHREAD_START_ROUTINE ThreadFunctionType;
00112 typedef HANDLE ThreadProcessIDType;
00113 #define ITK_THREAD_RETURN_VALUE 0
00114 #define ITK_THREAD_RETURN_TYPE DWORD __stdcall
00115 #endif
00116 
00117 #ifndef ITK_THREAD_RETURN_VALUE
00118 typedef void (*ThreadFunctionType)(void *);
00119 typedef int ThreadProcessIDType;
00120 #define ITK_THREAD_RETURN_VALUE
00121 #define ITK_THREAD_RETURN_TYPE void
00122 #endif
00123   
00124 class ITKCommon_EXPORT MultiThreader : public Object 
00125 {
00126 public:
00128   typedef MultiThreader         Self;
00129   typedef Object  Superclass;
00130   typedef SmartPointer<Self>  Pointer;
00131   typedef SmartPointer<const Self>  ConstPointer;
00132   
00134   itkNewMacro(Self);  
00135 
00137   itkTypeMacro(MultiThreader, Object);
00138 
00142   itkSetClampMacro( NumberOfThreads, int, 1, ITK_MAX_THREADS );
00143   itkGetMacro( NumberOfThreads, int );
00144     
00148   static void SetGlobalMaximumNumberOfThreads(int val);
00149   static int  GetGlobalMaximumNumberOfThreads();
00150   
00154   static void SetGlobalDefaultNumberOfThreads(int val);
00155   static int  GetGlobalDefaultNumberOfThreads();
00156   
00159   void SingleMethodExecute();
00160 
00164   void MultipleMethodExecute();
00165   
00171   void SetSingleMethod(ThreadFunctionType, void *data );
00172   
00175   void SetMultipleMethod( int index, ThreadFunctionType, void *data ); 
00176 
00180   int SpawnThread( ThreadFunctionType, void *data );
00181 
00183   void TerminateThread( int thread_id );
00184   
00185 #ifdef ITK_USE_SPROC
00186   static bool GetInitialized()
00187   { return m_Initialized; }
00188   static usptr_t * GetThreadArena()
00189   { return m_ThreadArena; }
00190 
00191   static void Initialize();
00192 #endif
00193   
00205 #ifdef ThreadInfoStruct
00206 #undef ThreadInfoStruct
00207 #endif
00208   struct ThreadInfoStruct
00209   {
00210 #ifdef ITK_USE_SPROC
00211     char Pad1[128];
00212 #endif
00213     int                 ThreadID;
00214     int                 NumberOfThreads;
00215     int                 *ActiveFlag;
00216     MutexLock::Pointer  ActiveFlagLock;
00217     void                *UserData;
00218 #ifdef ITK_USE_SPROC
00219     char Pad2[128];
00220 #endif
00221   };
00222   
00223 protected:
00224   MultiThreader();
00225   ~MultiThreader();
00226   void PrintSelf(std::ostream& os, Indent indent) const;
00227 
00228 private:
00229   
00230 #ifdef ITK_USE_SPROC
00231   static bool m_Initialized;
00232   static usptr_t * m_ThreadArena;
00233   static int m_DevzeroFd;
00234 #endif
00235 
00236   MultiThreader(const Self&); //purposely not implemented
00237   void operator=(const Self&); //purposely not implemented
00238 
00240   int                        m_NumberOfThreads;
00241 
00245   ThreadInfoStruct           m_ThreadInfoArray[ITK_MAX_THREADS];
00246 
00248   ThreadFunctionType         m_SingleMethod;
00249   ThreadFunctionType         m_MultipleMethod[ITK_MAX_THREADS];
00250   
00253   int                        m_SpawnedThreadActiveFlag    [ITK_MAX_THREADS];
00254   MutexLock::Pointer         m_SpawnedThreadActiveFlagLock[ITK_MAX_THREADS];
00255   ThreadProcessIDType        m_SpawnedThreadProcessID     [ITK_MAX_THREADS];
00256   ThreadInfoStruct           m_SpawnedThreadInfoArray     [ITK_MAX_THREADS];
00257   
00259   void                       *m_SingleData;
00260   void                       *m_MultipleData[ITK_MAX_THREADS];
00261 
00263   static int                  m_GlobalMaximumNumberOfThreads;
00264   static int                  m_GlobalDefaultNumberOfThreads;
00265   
00269   friend class ProcessObject;
00270 }; 
00271  
00272 }  // end namespace itk
00273 #endif

Generated at Wed Mar 30 00:04:23 2005 for ITK by doxygen 1.3.9.1 written by Dimitri van Heesch, © 1997-2000