| /* |
| * Copyright (C) 2008 The Android Open Source Project |
| * |
| * 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 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 android.renderscript; |
| |
| import java.io.IOException; |
| import java.io.InputStream; |
| |
| import android.content.res.Resources; |
| import android.content.res.AssetManager; |
| import android.graphics.Bitmap; |
| import android.graphics.BitmapFactory; |
| import android.util.Log; |
| import android.util.TypedValue; |
| |
| /** |
| * @hide |
| * |
| **/ |
| public class Allocation extends BaseObj { |
| Type mType; |
| Bitmap mBitmap; |
| int mUsage; |
| |
| public static final int USAGE_SCRIPT = 0x0001; |
| public static final int USAGE_GRAPHICS_TEXTURE = 0x0002; |
| public static final int USAGE_GRAPHICS_VERTEX = 0x0004; |
| public static final int USAGE_GRAPHICS_CONSTANTS = 0x0008; |
| |
| private static final int USAGE_ALL = 0x000F; |
| |
| |
| public enum CubemapLayout { |
| VERTICAL_FACE_LIST (0), |
| HORIZONTAL_FACE_LIST (1), |
| VERTICAL_CROSS (2), |
| HORIZONTAL_CROSS (3); |
| |
| int mID; |
| CubemapLayout(int id) { |
| mID = id; |
| } |
| } |
| |
| public enum MipmapControl { |
| MIPMAP_NONE(0), |
| MIPMAP_FULL(1), |
| MIPMAP_ON_SYNC_TO_TEXTURE(2); |
| |
| int mID; |
| MipmapControl(int id) { |
| mID = id; |
| } |
| } |
| |
| Allocation(int id, RenderScript rs, Type t, int usage) { |
| super(id, rs); |
| if (usage > USAGE_ALL) { |
| throw new RSIllegalArgumentException("Unknown usage specified."); |
| } |
| mType = t; |
| } |
| |
| @Override |
| void updateFromNative() { |
| super.updateFromNative(); |
| int typeID = mRS.nAllocationGetType(getID()); |
| if(typeID != 0) { |
| mType = new Type(typeID, mRS); |
| mType.updateFromNative(); |
| } |
| } |
| |
| public Type getType() { |
| return mType; |
| } |
| |
| public void syncAll(int srcLocation) { |
| switch (srcLocation) { |
| case USAGE_SCRIPT: |
| case USAGE_GRAPHICS_CONSTANTS: |
| case USAGE_GRAPHICS_TEXTURE: |
| case USAGE_GRAPHICS_VERTEX: |
| break; |
| default: |
| throw new RSIllegalArgumentException("Source must be exactly one usage type."); |
| } |
| mRS.validate(); |
| mRS.nAllocationSyncAll(getID(), srcLocation); |
| } |
| |
| public void uploadToBufferObject() { |
| mRS.validate(); |
| mRS.nAllocationUploadToBufferObject(getID()); |
| } |
| |
| |
| public void copyFrom(BaseObj[] d) { |
| mRS.validate(); |
| if (d.length != mType.getCount()) { |
| throw new RSIllegalArgumentException("Array size mismatch, allocation sizeX = " + |
| mType.getCount() + ", array length = " + d.length); |
| } |
| int i[] = new int[d.length]; |
| for (int ct=0; ct < d.length; ct++) { |
| i[ct] = d[ct].getID(); |
| } |
| subData1D(0, mType.getCount(), i); |
| } |
| |
| private void validateBitmap(Bitmap b) { |
| mRS.validate(); |
| if(mType.getX() != b.getWidth() || |
| mType.getY() != b.getHeight()) { |
| throw new RSIllegalArgumentException("Cannot update allocation from bitmap, sizes mismatch"); |
| } |
| } |
| |
| public void copyFrom(int[] d) { |
| mRS.validate(); |
| subData1D(0, mType.getCount(), d); |
| } |
| public void copyFrom(short[] d) { |
| mRS.validate(); |
| subData1D(0, mType.getCount(), d); |
| } |
| public void copyFrom(byte[] d) { |
| mRS.validate(); |
| subData1D(0, mType.getCount(), d); |
| } |
| public void copyFrom(float[] d) { |
| mRS.validate(); |
| subData1D(0, mType.getCount(), d); |
| } |
| public void copyFrom(Bitmap b) { |
| validateBitmap(b); |
| mRS.nAllocationCopyFromBitmap(getID(), b); |
| } |
| |
| public void copyTo(Bitmap b) { |
| validateBitmap(b); |
| mRS.nAllocationCopyToBitmap(getID(), b); |
| } |
| |
| |
| public void subData(int xoff, FieldPacker fp) { |
| int eSize = mType.mElement.getSizeBytes(); |
| final byte[] data = fp.getData(); |
| |
| int count = data.length / eSize; |
| if ((eSize * count) != data.length) { |
| throw new RSIllegalArgumentException("Field packer length " + data.length + |
| " not divisible by element size " + eSize + "."); |
| } |
| data1DChecks(xoff, count, data.length, data.length); |
| mRS.nAllocationSubData1D(getID(), xoff, count, data, data.length); |
| } |
| |
| |
| public void subElementData(int xoff, int component_number, FieldPacker fp) { |
| if (component_number >= mType.mElement.mElements.length) { |
| throw new RSIllegalArgumentException("Component_number " + component_number + " out of range."); |
| } |
| if(xoff < 0) { |
| throw new RSIllegalArgumentException("Offset must be >= 0."); |
| } |
| |
| final byte[] data = fp.getData(); |
| int eSize = mType.mElement.mElements[component_number].getSizeBytes(); |
| |
| if (data.length != eSize) { |
| throw new RSIllegalArgumentException("Field packer sizelength " + data.length + |
| " does not match component size " + eSize + "."); |
| } |
| |
| mRS.nAllocationSubElementData1D(getID(), xoff, component_number, data, data.length); |
| } |
| |
| private void data1DChecks(int off, int count, int len, int dataSize) { |
| mRS.validate(); |
| if(off < 0) { |
| throw new RSIllegalArgumentException("Offset must be >= 0."); |
| } |
| if(count < 1) { |
| throw new RSIllegalArgumentException("Count must be >= 1."); |
| } |
| if((off + count) > mType.getCount()) { |
| throw new RSIllegalArgumentException("Overflow, Available count " + mType.getCount() + |
| ", got " + count + " at offset " + off + "."); |
| } |
| if((len) < dataSize) { |
| throw new RSIllegalArgumentException("Array too small for allocation type."); |
| } |
| } |
| |
| public void subData1D(int off, int count, int[] d) { |
| int dataSize = mType.mElement.getSizeBytes() * count; |
| data1DChecks(off, count, d.length * 4, dataSize); |
| mRS.nAllocationSubData1D(getID(), off, count, d, dataSize); |
| } |
| public void subData1D(int off, int count, short[] d) { |
| int dataSize = mType.mElement.getSizeBytes() * count; |
| data1DChecks(off, count, d.length * 2, dataSize); |
| mRS.nAllocationSubData1D(getID(), off, count, d, dataSize); |
| } |
| public void subData1D(int off, int count, byte[] d) { |
| int dataSize = mType.mElement.getSizeBytes() * count; |
| data1DChecks(off, count, d.length, dataSize); |
| mRS.nAllocationSubData1D(getID(), off, count, d, dataSize); |
| } |
| public void subData1D(int off, int count, float[] d) { |
| int dataSize = mType.mElement.getSizeBytes() * count; |
| data1DChecks(off, count, d.length * 4, dataSize); |
| mRS.nAllocationSubData1D(getID(), off, count, d, dataSize); |
| } |
| |
| |
| public void subData2D(int xoff, int yoff, int w, int h, int[] d) { |
| mRS.validate(); |
| mRS.nAllocationSubData2D(getID(), xoff, yoff, w, h, d, d.length * 4); |
| } |
| |
| public void subData2D(int xoff, int yoff, int w, int h, float[] d) { |
| mRS.validate(); |
| mRS.nAllocationSubData2D(getID(), xoff, yoff, w, h, d, d.length * 4); |
| } |
| |
| public void readData(int[] d) { |
| mRS.validate(); |
| mRS.nAllocationRead(getID(), d); |
| } |
| |
| public void readData(float[] d) { |
| mRS.validate(); |
| mRS.nAllocationRead(getID(), d); |
| } |
| |
| public synchronized void resize(int dimX) { |
| if ((mType.getY() > 0)|| (mType.getZ() > 0) || mType.hasFaces() || mType.hasMipmaps()) { |
| throw new RSInvalidStateException("Resize only support for 1D allocations at this time."); |
| } |
| mRS.nAllocationResize1D(getID(), dimX); |
| mRS.finish(); // Necessary because resize is fifoed and update is async. |
| |
| int typeID = mRS.nAllocationGetType(getID()); |
| mType = new Type(typeID, mRS); |
| mType.updateFromNative(); |
| } |
| |
| /* |
| public void resize(int dimX, int dimY) { |
| if ((mType.getZ() > 0) || mType.getFaces() || mType.getLOD()) { |
| throw new RSIllegalStateException("Resize only support for 2D allocations at this time."); |
| } |
| if (mType.getY() == 0) { |
| throw new RSIllegalStateException("Resize only support for 2D allocations at this time."); |
| } |
| mRS.nAllocationResize2D(getID(), dimX, dimY); |
| } |
| */ |
| |
| /* |
| public class Adapter1D extends BaseObj { |
| Adapter1D(int id, RenderScript rs) { |
| super(id, rs); |
| } |
| |
| public void setConstraint(Dimension dim, int value) { |
| mRS.validate(); |
| mRS.nAdapter1DSetConstraint(getID(), dim.mID, value); |
| } |
| |
| public void data(int[] d) { |
| mRS.validate(); |
| mRS.nAdapter1DData(getID(), d); |
| } |
| |
| public void data(float[] d) { |
| mRS.validate(); |
| mRS.nAdapter1DData(getID(), d); |
| } |
| |
| public void subData(int off, int count, int[] d) { |
| mRS.validate(); |
| mRS.nAdapter1DSubData(getID(), off, count, d); |
| } |
| |
| public void subData(int off, int count, float[] d) { |
| mRS.validate(); |
| mRS.nAdapter1DSubData(getID(), off, count, d); |
| } |
| } |
| |
| public Adapter1D createAdapter1D() { |
| mRS.validate(); |
| int id = mRS.nAdapter1DCreate(); |
| if(id == 0) { |
| throw new RSRuntimeException("Adapter creation failed."); |
| } |
| mRS.nAdapter1DBindAllocation(id, getID()); |
| return new Adapter1D(id, mRS); |
| } |
| */ |
| |
| |
| public class Adapter2D extends BaseObj { |
| Adapter2D(int id, RenderScript rs) { |
| super(id, rs); |
| } |
| |
| public void setConstraint(Dimension dim, int value) { |
| mRS.validate(); |
| mRS.nAdapter2DSetConstraint(getID(), dim.mID, value); |
| } |
| |
| public void data(int[] d) { |
| mRS.validate(); |
| mRS.nAdapter2DData(getID(), d); |
| } |
| |
| public void data(float[] d) { |
| mRS.validate(); |
| mRS.nAdapter2DData(getID(), d); |
| } |
| |
| public void subData(int xoff, int yoff, int w, int h, int[] d) { |
| mRS.validate(); |
| mRS.nAdapter2DSubData(getID(), xoff, yoff, w, h, d); |
| } |
| |
| public void subData(int xoff, int yoff, int w, int h, float[] d) { |
| mRS.validate(); |
| mRS.nAdapter2DSubData(getID(), xoff, yoff, w, h, d); |
| } |
| } |
| |
| public Adapter2D createAdapter2D() { |
| mRS.validate(); |
| int id = mRS.nAdapter2DCreate(); |
| if(id == 0) { |
| throw new RSRuntimeException("allocation failed."); |
| } |
| mRS.nAdapter2DBindAllocation(id, getID()); |
| if(id == 0) { |
| throw new RSRuntimeException("Adapter creation failed."); |
| } |
| return new Adapter2D(id, mRS); |
| } |
| |
| |
| // creation |
| |
| private static BitmapFactory.Options mBitmapOptions = new BitmapFactory.Options(); |
| static { |
| mBitmapOptions.inScaled = false; |
| } |
| |
| static public Allocation createTyped(RenderScript rs, Type type, MipmapControl mc, int usage) { |
| rs.validate(); |
| if (type.getID() == 0) { |
| throw new RSInvalidStateException("Bad Type"); |
| } |
| int id = rs.nAllocationCreateTyped(type.getID(), mc.mID, usage); |
| if (id == 0) { |
| throw new RSRuntimeException("Allocation creation failed."); |
| } |
| return new Allocation(id, rs, type, usage); |
| } |
| |
| static public Allocation createTyped(RenderScript rs, Type type, int usage) { |
| return createTyped(rs, type, MipmapControl.MIPMAP_NONE, usage); |
| } |
| |
| static public Allocation createTyped(RenderScript rs, Type type) { |
| return createTyped(rs, type, MipmapControl.MIPMAP_NONE, USAGE_SCRIPT); |
| } |
| |
| static public Allocation createSized(RenderScript rs, Element e, |
| int count, int usage) { |
| rs.validate(); |
| Type.Builder b = new Type.Builder(rs, e); |
| b.setX(count); |
| Type t = b.create(); |
| |
| int id = rs.nAllocationCreateTyped(t.getID(), MipmapControl.MIPMAP_NONE.mID, usage); |
| if (id == 0) { |
| throw new RSRuntimeException("Allocation creation failed."); |
| } |
| return new Allocation(id, rs, t, usage); |
| } |
| |
| static public Allocation createSized(RenderScript rs, Element e, int count) { |
| return createSized(rs, e, count, USAGE_SCRIPT); |
| } |
| |
| static private Element elementFromBitmap(RenderScript rs, Bitmap b) { |
| final Bitmap.Config bc = b.getConfig(); |
| if (bc == Bitmap.Config.ALPHA_8) { |
| return Element.A_8(rs); |
| } |
| if (bc == Bitmap.Config.ARGB_4444) { |
| return Element.RGBA_4444(rs); |
| } |
| if (bc == Bitmap.Config.ARGB_8888) { |
| return Element.RGBA_8888(rs); |
| } |
| if (bc == Bitmap.Config.RGB_565) { |
| return Element.RGB_565(rs); |
| } |
| throw new RSInvalidStateException("Bad bitmap type: " + bc); |
| } |
| |
| static private Type typeFromBitmap(RenderScript rs, Bitmap b, |
| MipmapControl mip) { |
| Element e = elementFromBitmap(rs, b); |
| Type.Builder tb = new Type.Builder(rs, e); |
| tb.setX(b.getWidth()); |
| tb.setY(b.getHeight()); |
| tb.setMipmaps(mip == MipmapControl.MIPMAP_FULL); |
| return tb.create(); |
| } |
| |
| static public Allocation createFromBitmap(RenderScript rs, Bitmap b, |
| MipmapControl mips, |
| int usage) { |
| rs.validate(); |
| Type t = typeFromBitmap(rs, b, mips); |
| |
| int id = rs.nAllocationCreateFromBitmap(t.getID(), mips.mID, b, usage); |
| if (id == 0) { |
| throw new RSRuntimeException("Load failed."); |
| } |
| return new Allocation(id, rs, t, usage); |
| } |
| |
| static public Allocation createFromBitmap(RenderScript rs, Bitmap b) { |
| return createFromBitmap(rs, b, MipmapControl.MIPMAP_NONE, |
| USAGE_GRAPHICS_TEXTURE); |
| } |
| |
| static public Allocation createCubemapFromBitmap(RenderScript rs, Bitmap b, |
| MipmapControl mips, |
| CubemapLayout layout, |
| int usage) { |
| rs.validate(); |
| |
| int height = b.getHeight(); |
| int width = b.getWidth(); |
| |
| if (layout != CubemapLayout.VERTICAL_FACE_LIST) { |
| throw new RSIllegalArgumentException("Only vertical face list supported"); |
| } |
| if (height % 6 != 0) { |
| throw new RSIllegalArgumentException("Cubemap height must be multiple of 6"); |
| } |
| if (height / 6 != width) { |
| throw new RSIllegalArgumentException("Only square cobe map faces supported"); |
| } |
| boolean isPow2 = (width & (width - 1)) == 0; |
| if (!isPow2) { |
| throw new RSIllegalArgumentException("Only power of 2 cube faces supported"); |
| } |
| |
| Element e = elementFromBitmap(rs, b); |
| Type.Builder tb = new Type.Builder(rs, e); |
| tb.setX(width); |
| tb.setY(width); |
| tb.setFaces(true); |
| tb.setMipmaps(mips == MipmapControl.MIPMAP_FULL); |
| Type t = tb.create(); |
| |
| int id = rs.nAllocationCubeCreateFromBitmap(t.getID(), mips.mID, b, usage); |
| if(id == 0) { |
| throw new RSRuntimeException("Load failed for bitmap " + b + " element " + e); |
| } |
| return new Allocation(id, rs, t, usage); |
| } |
| |
| static public Allocation createCubemapFromBitmap(RenderScript rs, Bitmap b, |
| CubemapLayout layout) { |
| return createCubemapFromBitmap(rs, b, MipmapControl.MIPMAP_NONE, |
| layout, USAGE_GRAPHICS_TEXTURE); |
| } |
| |
| static public Allocation createFromBitmapResource(RenderScript rs, |
| Resources res, |
| int id, |
| MipmapControl mips, |
| int usage) { |
| |
| rs.validate(); |
| Bitmap b = BitmapFactory.decodeResource(res, id); |
| Allocation alloc = createFromBitmap(rs, b, mips, usage); |
| b.recycle(); |
| return alloc; |
| } |
| |
| static public Allocation createFromBitmapResource(RenderScript rs, |
| Resources res, |
| int id) { |
| return createFromBitmapResource(rs, res, id, |
| MipmapControl.MIPMAP_NONE, |
| USAGE_GRAPHICS_TEXTURE); |
| } |
| |
| /* |
| static public Allocation createFromBitmapResource(RenderScript rs, |
| Resources res, |
| int id, |
| Element dstFmt, |
| boolean genMips) { |
| MipmapControl mc = MipmapControl.MIPMAP_NONE; |
| if (genMips) { |
| mc = MipmapControl.MIPMAP_ON_SYNC_TO_TEXTURE; |
| } |
| return createFromBitmapResource(rs, res, id, mc, USAGE_GRAPHICS_TEXTURE); |
| } |
| */ |
| static public Allocation createFromString(RenderScript rs, |
| String str, |
| int usage) { |
| rs.validate(); |
| byte[] allocArray = null; |
| try { |
| allocArray = str.getBytes("UTF-8"); |
| Allocation alloc = Allocation.createSized(rs, Element.U8(rs), allocArray.length, usage); |
| alloc.copyFrom(allocArray); |
| return alloc; |
| } |
| catch (Exception e) { |
| throw new RSRuntimeException("Could not convert string to utf-8."); |
| } |
| } |
| } |
| |
| |