1 // Copyright (C) 2016 and later: Unicode, Inc. and others. 2 // License & terms of use: http://www.unicode.org/copyright.html 3 /* 4 ******************************************************************************* 5 * Copyright (C) 2012-2016, International Business Machines 6 * Corporation and others. All Rights Reserved. 7 ******************************************************************************* 8 * uitercollationiterator.h 9 * 10 * created on: 2012sep23 (from utf16collationiterator.h) 11 * created by: Markus W. Scherer 12 */ 13 14 #ifndef __UITERCOLLATIONITERATOR_H__ 15 #define __UITERCOLLATIONITERATOR_H__ 16 17 #include "unicode/utypes.h" 18 19 #if !UCONFIG_NO_COLLATION 20 21 #include "unicode/uiter.h" 22 #include "cmemory.h" 23 #include "collation.h" 24 #include "collationdata.h" 25 #include "collationiterator.h" 26 #include "normalizer2impl.h" 27 28 U_NAMESPACE_BEGIN 29 30 /** 31 * UCharIterator-based collation element and character iterator. 32 * Handles normalized text inline, with length or NUL-terminated. 33 * Unnormalized text is handled by a subclass. 34 */ 35 class U_I18N_API UIterCollationIterator : public CollationIterator { 36 public: UIterCollationIterator(const CollationData * d,UBool numeric,UCharIterator & ui)37 UIterCollationIterator(const CollationData *d, UBool numeric, UCharIterator &ui) 38 : CollationIterator(d, numeric), iter(ui) {} 39 40 virtual ~UIterCollationIterator(); 41 42 virtual void resetToOffset(int32_t newOffset); 43 44 virtual int32_t getOffset() const; 45 46 virtual UChar32 nextCodePoint(UErrorCode &errorCode); 47 48 virtual UChar32 previousCodePoint(UErrorCode &errorCode); 49 50 protected: 51 virtual uint32_t handleNextCE32(UChar32 &c, UErrorCode &errorCode); 52 53 virtual UChar handleGetTrailSurrogate(); 54 55 virtual void forwardNumCodePoints(int32_t num, UErrorCode &errorCode); 56 57 virtual void backwardNumCodePoints(int32_t num, UErrorCode &errorCode); 58 59 UCharIterator &iter; 60 }; 61 62 /** 63 * Incrementally checks the input text for FCD and normalizes where necessary. 64 */ 65 class U_I18N_API FCDUIterCollationIterator : public UIterCollationIterator { 66 public: FCDUIterCollationIterator(const CollationData * data,UBool numeric,UCharIterator & ui,int32_t startIndex)67 FCDUIterCollationIterator(const CollationData *data, UBool numeric, UCharIterator &ui, int32_t startIndex) 68 : UIterCollationIterator(data, numeric, ui), 69 state(ITER_CHECK_FWD), start(startIndex), 70 nfcImpl(data->nfcImpl) {} 71 72 virtual ~FCDUIterCollationIterator(); 73 74 virtual void resetToOffset(int32_t newOffset); 75 76 virtual int32_t getOffset() const; 77 78 virtual UChar32 nextCodePoint(UErrorCode &errorCode); 79 80 virtual UChar32 previousCodePoint(UErrorCode &errorCode); 81 82 protected: 83 virtual uint32_t handleNextCE32(UChar32 &c, UErrorCode &errorCode); 84 85 virtual UChar handleGetTrailSurrogate(); 86 87 virtual void forwardNumCodePoints(int32_t num, UErrorCode &errorCode); 88 89 virtual void backwardNumCodePoints(int32_t num, UErrorCode &errorCode); 90 91 private: 92 /** 93 * Switches to forward checking if possible. 94 */ 95 void switchToForward(); 96 97 /** 98 * Extends the FCD text segment forward or normalizes around pos. 99 * @return TRUE if success 100 */ 101 UBool nextSegment(UErrorCode &errorCode); 102 103 /** 104 * Switches to backward checking. 105 */ 106 void switchToBackward(); 107 108 /** 109 * Extends the FCD text segment backward or normalizes around pos. 110 * @return TRUE if success 111 */ 112 UBool previousSegment(UErrorCode &errorCode); 113 114 UBool normalize(const UnicodeString &s, UErrorCode &errorCode); 115 116 enum State { 117 /** 118 * The input text [start..(iter index)[ passes the FCD check. 119 * Moving forward checks incrementally. 120 * pos & limit are undefined. 121 */ 122 ITER_CHECK_FWD, 123 /** 124 * The input text [(iter index)..limit[ passes the FCD check. 125 * Moving backward checks incrementally. 126 * start & pos are undefined. 127 */ 128 ITER_CHECK_BWD, 129 /** 130 * The input text [start..limit[ passes the FCD check. 131 * pos tracks the current text index. 132 */ 133 ITER_IN_FCD_SEGMENT, 134 /** 135 * The input text [start..limit[ failed the FCD check and was normalized. 136 * pos tracks the current index in the normalized string. 137 * The text iterator is at the limit index. 138 */ 139 IN_NORM_ITER_AT_LIMIT, 140 /** 141 * The input text [start..limit[ failed the FCD check and was normalized. 142 * pos tracks the current index in the normalized string. 143 * The text iterator is at the start index. 144 */ 145 IN_NORM_ITER_AT_START 146 }; 147 148 State state; 149 150 int32_t start; 151 int32_t pos; 152 int32_t limit; 153 154 const Normalizer2Impl &nfcImpl; 155 UnicodeString normalized; 156 }; 157 158 U_NAMESPACE_END 159 160 #endif // !UCONFIG_NO_COLLATION 161 #endif // __UITERCOLLATIONITERATOR_H__ 162