1 /*
2 htop - RichString.c
3 (C) 2004,2011 Hisham H. Muhammad
4 Released under the GNU GPLv2+, see the COPYING file
5 in the source distribution for its full text.
6 */
7
8 #include "RichString.h"
9
10 #include <ctype.h>
11 #include <stdlib.h>
12 #include <string.h>
13
14 #include "Macros.h"
15 #include "XUtils.h"
16
17
18 #define charBytes(n) (sizeof(CharType) * (n))
19
RichString_extendLen(RichString * this,int len)20 static void RichString_extendLen(RichString* this, int len) {
21 if (this->chlen <= RICHSTRING_MAXLEN) {
22 if (len > RICHSTRING_MAXLEN) {
23 this->chptr = xMalloc(charBytes(len + 1));
24 memcpy(this->chptr, this->chstr, charBytes(this->chlen));
25 }
26 } else {
27 if (len <= RICHSTRING_MAXLEN) {
28 memcpy(this->chstr, this->chptr, charBytes(len));
29 free(this->chptr);
30 this->chptr = this->chstr;
31 } else {
32 this->chptr = xRealloc(this->chptr, charBytes(len + 1));
33 }
34 }
35
36 RichString_setChar(this, len, 0);
37 this->chlen = len;
38 }
39
RichString_setLen(RichString * this,int len)40 static void RichString_setLen(RichString* this, int len) {
41 if (len < RICHSTRING_MAXLEN && this->chlen < RICHSTRING_MAXLEN) {
42 RichString_setChar(this, len, 0);
43 this->chlen = len;
44 } else {
45 RichString_extendLen(this, len);
46 }
47 }
48
RichString_rewind(RichString * this,int count)49 void RichString_rewind(RichString* this, int count) {
50 RichString_setLen(this, this->chlen - count);
51 }
52
53 #ifdef HAVE_LIBNCURSESW
54
RichString_writeFromWide(RichString * this,int attrs,const char * data_c,int from,int len)55 static inline int RichString_writeFromWide(RichString* this, int attrs, const char* data_c, int from, int len) {
56 wchar_t data[len + 1];
57 len = mbstowcs(data, data_c, len);
58 if (len <= 0)
59 return 0;
60
61 int newLen = from + len;
62 RichString_setLen(this, newLen);
63 for (int i = from, j = 0; i < newLen; i++, j++) {
64 this->chptr[i] = (CharType) { .attr = attrs & 0xffffff, .chars = { (iswprint(data[j]) ? data[j] : '?') } };
65 }
66
67 return len;
68 }
69
RichString_appendnWideColumns(RichString * this,int attrs,const char * data_c,int len,int * columns)70 int RichString_appendnWideColumns(RichString* this, int attrs, const char* data_c, int len, int* columns) {
71 wchar_t data[len + 1];
72 len = mbstowcs(data, data_c, len);
73 if (len <= 0)
74 return 0;
75
76 int from = this->chlen;
77 int newLen = from + len;
78 RichString_setLen(this, newLen);
79 int columnsWritten = 0;
80 int pos = from;
81 for (int j = 0; j < len; j++) {
82 wchar_t c = iswprint(data[j]) ? data[j] : '?';
83 int cwidth = wcwidth(c);
84 if (cwidth > *columns)
85 break;
86
87 *columns -= cwidth;
88 columnsWritten += cwidth;
89
90 this->chptr[pos] = (CharType) { .attr = attrs & 0xffffff, .chars = { c, '\0' } };
91 pos++;
92 }
93
94 RichString_setLen(this, pos);
95 *columns = columnsWritten;
96
97 return pos - from;
98 }
99
RichString_writeFromAscii(RichString * this,int attrs,const char * data,int from,int len)100 static inline int RichString_writeFromAscii(RichString* this, int attrs, const char* data, int from, int len) {
101 int newLen = from + len;
102 RichString_setLen(this, newLen);
103 for (int i = from, j = 0; i < newLen; i++, j++) {
104 this->chptr[i] = (CharType) { .attr = attrs & 0xffffff, .chars = { (isprint(data[j]) ? data[j] : '?') } };
105 }
106
107 return len;
108 }
109
RichString_setAttrn(RichString * this,int attrs,int start,int charcount)110 inline void RichString_setAttrn(RichString* this, int attrs, int start, int charcount) {
111 int end = CLAMP(start + charcount, 0, this->chlen);
112 for (int i = start; i < end; i++) {
113 this->chptr[i].attr = attrs;
114 }
115 }
116
RichString_appendChr(RichString * this,int attrs,char c,int count)117 void RichString_appendChr(RichString* this, int attrs, char c, int count) {
118 int from = this->chlen;
119 int newLen = from + count;
120 RichString_setLen(this, newLen);
121 for (int i = from; i < newLen; i++) {
122 this->chptr[i] = (CharType) { .attr = attrs, .chars = { c, 0 } };
123 }
124 }
125
RichString_findChar(const RichString * this,char c,int start)126 int RichString_findChar(const RichString* this, char c, int start) {
127 const wchar_t wc = btowc(c);
128 const cchar_t* ch = this->chptr + start;
129 for (int i = start; i < this->chlen; i++) {
130 if (ch->chars[0] == wc)
131 return i;
132 ch++;
133 }
134 return -1;
135 }
136
137 #else /* HAVE_LIBNCURSESW */
138
RichString_writeFromWide(RichString * this,int attrs,const char * data_c,int from,int len)139 static inline int RichString_writeFromWide(RichString* this, int attrs, const char* data_c, int from, int len) {
140 int newLen = from + len;
141 RichString_setLen(this, newLen);
142 for (int i = from, j = 0; i < newLen; i++, j++) {
143 this->chptr[i] = (((unsigned char)data_c[j]) >= 32 ? ((unsigned char)data_c[j]) : '?') | attrs;
144 }
145 this->chptr[newLen] = 0;
146
147 return len;
148 }
149
RichString_appendnWideColumns(RichString * this,int attrs,const char * data_c,int len,int * columns)150 int RichString_appendnWideColumns(RichString* this, int attrs, const char* data_c, int len, int* columns) {
151 int written = RichString_writeFromWide(this, attrs, data_c, this->chlen, MINIMUM(len, *columns));
152 *columns = written;
153 return written;
154 }
155
RichString_writeFromAscii(RichString * this,int attrs,const char * data_c,int from,int len)156 static inline int RichString_writeFromAscii(RichString* this, int attrs, const char* data_c, int from, int len) {
157 return RichString_writeFromWide(this, attrs, data_c, from, len);
158 }
159
RichString_setAttrn(RichString * this,int attrs,int start,int charcount)160 void RichString_setAttrn(RichString* this, int attrs, int start, int charcount) {
161 int end = CLAMP(start + charcount, 0, this->chlen);
162 for (int i = start; i < end; i++) {
163 this->chptr[i] = (this->chptr[i] & 0xff) | attrs;
164 }
165 }
166
RichString_appendChr(RichString * this,int attrs,char c,int count)167 void RichString_appendChr(RichString* this, int attrs, char c, int count) {
168 int from = this->chlen;
169 int newLen = from + count;
170 RichString_setLen(this, newLen);
171 for (int i = from; i < newLen; i++) {
172 this->chptr[i] = c | attrs;
173 }
174 }
175
RichString_findChar(const RichString * this,char c,int start)176 int RichString_findChar(const RichString* this, char c, int start) {
177 const chtype* ch = this->chptr + start;
178 for (int i = start; i < this->chlen; i++) {
179 if ((*ch & 0xff) == (chtype) c)
180 return i;
181 ch++;
182 }
183 return -1;
184 }
185
186 #endif /* HAVE_LIBNCURSESW */
187
RichString_delete(RichString * this)188 void RichString_delete(RichString* this) {
189 if (this->chlen > RICHSTRING_MAXLEN) {
190 free(this->chptr);
191 this->chptr = this->chstr;
192 }
193 }
194
RichString_setAttr(RichString * this,int attrs)195 void RichString_setAttr(RichString* this, int attrs) {
196 RichString_setAttrn(this, attrs, 0, this->chlen);
197 }
198
RichString_appendWide(RichString * this,int attrs,const char * data)199 int RichString_appendWide(RichString* this, int attrs, const char* data) {
200 return RichString_writeFromWide(this, attrs, data, this->chlen, strlen(data));
201 }
202
RichString_appendnWide(RichString * this,int attrs,const char * data,int len)203 int RichString_appendnWide(RichString* this, int attrs, const char* data, int len) {
204 return RichString_writeFromWide(this, attrs, data, this->chlen, len);
205 }
206
RichString_writeWide(RichString * this,int attrs,const char * data)207 int RichString_writeWide(RichString* this, int attrs, const char* data) {
208 return RichString_writeFromWide(this, attrs, data, 0, strlen(data));
209 }
210
RichString_appendAscii(RichString * this,int attrs,const char * data)211 int RichString_appendAscii(RichString* this, int attrs, const char* data) {
212 return RichString_writeFromAscii(this, attrs, data, this->chlen, strlen(data));
213 }
214
RichString_appendnAscii(RichString * this,int attrs,const char * data,int len)215 int RichString_appendnAscii(RichString* this, int attrs, const char* data, int len) {
216 return RichString_writeFromAscii(this, attrs, data, this->chlen, len);
217 }
218
RichString_writeAscii(RichString * this,int attrs,const char * data)219 int RichString_writeAscii(RichString* this, int attrs, const char* data) {
220 return RichString_writeFromAscii(this, attrs, data, 0, strlen(data));
221 }
222