/*
* Copyright (C) 2015, 2016 higherfrequencytrading.com
* Copyright (C) 2016 Roman Leventov
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License as published by
* the Free Software Foundation, either version 3 of the License.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
package net.openhft.chronicle.values;
import com.squareup.javapoet.ArrayTypeName;
import com.squareup.javapoet.FieldSpec;
import com.squareup.javapoet.MethodSpec;
import java.lang.reflect.Method;
import static javax.lang.model.element.Modifier.FINAL;
import static javax.lang.model.element.Modifier.PRIVATE;
class ValueFieldModel extends ScalarFieldModel {
private ValueModel valueModel;
private ValueModel valueModel() {
if (valueModel == null)
valueModel = ValueModel.acquire(type);
return valueModel;
}
@Override
int sizeInBits() {
return valueModel().sizeInBytes() * 8;
}
@Override
int offsetAlignmentInBytes() {
if (offsetAlignment == Align.DEFAULT)
return valueModel().recommendedOffsetAlignment();
// Value fields should be aligned at least to a byte boundary
return Math.max(1, offsetAlignment);
}
private final NativeMemberGenerator nativeGenerator = new NativeMemberGenerator();
final class NativeMemberGenerator extends MemberGenerator {
private Class nativeType;
FieldSpec cachedValue;
NativeMemberGenerator() {
super(ValueFieldModel.this);
}
@Override
public void generateFields(ValueBuilder valueBuilder) {
nativeType = Values.nativeClassFor(type);
cachedValue = FieldSpec
.builder(nativeType, name + "CachedValue", PRIVATE, FINAL)
.initializer("new $T()", nativeType)
.build();
valueBuilder.typeBuilder.addField(cachedValue);
}
@Override
public void generateArrayElementFields(
ArrayFieldModel arrayFieldModel, ValueBuilder valueBuilder) {
generateFields(valueBuilder);
}
private void initCachedValue(ValueBuilder valueBuilder, MethodSpec.Builder methodBuilder) {
methodBuilder.addStatement("$T $N = this.$N", nativeType, cachedValue, cachedValue);
int byteOffset = verifiedByteOffset(valueBuilder);
methodBuilder.addStatement("$N.bytesStore(bs, offset + $L, $L)",
cachedValue, byteOffset, valueModel().sizeInBytes());
}
@Override
public void generateGet(ValueBuilder valueBuilder, MethodSpec.Builder methodBuilder) {
initCachedValue(valueBuilder, methodBuilder);
methodBuilder.addStatement("return $N", cachedValue);
}
@Override
public void generateArrayElementGet(
ArrayFieldModel arrayFieldModel, ValueBuilder valueBuilder,
MethodSpec.Builder methodBuilder) {
initArrayElementCachedValue(arrayFieldModel, valueBuilder, methodBuilder);
methodBuilder.addStatement("return $N", cachedValue);
}
private void initArrayElementCachedValue(
ArrayFieldModel arrayField, ValueBuilder valueBuilder,
MethodSpec.Builder methodBuilder) {
methodBuilder.addStatement("$T $N = this.$N", nativeType, cachedValue, cachedValue);
int arrayByteOffset = arrayField.verifiedByteOffset(valueBuilder);
genVerifiedElementOffset(arrayField, methodBuilder);
methodBuilder.addStatement("$N.bytesStore(bs, offset + $L + elementOffset, $L)",
cachedValue, arrayByteOffset, valueModel().sizeInBytes());
}
@Override
public void generateGetUsing(ValueBuilder valueBuilder, MethodSpec.Builder methodBuilder) {
//TODO if both are Byteable, should do a shallow ("pointer") copy or memcpy?
methodBuilder.beginControlFlow("if ($N instanceof $T)", varName(), nativeType);
int byteOffset = verifiedByteOffset(valueBuilder);
methodBuilder.addStatement("(($T) $N).bytesStore(bs, offset + $L, $L)",
nativeType, varName(), byteOffset, valueModel().sizeInBytes());
methodBuilder.nextControlFlow("else");
initCachedValue(valueBuilder, methodBuilder);
methodBuilder.addStatement("(($T) $N).copyFrom($N)",
Copyable.class, varName(), cachedValue);
methodBuilder.endControlFlow();
methodBuilder.addStatement("return $N", varName());
}
@Override
public void generateArrayElementGetUsing(
ArrayFieldModel arrayFieldModel, ValueBuilder valueBuilder,
MethodSpec.Builder methodBuilder) {
methodBuilder.beginControlFlow("if ($N instanceof $T)", varName(), nativeType);
genVerifiedElementOffset(arrayFieldModel, methodBuilder);
int arrayByteOffset = arrayFieldModel.verifiedByteOffset(valueBuilder);
methodBuilder.addStatement("(($T) $N).bytesStore(bs, offset + $L + elementOffset, $L)",
nativeType, varName(), arrayByteOffset, valueModel().sizeInBytes());
methodBuilder.nextControlFlow("else");
initArrayElementCachedValue(arrayFieldModel, valueBuilder, methodBuilder);
methodBuilder.addStatement(
"(($T) $N).copyFrom($N)", Copyable.class, varName(), cachedValue);
methodBuilder.endControlFlow();
methodBuilder.addStatement("return $N", varName());
}
@Override
public void generateSet(ValueBuilder valueBuilder, MethodSpec.Builder methodBuilder) {
methodBuilder.beginControlFlow("if ($N instanceof $T)", varName(), nativeType);
int byteOffset = verifiedByteOffset(valueBuilder);
methodBuilder.addStatement(
"bs.write(offset + $L, (($T) $N).bytesStore(), (($T) $N).offset(), $L)",
byteOffset, nativeType, varName(), nativeType, varName(),
valueModel().sizeInBytes());
methodBuilder.nextControlFlow("else");
initCachedValue(valueBuilder, methodBuilder);
methodBuilder.addStatement("$N.copyFrom($N)", cachedValue, varName());
methodBuilder.endControlFlow();
}
@Override
public void generateArrayElementSet(
ArrayFieldModel arrayFieldModel, ValueBuilder valueBuilder,
MethodSpec.Builder methodBuilder) {
methodBuilder.beginControlFlow("if ($N instanceof $T)", varName(), nativeType);
genVerifiedElementOffset(arrayFieldModel, methodBuilder);
int arrayByteOffset = arrayFieldModel.verifiedByteOffset(valueBuilder);
methodBuilder.addStatement(
"bs.write(offset + $L + elementOffset, " +
"(($T) $N).bytesStore(), (($T) $N).offset(), $L)",
arrayByteOffset, nativeType, varName(), nativeType, varName(),
valueModel().sizeInBytes());
methodBuilder.nextControlFlow("else");
initArrayElementCachedValue(arrayFieldModel, valueBuilder, methodBuilder);
methodBuilder.addStatement("$N.copyFrom($N)", cachedValue, varName());
methodBuilder.endControlFlow();
}
@Override
public void generateCopyFrom(ValueBuilder valueBuilder, MethodSpec.Builder methodBuilder) {
initCachedValue(valueBuilder, methodBuilder);
if (getUsing != null) {
methodBuilder.addStatement("from.$N($N)", getUsing.getName(), cachedValue);
} else {
methodBuilder.addStatement("$N.copyFrom(from.$N())", cachedValue, get.getName());
}
}
@Override
public void generateArrayElementCopyFrom(
ArrayFieldModel arrayFieldModel, ValueBuilder valueBuilder,
MethodSpec.Builder methodBuilder) {
initArrayElementCachedValue(arrayFieldModel, valueBuilder, methodBuilder);
Method getUsing = arrayFieldModel.getUsing;
if (getUsing != null) {
methodBuilder.addStatement("from.$N(index, $N)", getUsing.getName(), cachedValue);
} else {
methodBuilder.addStatement("$N.copyFrom(from.$N(index))",
cachedValue, arrayFieldModel.get.getName());
}
}
@Override
void generateWriteMarshallable(
ValueBuilder valueBuilder, MethodSpec.Builder methodBuilder) {
initCachedValue(valueBuilder, methodBuilder);
methodBuilder.addStatement("$N.writeMarshallable(bytes)", cachedValue);
}
@Override
void generateArrayElementWriteMarshallable(
ArrayFieldModel arrayFieldModel, ValueBuilder valueBuilder,
MethodSpec.Builder methodBuilder) {
initArrayElementCachedValue(arrayFieldModel, valueBuilder, methodBuilder);
methodBuilder.addStatement("$N.writeMarshallable(bytes)", cachedValue);
}
@Override
void generateReadMarshallable(ValueBuilder valueBuilder, MethodSpec.Builder methodBuilder) {
initCachedValue(valueBuilder, methodBuilder);
methodBuilder.addStatement("$N.readMarshallable(bytes)", cachedValue);
}
@Override
void generateArrayElementReadMarshallable(
ArrayFieldModel arrayFieldModel, ValueBuilder valueBuilder,
MethodSpec.Builder methodBuilder) {
initArrayElementCachedValue(arrayFieldModel, valueBuilder, methodBuilder);
methodBuilder.addStatement("$N.readMarshallable(bytes)", cachedValue);
}
@Override
void generateEquals(ValueBuilder valueBuilder, MethodSpec.Builder methodBuilder) {
initCachedValue(valueBuilder, methodBuilder);
methodBuilder.addCode("if (!$N.equals(other.$N())) return false;\n",
cachedValue, get.getName());
}
@Override
void generateArrayElementEquals(
ArrayFieldModel arrayFieldModel, ValueBuilder valueBuilder,
MethodSpec.Builder methodBuilder) {
initArrayElementCachedValue(arrayFieldModel, valueBuilder, methodBuilder);
methodBuilder.addCode("if (!$N.equals(other.$N(index))) return false;\n",
cachedValue, arrayFieldModel.get.getName());
}
@Override
String generateHashCode(ValueBuilder valueBuilder, MethodSpec.Builder methodBuilder) {
initCachedValue(valueBuilder, methodBuilder);
return cachedValue.name + ".hashCode()";
}
@Override
String generateArrayElementHashCode(
ArrayFieldModel arrayFieldModel, ValueBuilder valueBuilder,
MethodSpec.Builder methodBuilder) {
initArrayElementCachedValue(arrayFieldModel, valueBuilder, methodBuilder);
return cachedValue.name + ".hashCode()";
}
@Override
void generateToString(ValueBuilder valueBuilder, MethodSpec.Builder methodBuilder) {
initCachedValue(valueBuilder, methodBuilder);
genToString(methodBuilder, cachedValue.name);
}
@Override
void generateArrayElementToString(
ArrayFieldModel arrayFieldModel, ValueBuilder valueBuilder,
MethodSpec.Builder methodBuilder) {
initArrayElementCachedValue(arrayFieldModel, valueBuilder, methodBuilder);
genArrayElementToString(methodBuilder, cachedValue.name);
}
}
@Override
NativeMemberGenerator nativeGenerator() {
return nativeGenerator;
}
@Override
MemberGenerator createHeapGenerator() {
return new ObjectHeapMemberGenerator(this) {
@Override
void generateFields(ValueBuilder valueBuilder) {
field = FieldSpec.builder(valueModel().heapClass(), fieldName(), PRIVATE)
.initializer("new $T()", valueModel().heapClass())
.build();
valueBuilder.typeBuilder.addField(field);
}
@Override
void generateArrayElementFields(
ArrayFieldModel arrayFieldModel, ValueBuilder valueBuilder) {
field = FieldSpec.builder(ArrayTypeName.of(valueModel().heapClass()), fieldName())
.addModifiers(PRIVATE, FINAL)
.initializer("new $T[$L]",
valueModel().heapClass(), arrayFieldModel.array.length())
.build();
valueBuilder.typeBuilder.addField(field);
MethodSpec.Builder constructor = valueBuilder.defaultConstructorBuilder();
constructor.beginControlFlow("for (int index = 0; index < $L; index++)",
arrayFieldModel.array.length());
constructor.addStatement("$N[index] = new $T()",
fieldName(), valueModel().heapClass());
constructor.endControlFlow();
}
@Override
public void generateSet(ValueBuilder valueBuilder, MethodSpec.Builder methodBuilder) {
methodBuilder.addStatement("$N.copyFrom($N)", field, varName());
}
@Override
public void generateArrayElementSet(
ArrayFieldModel arrayFieldModel, ValueBuilder valueBuilder,
MethodSpec.Builder methodBuilder) {
methodBuilder.addStatement("$N[index].copyFrom($N)", field, varName());
}
@Override
public void generateSetVolatile(
ValueBuilder valueBuilder, MethodSpec.Builder methodBuilder) {
methodBuilder.addStatement("$N.copyFrom($N)", field, varName());
methodBuilder.addStatement("$N.storeFence()", valueBuilder.unsafe());
}
@Override
public void generateArrayElementSetVolatile(
ArrayFieldModel arrayFieldModel, ValueBuilder valueBuilder,
MethodSpec.Builder methodBuilder) {
methodBuilder.addStatement("$N[index].copyFrom($N)", field, varName());
methodBuilder.addStatement("$N.storeFence()", valueBuilder.unsafe());
}
@Override
public void generateSetOrdered(
ValueBuilder valueBuilder, MethodSpec.Builder methodBuilder) {
generateSetVolatile(valueBuilder, methodBuilder);
}
@Override
public void generateArrayElementSetOrdered(
ArrayFieldModel arrayFieldModel, ValueBuilder valueBuilder,
MethodSpec.Builder methodBuilder) {
generateArrayElementSetVolatile(arrayFieldModel, valueBuilder, methodBuilder);
}
@Override
public void generateCompareAndSwap(
ValueBuilder valueBuilder, MethodSpec.Builder methodBuilder) {
throw new UnsupportedOperationException(
"compareAndSwap() is not supported by value field " + name);
}
@Override
public void generateArrayElementCompareAndSwap(
ArrayFieldModel arrayFieldModel, ValueBuilder valueBuilder,
MethodSpec.Builder methodBuilder) {
throw new UnsupportedOperationException(
"compareAndSwap() is not supported by value field " + name);
}
@Override
public void generateCopyFrom(
ValueBuilder valueBuilder, MethodSpec.Builder methodBuilder) {
methodBuilder.addStatement("$N.copyFrom(from.$N())",
field, fieldModel.getOrGetVolatile().getName());
}
@Override
public void generateArrayElementCopyFrom(
ArrayFieldModel arrayFieldModel, ValueBuilder valueBuilder,
MethodSpec.Builder methodBuilder) {
methodBuilder.addStatement("this.$N[index].copyFrom(from.$N(index))",
field, arrayFieldModel.getOrGetVolatile().getName());
}
@Override
void generateWriteMarshallable(
ValueBuilder valueBuilder, MethodSpec.Builder methodBuilder) {
methodBuilder.addStatement("$N.writeMarshallable(bytes)", fieldName());
}
@Override
void generateArrayElementWriteMarshallable(
ArrayFieldModel arrayFieldModel, ValueBuilder valueBuilder,
MethodSpec.Builder methodBuilder) {
methodBuilder.addStatement("$N[index].writeMarshallable(bytes)", fieldName());
}
@Override
void generateReadMarshallable(
ValueBuilder valueBuilder, MethodSpec.Builder methodBuilder) {
methodBuilder.addStatement("$N.readMarshallable(bytes)", fieldName());
}
@Override
void generateArrayElementReadMarshallable(
ArrayFieldModel arrayFieldModel, ValueBuilder valueBuilder,
MethodSpec.Builder methodBuilder) {
methodBuilder.addStatement("$N[index].readMarshallable(bytes)", fieldName());
}
@Override
String generateHashCode(ValueBuilder valueBuilder, MethodSpec.Builder methodBuilder) {
return field.name + ".hashCode()";
}
@Override
String generateArrayElementHashCode(
ArrayFieldModel arrayFieldModel, ValueBuilder valueBuilder,
MethodSpec.Builder methodBuilder) {
return field.name + "[index].hashCode()";
}
};
}
}