1 files changed, 191 insertions, 0 deletions
diff --git a/src/String.h b/src/String.h
new file mode 100644
index 0000000..e3daa1d
--- /dev/null
+++ b/src/String.h
@@ -0,0 +1,191 @@
+/* vim:set et ts=4 sts=4:
+ *
+ * libpyzy - The Chinese PinYin and Bopomofo conversion library.
+ *
+ * Copyright (c) 2008-2010 Peng Huang <shawn.p.huang@gmail.com>
+ *
+ * This library is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU Lesser General Public License as
+ * published by the Free Software Foundation; either version 2.1 of the
+ * License, or (at your option) any later version.
+ *
+ * This library is distributed in the hope that it will be useful, but
+ * WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this library; if not, write to the Free Software
+ * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
+ * USA
+ */
+#ifndef __PYZY_STRING_H_
+#define __PYZY_STRING_H_
+
+#include <glib.h>
+#include <stdarg.h>
+#include <string>
+
+#include "Util.h"
+
+namespace PyZy {
+
+class String : public std::string {
+public:
+    String () : std::string () { }
+    String (const char *str) : std::string (str) { }
+    String (const std::string &str) : std::string (str) { }
+    String (size_t len) : std::string () { reserve (len); }
+
+    String & printf (const char *fmt, ...)
+    {
+        char *str;
+        va_list args;
+
+        va_start (args, fmt);
+        str = g_strdup_vprintf (fmt, args);
+        va_end (args);
+
+        assign (str);
+        g_free (str);
+        return *this;
+    }
+
+    String & appendPrintf (const char *fmt, ...)
+    {
+        char *str;
+        va_list args;
+
+        va_start (args, fmt);
+        str = g_strdup_vprintf (fmt, args);
+        va_end (args);
+
+        append (str);
+        g_free (str);
+
+        return *this;
+    }
+
+    String & appendUnichar (unichar ch)
+    {
+        char str[12];
+        size_t len;
+        len = g_unichar_to_utf8 (ch, str);
+        str[len] = 0;
+        append (str);
+        return *this;
+    }
+
+    String & insert (size_t i, char ch)
+    {
+        std::string::insert (i, 1, ch);
+        return *this;
+    }
+
+    String & truncate (size_t len)
+    {
+        erase(len);
+        return *this;
+    }
+
+    String & replace (const char *pattern, const char *str)
+    {
+        String result;
+        String::size_type pos = 0;
+        String::size_type pos_before = 0;
+        const String::size_type length = std::string (pattern).size ();
+
+        while ((pos = this->find (pattern, pos)) != String::npos) {
+            result.append (*this, pos_before, pos - pos_before);
+            result.append (str);
+            pos += length;
+            pos_before = pos;
+        }
+        result.append (*this, pos_before, this->size () - pos_before);
+        this->assign (result);
+        return *this;
+    }
+
+    size_t utf8Length (void) const
+    {
+        return g_utf8_strlen (c_str(), -1);
+    }
+
+    String & operator<< (int i)
+    {
+        return appendPrintf ("%d", i);
+    }
+
+    String & operator<< (unsigned int i)
+    {
+        return appendPrintf ("%u", i);
+    }
+
+    String & operator<< (unsigned long i)
+    {
+        return appendPrintf ("%lu", i);
+    }
+
+    String & operator<< (const char ch)
+    {
+        append (1, ch);
+        return *this;
+    }
+
+    String & operator<< (const char *str)
+    {
+        append (str);
+        return *this;
+    }
+
+    String & operator<< (const unichar *wstr)
+    {
+        char *str;
+        GError *error;
+        str = g_ucs4_to_utf8 (wstr, -1, NULL, NULL, &error);
+        if (str == NULL) {
+            g_warning ("convert ucs4 to utf8 failed: %s", error->message);
+            g_error_free (error);
+        }
+        else {
+            append (str);
+            g_free (str);
+        }
+        return *this;
+    }
+
+    char operator[] (size_t i)
+    {
+        return std::string::operator[] (i);
+    }
+
+    String & operator<< (const std::string &str)
+    {
+        return operator<< (str.c_str ());
+    }
+
+    String & operator<< (const String &str)
+    {
+        return operator<< ((const char *)str);
+    }
+
+    String & operator= (const char * str)
+    {
+        assign (str);
+        return *this;
+    }
+
+    operator const char *(void) const
+    {
+        return this->c_str ();
+    }
+
+    operator bool (void) const
+    {
+        return ! empty ();
+    }
+};
+
+};  // namespace PyZy
+
+#endif  // __PYZY_STRING_H_