Get current program path and number of CPUs under Windows and Linux

[CPP]View Plaincopy

2. #ifdef WIN32
4. #include <Windows.h>
6. #else
8. #include <stdio.h>
10. #include <unistd.h>
12. #endif
16. #include <assert.h>
20. std::string Getcurrentapppath ()
22. {
24. #ifdef WIN32
26. Char Path[max_path + 1] = {0};
28. if (GetModuleFileName (NULL, PATH, MAX_PATH)! = 0)
30. return std::string (path);
32. #else
34. char path[256] = {0};
36. char filepath[256] = {0};
38. char cmd[256] = {0};
40. file* fp = NULL;
44. //Set the proc path where the process is located
46. sprintf (filepath, "/proc/%d", Getpid ());
48. //Set current path to process path
50. if (chdir (filepath)! =-1)
52. {
54. //Specify the shell command to execute
56. snprintf (cmd, page, "Ls-l | grep exe |  awk ' {print $ *} ');
58. if (fp = Popen (cmd,"R")) = = = NULL)
60. {
62. return std::string ();
64. }
66. //Read shell command execution result into string path
68. if (fgets (path, sizeof (path)/sizeof (path[0]), fp) = = NULL)
70. {
72. Pclose (FP);
74. return std::string ();
76. }
80. //popen Open FD must be pclose closed
82. Pclose (FP);
84. return std::string (path);
86. }
88. #endif
90. return std::string ();
92. }
96. std::size_t getcpucount ()
98. {
100. #ifdef WIN32
102. System_info SysInfo;
104. GetSystemInfo (&sysinfo);
106. return sysinfo.dwnumberofprocessors;
108. #else
110. Long cpu_num = sysconf (_SC_NPROCESSORS_ONLN);
112. if (cpu_num = =-1)
114. {
116. ASSERT (false);
118. return 0;
120. }
122. //See if the two are equal
124. ASSERT (Cpu_num = = sysconf (_sc_nprocessors_conf));
126. return cpu_num;
128. #endif
130. }

Get current program path and number of CPUs under Windows and Linux