]> git.lyx.org Git - lyx.git/blob - src/support/unicode.cpp
abdel likes short code
[lyx.git] / src / support / unicode.cpp
1 /**
2  * \file unicode.cpp
3  * This file is part of LyX, the document processor.
4  * Licence details can be found in the file COPYING.
5  *
6  * \author Lars Gullik Bjønnes
7  *
8  * Full author contact details are available in file CREDITS.
9  *
10  * A collection of unicode conversion functions, using iconv.
11  */
12
13 #include <config.h>
14
15 #include "support/unicode.h"
16 #include "support/debug.h"
17
18 #include <iconv.h>
19
20 #include <boost/cstdint.hpp>
21
22 #include <cerrno>
23 #include <iomanip>
24 #include <ostream>
25 #include <map>
26 #include <string>
27
28 using namespace std;
29
30 namespace {
31
32 #ifdef WORDS_BIGENDIAN
33         char const * utf16_codeset = "UTF16-BE";
34 #else
35         char const * utf16_codeset = "UTF16-LE";
36 #endif
37
38 }
39
40
41 namespace lyx {
42
43 #ifdef WORDS_BIGENDIAN
44         char const * ucs4_codeset = "UCS-4BE";
45 #else
46         char const * ucs4_codeset = "UCS-4LE";
47 #endif
48
49 static const iconv_t invalid_cd = (iconv_t)(-1);
50
51
52 struct IconvProcessor::Impl
53 {
54         Impl(string const & to, string const & from)
55                 : cd(invalid_cd), tocode_(to), fromcode_(from)
56         {}
57
58         ~Impl()
59         {
60                 if (cd != invalid_cd && iconv_close(cd) == -1)
61                                 LYXERR0("Error returned from iconv_close(" << errno << ")");
62         }
63
64         iconv_t cd;
65         string tocode_;
66         string fromcode_;
67 };
68
69
70 IconvProcessor::IconvProcessor(char const * tocode, char const * fromcode)
71         : pimpl_(new IconvProcessor::Impl(tocode, fromcode))
72 {
73 }
74
75
76 IconvProcessor::IconvProcessor(IconvProcessor const & other)
77         : pimpl_(new IconvProcessor::Impl(other.pimpl_->tocode_, other.pimpl_->fromcode_))
78 {
79 }
80
81
82 IconvProcessor::~IconvProcessor()
83 {
84         delete pimpl_;
85 }
86
87
88 void IconvProcessor::operator=(IconvProcessor const & other)
89 {
90         if (&other != this)
91                 pimpl_ = new Impl(other.pimpl_->tocode_, other.pimpl_->fromcode_);
92 }
93
94
95 bool IconvProcessor::init()
96 {
97         if (pimpl_->cd != invalid_cd)
98                 return true;
99
100         pimpl_->cd = iconv_open(pimpl_->tocode_.c_str(), pimpl_->fromcode_.c_str());
101         if (pimpl_->cd != invalid_cd)
102                 return true;
103
104         lyxerr << "Error returned from iconv_open" << endl;
105         switch (errno) {
106                 case EINVAL:
107                         lyxerr << "EINVAL The conversion from " << pimpl_->fromcode_
108                                 << " to " << pimpl_->tocode_
109                                 << " is not supported by the implementation."
110                                 << endl;
111                         break;
112                 default:
113                         lyxerr << "\tSome other error: " << errno << endl;
114                         break;
115         }
116         return false;
117 }
118
119
120 int IconvProcessor::convert(char const * buf, size_t buflen,
121                 char * outbuf, size_t maxoutsize)
122 {
123         if (buflen == 0)
124                 return 0;
125
126         if (pimpl_->cd == invalid_cd) {
127                 if (!init())
128                         return -1;
129         }
130
131         char ICONV_CONST * inbuf = const_cast<char ICONV_CONST *>(buf);
132         size_t inbytesleft = buflen;
133         size_t outbytesleft = maxoutsize;
134
135         int res = iconv(pimpl_->cd, &inbuf, &inbytesleft, &outbuf, &outbytesleft);
136
137         // flush out remaining data. This is needed because iconv sometimes
138         // holds back chars in the stream, waiting for a combination character
139         // (see e.g. http://sources.redhat.com/bugzilla/show_bug.cgi?id=1124)
140         iconv(pimpl_->cd, NULL, NULL, &outbuf, &outbytesleft);
141
142         //lyxerr << dec;
143         //lyxerr << "Inbytesleft: " << inbytesleft << endl;
144         //lyxerr << "Outbytesleft: " << outbytesleft << endl;
145
146         if (res != -1)
147                 // Everything went well.
148                 return maxoutsize - outbytesleft;
149
150         // There are some errors in the conversion
151         lyxerr << "Error returned from iconv" << endl;
152         switch (errno) {
153                 case E2BIG:
154                         lyxerr << "E2BIG  There is not sufficient room at *outbuf." << endl;
155                         break;
156                 case EILSEQ:
157                         lyxerr << "EILSEQ An invalid multibyte sequence"
158                                 << " has been encountered in the input.\n"
159                                 << "When converting from " << pimpl_->fromcode_
160                                 << " to " << pimpl_->tocode_ << ".\n";
161                         lyxerr << "Input:" << hex;
162                         for (size_t i = 0; i < buflen; ++i) {
163                                 // char may be signed, avoid output of
164                                 // something like 0xffffffc2
165                                 boost::uint32_t const b =
166                                         *reinterpret_cast<unsigned char const *>(buf + i);
167                                 lyxerr << " 0x" << (unsigned int)b;
168                         }
169                         lyxerr << dec << endl;
170                         break;
171                 case EINVAL:
172                         lyxerr << "EINVAL An incomplete multibyte sequence"
173                                 << " has been encountered in the input.\n"
174                                 << "When converting from " << pimpl_->fromcode_
175                                 << " to " << pimpl_->tocode_ << ".\n";
176                         lyxerr << "Input:" << hex;
177                         for (size_t i = 0; i < buflen; ++i) {
178                                 // char may be signed, avoid output of
179                                 // something like 0xffffffc2
180                                 boost::uint32_t const b =
181                                         *reinterpret_cast<unsigned char const *>(buf + i);
182                                 lyxerr << " 0x" << (unsigned int)b;
183                         }
184                         lyxerr << dec << endl;
185                         break;
186                 default:
187                         lyxerr << "\tSome other error: " << errno << endl;
188                         break;
189         }
190         // We got an error so we close down the conversion engine
191         if (iconv_close(pimpl_->cd) == -1) {
192                 lyxerr << "Error returned from iconv_close("
193                         << errno << ")" << endl;
194         }
195         pimpl_->cd = invalid_cd;
196         return -1;
197 }
198
199
200 namespace {
201
202
203 template<typename RetType, typename InType>
204 vector<RetType>
205 iconv_convert(IconvProcessor & processor, InType const * buf, size_t buflen)
206 {
207         if (buflen == 0)
208                 return vector<RetType>();
209
210         char const * inbuf = reinterpret_cast<char const *>(buf);
211         size_t inbytesleft = buflen * sizeof(InType);
212
213         size_t const outsize = 32768;
214         static char out[outsize];
215         char * outbuf = out;
216
217         int bytes = processor.convert(inbuf, inbytesleft, outbuf, outsize);
218         if (bytes <= 0)
219                 // Conversion failed
220                 // FIXME Maybe throw an exception and handle that in the caller?
221                 return vector<RetType>();
222
223         RetType const * tmp = reinterpret_cast<RetType const *>(out);
224         return vector<RetType>(tmp, tmp + bytes / sizeof(RetType));
225 }
226
227 } // anon namespace
228
229
230 vector<char_type> utf8_to_ucs4(vector<char> const & utf8str)
231 {
232         if (utf8str.empty())
233                 return vector<char_type>();
234
235         return utf8_to_ucs4(&utf8str[0], utf8str.size());
236 }
237
238
239 vector<char_type>
240 utf8_to_ucs4(char const * utf8str, size_t ls)
241 {
242         static IconvProcessor processor(ucs4_codeset, "UTF-8");
243         return iconv_convert<char_type>(processor, utf8str, ls);
244 }
245
246
247 vector<char_type>
248 utf16_to_ucs4(unsigned short const * s, size_t ls)
249 {
250         static IconvProcessor processor(ucs4_codeset, utf16_codeset);
251         return iconv_convert<char_type>(processor, s, ls);
252 }
253
254
255 vector<unsigned short>
256 ucs4_to_utf16(char_type const * s, size_t ls)
257 {
258         static IconvProcessor processor(utf16_codeset, ucs4_codeset);
259         return iconv_convert<unsigned short>(processor, s, ls);
260 }
261
262
263 vector<char>
264 ucs4_to_utf8(char_type c)
265 {
266         static IconvProcessor processor("UTF-8", ucs4_codeset);
267         return iconv_convert<char>(processor, &c, 1);
268 }
269
270
271 vector<char>
272 ucs4_to_utf8(vector<char_type> const & ucs4str)
273 {
274         if (ucs4str.empty())
275                 return vector<char>();
276
277         return ucs4_to_utf8(&ucs4str[0], ucs4str.size());
278 }
279
280
281 vector<char>
282 ucs4_to_utf8(char_type const * ucs4str, size_t ls)
283 {
284         static IconvProcessor processor("UTF-8", ucs4_codeset);
285         return iconv_convert<char>(processor, ucs4str, ls);
286 }
287
288
289 vector<char_type>
290 eightbit_to_ucs4(char const * s, size_t ls, string const & encoding)
291 {
292         static map<string, IconvProcessor> processors;
293         if (processors.find(encoding) == processors.end()) {
294                 IconvProcessor processor(ucs4_codeset, encoding.c_str());
295                 processors.insert(make_pair(encoding, processor));
296         }
297         return iconv_convert<char_type>(processors[encoding], s, ls);
298 }
299
300
301 vector<char>
302 ucs4_to_eightbit(char_type const * ucs4str, size_t ls, string const & encoding)
303 {
304         static map<string, IconvProcessor> processors;
305         if (processors.find(encoding) == processors.end()) {
306                 IconvProcessor processor(encoding.c_str(), ucs4_codeset);
307                 processors.insert(make_pair(encoding, processor));
308         }
309         return iconv_convert<char>(processors[encoding], ucs4str, ls);
310 }
311
312
313 char ucs4_to_eightbit(char_type ucs4, string const & encoding)
314 {
315         static map<string, IconvProcessor> processors;
316         map<string, IconvProcessor>::iterator it = processors.find(encoding);
317         if (it == processors.end()) {
318                 IconvProcessor processor(encoding.c_str(), ucs4_codeset);
319                 it = processors.insert(make_pair(encoding, processor)).first;
320         }
321
322         char out;
323         int const bytes = it->second.convert((char *)(&ucs4), 4, &out, 1);
324         if (bytes > 0)
325                 return out;
326         return 0;
327 }
328
329
330 void ucs4_to_multibytes(char_type ucs4, vector<char> & out,
331         string const & encoding)
332 {
333         static map<string, IconvProcessor> processors;
334         map<string, IconvProcessor>::iterator it = processors.find(encoding);
335         if (it == processors.end()) {
336                 IconvProcessor processor(encoding.c_str(), ucs4_codeset);
337                 it = processors.insert(make_pair(encoding, processor)).first;
338         }
339
340         out.resize(4);
341         int bytes = it->second.convert((char *)(&ucs4), 4, &out[0], 4);
342         if (bytes > 0)
343                 out.resize(bytes);
344         else
345                 out.clear();
346 }
347
348 } // namespace lyx