Commit 933d6755 authored by Laurent Aimar's avatar Laurent Aimar

Added vlc_GetCPUCount() to retreive the number of available CPU.

It is implemented for win32 and when sched_getaffinity() is available.
parent 529a7a1a
...@@ -556,7 +556,7 @@ AC_CHECK_FUNCS(fdatasync,, ...@@ -556,7 +556,7 @@ AC_CHECK_FUNCS(fdatasync,,
]) ])
dnl Check for non-standard system calls dnl Check for non-standard system calls
AC_CHECK_FUNCS([accept4 dup3 eventfd fstatfs vmsplice]) AC_CHECK_FUNCS([accept4 dup3 eventfd fstatfs vmsplice sched_getaffinity])
AH_BOTTOM([#include <vlc_fixups.h>]) AH_BOTTOM([#include <vlc_fixups.h>])
......
...@@ -63,6 +63,7 @@ ...@@ -63,6 +63,7 @@
# endif # endif
VLC_EXPORT( unsigned, vlc_CPU, ( void ) ); VLC_EXPORT( unsigned, vlc_CPU, ( void ) );
VLC_EXPORT( unsigned, vlc_GetCPUCount, ( void ) );
/** Are floating point operations fast? /** Are floating point operations fast?
* If this bit is not set, you should try to use fixed-point instead. * If this bit is not set, you should try to use fixed-point instead.
......
...@@ -493,6 +493,7 @@ vlc_sem_destroy ...@@ -493,6 +493,7 @@ vlc_sem_destroy
vlc_sem_post vlc_sem_post
vlc_sem_wait vlc_sem_wait
vlc_control_cancel vlc_control_cancel
vlc_GetCPUCount
vlc_CPU vlc_CPU
vlc_error vlc_error
vlc_event_attach vlc_event_attach
......
...@@ -318,6 +318,38 @@ const struct ...@@ -318,6 +318,38 @@ const struct
#endif #endif
}; };
/**
* Return the number of available logical CPU.
*/
unsigned vlc_GetCPUCount(void)
{
#ifdef WIN32
DWORD process_mask;
DWORD system_mask;
if (!GetProcessAffinityMask(GetCurrentProcess(), &process_mask, &system_mask))
return 1;
unsigned count = 0;
while (system_mask) {
count++;
system_mask >>= 1;
}
return count;
#elif HAVE_SCHED_GETAFFINITY
cpu_set_t cpu;
CPU_ZERO(&cpu);
if (sched_getaffinity(0, sizeof(cpu), &cpu) < 0)
return 1;
unsigned count = 0;
for (unsigned i = 0; i < CPU_SETSIZE; i++)
count += CPU_ISSET(i, &cpu) != 0;
return count;
#else
# warning "vlc_GetCPUCount is not implemented for your platform"
return 1;
#endif
}
/** /**
* Check if a directory name contains usable plugins w.r.t. the hardware * Check if a directory name contains usable plugins w.r.t. the hardware
* capabilities. Loading a plugin when the hardware has insufficient * capabilities. Loading a plugin when the hardware has insufficient
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment