Changeset 5876

Timestamp:

01/03/07 17:49:46

Author:

robert

Message:

Added SetProcessorAffinityOfCurrentThread(int) method to allow one to set
the affinity of the current thread including the main application thread.

Files:

• OpenThreads/trunk/include/OpenThreads/Thread (modified) (1 diff)
• OpenThreads/trunk/pthread_src/PThread.c++ (modified) (1 diff)
• OpenThreads/trunk/sproc_src/SprocThread.c++ (modified) (1 diff)
• OpenThreads/trunk/win32_src/Win32Thread.cpp (modified) (1 diff)

OpenThreads/trunk/include/OpenThreads/Thread

@@ -39 +39 @@
-extern OPENTHREAD_EXPORT_DIRECTIVE int GetNumberOfProcessors();
-
+/**
+ *  Set the processor affinity of current thread.
+ *
+ *  Note, systems where no support exists no affinity will be set, and -1 will be returned. 
+ *
+ */
+extern OPENTHREAD_EXPORT_DIRECTIVE int SetProcessorAffinityOfCurrentThread(unsigned int cpunum);
-
-/**

OpenThreads/trunk/pthread_src/PThread.c++

@@ -901 +901 @@
-#endif    
-}
+
+int OpenThreads::SetProcessorAffinityOfCurrentThread(unsigned int cpunum)
+{
+    if (cpunum<0) return -1;
+
+    Thread* thread = Thread::CurrentThread();
+    if (thread) 
+    {
+        return thread->setProcessorAffinity(cpunum);
+    }
+    else
+    {
+    #if defined (__linux__) && defined(CPU_SET)
+
+        cpu_set_t cpumask;
+        CPU_ZERO( &cpumask );
+        CPU_SET( cpunum, &cpumask );
+
+        #if defined(COMPILE_USING_TWO_PARAM_sched_setaffinity)
+            return sched_setaffinity( 0, &cpumask );
+        #else
+            return sched_setaffinity( 0, sizeof(cpumask), &cpumask );
+        #endif
+
+    #endif
+    }
+    
+    return -1;
+}

OpenThreads/trunk/sproc_src/SprocThread.c++

@@ -788 +788 @@
-    return 1;
-}
+
+int OpenThreads::SetProcessorAffinityOfCurrentThread(unsigned int cpunum)
+{
+    if (cpunum<0) return -1;
+
+    Thread* thread = Thread::CurrentThread();
+    if (thread) 
+    {
+        return thread->setProcessorAffinity(cpunum);
+    }
+    else
+    {
+        // non op right now, needs implementation.
+    }
+}

OpenThreads/trunk/win32_src/Win32Thread.cpp

@@ -646 +646 @@
-    return sysInfo.dwNumberOfProcessors;
-}
+
+int OpenThreads::SetProcessorAffinityOfCurrentThread(unsigned int cpunum)
+{
+    if (cpunum<0) return -1;
+
+    Thread* thread = Thread::CurrentThread();
+    if (thread) 
+    {
+        return thread->setProcessorAffinity(cpunum);
+    }
+    else
+    {
+        // non op right now, needs implementation.
+    }
+}