// Copyright 2016 the V8 project authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "src/strings/string-case.h"

#include "src/base/logging.h"
#include "src/common/assert-scope.h"
#include "src/common/globals.h"
#include "src/strings/unicode.h"
#include "src/utils/utils.h"

namespace v8 {
namespace internal {

// FastAsciiConvert tries to do character processing on a word_t basis if
// source and destination strings are properly aligned. Natural alignment of
// string data depends on kTaggedSize so we define word_t via Tagged_t.
using word_t = std::make_unsigned_t<Tagged_t>;

constexpr word_t kWordTAllBitsSet = std::numeric_limits<word_t>::max();
constexpr word_t kOneInEveryByte = kWordTAllBitsSet / 0xFF;
constexpr word_t kAsciiMask = kOneInEveryByte << 7;

#ifdef DEBUG
template <class CaseMapping>
void CheckFastAsciiConvert(char* dst, const char* src, uint32_t length) {
  constexpr bool is_to_lower = CaseMapping::kIsToLower;
  constexpr char lo = is_to_lower ? 'A' : 'a';
  constexpr char hi = is_to_lower ? 'Z' : 'z';
  constexpr int diff = is_to_lower ? ('a' - 'A') : ('A' - 'a');
  for (uint32_t i = 0; i < length; i++) {
    if (lo <= src[i] && src[i] <= hi) {
      DCHECK_EQ(dst[i], src[i] + diff);
    } else {
      DCHECK_EQ(src[i], dst[i]);
    }
  }
}
#endif

// Given a word and two range boundaries returns a word with high bit
// set in every byte iff the corresponding input byte was strictly in
// the range (m, n). All the other bits in the result are cleared.
// This function is only useful when it can be inlined and the
// boundaries are statically known.
// Requires: all bytes in the input word and the boundaries must be
// ASCII (less than 0x7F).
template <char low, char high>
static inline word_t AsciiRangeMask(word_t w) {
  // Use strict inequalities since in edge cases the function could be
  // further simplified.
  DCHECK_LT(0, low);
  DCHECK_LT(low, high);
  // Has high bit set in every w byte less than n.
  word_t tmp1 = kOneInEveryByte * (0x7F + high) - w;
  // Has high bit set in every w byte greater than m.
  word_t tmp2 = w + kOneInEveryByte * (0x7F - low);
  return (tmp1 & tmp2 & (kOneInEveryByte * 0x80));
}

template <class CaseMapping>
uint32_t FastAsciiCasePrefixLength(const char* src, uint32_t length) {
  const char* saved_src = src;
  DisallowGarbageCollection no_gc;
  // We rely on the distance between upper and lower case letters
  // being a known power of 2.
  DCHECK_EQ('a' - 'A', 1 << 5);
  // Boundaries for the range of input characters that require conversion.
  constexpr bool is_lower = CaseMapping::kIsToLower;
  constexpr char lo = is_lower ? 'A' - 1 : 'a' - 1;
  constexpr char hi = is_lower ? 'Z' + 1 : 'z' + 1;
  const char* const limit = src + length;

  // Only attempt processing one word at a time if src is also aligned.
  if (IsAligned(reinterpret_cast<Address>(src), sizeof(word_t))) {
    // Process the prefix of the input that requires no conversion one aligned
    // (machine) word at a time.
    while (src <= limit - sizeof(word_t)) {
      const word_t w = *reinterpret_cast<const word_t*>(src);
      if ((w & kAsciiMask) != 0) return static_cast<int>(src - saved_src);
      if (AsciiRangeMask<lo, hi>(w) != 0) {
        return static_cast<int>(src - saved_src);
      }
      src += sizeof(word_t);
    }
  }
  // Process the last few bytes of the input (or the whole input if
  // unaligned access is not supported).
  for (; src < limit; ++src) {
    char c = *src;
    if ((c & kAsciiMask) != 0 || (lo < c && c < hi)) {
      return static_cast<int>(src - saved_src);
    }
  }
  return length;
}

template uint32_t FastAsciiCasePrefixLength<unibrow::ToLowercase>(
    const char* src, uint32_t length);
template uint32_t FastAsciiCasePrefixLength<unibrow::ToUppercase>(
    const char* src, uint32_t length);

template <class CaseMapping>
uint32_t FastAsciiConvert(char* dst, const char* src, uint32_t length) {
#ifdef DEBUG
  char* saved_dst = dst;
#endif
  const char* saved_src = src;
  DisallowGarbageCollection no_gc;
  // We rely on the distance between upper and lower case letters
  // being a known power of 2.
  DCHECK_EQ('a' - 'A', 1 << 5);
  // Boundaries for the range of input characters that require conversion.
  constexpr bool is_lower = CaseMapping::kIsToLower;
  constexpr char lo = is_lower ? 'A' - 1 : 'a' - 1;
  constexpr char hi = is_lower ? 'Z' + 1 : 'z' + 1;
  const char* const limit = src + length;

  // dst is newly allocated and always aligned.
  // Only attempt processing one word at a time if src and dst are aligned.

  if (IsAligned(reinterpret_cast<Address>(dst), sizeof(word_t)) &&
      IsAligned(reinterpret_cast<Address>(src), sizeof(word_t))) {
    // Process the prefix of the input that requires no conversion one aligned
    // (machine) word at a time.
    while (src <= limit - sizeof(word_t)) {
      const word_t w = *reinterpret_cast<const word_t*>(src);
      if ((w & kAsciiMask) != 0) return static_cast<int>(src - saved_src);
      if (AsciiRangeMask<lo, hi>(w) != 0) break;
      *reinterpret_cast<word_t*>(dst) = w;
      src += sizeof(word_t);
      dst += sizeof(word_t);
    }
    // Process the remainder of the input performing conversion when
    // required one word at a time.
    while (src <= limit - sizeof(word_t)) {
      const word_t w = *reinterpret_cast<const word_t*>(src);
      if ((w & kAsciiMask) != 0) return static_cast<int>(src - saved_src);
      word_t m = AsciiRangeMask<lo, hi>(w);
      // The mask has high (7th) bit set in every byte that needs
      // conversion and we know that the distance between cases is
      // 1 << 5.
      *reinterpret_cast<word_t*>(dst) = w ^ (m >> 2);
      src += sizeof(word_t);
      dst += sizeof(word_t);
    }
  }
  // Process the last few bytes of the input (or the whole input if
  // unaligned access is not supported).
  while (src < limit) {
    char c = *src;
    if ((c & kAsciiMask) != 0) return static_cast<int>(src - saved_src);
    if (lo < c && c < hi) {
      c ^= (1 << 5);
    }
    *dst = c;
    ++src;
    ++dst;
  }

#ifdef DEBUG
  CheckFastAsciiConvert<CaseMapping>(saved_dst, saved_src, length);
#endif

  return length;
}

template uint32_t FastAsciiConvert<unibrow::ToLowercase>(char* dst,
                                                         const char* src,
                                                         uint32_t length);
template uint32_t FastAsciiConvert<unibrow::ToUppercase>(char* dst,
                                                         const char* src,
                                                         uint32_t length);

}  // namespace internal
}  // namespace v8