strfunc.h 1.9 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455
  1. // Tencent is pleased to support the open source community by making RapidJSON available.
  2. //
  3. // Copyright (C) 2015 THL A29 Limited, a Tencent company, and Milo Yip. All rights reserved.
  4. //
  5. // Licensed under the MIT License (the "License"); you may not use this file except
  6. // in compliance with the License. You may obtain a copy of the License at
  7. //
  8. // http://opensource.org/licenses/MIT
  9. //
  10. // Unless required by applicable law or agreed to in writing, software distributed
  11. // under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
  12. // CONDITIONS OF ANY KIND, either express or implied. See the License for the
  13. // specific language governing permissions and limitations under the License.
  14. #ifndef RAPIDJSON_INTERNAL_STRFUNC_H_
  15. #define RAPIDJSON_INTERNAL_STRFUNC_H_
  16. #include "../stream.h"
  17. RAPIDJSON_NAMESPACE_BEGIN
  18. namespace internal {
  19. //! Custom strlen() which works on different character types.
  20. /*! \tparam Ch Character type (e.g. char, wchar_t, short)
  21. \param s Null-terminated input string.
  22. \return Number of characters in the string.
  23. \note This has the same semantics as strlen(), the return value is not number of Unicode codepoints.
  24. */
  25. template <typename Ch>
  26. inline SizeType StrLen(const Ch* s) {
  27. const Ch* p = s;
  28. while (*p) ++p;
  29. return SizeType(p - s);
  30. }
  31. //! Returns number of code points in a encoded string.
  32. template<typename Encoding>
  33. bool CountStringCodePoint(const typename Encoding::Ch* s, SizeType length, SizeType* outCount) {
  34. GenericStringStream<Encoding> is(s);
  35. const typename Encoding::Ch* end = s + length;
  36. SizeType count = 0;
  37. while (is.src_ < end) {
  38. unsigned codepoint;
  39. if (!Encoding::Decode(is, &codepoint))
  40. return false;
  41. count++;
  42. }
  43. *outCount = count;
  44. return true;
  45. }
  46. } // namespace internal
  47. RAPIDJSON_NAMESPACE_END
  48. #endif // RAPIDJSON_INTERNAL_STRFUNC_H_