Wide character support in bar meter text

Author: Explorer09

Meter.c

@@ -109,71 +109,146 @@ static void BarMeterMode_draw(Meter* this, int x, int y, int w) {
    }
    x++;
 
-   // The text in the bar is right aligned;
-   // Pad with maximal spaces and then calculate needed starting position offset
-   RichString_begin(bar);
-   RichString_appendChr(&bar, 0, ' ', w);
-   RichString_appendWide(&bar, 0, this->txtBuffer);
-
-   int startPos = RichString_sizeVal(bar) - w;
-   if (startPos > w) {
-      // Text is too large for bar
-      // Truncate meter text at a space character
-      for (int pos = 2 * w; pos > w; pos--) {
-         if (RichString_getCharVal(bar, pos) == ' ') {
-            while (pos > w && RichString_getCharVal(bar, pos - 1) == ' ')
-               pos--;
-            startPos = pos - w;
-            break;
+   // Calculate the number of terminal columns needed for the meter text.
+
+   // The text in the bar is right aligned
+
+   MBStringDecoderState state;
+   memset(&state, 0, sizeof(state));
+   state.str = this->txtBuffer;
+   state.maxLen = sizeof(this->txtBuffer) - 1;
+
+   int nColsLeft = w; // pun intended
+   int savedCols = nColsLeft;
+   size_t len = 0;
+   size_t savedLen = 0;
+
+   while (String_decodeNextWChar(&state)) {
+      if (state.ch == 0)
+         break;
+
+      if (state.ch == ' ') {
+         savedLen = len;
+         savedCols = nColsLeft;
+      }
+
+#ifdef HAVE_LIBNCURSESW
+      int nCols = wcwidth((wchar_t)state.ch);
+      if (nCols < 0) {
+         assert(nCols >= 0);
+         break;
+      }
+#else
+      int nCols = 1;
+#endif
+
+      if (nCols > nColsLeft) {
+         // Text is too large for bar
+         // Truncate meter text at a space character
+         if (savedLen > 0) {
+            len = savedLen;
+            nColsLeft = savedCols;
          }
+         break;
       }
 
-      // If still too large, print the start not the end
-      startPos = MINIMUM(startPos, w);
-   }
+      nColsLeft -= nCols;
+
+      if (state.ch == ' ') {
+         continue;
+      }
+
+#ifdef HAVE_LIBNCURSESW
+      // If the character takes zero columns, include the character in the
+      // substring if the working encoding is UTF-8, and ignore it otherwise.
+      if (nCols <= 0 && !CRT_utf8) {
+         continue;
+      }
+#endif
 
-   assert(startPos >= 0);
-   assert(startPos <= w);
-   assert(startPos + w <= RichString_sizeVal(bar));
+      len = (size_t)(state.str - this->txtBuffer);
+   }
 
-   int blockSizes[10];
+   RichString_begin(bar);
+   RichString_appendChr(&bar, 0, ' ', nColsLeft);
+   RichString_appendnWide(&bar, 0, this->txtBuffer, len);
 
-   // First draw in the bar[] buffer...
+   size_t charPos = 0;
    int offset = 0;
    for (uint8_t i = 0; i < this->curItems; i++) {
+      if (!(this->total > 0.0)) {
+         break;
+      }
+      if (offset >= w) {
+         break;
+      }
+
       double value = this->values[i];
-      if (isPositive(value) && this->total > 0.0) {
-         value = MINIMUM(value, this->total);
-         blockSizes[i] = ceil((value / this->total) * w);
-         blockSizes[i] = MINIMUM(blockSizes[i], w - offset);
-      } else {
-         blockSizes[i] = 0;
+      if (!isPositive(value)) {
+         continue;
       }
-      int nextOffset = offset + blockSizes[i];
-      for (int j = offset; j < nextOffset; j++)
-         if (RichString_getCharVal(bar, startPos + j) == ' ') {
+      value = MINIMUM(value, this->total);
+      int blockSize = ceil((value / this->total) * w);
+      blockSize = MINIMUM(blockSize, w - offset);
+      if (blockSize < 1) {
+         continue;
+      }
+
+      int nextOffset = offset + blockSize;
+      assert(offset < nextOffset);
+
+      size_t startPos = charPos;
+      while (true) {
+         if (offset >= nextOffset) {
+#ifdef HAVE_LIBNCURSESW
+            if (!CRT_utf8) {
+               break;
+            }
+#else
+            break;
+#endif
+         }
+
+#ifdef HAVE_LIBNCURSESW
+         wchar_t ch = RichString_getCharVal(bar, charPos);
+         if (ch == 0)
+            break;
+
+         int nCols = wcwidth(ch);
+         assert(nCols >= 0);
+
+         if (offset >= nextOffset && nCols > 0) {
+            // This break condition is for UTF-8.
+            break;
+         }
+#else
+         char ch = RichString_getCharVal(bar, charPos);
+         int nCols = 1;
+
+         assert(offset < nextOffset);
+#endif
+         if (ch == ' ') {
             if (CRT_colorScheme == COLORSCHEME_MONOCHROME) {
                assert(i < strlen(BarMeterMode_characters));
-               RichString_setChar(&bar, startPos + j, BarMeterMode_characters[i]);
+               RichString_setChar(&bar, charPos, BarMeterMode_characters[i]);
             } else {
-               RichString_setChar(&bar, startPos + j, '|');
+               RichString_setChar(&bar, charPos, '|');
             }
          }
-      offset = nextOffset;
-   }
 
-   // ...then print the buffer.
-   offset = 0;
-   for (uint8_t i = 0; i < this->curItems; i++) {
+         offset += nCols;
+         charPos++;
+      }
+
       int attr = this->curAttributes ? this->curAttributes[i] : Meter_attributes(this)[i];
-      RichString_setAttrn(&bar, CRT_colors[attr], startPos + offset, blockSizes[i]);
-      RichString_printoffnVal(bar, y, x + offset, startPos + offset, blockSizes[i]);
-      offset += blockSizes[i];
+      RichString_setAttrn(&bar, CRT_colors[attr], startPos, charPos - startPos);
    }
-   if (offset < w) {
-      RichString_setAttrn(&bar, CRT_colors[BAR_SHADOW], startPos + offset, w - offset);
-      RichString_printoffnVal(bar, y, x + offset, startPos + offset, w - offset);
+
+   len = RichString_sizeVal(bar);
+   if (charPos < len) {
+      RichString_setAttrn(&bar, CRT_colors[BAR_SHADOW], charPos, len - charPos);
    }
+   RichString_printVal(bar, y, x);
 
    RichString_delete(&bar);