| /* |
| ******************************************************************************* |
| * Copyright (C) 2010-2015, International Business Machines |
| * Corporation and others. All Rights Reserved. |
| ******************************************************************************* |
| * file name: charstr.cpp |
| * encoding: US-ASCII |
| * tab size: 8 (not used) |
| * indentation:4 |
| * |
| * created on: 2010may19 |
| * created by: Markus W. Scherer |
| */ |
| |
| #include "unicode/utypes.h" |
| #include "charstr.h" |
| #include "cmemory.h" |
| #include "cstring.h" |
| #include "uinvchar.h" |
| |
| U_NAMESPACE_BEGIN |
| |
| CharString &CharString::copyFrom(const CharString &s, UErrorCode &errorCode) { |
| if(U_SUCCESS(errorCode) && this!=&s && ensureCapacity(s.len+1, 0, errorCode)) { |
| len=s.len; |
| uprv_memcpy(buffer.getAlias(), s.buffer.getAlias(), len+1); |
| } |
| return *this; |
| } |
| |
| int32_t CharString::lastIndexOf(char c) const { |
| for(int32_t i=len; i>0;) { |
| if(buffer[--i]==c) { |
| return i; |
| } |
| } |
| return -1; |
| } |
| |
| CharString &CharString::truncate(int32_t newLength) { |
| if(newLength<0) { |
| newLength=0; |
| } |
| if(newLength<len) { |
| buffer[len=newLength]=0; |
| } |
| return *this; |
| } |
| |
| CharString &CharString::append(char c, UErrorCode &errorCode) { |
| if(ensureCapacity(len+2, 0, errorCode)) { |
| buffer[len++]=c; |
| buffer[len]=0; |
| } |
| return *this; |
| } |
| |
| CharString &CharString::append(const char *s, int32_t sLength, UErrorCode &errorCode) { |
| if(U_FAILURE(errorCode)) { |
| return *this; |
| } |
| if(sLength<-1 || (s==NULL && sLength!=0)) { |
| errorCode=U_ILLEGAL_ARGUMENT_ERROR; |
| return *this; |
| } |
| if(sLength<0) { |
| sLength=uprv_strlen(s); |
| } |
| if(sLength>0) { |
| if(s==(buffer.getAlias()+len)) { |
| // The caller wrote into the getAppendBuffer(). |
| if(sLength>=(buffer.getCapacity()-len)) { |
| // The caller wrote too much. |
| errorCode=U_INTERNAL_PROGRAM_ERROR; |
| } else { |
| buffer[len+=sLength]=0; |
| } |
| } else if(buffer.getAlias()<=s && s<(buffer.getAlias()+len) && |
| sLength>=(buffer.getCapacity()-len) |
| ) { |
| // (Part of) this string is appended to itself which requires reallocation, |
| // so we have to make a copy of the substring and append that. |
| return append(CharString(s, sLength, errorCode), errorCode); |
| } else if(ensureCapacity(len+sLength+1, 0, errorCode)) { |
| uprv_memcpy(buffer.getAlias()+len, s, sLength); |
| buffer[len+=sLength]=0; |
| } |
| } |
| return *this; |
| } |
| |
| char *CharString::getAppendBuffer(int32_t minCapacity, |
| int32_t desiredCapacityHint, |
| int32_t &resultCapacity, |
| UErrorCode &errorCode) { |
| if(U_FAILURE(errorCode)) { |
| resultCapacity=0; |
| return NULL; |
| } |
| int32_t appendCapacity=buffer.getCapacity()-len-1; // -1 for NUL |
| if(appendCapacity>=minCapacity) { |
| resultCapacity=appendCapacity; |
| return buffer.getAlias()+len; |
| } |
| if(ensureCapacity(len+minCapacity+1, len+desiredCapacityHint+1, errorCode)) { |
| resultCapacity=buffer.getCapacity()-len-1; |
| return buffer.getAlias()+len; |
| } |
| resultCapacity=0; |
| return NULL; |
| } |
| |
| CharString &CharString::appendInvariantChars(const UnicodeString &s, UErrorCode &errorCode) { |
| if(U_FAILURE(errorCode)) { |
| return *this; |
| } |
| if (!uprv_isInvariantUnicodeString(s)) { |
| errorCode = U_INVARIANT_CONVERSION_ERROR; |
| return *this; |
| } |
| if(ensureCapacity(len+s.length()+1, 0, errorCode)) { |
| len+=s.extract(0, 0x7fffffff, buffer.getAlias()+len, buffer.getCapacity()-len, US_INV); |
| } |
| return *this; |
| } |
| |
| UBool CharString::ensureCapacity(int32_t capacity, |
| int32_t desiredCapacityHint, |
| UErrorCode &errorCode) { |
| if(U_FAILURE(errorCode)) { |
| return FALSE; |
| } |
| if(capacity>buffer.getCapacity()) { |
| if(desiredCapacityHint==0) { |
| desiredCapacityHint=capacity+buffer.getCapacity(); |
| } |
| if( (desiredCapacityHint<=capacity || buffer.resize(desiredCapacityHint, len+1)==NULL) && |
| buffer.resize(capacity, len+1)==NULL |
| ) { |
| errorCode=U_MEMORY_ALLOCATION_ERROR; |
| return FALSE; |
| } |
| } |
| return TRUE; |
| } |
| |
| CharString &CharString::appendPathPart(const StringPiece &s, UErrorCode &errorCode) { |
| if(U_FAILURE(errorCode)) { |
| return *this; |
| } |
| if(s.length()==0) { |
| return *this; |
| } |
| char c; |
| if(len>0 && (c=buffer[len-1])!=U_FILE_SEP_CHAR && c!=U_FILE_ALT_SEP_CHAR) { |
| append(U_FILE_SEP_CHAR, errorCode); |
| } |
| append(s, errorCode); |
| return *this; |
| } |
| |
| CharString &CharString::ensureEndsWithFileSeparator(UErrorCode &errorCode) { |
| char c; |
| if(U_SUCCESS(errorCode) && len>0 && |
| (c=buffer[len-1])!=U_FILE_SEP_CHAR && c!=U_FILE_ALT_SEP_CHAR) { |
| append(U_FILE_SEP_CHAR, errorCode); |
| } |
| return *this; |
| } |
| |
| U_NAMESPACE_END |