src/util/LocaleString.cxx

   1 /*
   2  * Copyright 2018 Max Kellermann <max.kellermann@gmail.com>
   3  *
   4  * Redistribution and use in source and binary forms, with or without
   5  * modification, are permitted provided that the following conditions
   6  * are met:
   7  *
   8  * - Redistributions of source code must retain the above copyright
   9  * notice, this list of conditions and the following disclaimer.
  10  *
  11  * - Redistributions in binary form must reproduce the above copyright
  12  * notice, this list of conditions and the following disclaimer in the
  13  * documentation and/or other materials provided with the
  14  * distribution.
  15  *
  16  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  17  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  18  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
  19  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE
  20  * FOUNDATION OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
  21  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  22  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
  23  * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  24  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
  25  * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  26  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
  27  * OF THE POSSIBILITY OF SUCH DAMAGE.
  28  */
  29
  30 #include "LocaleString.hxx"
  31
  32 #include <cwchar>
  33
  34 #include <string.h>
  35
  36 bool
  37 IsIncompleteCharMB(const char *s, size_t n)
  38 {
  39         auto mb = std::mbstate_t();
  40         const std::size_t length = std::mbrlen(s, n, &mb);
  41         return length == std::size_t(-2);
  42 }
  43
  44 std::size_t
  45 CharSizeMB(const char *s, size_t n)
  46 {
  47         auto mb = std::mbstate_t();
  48         const std::size_t length = std::mbrlen(s, n, &mb);
  49         if (length == std::size_t(-2))
  50                 return n;
  51
  52         if (length == std::size_t(-1))
  53                 return 1;
  54
  55         return length;
  56 }
  57
  58 const char *
  59 PrevCharMB(const char *start, const char *reference)
  60 {
  61         const char *p = reference;
  62
  63         while (p > start) {
  64                 --p;
  65
  66                 auto mb = std::mbstate_t();
  67                 const std::size_t length = std::mbrlen(p, reference - p, &mb);
  68                 if (length != std::size_t(-1))
  69                         break;
  70         }
  71
  72         return p;
  73 }
  74
  75 size_t
  76 StringWidthMB(const char *s, size_t length)
  77 {
  78         const char *const end = s + length;
  79         auto state = std::mbstate_t();
  80
  81         size_t width = 0;
  82         while (s < end) {
  83                 wchar_t w;
  84                 std::size_t n = std::mbrtowc(&w, s, end - s, &state);
  85                 if (n == std::size_t(-2))
  86                         break;
  87
  88                 if (n == std::size_t(-1) || n == 0) {
  89                         ++s;
  90                 } else {
  91                         s += n;
  92                         int cw = wcwidth(w);
  93                         if (cw > 0)
  94                                 width += cw;
  95                 }
  96         }
  97
  98         return width;
  99 }
 100
 101 size_t
 102 StringWidthMB(const char *s)
 103 {
 104         return StringWidthMB(s, strlen(s));
 105 }