Merge "RenderScript: implement a Script entry point for calling a reduce-style kernel." (6430812a) · Commits · e / os / android_frameworks_base

rs/java/android/renderscript/RenderScript.java

+10 −0

Original line number	Diff line number	Diff line
		@@ -31,6 +31,8 @@ import android.os.SystemProperties;
		import android.os.Trace;
		import java.util.ArrayList;

		// TODO: Clean up the whitespace that separates methods in this class.

		/**
		* This class provides access to a RenderScript context, which controls RenderScript
		* initialization, resource management, and teardown. An instance of the RenderScript
		@@ -727,6 +729,14 @@ public class RenderScript {
		rsnScriptForEach(mContext, id, slot, ains, aout, params, limits);
		}

		native void rsnScriptReduce(long con, long id, int slot, long ain,
		long aout, int[] limits);
		synchronized void nScriptReduce(long id, int slot, long ain, long aout,
		int[] limits) {
		validate();
		rsnScriptReduce(mContext, id, slot, ain, aout, limits);
		}

		native void rsnScriptInvokeV(long con, long id, int slot, byte[] params);
		synchronized void nScriptInvokeV(long id, int slot, byte[] params) {
		validate();

rs/java/android/renderscript/Script.java

+29 −1

Original line number	Diff line number	Diff line
		@@ -284,6 +284,35 @@ public class Script extends BaseObj {
		mRS.nScriptForEach(getID(mRS), slot, in_ids, out_id, params, limits);
		}

		/**
		* Only intended for use by generated reflected code.
		*
		* @hide
		*/
		protected void reduce(int slot, Allocation ain, Allocation aout, LaunchOptions sc) {
		mRS.validate();
		mRS.validateObject(ain);
		mRS.validateObject(aout);

		if (ain == null \|\| aout == null) {
		throw new RSIllegalArgumentException(
		"Both ain and aout are required to be non-null.");
		}

		long in_id = ain.getID(mRS);
		long out_id = aout.getID(mRS);

		int[] limits = null;
		if (sc != null) {
		limits = new int[2];

		limits[0] = sc.xstart;
		limits[1] = sc.xend;
		}

		mRS.nScriptReduce(getID(mRS), slot, in_id, out_id, limits);
		}

		long[] mInIdsBuffer;

		Script(long id, RenderScript rs) {
		@@ -292,7 +321,6 @@ public class Script extends BaseObj {
		mInIdsBuffer = new long[1];
		}


		/**
		* Only intended for use by generated reflected code.
		*

rs/jni/android_renderscript_RenderScript.cpp

+54 −0

Original line number	Diff line number	Diff line
		@@ -1958,6 +1958,59 @@ nScriptForEach(JNIEnv *_env, jobject _this, jlong con, jlong script, jint slot,
		}
		}

		static void
		nScriptReduce(JNIEnv *_env, jobject _this, jlong con, jlong script, jint slot,
		jlong ain, jlong aout, jintArray limits)
		{
		if (kLogApi) {
		ALOGD("nScriptReduce, con(%p), s(%p), slot(%i) ain(%" PRId64 ") aout(%" PRId64 ")", (RsContext)con, (void *)script, slot, ain, aout);
		}

		RsScriptCall sc, *sca = nullptr;
		uint32_t sc_size = 0;

		jint limit_len = 0;
		jint *limit_ptr = nullptr;

		// If the caller passed limits, reflect them in the RsScriptCall.
		if (limits != nullptr) {
		limit_len = _env->GetArrayLength(limits);
		limit_ptr = _env->GetIntArrayElements(limits, nullptr);

		// We expect to be passed an array [x1, x2] which specifies
		// the sub-range for a 1-dimensional reduction.
		assert(limit_len == 2);
		UNUSED(limit_len); // As the assert might not be compiled.

		sc.xStart = limit_ptr[0];
		sc.xEnd = limit_ptr[1];
		sc.yStart = 0;
		sc.yEnd = 0;
		sc.zStart = 0;
		sc.zEnd = 0;
		sc.strategy = RS_FOR_EACH_STRATEGY_DONT_CARE;
		sc.arrayStart = 0;
		sc.arrayEnd = 0;
		sc.array2Start = 0;
		sc.array2End = 0;
		sc.array3Start = 0;
		sc.array3End = 0;
		sc.array4Start = 0;
		sc.array4End = 0;

		sca = &sc;
		sc_size = sizeof(sc);
		}

		rsScriptReduce((RsContext)con, (RsScript)script, slot,
		(RsAllocation)ain, (RsAllocation)aout,
		sca, sc_size);

		if (limits != nullptr) {
		_env->ReleaseIntArrayElements(limits, limit_ptr, JNI_ABORT);
		}
		}

		// -----------------------------------

		static jlong
		@@ -2541,6 +2594,7 @@ static JNINativeMethod methods[] = {
		{"rsnScriptInvokeV", "(JJI[B)V", (void*)nScriptInvokeV },

		{"rsnScriptForEach", "(JJI[JJ[B[I)V", (void*)nScriptForEach },
		{"rsnScriptReduce", "(JJIJJ[I)V", (void*)nScriptReduce },

		{"rsnScriptSetVarI", "(JJII)V", (void*)nScriptSetVarI },
		{"rsnScriptGetVarI", "(JJI)I", (void*)nScriptGetVarI },