graphics/java/android/renderscript/Allocation.java - LeafOS-Project/android_frameworks_base - Gitiles

 /*
  * 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;

     Allocation(int id, RenderScript rs, Type t) {
         super(rs);
         mID = id;
         mType = t;
     }

     Allocation(int id, RenderScript rs) {
         super(rs);
         mID = id;
     }

     public Type getType() {
         return mType;
     }

     public void uploadToTexture(int baseMipLevel) {
         mRS.validate();
         mRS.nAllocationUploadToTexture(mID, false, baseMipLevel);
     }

     public void uploadToTexture(boolean genMips, int baseMipLevel) {
         mRS.validate();
         mRS.nAllocationUploadToTexture(mID, genMips, baseMipLevel);
     }

     public void uploadToBufferObject() {
         mRS.validate();
         mRS.nAllocationUploadToBufferObject(mID);
     }

     public void data(int[] d) {
         mRS.validate();
         subData1D(0, mType.getElementCount(), d);
     }
     public void data(short[] d) {
         mRS.validate();
         subData1D(0, mType.getElementCount(), d);
     }
     public void data(byte[] d) {
         mRS.validate();
         subData1D(0, mType.getElementCount(), d);
     }
     public void data(float[] d) {
         mRS.validate();
         subData1D(0, mType.getElementCount(), d);
     }

     public void subData(int off, FieldPacker fp) {
         int eSize = mType.mElement.getSizeBytes();
         final byte[] data = fp.getData();

         int count = data.length / eSize;
         if ((eSize * count) != data.length) {
             throw new IllegalArgumentException("Field packer length " + data.length +
                                                " not divisible by element size " + eSize + ".");
         }
         data1DChecks(off, count, data.length, data.length);
         mRS.nAllocationSubData1D(mID, off, count, data, data.length);
     }

     private void data1DChecks(int off, int count, int len, int dataSize) {
         mRS.validate();
         if(off < 0) {
             throw new IllegalArgumentException("Offset must be >= 0.");
         }
         if(count < 1) {
             throw new IllegalArgumentException("Count must be >= 1.");
         }
         if((off + count) > mType.getElementCount()) {
             throw new IllegalArgumentException("Overflow, Available count " + mType.getElementCount() +
                                                ", got " + count + " at offset " + off + ".");
         }
         if((len) < dataSize) {
             throw new IllegalArgumentException("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(mID, 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(mID, 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(mID, 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(mID, off, count, d, dataSize);
     }


     public void subData2D(int xoff, int yoff, int w, int h, int[] d) {
         mRS.validate();
         mRS.nAllocationSubData2D(mID, 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(mID, xoff, yoff, w, h, d, d.length * 4);
     }

     public void readData(int[] d) {
         mRS.validate();
         mRS.nAllocationRead(mID, d);
     }

     public void readData(float[] d) {
         mRS.validate();
         mRS.nAllocationRead(mID, d);
     }

     public void data(Object o) {
         mRS.validate();
         mRS.nAllocationSubDataFromObject(mID, mType, 0, o);
     }

     public void read(Object o) {
         mRS.validate();
         mRS.nAllocationSubReadFromObject(mID, mType, 0, o);
     }

     public void subData(int offset, Object o) {
         mRS.validate();
         mRS.nAllocationSubDataFromObject(mID, mType, offset, o);
     }

     public class Adapter1D extends BaseObj {
         Adapter1D(int id, RenderScript rs) {
             super(rs);
             mID = id;
         }

         public void setConstraint(Dimension dim, int value) {
             mRS.validate();
             mRS.nAdapter1DSetConstraint(mID, dim.mID, value);
         }

         public void data(int[] d) {
             mRS.validate();
             mRS.nAdapter1DData(mID, d);
         }

         public void data(float[] d) {
             mRS.validate();
             mRS.nAdapter1DData(mID, d);
         }

         public void subData(int off, int count, int[] d) {
             mRS.validate();
             mRS.nAdapter1DSubData(mID, off, count, d);
         }

         public void subData(int off, int count, float[] d) {
             mRS.validate();
             mRS.nAdapter1DSubData(mID, off, count, d);
         }
     }

     public Adapter1D createAdapter1D() {
         mRS.validate();
         int id = mRS.nAdapter1DCreate();
         if(id == 0) {
             throw new IllegalStateException("allocation failed.");
         }
         mRS.nAdapter1DBindAllocation(id, mID);
         return new Adapter1D(id, mRS);
     }


     public class Adapter2D extends BaseObj {
         Adapter2D(int id, RenderScript rs) {
             super(rs);
             mID = id;
         }

         public void setConstraint(Dimension dim, int value) {
             mRS.validate();
             mRS.nAdapter2DSetConstraint(mID, dim.mID, value);
         }

         public void data(int[] d) {
             mRS.validate();
             mRS.nAdapter2DData(mID, d);
         }

         public void data(float[] d) {
             mRS.validate();
             mRS.nAdapter2DData(mID, d);
         }

         public void subData(int xoff, int yoff, int w, int h, int[] d) {
             mRS.validate();
             mRS.nAdapter2DSubData(mID, xoff, yoff, w, h, d);
         }

         public void subData(int xoff, int yoff, int w, int h, float[] d) {
             mRS.validate();
             mRS.nAdapter2DSubData(mID, xoff, yoff, w, h, d);
         }
     }

     public Adapter2D createAdapter2D() {
         mRS.validate();
         int id = mRS.nAdapter2DCreate();
         if(id == 0) {
             throw new IllegalStateException("allocation failed.");
         }
         mRS.nAdapter2DBindAllocation(id, mID);
         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)
         throws IllegalArgumentException {

         rs.validate();
         if(type.mID == 0) {
             throw new IllegalStateException("Bad Type");
         }
         int id = rs.nAllocationCreateTyped(type.mID);
         return new Allocation(id, rs, type);
     }

     static public Allocation createSized(RenderScript rs, Element e, int count)
         throws IllegalArgumentException {

         rs.validate();
         Type.Builder b = new Type.Builder(rs, e);
         b.add(Dimension.X, count);
         Type t = b.create();

         int id = rs.nAllocationCreateTyped(t.mID);
         if(id == 0) {
             throw new IllegalStateException("Bad element.");
         }
         return new Allocation(id, rs, t);
     }

     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 IllegalStateException("Bad bitmap type.");
     }

     static private Type typeFromBitmap(RenderScript rs, Bitmap b) {
         Element e = elementFromBitmap(rs, b);
         Type.Builder tb = new Type.Builder(rs, e);
         tb.add(Dimension.X, b.getWidth());
         tb.add(Dimension.Y, b.getHeight());
         return tb.create();
     }

     static public Allocation createFromBitmap(RenderScript rs, Bitmap b, Element dstFmt, boolean genMips)
         throws IllegalArgumentException {

         rs.validate();
         Type t = typeFromBitmap(rs, b);

         int id = rs.nAllocationCreateFromBitmap(dstFmt.mID, genMips, b);
         if(id == 0) {
             throw new IllegalStateException("Load failed.");
         }
         return new Allocation(id, rs, t);
     }

     static public Allocation createBitmapRef(RenderScript rs, Bitmap b)
         throws IllegalArgumentException {

         rs.validate();
         Type t = typeFromBitmap(rs, b);

         int id = rs.nAllocationCreateBitmapRef(t.getID(), b);
         if(id == 0) {
             throw new IllegalStateException("Load failed.");
         }

         Allocation a = new Allocation(id, rs, t);
         a.mBitmap = b;
         return a;
     }

     static Allocation createFromBitmapBoxed(RenderScript rs, Bitmap b, Element dstFmt, boolean genMips)
         throws IllegalArgumentException {

         rs.validate();
         int id = rs.nAllocationCreateFromBitmapBoxed(dstFmt.mID, genMips, b);
         if(id == 0) {
             throw new IllegalStateException("Load failed.");
         }
         return new Allocation(id, rs, null);
     }

     static public Allocation createFromBitmapResource(RenderScript rs, Resources res, int id, Element dstFmt, boolean genMips)
         throws IllegalArgumentException {

         rs.validate();
         InputStream is = null;
         try {
             final TypedValue value = new TypedValue();
             is = res.openRawResource(id, value);

             int asset = ((AssetManager.AssetInputStream) is).getAssetInt();
             int allocationId = rs.nAllocationCreateFromAssetStream(dstFmt.mID, genMips,
                     asset);

             if(allocationId == 0) {
                 throw new IllegalStateException("Load failed.");
             }
             return new Allocation(allocationId, rs, null);
         } catch (Exception e) {
             // Ignore
         } finally {
             if (is != null) {
                 try {
                     is.close();
                 } catch (IOException e) {
                     // Ignore
                 }
             }
         }

         return null;
     }

     static public Allocation createFromBitmapResourceBoxed(RenderScript rs, Resources res, int id, Element dstFmt, boolean genMips)
         throws IllegalArgumentException {

         Bitmap b = BitmapFactory.decodeResource(res, id, mBitmapOptions);
         return createFromBitmapBoxed(rs, b, dstFmt, genMips);
     }

     static public Allocation createFromString(RenderScript rs, String str)
         throws IllegalArgumentException {
         byte[] allocArray = null;
         try {
             allocArray = str.getBytes("UTF-8");
             Allocation alloc = Allocation.createSized(rs, Element.U8(rs), allocArray.length);
             alloc.data(allocArray);
             return alloc;
         }
         catch (Exception e) {
             Log.e("rs", "could not convert string to utf-8");
         }
         return null;
     }
 }
	/*
	* 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;

	Allocation(int id, RenderScript rs, Type t) {
	super(rs);
	mID = id;
	mType = t;
	}

	Allocation(int id, RenderScript rs) {
	super(rs);
	mID = id;
	}

	public Type getType() {
	return mType;
	}

	public void uploadToTexture(int baseMipLevel) {
	mRS.validate();
	mRS.nAllocationUploadToTexture(mID, false, baseMipLevel);
	}

	public void uploadToTexture(boolean genMips, int baseMipLevel) {
	mRS.validate();
	mRS.nAllocationUploadToTexture(mID, genMips, baseMipLevel);
	}

	public void uploadToBufferObject() {
	mRS.validate();
	mRS.nAllocationUploadToBufferObject(mID);
	}

	public void data(int[] d) {
	mRS.validate();
	subData1D(0, mType.getElementCount(), d);
	}
	public void data(short[] d) {
	mRS.validate();
	subData1D(0, mType.getElementCount(), d);
	}
	public void data(byte[] d) {
	mRS.validate();
	subData1D(0, mType.getElementCount(), d);
	}
	public void data(float[] d) {
	mRS.validate();
	subData1D(0, mType.getElementCount(), d);
	}

	public void subData(int off, FieldPacker fp) {
	int eSize = mType.mElement.getSizeBytes();
	final byte[] data = fp.getData();

	int count = data.length / eSize;
	if ((eSize * count) != data.length) {
	throw new IllegalArgumentException("Field packer length " + data.length +
	" not divisible by element size " + eSize + ".");
	}
	data1DChecks(off, count, data.length, data.length);
	mRS.nAllocationSubData1D(mID, off, count, data, data.length);
	}

	private void data1DChecks(int off, int count, int len, int dataSize) {
	mRS.validate();
	if(off < 0) {
	throw new IllegalArgumentException("Offset must be >= 0.");
	}
	if(count < 1) {
	throw new IllegalArgumentException("Count must be >= 1.");
	}
	if((off + count) > mType.getElementCount()) {
	throw new IllegalArgumentException("Overflow, Available count " + mType.getElementCount() +
	", got " + count + " at offset " + off + ".");
	}
	if((len) < dataSize) {
	throw new IllegalArgumentException("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(mID, 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(mID, 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(mID, 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(mID, off, count, d, dataSize);
	}



	public void subData2D(int xoff, int yoff, int w, int h, int[] d) {
	mRS.validate();
	mRS.nAllocationSubData2D(mID, 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(mID, xoff, yoff, w, h, d, d.length * 4);
	}

	public void readData(int[] d) {
	mRS.validate();
	mRS.nAllocationRead(mID, d);
	}

	public void readData(float[] d) {
	mRS.validate();
	mRS.nAllocationRead(mID, d);
	}

	public void data(Object o) {
	mRS.validate();
	mRS.nAllocationSubDataFromObject(mID, mType, 0, o);
	}

	public void read(Object o) {
	mRS.validate();
	mRS.nAllocationSubReadFromObject(mID, mType, 0, o);
	}

	public void subData(int offset, Object o) {
	mRS.validate();
	mRS.nAllocationSubDataFromObject(mID, mType, offset, o);
	}

	public class Adapter1D extends BaseObj {
	Adapter1D(int id, RenderScript rs) {
	super(rs);
	mID = id;
	}

	public void setConstraint(Dimension dim, int value) {
	mRS.validate();
	mRS.nAdapter1DSetConstraint(mID, dim.mID, value);
	}

	public void data(int[] d) {
	mRS.validate();
	mRS.nAdapter1DData(mID, d);
	}

	public void data(float[] d) {
	mRS.validate();
	mRS.nAdapter1DData(mID, d);
	}

	public void subData(int off, int count, int[] d) {
	mRS.validate();
	mRS.nAdapter1DSubData(mID, off, count, d);
	}

	public void subData(int off, int count, float[] d) {
	mRS.validate();
	mRS.nAdapter1DSubData(mID, off, count, d);
	}
	}

	public Adapter1D createAdapter1D() {
	mRS.validate();
	int id = mRS.nAdapter1DCreate();
	if(id == 0) {
	throw new IllegalStateException("allocation failed.");
	}
	mRS.nAdapter1DBindAllocation(id, mID);
	return new Adapter1D(id, mRS);
	}


	public class Adapter2D extends BaseObj {
	Adapter2D(int id, RenderScript rs) {
	super(rs);
	mID = id;
	}

	public void setConstraint(Dimension dim, int value) {
	mRS.validate();
	mRS.nAdapter2DSetConstraint(mID, dim.mID, value);
	}

	public void data(int[] d) {
	mRS.validate();
	mRS.nAdapter2DData(mID, d);
	}

	public void data(float[] d) {
	mRS.validate();
	mRS.nAdapter2DData(mID, d);
	}

	public void subData(int xoff, int yoff, int w, int h, int[] d) {
	mRS.validate();
	mRS.nAdapter2DSubData(mID, xoff, yoff, w, h, d);
	}

	public void subData(int xoff, int yoff, int w, int h, float[] d) {
	mRS.validate();
	mRS.nAdapter2DSubData(mID, xoff, yoff, w, h, d);
	}
	}

	public Adapter2D createAdapter2D() {
	mRS.validate();
	int id = mRS.nAdapter2DCreate();
	if(id == 0) {
	throw new IllegalStateException("allocation failed.");
	}
	mRS.nAdapter2DBindAllocation(id, mID);
	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)
	throws IllegalArgumentException {

	rs.validate();
	if(type.mID == 0) {
	throw new IllegalStateException("Bad Type");
	}
	int id = rs.nAllocationCreateTyped(type.mID);
	return new Allocation(id, rs, type);
	}

	static public Allocation createSized(RenderScript rs, Element e, int count)
	throws IllegalArgumentException {

	rs.validate();
	Type.Builder b = new Type.Builder(rs, e);
	b.add(Dimension.X, count);
	Type t = b.create();

	int id = rs.nAllocationCreateTyped(t.mID);
	if(id == 0) {
	throw new IllegalStateException("Bad element.");
	}
	return new Allocation(id, rs, t);
	}

	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 IllegalStateException("Bad bitmap type.");
	}

	static private Type typeFromBitmap(RenderScript rs, Bitmap b) {
	Element e = elementFromBitmap(rs, b);
	Type.Builder tb = new Type.Builder(rs, e);
	tb.add(Dimension.X, b.getWidth());
	tb.add(Dimension.Y, b.getHeight());
	return tb.create();
	}

	static public Allocation createFromBitmap(RenderScript rs, Bitmap b, Element dstFmt, boolean genMips)
	throws IllegalArgumentException {

	rs.validate();
	Type t = typeFromBitmap(rs, b);

	int id = rs.nAllocationCreateFromBitmap(dstFmt.mID, genMips, b);
	if(id == 0) {
	throw new IllegalStateException("Load failed.");
	}
	return new Allocation(id, rs, t);
	}

	static public Allocation createBitmapRef(RenderScript rs, Bitmap b)
	throws IllegalArgumentException {

	rs.validate();
	Type t = typeFromBitmap(rs, b);

	int id = rs.nAllocationCreateBitmapRef(t.getID(), b);
	if(id == 0) {
	throw new IllegalStateException("Load failed.");
	}

	Allocation a = new Allocation(id, rs, t);
	a.mBitmap = b;
	return a;
	}

	static Allocation createFromBitmapBoxed(RenderScript rs, Bitmap b, Element dstFmt, boolean genMips)
	throws IllegalArgumentException {

	rs.validate();
	int id = rs.nAllocationCreateFromBitmapBoxed(dstFmt.mID, genMips, b);
	if(id == 0) {
	throw new IllegalStateException("Load failed.");
	}
	return new Allocation(id, rs, null);
	}

	static public Allocation createFromBitmapResource(RenderScript rs, Resources res, int id, Element dstFmt, boolean genMips)
	throws IllegalArgumentException {

	rs.validate();
	InputStream is = null;
	try {
	final TypedValue value = new TypedValue();
	is = res.openRawResource(id, value);

	int asset = ((AssetManager.AssetInputStream) is).getAssetInt();
	int allocationId = rs.nAllocationCreateFromAssetStream(dstFmt.mID, genMips,
	asset);

	if(allocationId == 0) {
	throw new IllegalStateException("Load failed.");
	}
	return new Allocation(allocationId, rs, null);
	} catch (Exception e) {
	// Ignore
	} finally {
	if (is != null) {
	try {
	is.close();
	} catch (IOException e) {
	// Ignore
	}
	}
	}

	return null;
	}

	static public Allocation createFromBitmapResourceBoxed(RenderScript rs, Resources res, int id, Element dstFmt, boolean genMips)
	throws IllegalArgumentException {

	Bitmap b = BitmapFactory.decodeResource(res, id, mBitmapOptions);
	return createFromBitmapBoxed(rs, b, dstFmt, genMips);
	}

	static public Allocation createFromString(RenderScript rs, String str)
	throws IllegalArgumentException {
	byte[] allocArray = null;
	try {
	allocArray = str.getBytes("UTF-8");
	Allocation alloc = Allocation.createSized(rs, Element.U8(rs), allocArray.length);
	alloc.data(allocArray);
	return alloc;
	}
	catch (Exception e) {
	Log.e("rs", "could not convert string to utf-8");
	}
	return null;
	}
	}