package com.jetbrains.python.console.pydev; import java.util.Iterator; /** * This is a custom string buffer optimized for append(), clear() and deleteLast(). * * Basically it aims at being created once, being used for something, having clear() called and then reused * (ultimately providing minimum allocation/garbage collection overhead for that use-case). * * append() is optimizing by doing less checks (so, exceptions thrown may be uglier on invalid operations * and null is not checked for in the common case -- use appendObject if it may be null). * * clear() and deleteLast() only change the internal count and have almost zero overhead. * * Note that it's also not synchronized. * * @author Fabio */ public final class FastStringBuffer { /** * Holds the actual chars */ private char[] value; /** * Count for which chars are actually used */ private int count; /** * Initializes with a default initial size (128 chars) */ public FastStringBuffer() { this(128); } /** * An initial size can be specified (if available and given for no allocations it can be more efficient) */ public FastStringBuffer(int initialSize) { this.value = new char[initialSize]; this.count = 0; } /** * initializes from a string and the additional size for the buffer * * @param s string with the initial contents * @param additionalSize the additional size for the buffer */ public FastStringBuffer(String s, int additionalSize) { this.count = s.length(); value = new char[this.count + additionalSize]; s.getChars(0, this.count, value, 0); } /** * Appends a string to the buffer. Passing a null string will throw an exception. */ public FastStringBuffer append(String string) { int strLen = string.length(); int newCount = count + strLen; if (newCount > this.value.length) { resizeForMinimum(newCount); } string.getChars(0, strLen, value, this.count); this.count = newCount; return this; } /** * Resizes the internal buffer to have at least the minimum capacity passed (but may be more) */ private void resizeForMinimum(int minimumCapacity) { int newCapacity = (value.length + 1) * 2; if (minimumCapacity > newCapacity) { newCapacity = minimumCapacity; } char newValue[] = new char[newCapacity]; System.arraycopy(value, 0, newValue, 0, count); value = newValue; } /** * Appends an int to the buffer. */ public FastStringBuffer append(int n) { append(String.valueOf(n)); return this; } /** * Appends a char to the buffer. */ public FastStringBuffer append(char n) { if (count + 1 > value.length) { resizeForMinimum(count + 1); } value[count] = n; count++; return this; } /** * Appends a long to the buffer. */ public FastStringBuffer append(long n) { append(String.valueOf(n)); return this; } /** * Appends a boolean to the buffer. */ public FastStringBuffer append(boolean b) { append(String.valueOf(b)); return this; } /** * Appends an array of chars to the buffer. */ public FastStringBuffer append(char[] chars) { int newCount = count + chars.length; if (newCount > value.length) { resizeForMinimum(newCount); } System.arraycopy(chars, 0, value, count, chars.length); count = newCount; return this; } /** * Appends another buffer to this buffer. */ public FastStringBuffer append(FastStringBuffer other) { append(other.value, 0, other.count); return this; } /** * Appends an array of chars to this buffer, starting at the offset passed with the length determined. */ public FastStringBuffer append(char[] chars, int offset, int len) { int newCount = count + len; if (newCount > value.length) { resizeForMinimum(newCount); } System.arraycopy(chars, offset, value, count, len); count = newCount; return this; } /** * Reverses the contents on this buffer */ public FastStringBuffer reverse() { final int limit = count / 2; for (int i = 0; i < limit; ++i) { char c = value[i]; value[i] = value[count - i - 1]; value[count - i - 1] = c; } return this; } /** * Clears this buffer. */ public FastStringBuffer clear() { this.count = 0; return this; } /** * @return the length of this buffer */ public int length() { return this.count; } /** * @return a new string with the contents of this buffer. */ @Override public String toString() { return new String(value, 0, count); } /** * @return a new char array with the contents of this buffer. */ public char[] toCharArray() { char[] v = new char[count]; System.arraycopy(value, 0, v, 0, count); return v; } /** * Erases the last char in this buffer */ public void deleteLast() { if (this.count > 0) { this.count--; } } /** * @return the char given at a specific position of the buffer (no bounds check) */ public char charAt(int i) { return this.value[i]; } /** * Inserts a string at a given position in the buffer. */ public FastStringBuffer insert(int offset, String str) { int len = str.length(); int newCount = count + len; if (newCount > value.length) { resizeForMinimum(newCount); } System.arraycopy(value, offset, value, offset + len, count - offset); str.getChars(0, len, value, offset); count = newCount; return this; } /** * Inserts a char at a given position in the buffer. */ public FastStringBuffer insert(int offset, char c) { int newCount = count + 1; if (newCount > value.length) { resizeForMinimum(newCount); } System.arraycopy(value, offset, value, offset + 1, count - offset); value[offset] = c; count = newCount; return this; } /** * Appends object.toString(). If null, "null" is appended. */ public FastStringBuffer appendObject(Object object) { return append(object != null ? object.toString() : "null"); } /** * Sets the new size of this buffer (warning: use with care: no validation is done of the len passed) */ public void setCount(int newLen) { this.count = newLen; } public FastStringBuffer delete(int start, int end) { if (start < 0) throw new StringIndexOutOfBoundsException(start); if (end > count) end = count; if (start > end) throw new StringIndexOutOfBoundsException(); int len = end - start; if (len > 0) { System.arraycopy(value, start + len, value, start, count - end); count -= len; } return this; } public FastStringBuffer replace(int start, int end, String str) { if (start < 0) throw new StringIndexOutOfBoundsException(start); if (start > count) throw new StringIndexOutOfBoundsException("start > length()"); if (start > end) throw new StringIndexOutOfBoundsException("start > end"); if (end > count) end = count; if (end > count) end = count; int len = str.length(); int newCount = count + len - (end - start); if (newCount > value.length) { resizeForMinimum(newCount); } System.arraycopy(value, end, value, start + len, count - end); str.getChars(0, len, value, start); count = newCount; return this; } /** * Replaces all the occurrences of a string in this buffer for another string and returns the * altered version. */ public FastStringBuffer replaceAll(String replace, String with) { int replaceLen = replace.length(); int withLen = with.length(); int matchPos = 0; for (int i = 0; i < this.count; i++) { if(this.value[i] == replace.charAt(matchPos)){ matchPos ++; if(matchPos == replaceLen){ this.replace(i-(replaceLen-1), i+1, with); matchPos = 0; i = i-(replaceLen-withLen); } continue; }else{ matchPos = 0; } } return this; } public FastStringBuffer deleteCharAt(int index) { if ((index < 0) || (index >= count)) { throw new StringIndexOutOfBoundsException(index); } System.arraycopy(value, index + 1, value, index, count - index - 1); count--; return this; } public int indexOf(char c) { for(int i=0;i<this.count;i++){ if(c == this.value[i]){ return i; } } return -1; } public char firstChar() { return this.value[0]; } public char lastChar() { return this.value[this.count-1]; } public final static class BackwardCharIterator implements Iterable<Character>{ private int i; private FastStringBuffer fastStringBuffer; public BackwardCharIterator(FastStringBuffer fastStringBuffer) { this.fastStringBuffer = fastStringBuffer; i = fastStringBuffer.length(); } public Iterator<Character> iterator() { return new Iterator<Character>(){ public boolean hasNext() { return i > 0; } public Character next() { return fastStringBuffer.value[--i]; } public void remove() { throw new RuntimeException("Not implemented"); } }; } } public BackwardCharIterator reverseIterator() { return new BackwardCharIterator(this); } public void rightTrim() { char c; while(((c=this.lastChar()) == ' ' || c == '\t' )){ this.deleteLast(); } } public char deleteFirst(){ char ret = this.value[0]; this.deleteCharAt(0); return ret; } public FastStringBuffer appendN(String val, int n){ int min = count + (n*val.length()); if (min > value.length) { resizeForMinimum(min); } int strLen = val.length(); while (n-- > 0){ val.getChars(0, strLen, value, this.count); this.count += strLen; } return this; } public FastStringBuffer appendN(char val, int n){ if (count + n > value.length) { resizeForMinimum(count + n); } while (n-- > 0){ value[count] = val; count++; } return this; } public boolean endsWith(String string) { return startsWith(string, count - string.length()); } public boolean startsWith(String prefix) { return startsWith(prefix, 0); } public boolean startsWith(String prefix, int offset) { char ta[] = value; int to = offset; char pa[] = prefix.toCharArray(); int po = 0; int pc = pa.length; // Note: toffset might be near -1>>>1. if ((offset < 0) || (offset > count - pc)) { return false; } while (--pc >= 0) { if (ta[to++] != pa[po++]) { return false; } } return true; } public void setCharAt(int i, char c) { this.value[i] = c; } /** * Careful: it doesn't check anything. Just sets the internal length. */ public void setLength(int i) { this.count = i; } }