shader-nodes-enhancement
Rémy Bouquet
7 years ago
committed by
Nehon
parent
302e746a94
commit
266d8b0828
File diff suppressed because it is too large
Load Diff
@ -0,0 +1,828 @@ |
||||
package com.jme3.shader; |
||||
|
||||
import com.jme3.math.*; |
||||
import com.jme3.renderer.Caps; |
||||
import com.jme3.renderer.Renderer; |
||||
import com.jme3.util.BufferUtils; |
||||
import com.jme3.util.NativeObject; |
||||
import com.jme3.util.SafeArrayList; |
||||
|
||||
import java.nio.ByteBuffer; |
||||
import java.util.Collection; |
||||
import java.util.HashMap; |
||||
import java.util.List; |
||||
import java.util.Map; |
||||
|
||||
/** |
||||
* The base implementation of BO. |
||||
* |
||||
* @author JavaSaBr |
||||
*/ |
||||
public class BufferObject extends NativeObject { |
||||
|
||||
private static final Map<Class<?>, VarType> CLASS_TO_VAR_TYPE = new HashMap<>(); |
||||
|
||||
static { |
||||
CLASS_TO_VAR_TYPE.put(Float.class, VarType.Float); |
||||
CLASS_TO_VAR_TYPE.put(Integer.class, VarType.Int); |
||||
CLASS_TO_VAR_TYPE.put(Boolean.class, VarType.Boolean); |
||||
CLASS_TO_VAR_TYPE.put(Vector2f.class, VarType.Vector2); |
||||
CLASS_TO_VAR_TYPE.put(Vector3f.class, VarType.Vector3); |
||||
CLASS_TO_VAR_TYPE.put(ColorRGBA.class, VarType.Vector4); |
||||
CLASS_TO_VAR_TYPE.put(Quaternion.class, VarType.Vector4); |
||||
CLASS_TO_VAR_TYPE.put(Vector4f.class, VarType.Vector4); |
||||
|
||||
CLASS_TO_VAR_TYPE.put(Vector2f[].class, VarType.Vector2Array); |
||||
CLASS_TO_VAR_TYPE.put(Vector3f[].class, VarType.Vector3Array); |
||||
CLASS_TO_VAR_TYPE.put(Vector4f[].class, VarType.Vector4Array); |
||||
CLASS_TO_VAR_TYPE.put(ColorRGBA[].class, VarType.Vector4Array); |
||||
CLASS_TO_VAR_TYPE.put(Quaternion[].class, VarType.Vector4Array); |
||||
|
||||
CLASS_TO_VAR_TYPE.put(Matrix3f.class, VarType.Matrix3); |
||||
CLASS_TO_VAR_TYPE.put(Matrix4f.class, VarType.Matrix4); |
||||
CLASS_TO_VAR_TYPE.put(Matrix3f[].class, VarType.Matrix3Array); |
||||
CLASS_TO_VAR_TYPE.put(Matrix4f[].class, VarType.Matrix4Array); |
||||
} |
||||
|
||||
protected static VarType getVarTypeByValue(final Object value) { |
||||
|
||||
final VarType varType = CLASS_TO_VAR_TYPE.get(value.getClass()); |
||||
if (varType != null) { |
||||
return varType; |
||||
} else if (value instanceof Collection<?> && ((Collection) value).isEmpty()) { |
||||
throw new IllegalArgumentException("Can't calculate a var type for the empty collection value[" + value + "]."); |
||||
} else if (value instanceof List<?>) { |
||||
return getVarTypeByValue(((List) value).get(0)); |
||||
} else if (value instanceof Collection<?>) { |
||||
return getVarTypeByValue(((Collection) value).iterator().next()); |
||||
} |
||||
|
||||
throw new IllegalArgumentException("Can't calculate a var type for the value " + value); |
||||
} |
||||
|
||||
public enum Layout { |
||||
std140, |
||||
/** unsupported yet */ |
||||
@Deprecated |
||||
std430, |
||||
} |
||||
|
||||
public enum BufferType { |
||||
ShaderStorageBufferObject(Caps.ShaderStorageBufferObject), |
||||
UniformBufferObject(Caps.UniformBufferObject), |
||||
; |
||||
|
||||
private final Caps requiredCaps; |
||||
|
||||
BufferType(final Caps requiredCaps) { |
||||
this.requiredCaps = requiredCaps; |
||||
} |
||||
|
||||
/** |
||||
* Get the required caps. |
||||
* |
||||
* @return the required caps. |
||||
*/ |
||||
public Caps getRequiredCaps() { |
||||
return requiredCaps; |
||||
} |
||||
} |
||||
|
||||
/** |
||||
* The fields of this BO. |
||||
*/ |
||||
private final Map<String, BufferObjectField> fields; |
||||
|
||||
/** |
||||
* The field's array. |
||||
*/ |
||||
private final SafeArrayList<BufferObjectField> fieldArray; |
||||
|
||||
/** |
||||
* The buffer's data layout. |
||||
*/ |
||||
private final Layout layout; |
||||
|
||||
/** |
||||
* The binding number. |
||||
*/ |
||||
private final int binding; |
||||
|
||||
/** |
||||
* The buffer's type. |
||||
*/ |
||||
private BufferType bufferType; |
||||
|
||||
/** |
||||
* The previous data buffer. |
||||
*/ |
||||
private ByteBuffer previousData; |
||||
|
||||
public BufferObject(final int binding, final Layout layout, final BufferType bufferType) { |
||||
this.handleRef = new Object(); |
||||
this.bufferType = bufferType; |
||||
this.binding = binding; |
||||
this.layout = layout; |
||||
this.fields = new HashMap<>(); |
||||
this.fieldArray = new SafeArrayList<>(BufferObjectField.class); |
||||
} |
||||
|
||||
public BufferObject(final int binding, final Layout layout) { |
||||
this(binding, layout, BufferType.UniformBufferObject); |
||||
} |
||||
|
||||
public BufferObject(final int binding, final BufferType bufferType) { |
||||
this(binding, Layout.std140, bufferType); |
||||
} |
||||
|
||||
public BufferObject(final BufferType bufferType) { |
||||
this(1, Layout.std140, bufferType); |
||||
} |
||||
|
||||
public BufferObject(final Layout layout) { |
||||
this(1, layout, BufferType.UniformBufferObject); |
||||
} |
||||
|
||||
public BufferObject(final int binding) { |
||||
this(binding, Layout.std140, BufferType.UniformBufferObject); |
||||
} |
||||
|
||||
public BufferObject() { |
||||
this(1, Layout.std140, BufferType.UniformBufferObject); |
||||
} |
||||
|
||||
private BufferObject(final Void unused, final int id) { |
||||
super(id); |
||||
this.fieldArray = null; |
||||
this.fields = null; |
||||
this.layout = null; |
||||
this.binding = 0; |
||||
} |
||||
|
||||
/** |
||||
* Declares a filed in this BO. |
||||
* |
||||
* @param name the field's name. |
||||
* @param varType the field's type. |
||||
*/ |
||||
public void declareField(final String name, final VarType varType) { |
||||
|
||||
if (fields.containsKey(name)) { |
||||
throw new IllegalArgumentException("The field " + name + " is already declared."); |
||||
} |
||||
|
||||
final BufferObjectField field = new BufferObjectField(name, varType); |
||||
|
||||
fields.put(name, field); |
||||
fieldArray.add(field); |
||||
} |
||||
|
||||
/** |
||||
* Gets the buffer's type. |
||||
* |
||||
* @return the buffer's type. |
||||
*/ |
||||
public BufferType getBufferType() { |
||||
return bufferType; |
||||
} |
||||
|
||||
/** |
||||
* Sets the buffer's type. |
||||
* |
||||
* @param bufferType the buffer's type. |
||||
*/ |
||||
public void setBufferType(final BufferType bufferType) { |
||||
|
||||
if (getId() != -1) { |
||||
throw new IllegalStateException("Can't change buffer's type when this buffer is already initialized."); |
||||
} |
||||
|
||||
this.bufferType = bufferType; |
||||
} |
||||
|
||||
/** |
||||
* Sets the value to the filed by the field's name. |
||||
* |
||||
* @param name the field's name. |
||||
* @param value the value. |
||||
*/ |
||||
public void setFieldValue(final String name, final Object value) { |
||||
|
||||
BufferObjectField field = fields.get(name); |
||||
|
||||
if (field == null) { |
||||
declareField(name, getVarTypeByValue(value)); |
||||
field = fields.get(name); |
||||
} |
||||
|
||||
field.setValue(value); |
||||
setUpdateNeeded(); |
||||
} |
||||
|
||||
/** |
||||
* Gets the current value of the field by the name. |
||||
* |
||||
* @param name the field name. |
||||
* @param <T> the value's type. |
||||
* @return the current value. |
||||
*/ |
||||
public <T> T getFieldValue(final String name) { |
||||
|
||||
final BufferObjectField field = fields.get(name); |
||||
if (field == null) { |
||||
throw new IllegalArgumentException("Unknown a field with the name " + name); |
||||
} |
||||
|
||||
return (T) field.getValue(); |
||||
} |
||||
|
||||
/** |
||||
* Get the binding number. |
||||
* |
||||
* @return the binding number. |
||||
*/ |
||||
public int getBinding() { |
||||
return binding; |
||||
} |
||||
|
||||
@Override |
||||
public void resetObject() { |
||||
this.id = -1; |
||||
setUpdateNeeded(); |
||||
} |
||||
|
||||
/** |
||||
* Computes the current binary data of this BO. |
||||
* |
||||
* @param maxSize the max data size. |
||||
* @return the current binary data of this BO. |
||||
*/ |
||||
public ByteBuffer computeData(final int maxSize) { |
||||
|
||||
int estimateSize = 0; |
||||
|
||||
for (final BufferObjectField field : fieldArray) { |
||||
estimateSize += estimateSize(field); |
||||
} |
||||
|
||||
if(maxSize < estimateSize) { |
||||
throw new IllegalStateException("The estimated size(" + estimateSize + ") of this BO is bigger than " + |
||||
"maximum available size " + maxSize); |
||||
} |
||||
|
||||
if (previousData != null) { |
||||
if (previousData.capacity() < estimateSize) { |
||||
BufferUtils.destroyDirectBuffer(previousData); |
||||
previousData = null; |
||||
} else { |
||||
previousData.clear(); |
||||
} |
||||
} |
||||
|
||||
final ByteBuffer data = previousData == null ? BufferUtils.createByteBuffer(estimateSize) : previousData; |
||||
|
||||
for (final BufferObjectField field : fieldArray) { |
||||
writeField(field, data); |
||||
} |
||||
|
||||
data.flip(); |
||||
|
||||
this.previousData = data; |
||||
|
||||
return data; |
||||
} |
||||
|
||||
/** |
||||
* Estimates size of the field. |
||||
* |
||||
* @param field the field. |
||||
* @return the estimated size. |
||||
*/ |
||||
protected int estimateSize(final BufferObjectField field) { |
||||
|
||||
switch (field.getType()) { |
||||
case Float: |
||||
case Int: { |
||||
if (layout == Layout.std140) { |
||||
return 16; |
||||
} |
||||
return 4; |
||||
} |
||||
case Boolean: { |
||||
if (layout == Layout.std140) { |
||||
return 16; |
||||
} |
||||
return 1; |
||||
} |
||||
case Vector2: { |
||||
return 4 * 2; |
||||
} |
||||
case Vector3: { |
||||
final int multiplier = layout == Layout.std140 ? 4 : 3; |
||||
return 4 * multiplier; |
||||
} |
||||
case Vector4: |
||||
return 16; |
||||
case IntArray: { |
||||
return estimate((int[]) field.getValue()); |
||||
} |
||||
case FloatArray: { |
||||
return estimate((float[]) field.getValue()); |
||||
} |
||||
case Vector2Array: { |
||||
return estimateArray(field.getValue(), 8); |
||||
} |
||||
case Vector3Array: { |
||||
final int multiplier = layout == Layout.std140 ? 16 : 12; |
||||
return estimateArray(field.getValue(), multiplier); |
||||
} |
||||
case Vector4Array: { |
||||
return estimateArray(field.getValue(), 16); |
||||
} |
||||
case Matrix3: { |
||||
final int multiplier = layout == Layout.std140 ? 16 : 12; |
||||
return 3 * 3 * multiplier; |
||||
} |
||||
case Matrix4: { |
||||
return 4 * 4 * 4; |
||||
} |
||||
case Matrix3Array: { |
||||
int multiplier = layout == Layout.std140 ? 16 : 12; |
||||
multiplier = 3 * 3 * multiplier; |
||||
return estimateArray(field.getValue(), multiplier); |
||||
} |
||||
case Matrix4Array: { |
||||
final int multiplier = 4 * 4 * 16; |
||||
return estimateArray(field.getValue(), multiplier); |
||||
} |
||||
default: { |
||||
throw new IllegalArgumentException("The type of BO field " + field.getType() + " doesn't support."); |
||||
} |
||||
} |
||||
} |
||||
|
||||
/** |
||||
* Estimates bytes count to present the value on GPU. |
||||
* |
||||
* @param value the value. |
||||
* @param multiplier the multiplier. |
||||
* @return the estimated bytes cunt. |
||||
*/ |
||||
protected int estimateArray(final Object value, final int multiplier) { |
||||
|
||||
if (value instanceof Object[]) { |
||||
return ((Object[]) value).length * multiplier; |
||||
} else if (value instanceof Collection) { |
||||
return ((Collection) value).size() * multiplier; |
||||
} |
||||
|
||||
throw new IllegalArgumentException("Unexpected value " + value); |
||||
} |
||||
|
||||
/** |
||||
* Estimates bytes count to present the values on GPU. |
||||
* |
||||
* @param values the values. |
||||
* @return the estimated bytes cunt. |
||||
*/ |
||||
protected int estimate(final float[] values) { |
||||
return values.length * 4; |
||||
} |
||||
|
||||
/** |
||||
* Estimates bytes count to present the values on GPU. |
||||
* |
||||
* @param values the values. |
||||
* @return the estimated bytes cunt. |
||||
*/ |
||||
protected int estimate(final int[] values) { |
||||
return values.length * 4; |
||||
} |
||||
|
||||
/** |
||||
* Writes the field to the data buffer. |
||||
* |
||||
* @param field the field. |
||||
* @param data the data buffer. |
||||
*/ |
||||
protected void writeField(final BufferObjectField field, final ByteBuffer data) { |
||||
|
||||
final Object value = field.getValue(); |
||||
|
||||
switch (field.getType()) { |
||||
case Int: { |
||||
data.putInt(((Number) value).intValue()); |
||||
if (layout == Layout.std140) { |
||||
data.putInt(0); |
||||
data.putLong(0); |
||||
} |
||||
break; |
||||
} |
||||
case Float: { |
||||
data.putFloat(((Number) value).floatValue()); |
||||
if (layout == Layout.std140) { |
||||
data.putInt(0); |
||||
data.putLong(0); |
||||
} |
||||
break; |
||||
} |
||||
case Boolean: |
||||
data.put((byte) (((Boolean) value) ? 1 : 0)); |
||||
if (layout == Layout.std140) { |
||||
data.putInt(0); |
||||
data.putLong(0); |
||||
data.putShort((short) 0); |
||||
data.put((byte) 0); |
||||
} |
||||
break; |
||||
case Vector2: |
||||
write(data, (Vector2f) value); |
||||
break; |
||||
case Vector3: |
||||
write(data, (Vector3f) value); |
||||
break; |
||||
case Vector4: |
||||
writeVec4(data, value); |
||||
break; |
||||
case IntArray: { |
||||
write(data, (int[]) value); |
||||
break; |
||||
} |
||||
case FloatArray: { |
||||
write(data, (float[]) value); |
||||
break; |
||||
} |
||||
case Vector2Array: { |
||||
writeVec2Array(data, value); |
||||
break; |
||||
} |
||||
case Vector3Array: { |
||||
writeVec3Array(data, value); |
||||
break; |
||||
} |
||||
case Vector4Array: { |
||||
writeVec4Array(data, value); |
||||
break; |
||||
} |
||||
case Matrix3: { |
||||
write(data, (Matrix3f) value); |
||||
break; |
||||
} |
||||
case Matrix4: { |
||||
write(data, (Matrix4f) value); |
||||
break; |
||||
} |
||||
case Matrix3Array: { |
||||
writeMat3Array(data, value); |
||||
break; |
||||
} |
||||
case Matrix4Array: { |
||||
writeMat4Array(data, value); |
||||
break; |
||||
} |
||||
default: { |
||||
throw new IllegalArgumentException("The type of BO field " + field.getType() + " doesn't support."); |
||||
} |
||||
} |
||||
} |
||||
|
||||
/** |
||||
* Writes the value to the data buffer. |
||||
* |
||||
* @param data the data buffer. |
||||
* @param value the value. |
||||
*/ |
||||
protected void writeMat3Array(final ByteBuffer data, final Object value) { |
||||
|
||||
if (value instanceof Matrix3f[]) { |
||||
|
||||
final Matrix3f[] values = (Matrix3f[]) value; |
||||
for (final Matrix3f mat : values) { |
||||
write(data, mat); |
||||
} |
||||
|
||||
} else if(value instanceof SafeArrayList) { |
||||
|
||||
final SafeArrayList<Matrix3f> values = (SafeArrayList<Matrix3f>) value; |
||||
for (final Matrix3f mat : values.getArray()) { |
||||
write(data, mat); |
||||
} |
||||
|
||||
} else if(value instanceof Collection) { |
||||
|
||||
final Collection<Matrix3f> values = (Collection<Matrix3f>) value; |
||||
for (final Matrix3f mat : values) { |
||||
write(data, mat); |
||||
} |
||||
} |
||||
} |
||||
|
||||
/** |
||||
* Writes the value to the data buffer. |
||||
* |
||||
* @param data the data buffer. |
||||
* @param value the value. |
||||
*/ |
||||
protected void writeMat4Array(final ByteBuffer data, final Object value) { |
||||
|
||||
if (value instanceof Matrix4f[]) { |
||||
|
||||
final Matrix4f[] values = (Matrix4f[]) value; |
||||
for (final Matrix4f mat : values) { |
||||
write(data, mat); |
||||
} |
||||
|
||||
} else if(value instanceof SafeArrayList) { |
||||
|
||||
final SafeArrayList<Matrix4f> values = (SafeArrayList<Matrix4f>) value; |
||||
for (final Matrix4f mat : values.getArray()) { |
||||
write(data, mat); |
||||
} |
||||
|
||||
} else if(value instanceof Collection) { |
||||
|
||||
final Collection<Matrix4f> values = (Collection<Matrix4f>) value; |
||||
for (final Matrix4f mat : values) { |
||||
write(data, mat); |
||||
} |
||||
} |
||||
} |
||||
|
||||
|
||||
/** |
||||
* Writes the value to the data buffer. |
||||
* |
||||
* @param data the data buffer. |
||||
* @param value the value. |
||||
*/ |
||||
protected void writeVec4Array(final ByteBuffer data, final Object value) { |
||||
|
||||
if (value instanceof Object[]) { |
||||
|
||||
final Object[] values = (Object[]) value; |
||||
for (final Object vec : values) { |
||||
writeVec4(data, vec); |
||||
} |
||||
|
||||
} else if(value instanceof SafeArrayList) { |
||||
|
||||
final SafeArrayList<Object> values = (SafeArrayList<Object>) value; |
||||
for (final Object vec : values.getArray()) { |
||||
writeVec4(data, vec); |
||||
} |
||||
|
||||
} else if(value instanceof Collection) { |
||||
|
||||
final Collection<Object> values = (Collection<Object>) value; |
||||
for (final Object vec : values) { |
||||
writeVec4(data, vec); |
||||
} |
||||
} |
||||
} |
||||
|
||||
/** |
||||
* Writes the value to the data buffer. |
||||
* |
||||
* @param data the data buffer. |
||||
* @param value the value. |
||||
*/ |
||||
protected void writeVec3Array(final ByteBuffer data, final Object value) { |
||||
|
||||
if (value instanceof Vector3f[]) { |
||||
|
||||
final Vector3f[] values = (Vector3f[]) value; |
||||
for (final Vector3f vec : values) { |
||||
write(data, vec); |
||||
} |
||||
|
||||
} else if(value instanceof SafeArrayList) { |
||||
|
||||
final SafeArrayList<Vector3f> values = (SafeArrayList<Vector3f>) value; |
||||
for (final Vector3f vec : values.getArray()) { |
||||
write(data, vec); |
||||
} |
||||
|
||||
} else if(value instanceof Collection) { |
||||
|
||||
final Collection<Vector3f> values = (Collection<Vector3f>) value; |
||||
for (final Vector3f vec : values) { |
||||
write(data, vec); |
||||
} |
||||
} |
||||
} |
||||
|
||||
/** |
||||
* Writes the value to the data buffer. |
||||
* |
||||
* @param data the data buffer. |
||||
* @param value the value. |
||||
*/ |
||||
protected void writeVec2Array(final ByteBuffer data, final Object value) { |
||||
|
||||
if (value instanceof Vector2f[]) { |
||||
|
||||
final Vector2f[] values = (Vector2f[]) value; |
||||
for (final Vector2f vec : values) { |
||||
write(data, vec); |
||||
} |
||||
|
||||
} else if(value instanceof SafeArrayList) { |
||||
|
||||
final SafeArrayList<Vector2f> values = (SafeArrayList<Vector2f>) value; |
||||
for (final Vector2f vec : values.getArray()) { |
||||
write(data, vec); |
||||
} |
||||
|
||||
} else if(value instanceof Collection) { |
||||
|
||||
final Collection<Vector2f> values = (Collection<Vector2f>) value; |
||||
for (final Vector2f vec : values) { |
||||
write(data, vec); |
||||
} |
||||
} |
||||
} |
||||
|
||||
/** |
||||
* Writes the value to the data buffer. |
||||
* |
||||
* @param data the data buffer. |
||||
* @param value the value. |
||||
*/ |
||||
protected void write(final ByteBuffer data, final float[] value) { |
||||
for (float val : value) { |
||||
data.putFloat(val); |
||||
} |
||||
} |
||||
|
||||
/** |
||||
* Writes the value to the data buffer. |
||||
* |
||||
* @param data the data buffer. |
||||
* @param value the value. |
||||
*/ |
||||
protected void write(final ByteBuffer data, final int[] value) { |
||||
for (int val : value) { |
||||
data.putInt(val); |
||||
} |
||||
} |
||||
|
||||
/** |
||||
* Writes the value to the data buffer. |
||||
* |
||||
* @param data the data buffer. |
||||
* @param value the value. |
||||
*/ |
||||
protected void writeVec4(final ByteBuffer data, final Object value) { |
||||
|
||||
if (value == null) { |
||||
data.putLong(0).putLong(0); |
||||
} else if (value instanceof Vector4f) { |
||||
|
||||
final Vector4f vec4 = (Vector4f) value; |
||||
data.putFloat(vec4.getX()) |
||||
.putFloat(vec4.getY()) |
||||
.putFloat(vec4.getZ()) |
||||
.putFloat(vec4.getW()); |
||||
|
||||
} else if(value instanceof Quaternion) { |
||||
|
||||
final Quaternion vec4 = (Quaternion) value; |
||||
data.putFloat(vec4.getX()) |
||||
.putFloat(vec4.getY()) |
||||
.putFloat(vec4.getZ()) |
||||
.putFloat(vec4.getW()); |
||||
|
||||
} else if(value instanceof ColorRGBA) { |
||||
|
||||
final ColorRGBA vec4 = (ColorRGBA) value; |
||||
data.putFloat(vec4.getRed()) |
||||
.putFloat(vec4.getGreen()) |
||||
.putFloat(vec4.getBlue()) |
||||
.putFloat(vec4.getAlpha()); |
||||
} |
||||
} |
||||
|
||||
/** |
||||
* Writes the value to the data buffer. |
||||
* |
||||
* @param data the data buffer. |
||||
* @param value the value. |
||||
*/ |
||||
protected void write(final ByteBuffer data, final Vector3f value) { |
||||
|
||||
if (value == null) { |
||||
data.putLong(0).putInt(0); |
||||
} else { |
||||
data.putFloat(value.getX()) |
||||
.putFloat(value.getY()) |
||||
.putFloat(value.getZ()); |
||||
} |
||||
|
||||
if (layout == Layout.std140) { |
||||
data.putInt(0); |
||||
} |
||||
} |
||||
|
||||
/** |
||||
* Writes the value to the data buffer. |
||||
* |
||||
* @param data the data buffer. |
||||
* @param x the x value. |
||||
* @param y the y value. |
||||
* @param z the z value. |
||||
*/ |
||||
protected void write(final ByteBuffer data, final float x, final float y, final float z) { |
||||
|
||||
data.putFloat(x) |
||||
.putFloat(y) |
||||
.putFloat(z); |
||||
|
||||
if (layout == Layout.std140) { |
||||
data.putInt(0); |
||||
} |
||||
} |
||||
|
||||
/** |
||||
* Writes the value to the data buffer. |
||||
* |
||||
* @param data the data buffer. |
||||
* @param x the x value. |
||||
* @param y the y value. |
||||
* @param z the z value. |
||||
* @param w the w value. |
||||
*/ |
||||
protected void write(final ByteBuffer data, final float x, final float y, final float z, final float w) { |
||||
data.putFloat(x) |
||||
.putFloat(y) |
||||
.putFloat(z) |
||||
.putFloat(w); |
||||
} |
||||
|
||||
/** |
||||
* Writes the value to the data buffer. |
||||
* |
||||
* @param data the data buffer. |
||||
* @param value the value. |
||||
*/ |
||||
protected void write(final ByteBuffer data, final Vector2f value) { |
||||
if (value == null) { |
||||
data.putLong(0); |
||||
} else { |
||||
data.putFloat(value.getX()).putFloat(value.getY()); |
||||
} |
||||
} |
||||
|
||||
/** |
||||
* Writes the value to the data buffer. |
||||
* |
||||
* @param data the data buffer. |
||||
* @param value the value. |
||||
*/ |
||||
protected void write(final ByteBuffer data, final Matrix3f value) { |
||||
write(data, value.get(0, 0), value.get(1, 0), value.get(2, 0)); |
||||
write(data, value.get(0, 1), value.get(1, 1), value.get(2, 1)); |
||||
write(data, value.get(0, 2), value.get(1, 2), value.get(2, 2)); |
||||
} |
||||
|
||||
/** |
||||
* Writes the value to the data buffer. |
||||
* |
||||
* @param data the data buffer. |
||||
* @param value the value. |
||||
*/ |
||||
protected void write(final ByteBuffer data, final Matrix4f value) { |
||||
write(data, value.get(0, 0), value.get(1, 0), value.get(2, 0), value.get(3, 0)); |
||||
write(data, value.get(0, 1), value.get(1, 1), value.get(2, 1), value.get(3, 1)); |
||||
write(data, value.get(0, 2), value.get(1, 2), value.get(2, 2), value.get(3, 2)); |
||||
write(data, value.get(0, 3), value.get(1, 3), value.get(2, 3), value.get(3, 3)); |
||||
} |
||||
|
||||
@Override |
||||
public void deleteObject(final Object rendererObject) { |
||||
|
||||
if (!(rendererObject instanceof Renderer)) { |
||||
throw new IllegalArgumentException("This bo can't be deleted from " + rendererObject); |
||||
} |
||||
|
||||
((Renderer) rendererObject).deleteBuffer(this); |
||||
} |
||||
|
||||
@Override |
||||
public NativeObject createDestructableClone() { |
||||
return new BufferObject(null, getId()); |
||||
} |
||||
|
||||
@Override |
||||
protected void deleteNativeBuffers() { |
||||
super.deleteNativeBuffers(); |
||||
if (previousData != null) { |
||||
BufferUtils.destroyDirectBuffer(previousData); |
||||
previousData = null; |
||||
} |
||||
} |
||||
|
||||
@Override |
||||
public long getUniqueId() { |
||||
return ((long) OBJTYPE_BO << 32) | ((long) id); |
||||
} |
||||
} |
||||
@ -0,0 +1,77 @@ |
||||
package com.jme3.shader; |
||||
|
||||
import static java.util.Objects.requireNonNull; |
||||
|
||||
/** |
||||
* The class to describe a filed in BO. |
||||
* |
||||
* @author JavaSaBr |
||||
*/ |
||||
public class BufferObjectField { |
||||
|
||||
|
||||
/** |
||||
* The field name. |
||||
*/ |
||||
private final String name; |
||||
|
||||
/** |
||||
* The field type. |
||||
*/ |
||||
private final VarType type; |
||||
|
||||
/** |
||||
* The field value. |
||||
*/ |
||||
private Object value; |
||||
|
||||
public BufferObjectField(final String name, final VarType type) { |
||||
this.name = name; |
||||
this.type = type; |
||||
} |
||||
|
||||
/** |
||||
* Get the field name. |
||||
* |
||||
* @return the field name. |
||||
*/ |
||||
public String getName() { |
||||
return name; |
||||
} |
||||
|
||||
/** |
||||
* Gets the field type. |
||||
* |
||||
* @return the field type. |
||||
*/ |
||||
public VarType getType() { |
||||
return type; |
||||
} |
||||
|
||||
/** |
||||
* Gets the field value. |
||||
* |
||||
* @return the field value. |
||||
*/ |
||||
public Object getValue() { |
||||
return value; |
||||
} |
||||
|
||||
/** |
||||
* Sets the field value. |
||||
* |
||||
* @param value the field value. |
||||
*/ |
||||
public void setValue(final Object value) { |
||||
this.value = requireNonNull(value, "The field's value can't be null."); |
||||
} |
||||
|
||||
@Override |
||||
public String toString() { |
||||
return "BufferObjectField{" + |
||||
"name='" + name + '\'' + |
||||
", type=" + type + |
||||
", value=" + value + |
||||
'}'; |
||||
} |
||||
} |
||||
@ -0,0 +1,93 @@ |
||||
/* |
||||
* Copyright (c) 2009-2018 jMonkeyEngine |
||||
* All rights reserved. |
||||
* |
||||
* Redistribution and use in source and binary forms, with or without |
||||
* modification, are permitted provided that the following conditions are |
||||
* met: |
||||
* |
||||
* * Redistributions of source code must retain the above copyright |
||||
* notice, this list of conditions and the following disclaimer. |
||||
* |
||||
* * Redistributions in binary form must reproduce the above copyright |
||||
* notice, this list of conditions and the following disclaimer in the |
||||
* documentation and/or other materials provided with the distribution. |
||||
* |
||||
* * Neither the name of 'jMonkeyEngine' nor the names of its contributors |
||||
* may be used to endorse or promote products derived from this software |
||||
* without specific prior written permission. |
||||
* |
||||
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
||||
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED |
||||
* TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR |
||||
* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR |
||||
* CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, |
||||
* EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, |
||||
* PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR |
||||
* PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF |
||||
* LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING |
||||
* NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS |
||||
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
||||
*/ |
||||
package com.jme3.shader; |
||||
|
||||
/** |
||||
* Implementation of shader's buffer block. |
||||
* |
||||
* @author JavaSaBr |
||||
*/ |
||||
public class ShaderBufferBlock extends ShaderVariable { |
||||
|
||||
/** |
||||
* Current used buffer object. |
||||
*/ |
||||
protected BufferObject bufferObject; |
||||
|
||||
/** |
||||
* Set the new buffer object. |
||||
* |
||||
* @param bufferObject the new buffer object. |
||||
*/ |
||||
public void setBufferObject(final BufferObject bufferObject) { |
||||
|
||||
if (bufferObject == null) { |
||||
throw new IllegalArgumentException("for storage block " + name + ": storageData cannot be null"); |
||||
} |
||||
|
||||
this.bufferObject = bufferObject; |
||||
|
||||
updateNeeded = true; |
||||
} |
||||
|
||||
/** |
||||
* Return true if need to update this storage block. |
||||
* |
||||
* @return true if need to update this storage block. |
||||
*/ |
||||
public boolean isUpdateNeeded(){ |
||||
return updateNeeded; |
||||
} |
||||
|
||||
/** |
||||
* Clear the flag {@link #isUpdateNeeded()}. |
||||
*/ |
||||
public void clearUpdateNeeded(){ |
||||
updateNeeded = false; |
||||
} |
||||
|
||||
/** |
||||
* Reset this storage block. |
||||
*/ |
||||
public void reset(){ |
||||
updateNeeded = true; |
||||
} |
||||
|
||||
/** |
||||
* Get the current storage data. |
||||
* |
||||
* @return the current storage data. |
||||
*/ |
||||
public BufferObject getBufferObject() { |
||||
return bufferObject; |
||||
} |
||||
} |
||||
Loading…
Reference in new issue