Home | History | Annotate | Download | only in win32
      1 // -*- C++ -*-
      2 //===----------------------- support/win32/support.h ----------------------===//
      3 //
      4 //                     The LLVM Compiler Infrastructure
      5 //
      6 // This file is dual licensed under the MIT and the University of Illinois Open
      7 // Source Licenses. See LICENSE.TXT for details.
      8 //
      9 //===----------------------------------------------------------------------===//
     10 
     11 #ifndef _LIBCPP_SUPPORT_WIN32_SUPPORT_H
     12 #define _LIBCPP_SUPPORT_WIN32_SUPPORT_H
     13 
     14 /*
     15    Functions and constants used in libc++ that are missing from the Windows C library.
     16   */
     17 
     18 #include <cwchar>  // mbstate_t
     19 #include <cstdarg> // va_ macros
     20 #define swprintf _snwprintf
     21 #define vswprintf _vsnwprintf
     22 
     23 extern "C" {
     24 
     25 int vasprintf( char **sptr, const char *__restrict fmt, va_list ap );
     26 int asprintf( char **sptr, const char *__restrict fmt, ...);
     27 size_t mbsnrtowcs( wchar_t *__restrict dst, const char **__restrict src,
     28                    size_t nmc, size_t len, mbstate_t *__restrict ps );
     29 size_t wcsnrtombs( char *__restrict dst, const wchar_t **__restrict src,
     30                    size_t nwc, size_t len, mbstate_t *__restrict ps );
     31 }
     32 
     33 #if defined(_MSC_VER)
     34 #define snprintf _snprintf
     35 #include <xlocinfo.h>
     36 #define atoll _atoi64
     37 #define strtoll _strtoi64
     38 #define strtoull _strtoui64
     39 #define wcstoll _wcstoi64
     40 #define wcstoull _wcstoui64
     41 _LIBCPP_ALWAYS_INLINE float strtof( const char *nptr, char **endptr )
     42 { return _Stof(nptr, endptr, 0); }
     43 _LIBCPP_ALWAYS_INLINE double strtod( const char *nptr, char **endptr )
     44 { return _Stod(nptr, endptr, 0); }
     45 _LIBCPP_ALWAYS_INLINE long double strtold( const char *nptr, char **endptr )
     46 { return _Stold(nptr, endptr, 0); }
     47 
     48 #define _Exit _exit
     49 
     50 #ifndef __clang__ // MSVC-based Clang also defines _MSC_VER
     51 #include <intrin.h>
     52 
     53 _LIBCPP_ALWAYS_INLINE int __builtin_popcount(unsigned int x) {
     54    static const unsigned int m1 = 0x55555555; //binary: 0101...
     55    static const unsigned int m2 = 0x33333333; //binary: 00110011..
     56    static const unsigned int m4 = 0x0f0f0f0f; //binary:  4 zeros,  4 ones ...
     57    static const unsigned int h01= 0x01010101; //the sum of 256 to the power of 0,1,2,3...
     58    x -= (x >> 1) & m1;             //put count of each 2 bits into those 2 bits
     59    x = (x & m2) + ((x >> 2) & m2); //put count of each 4 bits into those 4 bits
     60    x = (x + (x >> 4)) & m4;        //put count of each 8 bits into those 8 bits
     61    return (x * h01) >> 24;  //returns left 8 bits of x + (x<<8) + (x<<16) + (x<<24)
     62 }
     63 
     64 _LIBCPP_ALWAYS_INLINE int __builtin_popcountl(unsigned long x) {
     65   return __builtin_popcount(static_cast<int>(x));
     66 }
     67 
     68 _LIBCPP_ALWAYS_INLINE int __builtin_popcountll(unsigned long long x) {
     69    static const unsigned long long m1  = 0x5555555555555555; //binary: 0101...
     70    static const unsigned long long m2  = 0x3333333333333333; //binary: 00110011..
     71    static const unsigned long long m4  = 0x0f0f0f0f0f0f0f0f; //binary:  4 zeros,  4 ones ...
     72    static const unsigned long long h01 = 0x0101010101010101; //the sum of 256 to the power of 0,1,2,3...
     73    x -= (x >> 1) & m1;             //put count of each 2 bits into those 2 bits
     74    x = (x & m2) + ((x >> 2) & m2); //put count of each 4 bits into those 4 bits
     75    x = (x + (x >> 4)) & m4;        //put count of each 8 bits into those 8 bits
     76    return static_cast<int>((x * h01)>>56);  //returns left 8 bits of x + (x<<8) + (x<<16) + (x<<24) + ...
     77 }
     78 
     79 _LIBCPP_ALWAYS_INLINE int __builtin_ctz( unsigned int x )
     80 {
     81    DWORD r = 0;
     82    _BitScanReverse(&r, x);
     83    return static_cast<int>(r);
     84 }
     85 
     86 // sizeof(long) == sizeof(int) on Windows
     87 _LIBCPP_ALWAYS_INLINE int __builtin_ctzl( unsigned long x )
     88 { return __builtin_ctz( static_cast<int>(x) ); }
     89 
     90 _LIBCPP_ALWAYS_INLINE int __builtin_ctzll( unsigned long long x )
     91 {
     92     DWORD r = 0;
     93     _BitScanReverse64(&r, x);
     94     return static_cast<int>(r);
     95 }
     96 _LIBCPP_ALWAYS_INLINE int __builtin_clz( unsigned int x )
     97 {
     98    DWORD r = 0;
     99    _BitScanForward(&r, x);
    100    return static_cast<int>(r);
    101 }
    102 // sizeof(long) == sizeof(int) on Windows
    103 _LIBCPP_ALWAYS_INLINE int __builtin_clzl( unsigned long x )
    104 { return __builtin_clz( static_cast<int>(x) ); }
    105 _LIBCPP_ALWAYS_INLINE int __builtin_clzll( unsigned long long x )
    106 {
    107     DWORD r = 0;
    108     _BitScanForward64(&r, x);
    109     return static_cast<int>(r);
    110 }
    111 #endif // !__clang__
    112 #endif // _MSC_VER
    113 
    114 #endif // _LIBCPP_SUPPORT_WIN32_SUPPORT_H
    115