src/corelib/tools/qlocale_icu.cpp

   1 /****************************************************************************
   2 **
   3 ** Copyright (C) 2012 Nokia Corporation and/or its subsidiary(-ies).
   4 ** Contact: http://www.qt-project.org/
   5 **
   6 ** This file is part of the QtCore module of the Qt Toolkit.
   7 **
   8 ** $QT_BEGIN_LICENSE:LGPL$
   9 ** GNU Lesser General Public License Usage
  10 ** This file may be used under the terms of the GNU Lesser General Public
  11 ** License version 2.1 as published by the Free Software Foundation and
  12 ** appearing in the file LICENSE.LGPL included in the packaging of this
  13 ** file. Please review the following information to ensure the GNU Lesser
  14 ** General Public License version 2.1 requirements will be met:
  15 ** http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.
  16 **
  17 ** In addition, as a special exception, Nokia gives you certain additional
  18 ** rights. These rights are described in the Nokia Qt LGPL Exception
  19 ** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.
  20 **
  21 ** GNU General Public License Usage
  22 ** Alternatively, this file may be used under the terms of the GNU General
  23 ** Public License version 3.0 as published by the Free Software Foundation
  24 ** and appearing in the file LICENSE.GPL included in the packaging of this
  25 ** file. Please review the following information to ensure the GNU General
  26 ** Public License version 3.0 requirements will be met:
  27 ** http://www.gnu.org/copyleft/gpl.html.
  28 **
  29 ** Other Usage
  30 ** Alternatively, this file may be used in accordance with the terms and
  31 ** conditions contained in a signed written agreement between you and Nokia.
  32 **
  33 **
  34 **
  35 **
  36 **
  37 **
  38 ** $QT_END_LICENSE$
  39 **
  40 ****************************************************************************/
  41
  42 #include "qglobal.h"
  43 #include "qlibrary.h"
  44 #include "qdebug.h"
  45 #include "qlocale_p.h"
  46 #include "qmutex.h"
  47
  48 #include "unicode/uversion.h"
  49 #include "unicode/ucol.h"
  50 #include "unicode/uloc.h"
  51 #include "unicode/ustring.h"
  52
  53 QT_BEGIN_NAMESPACE
  54
  55 typedef int32_t (*Ptr_u_strToCase)(UChar *dest, int32_t destCapacity, const UChar *src, int32_t srcLength, const char *locale, UErrorCode *pErrorCode);
  56
  57 bool QIcu::strcoll(const QByteArray &localeID,
  58                    const QChar *source, int sourceLength, const QChar *target, int targetLength, int *result)
  59 {
  60     Q_ASSERT(result);
  61     Q_ASSERT(source);
  62     Q_ASSERT(target);
  63
  64     UErrorCode icuStatus = U_ZERO_ERROR;
  65     UCollator *collator = ucol_open(localeID, &icuStatus);
  66
  67     if (U_FAILURE((icuStatus)))
  68         return false;
  69
  70     *result = ucol_strcoll(collator,
  71                            reinterpret_cast<const UChar *>(source), int32_t(sourceLength),
  72                            reinterpret_cast<const UChar *>(target), int32_t(targetLength));
  73
  74     ucol_close(collator);
  75
  76     return true;
  77 }
  78
  79 // caseFunc can either be u_strToUpper or u_strToLower
  80 static bool qt_u_strToCase(const QString &str, QString *out, const char *localeID, Ptr_u_strToCase caseFunc)
  81 {
  82     Q_ASSERT(out);
  83
  84     int32_t size = str.size();
  85     size += size >> 2; // add 25% for possible expansions
  86     QString result(size, Qt::Uninitialized);
  87
  88     UErrorCode status = U_ZERO_ERROR;
  89
  90     size = caseFunc(reinterpret_cast<UChar *>(result.data()), result.size(),
  91             reinterpret_cast<const UChar *>(str.constData()), str.size(),
  92             localeID, &status);
  93
  94     if (U_FAILURE(status) && status != U_BUFFER_OVERFLOW_ERROR)
  95         return false;
  96
  97     if (size < result.size()) {
  98         result.resize(size);
  99     } else if (size > result.size()) {
 100         // the resulting string is larger than our source string
 101         result.resize(size);
 102
 103         status = U_ZERO_ERROR;
 104         size = caseFunc(reinterpret_cast<UChar *>(result.data()), result.size(),
 105             reinterpret_cast<const UChar *>(str.constData()), str.size(),
 106             localeID, &status);
 107
 108         if (U_FAILURE(status))
 109             return false;
 110
 111         // if the sizes don't match now, we give up.
 112         if (size != result.size())
 113             return false;
 114     }
 115
 116     *out = result;
 117     return true;
 118 }
 119
 120 QString QIcu::toUpper(const QByteArray &localeID, const QString &str, bool *ok)
 121 {
 122     QString out;
 123     bool err = qt_u_strToCase(str, &out, localeID, u_strToUpper);
 124     if (ok)
 125         *ok = err;
 126     return out;
 127 }
 128
 129 QString QIcu::toLower(const QByteArray &localeID, const QString &str, bool *ok)
 130 {
 131     QString out;
 132     bool err = qt_u_strToCase(str, &out, localeID, u_strToLower);
 133     if (ok)
 134         *ok = err;
 135     return out;
 136 }
 137
 138 QT_END_NAMESPACE