| /* |
| ******************************************************************************* |
| * Copyright (C) 2010-2011, International Business Machines |
| * Corporation and others. All Rights Reserved. |
| ******************************************************************************* |
| * file name: ucharstriebuilder.h |
| * encoding: US-ASCII |
| * tab size: 8 (not used) |
| * indentation:4 |
| * |
| * created on: 2010nov14 |
| * created by: Markus W. Scherer |
| */ |
| |
| #ifndef __UCHARSTRIEBUILDER_H__ |
| #define __UCHARSTRIEBUILDER_H__ |
| |
| #include "unicode/utypes.h" |
| #include "unicode/stringtriebuilder.h" |
| #include "unicode/ucharstrie.h" |
| #include "unicode/unistr.h" |
| |
| U_NAMESPACE_BEGIN |
| |
| class UCharsTrieElement; |
| |
| /** |
| * Builder class for UCharsTrie. |
| * |
| * This class is not intended for public subclassing. |
| * @draft ICU 4.8 |
| */ |
| class U_COMMON_API UCharsTrieBuilder : public StringTrieBuilder { |
| public: |
| /** |
| * Constructs an empty builder. |
| * @param errorCode Standard ICU error code. |
| * @draft ICU 4.8 |
| */ |
| UCharsTrieBuilder(UErrorCode &errorCode); |
| |
| /** |
| * Destructor. |
| * @draft ICU 4.8 |
| */ |
| virtual ~UCharsTrieBuilder(); |
| |
| /** |
| * Adds a (string, value) pair. |
| * The string must be unique. |
| * The string contents will be copied; the builder does not keep |
| * a reference to the input UnicodeString or its buffer. |
| * @param s The input string. |
| * @param value The value associated with this string. |
| * @param errorCode Standard ICU error code. Its input value must |
| * pass the U_SUCCESS() test, or else the function returns |
| * immediately. Check for U_FAILURE() on output or use with |
| * function chaining. (See User Guide for details.) |
| * @return *this |
| * @draft ICU 4.8 |
| */ |
| UCharsTrieBuilder &add(const UnicodeString &s, int32_t value, UErrorCode &errorCode); |
| |
| /** |
| * Builds a UCharsTrie for the add()ed data. |
| * Once built, no further data can be add()ed until clear() is called. |
| * |
| * This method passes ownership of the builder's internal result array to the new trie object. |
| * Another call to any build() variant will re-serialize the trie. |
| * After clear() has been called, a new array will be used as well. |
| * @param buildOption Build option, see UStringTrieBuildOption. |
| * @param errorCode Standard ICU error code. Its input value must |
| * pass the U_SUCCESS() test, or else the function returns |
| * immediately. Check for U_FAILURE() on output or use with |
| * function chaining. (See User Guide for details.) |
| * @return A new UCharsTrie for the add()ed data. |
| * @draft ICU 4.8 |
| */ |
| UCharsTrie *build(UStringTrieBuildOption buildOption, UErrorCode &errorCode); |
| |
| /** |
| * Builds a UCharsTrie for the add()ed data and UChar-serializes it. |
| * Once built, no further data can be add()ed until clear() is called. |
| * |
| * Multiple calls to buildUnicodeString() set the UnicodeStrings to the |
| * builder's same UChar array, without rebuilding. |
| * If buildUnicodeString() is called after build(), the trie will be |
| * re-serialized into a new array. |
| * If build() is called after buildUnicodeString(), the trie object will become |
| * the owner of the previously returned array. |
| * After clear() has been called, a new array will be used as well. |
| * @param buildOption Build option, see UStringTrieBuildOption. |
| * @param result A UnicodeString which will be set to the UChar-serialized |
| * UCharsTrie for the add()ed data. |
| * @param errorCode Standard ICU error code. Its input value must |
| * pass the U_SUCCESS() test, or else the function returns |
| * immediately. Check for U_FAILURE() on output or use with |
| * function chaining. (See User Guide for details.) |
| * @return result |
| * @draft ICU 4.8 |
| */ |
| UnicodeString &buildUnicodeString(UStringTrieBuildOption buildOption, UnicodeString &result, |
| UErrorCode &errorCode); |
| |
| /** |
| * Removes all (string, value) pairs. |
| * New data can then be add()ed and a new trie can be built. |
| * @return *this |
| * @draft ICU 4.8 |
| */ |
| UCharsTrieBuilder &clear() { |
| strings.remove(); |
| elementsLength=0; |
| ucharsLength=0; |
| return *this; |
| } |
| |
| private: |
| UCharsTrieBuilder(const UCharsTrieBuilder &other); // no copy constructor |
| UCharsTrieBuilder &operator=(const UCharsTrieBuilder &other); // no assignment operator |
| |
| void buildUChars(UStringTrieBuildOption buildOption, UErrorCode &errorCode); |
| |
| virtual int32_t getElementStringLength(int32_t i) const; |
| virtual UChar getElementUnit(int32_t i, int32_t unitIndex) const; |
| virtual int32_t getElementValue(int32_t i) const; |
| |
| virtual int32_t getLimitOfLinearMatch(int32_t first, int32_t last, int32_t unitIndex) const; |
| |
| virtual int32_t countElementUnits(int32_t start, int32_t limit, int32_t unitIndex) const; |
| virtual int32_t skipElementsBySomeUnits(int32_t i, int32_t unitIndex, int32_t count) const; |
| virtual int32_t indexOfElementWithNextUnit(int32_t i, int32_t unitIndex, UChar unit) const; |
| |
| virtual UBool matchNodesCanHaveValues() const { return TRUE; } |
| |
| virtual int32_t getMaxBranchLinearSubNodeLength() const { return UCharsTrie::kMaxBranchLinearSubNodeLength; } |
| virtual int32_t getMinLinearMatch() const { return UCharsTrie::kMinLinearMatch; } |
| virtual int32_t getMaxLinearMatchLength() const { return UCharsTrie::kMaxLinearMatchLength; } |
| |
| class UCTLinearMatchNode : public LinearMatchNode { |
| public: |
| UCTLinearMatchNode(const UChar *units, int32_t len, Node *nextNode); |
| virtual UBool operator==(const Node &other) const; |
| virtual void write(StringTrieBuilder &builder); |
| private: |
| const UChar *s; |
| }; |
| |
| virtual Node *createLinearMatchNode(int32_t i, int32_t unitIndex, int32_t length, |
| Node *nextNode) const; |
| |
| UBool ensureCapacity(int32_t length); |
| virtual int32_t write(int32_t unit); |
| int32_t write(const UChar *s, int32_t length); |
| virtual int32_t writeElementUnits(int32_t i, int32_t unitIndex, int32_t length); |
| virtual int32_t writeValueAndFinal(int32_t i, UBool isFinal); |
| virtual int32_t writeValueAndType(UBool hasValue, int32_t value, int32_t node); |
| virtual int32_t writeDeltaTo(int32_t jumpTarget); |
| |
| UnicodeString strings; |
| UCharsTrieElement *elements; |
| int32_t elementsCapacity; |
| int32_t elementsLength; |
| |
| // UChar serialization of the trie. |
| // Grows from the back: ucharsLength measures from the end of the buffer! |
| UChar *uchars; |
| int32_t ucharsCapacity; |
| int32_t ucharsLength; |
| }; |
| |
| U_NAMESPACE_END |
| |
| #endif // __UCHARSTRIEBUILDER_H__ |