00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
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
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
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&);
00237 void operator=(const Self&);
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 }
00273 #endif