folly/portability/Constexpr.h

   1 /*
   2  * Copyright 2017 Facebook, Inc.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *   http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 #pragma once
  18
  19 #include <cstdint>
  20 #include <cstring>
  21 #include <type_traits>
  22
  23 namespace folly {
  24
  25 // TLDR: Prefer using operator< for ordering. And when
  26 // a and b are equivalent objects, we return b to make
  27 // sorting stable.
  28 // See http://stepanovpapers.com/notes.pdf for details.
  29 template <typename T>
  30 constexpr T constexpr_max(T a, T b) {
  31   return b < a ? a : b;
  32 }
  33
  34 // When a and b are equivalent objects, we return a to
  35 // make sorting stable.
  36 template <typename T>
  37 constexpr T constexpr_min(T a, T b) {
  38   return b < a ? b : a;
  39 }
  40
  41 namespace detail {
  42
  43 template <typename T, typename = void>
  44 struct constexpr_abs_helper {};
  45
  46 template <typename T>
  47 struct constexpr_abs_helper<
  48     T,
  49     typename std::enable_if<std::is_floating_point<T>::value>::type> {
  50   static constexpr T go(T t) {
  51     return t < static_cast<T>(0) ? -t : t;
  52   }
  53 };
  54
  55 template <typename T>
  56 struct constexpr_abs_helper<
  57     T,
  58     typename std::enable_if<
  59         std::is_integral<T>::value && !std::is_same<T, bool>::value &&
  60         std::is_unsigned<T>::value>::type> {
  61   static constexpr T go(T t) {
  62     return t;
  63   }
  64 };
  65
  66 template <typename T>
  67 struct constexpr_abs_helper<
  68     T,
  69     typename std::enable_if<
  70         std::is_integral<T>::value && !std::is_same<T, bool>::value &&
  71         std::is_signed<T>::value>::type> {
  72   static constexpr typename std::make_unsigned<T>::type go(T t) {
  73     return typename std::make_unsigned<T>::type(t < static_cast<T>(0) ? -t : t);
  74   }
  75 };
  76 } // namespace detail
  77
  78 template <typename T>
  79 constexpr auto constexpr_abs(T t)
  80     -> decltype(detail::constexpr_abs_helper<T>::go(t)) {
  81   return detail::constexpr_abs_helper<T>::go(t);
  82 }
  83
  84 template <typename T>
  85 constexpr T constexpr_log2(T t) {
  86   return t == T(1) ? T(0) : T(1) + constexpr_log2(t / T(2));
  87 }
  88
  89 namespace detail {
  90
  91 template <typename Char>
  92 constexpr size_t constexpr_strlen_internal(const Char* s, size_t len) {
  93   return *s == Char(0) ? len : constexpr_strlen_internal(s + 1, len + 1);
  94 }
  95 static_assert(
  96     constexpr_strlen_internal("123456789", 0) == 9,
  97     "Someone appears to have broken constexpr_strlen...");
  98 } // namespace detail
  99
 100 template <typename Char>
 101 constexpr size_t constexpr_strlen(const Char* s) {
 102   return detail::constexpr_strlen_internal(s, 0);
 103 }
 104
 105 template <>
 106 constexpr size_t constexpr_strlen(const char* s) {
 107 #if defined(__clang__)
 108   return __builtin_strlen(s);
 109 #elif defined(_MSC_VER) || defined(__CUDACC__)
 110   return detail::constexpr_strlen_internal(s, 0);
 111 #else
 112   return std::strlen(s);
 113 #endif
 114 }
 115 }