| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You under the Apache License, Version 2.0 |
| * (the "License"); you may not use this file except in compliance with |
| * the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| package java.nio; |
| |
| import libcore.io.SizeOf; |
| import libcore.io.Memory; |
| |
| /** |
| * DirectByteBuffer, ReadWriteDirectByteBuffer and ReadOnlyDirectByteBuffer |
| * compose the implementation of platform memory based byte buffers. |
| * <p> |
| * ReadWriteDirectByteBuffer extends DirectByteBuffer with all the write |
| * methods. |
| * </p> |
| * <p> |
| * This class is marked final for runtime performance. |
| * </p> |
| */ |
| final class ReadWriteDirectByteBuffer extends DirectByteBuffer { |
| static ReadWriteDirectByteBuffer copy(DirectByteBuffer other, int markOfOther) { |
| ReadWriteDirectByteBuffer buf = |
| new ReadWriteDirectByteBuffer(other.block, other.capacity(), other.offset); |
| buf.limit = other.limit; |
| buf.position = other.position(); |
| buf.mark = markOfOther; |
| return buf; |
| } |
| |
| // Used by ByteBuffer.allocateDirect. |
| ReadWriteDirectByteBuffer(int capacity) { |
| super(MemoryBlock.allocate(capacity), capacity, 0); |
| } |
| |
| // Used by the JNI NewDirectByteBuffer function. |
| ReadWriteDirectByteBuffer(int address, int capacity) { |
| super(MemoryBlock.wrapFromJni(address, capacity), capacity, 0); |
| } |
| |
| ReadWriteDirectByteBuffer(MemoryBlock block, int capacity, int offset) { |
| super(block, capacity, offset); |
| } |
| |
| @Override |
| public ByteBuffer asReadOnlyBuffer() { |
| return ReadOnlyDirectByteBuffer.copy(this, mark); |
| } |
| |
| @Override |
| public ByteBuffer compact() { |
| Memory.memmove(this, 0, this, position, remaining()); |
| position = limit - position; |
| limit = capacity; |
| mark = UNSET_MARK; |
| return this; |
| } |
| |
| @Override |
| public ByteBuffer duplicate() { |
| return copy(this, mark); |
| } |
| |
| @Override |
| public boolean isReadOnly() { |
| return false; |
| } |
| |
| @Override |
| public ByteBuffer put(byte value) { |
| if (position == limit) { |
| throw new BufferOverflowException(); |
| } |
| this.block.pokeByte(offset + position++, value); |
| return this; |
| } |
| |
| @Override |
| public ByteBuffer put(int index, byte value) { |
| checkIndex(index); |
| this.block.pokeByte(offset + index, value); |
| return this; |
| } |
| |
| @Override |
| public ByteBuffer put(byte[] src, int srcOffset, int byteCount) { |
| checkPutBounds(1, src.length, srcOffset, byteCount); |
| this.block.pokeByteArray(offset + position, src, srcOffset, byteCount); |
| position += byteCount; |
| return this; |
| } |
| |
| final void put(char[] src, int srcOffset, int charCount) { |
| int byteCount = checkPutBounds(SizeOf.CHAR, src.length, srcOffset, charCount); |
| this.block.pokeCharArray(offset + position, src, srcOffset, charCount, order.needsSwap); |
| position += byteCount; |
| } |
| |
| final void put(double[] src, int srcOffset, int doubleCount) { |
| int byteCount = checkPutBounds(SizeOf.DOUBLE, src.length, srcOffset, doubleCount); |
| this.block.pokeDoubleArray(offset + position, src, srcOffset, doubleCount, order.needsSwap); |
| position += byteCount; |
| } |
| |
| final void put(float[] src, int srcOffset, int floatCount) { |
| int byteCount = checkPutBounds(SizeOf.FLOAT, src.length, srcOffset, floatCount); |
| this.block.pokeFloatArray(offset + position, src, srcOffset, floatCount, order.needsSwap); |
| position += byteCount; |
| } |
| |
| final void put(int[] src, int srcOffset, int intCount) { |
| int byteCount = checkPutBounds(SizeOf.INT, src.length, srcOffset, intCount); |
| this.block.pokeIntArray(offset + position, src, srcOffset, intCount, order.needsSwap); |
| position += byteCount; |
| } |
| |
| final void put(long[] src, int srcOffset, int longCount) { |
| int byteCount = checkPutBounds(SizeOf.LONG, src.length, srcOffset, longCount); |
| this.block.pokeLongArray(offset + position, src, srcOffset, longCount, order.needsSwap); |
| position += byteCount; |
| } |
| |
| final void put(short[] src, int srcOffset, int shortCount) { |
| int byteCount = checkPutBounds(SizeOf.SHORT, src.length, srcOffset, shortCount); |
| this.block.pokeShortArray(offset + position, src, srcOffset, shortCount, order.needsSwap); |
| position += byteCount; |
| } |
| |
| @Override |
| public ByteBuffer putChar(char value) { |
| int newPosition = position + SizeOf.CHAR; |
| if (newPosition > limit) { |
| throw new BufferOverflowException(); |
| } |
| this.block.pokeShort(offset + position, (short) value, order); |
| position = newPosition; |
| return this; |
| } |
| |
| @Override |
| public ByteBuffer putChar(int index, char value) { |
| checkIndex(index, SizeOf.CHAR); |
| this.block.pokeShort(offset + index, (short) value, order); |
| return this; |
| } |
| |
| @Override |
| public ByteBuffer putDouble(double value) { |
| int newPosition = position + SizeOf.DOUBLE; |
| if (newPosition > limit) { |
| throw new BufferOverflowException(); |
| } |
| this.block.pokeLong(offset + position, Double.doubleToRawLongBits(value), order); |
| position = newPosition; |
| return this; |
| } |
| |
| @Override |
| public ByteBuffer putDouble(int index, double value) { |
| checkIndex(index, SizeOf.DOUBLE); |
| this.block.pokeLong(offset + index, Double.doubleToRawLongBits(value), order); |
| return this; |
| } |
| |
| @Override |
| public ByteBuffer putFloat(float value) { |
| int newPosition = position + SizeOf.FLOAT; |
| if (newPosition > limit) { |
| throw new BufferOverflowException(); |
| } |
| this.block.pokeInt(offset + position, Float.floatToRawIntBits(value), order); |
| position = newPosition; |
| return this; |
| } |
| |
| @Override |
| public ByteBuffer putFloat(int index, float value) { |
| checkIndex(index, SizeOf.FLOAT); |
| this.block.pokeInt(offset + index, Float.floatToRawIntBits(value), order); |
| return this; |
| } |
| |
| @Override |
| public ByteBuffer putInt(int value) { |
| int newPosition = position + SizeOf.INT; |
| if (newPosition > limit) { |
| throw new BufferOverflowException(); |
| } |
| this.block.pokeInt(offset + position, value, order); |
| position = newPosition; |
| return this; |
| } |
| |
| @Override |
| public ByteBuffer putInt(int index, int value) { |
| checkIndex(index, SizeOf.INT); |
| this.block.pokeInt(offset + index, value, order); |
| return this; |
| } |
| |
| @Override |
| public ByteBuffer putLong(long value) { |
| int newPosition = position + SizeOf.LONG; |
| if (newPosition > limit) { |
| throw new BufferOverflowException(); |
| } |
| this.block.pokeLong(offset + position, value, order); |
| position = newPosition; |
| return this; |
| } |
| |
| @Override |
| public ByteBuffer putLong(int index, long value) { |
| checkIndex(index, SizeOf.LONG); |
| this.block.pokeLong(offset + index, value, order); |
| return this; |
| } |
| |
| @Override |
| public ByteBuffer putShort(short value) { |
| int newPosition = position + SizeOf.SHORT; |
| if (newPosition > limit) { |
| throw new BufferOverflowException(); |
| } |
| this.block.pokeShort(offset + position, value, order); |
| position = newPosition; |
| return this; |
| } |
| |
| @Override |
| public ByteBuffer putShort(int index, short value) { |
| checkIndex(index, SizeOf.SHORT); |
| this.block.pokeShort(offset + index, value, order); |
| return this; |
| } |
| |
| @Override |
| public ByteBuffer slice() { |
| return new ReadWriteDirectByteBuffer(block, remaining(), offset + position); |
| } |
| |
| } |