| /* |
| * Copyright 2009-2010 by The Regents of the University of California |
| * Licensed 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 from |
| * |
| * 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 edu.uci.ics.hyracks.storage.am.common.tuples; |
| |
| import java.nio.ByteBuffer; |
| |
| import edu.uci.ics.hyracks.api.dataflow.value.ITypeTrait; |
| import edu.uci.ics.hyracks.storage.am.common.api.ITreeIndexFrame; |
| import edu.uci.ics.hyracks.storage.am.common.api.ITreeIndexTupleReference; |
| |
| public class TypeAwareTupleReference implements ITreeIndexTupleReference { |
| protected ByteBuffer buf; |
| protected int fieldStartIndex; |
| protected int fieldCount; |
| protected int tupleStartOff; |
| protected int nullFlagsBytes; |
| protected int dataStartOff; |
| |
| protected ITypeTrait[] typeTraits; |
| protected VarLenIntEncoderDecoder encDec = new VarLenIntEncoderDecoder(); |
| protected int[] decodedFieldSlots; |
| |
| public TypeAwareTupleReference(ITypeTrait[] typeTraits) { |
| this.typeTraits = typeTraits; |
| this.fieldStartIndex = 0; |
| } |
| |
| @Override |
| public void resetByTupleOffset(ByteBuffer buf, int tupleStartOff) { |
| this.buf = buf; |
| this.tupleStartOff = tupleStartOff; |
| |
| // decode field slots |
| int field = 0; |
| int cumul = 0; |
| int end = fieldStartIndex + fieldCount; |
| encDec.reset(buf.array(), tupleStartOff + nullFlagsBytes); |
| for (int i = fieldStartIndex; i < end; i++) { |
| int staticDataLen = typeTraits[i].getStaticallyKnownDataLength(); |
| if (staticDataLen == ITypeTrait.VARIABLE_LENGTH) { |
| cumul += encDec.decode(); |
| decodedFieldSlots[field++] = cumul; |
| } else { |
| cumul += staticDataLen; |
| decodedFieldSlots[field++] = cumul; |
| } |
| } |
| dataStartOff = encDec.getPos(); |
| } |
| |
| @Override |
| public void resetByTupleIndex(ITreeIndexFrame frame, int tupleIndex) { |
| resetByTupleOffset(frame.getBuffer(), frame.getTupleOffset(tupleIndex)); |
| } |
| |
| @Override |
| public void setFieldCount(int fieldCount) { |
| this.fieldCount = fieldCount; |
| if (decodedFieldSlots == null) { |
| decodedFieldSlots = new int[fieldCount]; |
| } else { |
| if (fieldCount > decodedFieldSlots.length) { |
| decodedFieldSlots = new int[fieldCount]; |
| } |
| } |
| nullFlagsBytes = getNullFlagsBytes(); |
| this.fieldStartIndex = 0; |
| } |
| |
| @Override |
| public void setFieldCount(int fieldStartIndex, int fieldCount) { |
| setFieldCount(fieldCount); |
| this.fieldStartIndex = fieldStartIndex; |
| } |
| |
| @Override |
| public int getFieldCount() { |
| return fieldCount; |
| } |
| |
| @Override |
| public byte[] getFieldData(int fIdx) { |
| return buf.array(); |
| } |
| |
| @Override |
| public int getFieldLength(int fIdx) { |
| if (fIdx == 0) { |
| return decodedFieldSlots[0]; |
| } else { |
| return decodedFieldSlots[fIdx] - decodedFieldSlots[fIdx - 1]; |
| } |
| } |
| |
| @Override |
| public int getFieldStart(int fIdx) { |
| if (fIdx == 0) { |
| return dataStartOff; |
| } else { |
| return dataStartOff + decodedFieldSlots[fIdx - 1]; |
| } |
| } |
| |
| protected int getNullFlagsBytes() { |
| return (int) Math.ceil(fieldCount / 8.0); |
| } |
| } |