【Darwin学习笔记】之获取系统处理器数量的方法

阅读Darwin源码的时候看到这个方法，感觉挺有用处，且考虑了多种平台下的实现方式，直接贴代码，以后说不定会用到~

单一种平台下的实现方法可能很容易，但是把这些个系统都收集在一起，在一个函数中实现还是极好的

【转载请注明出处】：http://blog.csdn.NET/longlong530

[cpp] view plain copy

 print?

1. UInt32  OS::GetNumProcessors()  
2. {  
3. #if (__Win32__)  
4.     SYSTEM_INFO theSystemInfo;  
5.     ::GetSystemInfo(&theSystemInfo);  
6.       
7.     return (UInt32)theSystemInfo.dwNumberOfProcessors;  
8. #endif  
9.   
10. #if (__MacOSX__ || __FreeBSD__)  
11.     int numCPUs = 1;  
12.     size_t len = sizeof(numCPUs);  
13.     int mib[2];  
14.     mib[0] = CTL_HW;  
15.     mib[1] = HW_NCPU;  
16.     (void) ::sysctl(mib,2,&numCPUs,&len,NULL,0);  
17.     if (numCPUs < 1)   
18.         numCPUs = 1;  
19.     return (UInt32) numCPUs;  
20. #endif  
21.   
22. #if(__linux__ || __linuxppc__)  
23.       
24.     char cpuBuffer[8192] = "";  
25.     StrPtrLen cpuInfoBuf(cpuBuffer, sizeof(cpuBuffer));  
26.     FILE    *cpuFile = ::fopen( "/proc/cpuinfo", "r" );  
27.     if (cpuFile)  
28.     {   cpuInfoBuf.Len = ::fread(cpuInfoBuf.Ptr, sizeof(char),  cpuInfoBuf.Len, cpuFile);  
29.         ::fclose(cpuFile);  
30.     }  
31.       
32.     StringParser cpuInfoFileParser(&cpuInfoBuf);  
33.     StrPtrLen line;  
34.     StrPtrLen word;  
35.     UInt32 numCPUs = 0;  
36.       
37.     while( cpuInfoFileParser.GetDataRemaining() != 0 )   
38.     {  
39.         cpuInfoFileParser.GetThruEOL(&line);    // Read each line     
40.         StringParser lineParser(&line);  
41.         lineParser.ConsumeWhitespace();         //skip over leading whitespace  
42.   
43.         if (lineParser.GetDataRemaining() == 0) // must be an empty line  
44.             continue;  
45.   
46.         lineParser.ConsumeUntilWhitespace(&word);  
47.                  
48.         if ( word.Equal("processor") ) // found a processor as first word in line  
49.         {   numCPUs ++;   
50.         }  
51.     }  
52.       
53.     if (numCPUs == 0)  
54.         numCPUs = 1;  
55.           
56.     return numCPUs;  
57. #endif  
58.   
59. #if(__solaris__)  
60. {  
61.     UInt32 numCPUs = 0;  
62.     char linebuff[512] = "";  
63.     StrPtrLen line(linebuff, sizeof(linebuff));  
64.     StrPtrLen word;  
65.   
66.     FILE *p = ::popen("uname -X","r");  
67.     while((::fgets(linebuff, sizeof(linebuff -1), p)) > 0)  
68.     {  
69.         StringParser lineParser(&line);  
70.         lineParser.ConsumeWhitespace(); //skip over leading whitespace  
71.   
72.         if (lineParser.GetDataRemaining() == 0) // must be an empty line  
73.             continue;  
74.   
75.         lineParser.ConsumeUntilWhitespace(&word);  
76.   
77.         if ( word.Equal("NumCPU")) // found a tag as first word in line  
78.         {  
79.             lineParser.GetThru(NULL,'=');  
80.             lineParser.ConsumeWhitespace();  //skip over leading whitespace  
81.             lineParser.ConsumeUntilWhitespace(&word); //read the number of cpus  
82.             if (word.Len > 0)  
83.                 ::sscanf(word.Ptr, "%"_U32BITARG_"", &numCPUs);  
84.   
85.             break;  
86.         }  
87.     }  
88.     if (numCPUs == 0)  
89.         numCPUs = 1;  
90.           
91.     ::pclose(p);  
92.       
93.     return numCPUs;  
94. }  
95. #endif  
96.   
97. #if(__sgi__)   
98.     UInt32 numCPUs = 0;  
99.   
100.     numCPUs = sysconf(_SC_NPROC_ONLN);  
101.       
102.     return numCPUs;  
103. #endif        
104.   
105.   
106.     return 1;  
107. }  

PS:  函数摘自Darwin源码：\StreamServer\CommonUtilitiesLib\OS.cpp