1572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams/* 2572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams * Copyright (C) 2012 The Android Open Source Project 3572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams * 4572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams * Licensed under the Apache License, Version 2.0 (the "License"); 5572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams * you may not use this file except in compliance with the License. 6572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams * You may obtain a copy of the License at 7572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams * 8572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams * http://www.apache.org/licenses/LICENSE-2.0 9572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams * 10572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams * Unless required by applicable law or agreed to in writing, software 11572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams * distributed under the License is distributed on an "AS IS" BASIS, 12572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams * See the License for the specific language governing permissions and 14572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams * limitations under the License. 15572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams */ 16572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams 17572a5031a5d8602db0bec0b253428a034bd4dd59Jason Samspackage com.android.rs.image; 18572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams 19572a5031a5d8602db0bec0b253428a034bd4dd59Jason Samsimport java.lang.Math; 20572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams 21572a5031a5d8602db0bec0b253428a034bd4dd59Jason Samsimport android.renderscript.Allocation; 22572a5031a5d8602db0bec0b253428a034bd4dd59Jason Samsimport android.renderscript.Element; 23572a5031a5d8602db0bec0b253428a034bd4dd59Jason Samsimport android.renderscript.RenderScript; 24572a5031a5d8602db0bec0b253428a034bd4dd59Jason Samsimport android.renderscript.ScriptIntrinsicConvolve3x3; 25572a5031a5d8602db0bec0b253428a034bd4dd59Jason Samsimport android.renderscript.ScriptIntrinsicColorMatrix; 26572a5031a5d8602db0bec0b253428a034bd4dd59Jason Samsimport android.renderscript.Type; 27572a5031a5d8602db0bec0b253428a034bd4dd59Jason Samsimport android.renderscript.Matrix4f; 28572a5031a5d8602db0bec0b253428a034bd4dd59Jason Samsimport android.renderscript.ScriptGroup; 29572a5031a5d8602db0bec0b253428a034bd4dd59Jason Samsimport android.util.Log; 30572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams 31572a5031a5d8602db0bec0b253428a034bd4dd59Jason Samspublic class GroupTest extends TestBase { 32572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams private ScriptIntrinsicConvolve3x3 mConvolve; 33572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams private ScriptIntrinsicColorMatrix mMatrix; 34572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams 35572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams private Allocation mScratchPixelsAllocation1; 36572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams private ScriptGroup mGroup; 37572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams 38572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams private int mWidth; 39572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams private int mHeight; 40572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams private boolean mUseNative; 41572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams 42572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams 43572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams public GroupTest(boolean useNative) { 44572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams mUseNative = useNative; 45572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams } 46572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams 47572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams public void createTest(android.content.res.Resources res) { 48572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams mWidth = mInPixelsAllocation.getType().getX(); 49572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams mHeight = mInPixelsAllocation.getType().getY(); 50572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams 51572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams mConvolve = ScriptIntrinsicConvolve3x3.create(mRS, Element.U8_4(mRS)); 52572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams mMatrix = ScriptIntrinsicColorMatrix.create(mRS, Element.U8_4(mRS)); 53572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams 54572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams float f[] = new float[9]; 55572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams f[0] = 0.f; f[1] = -1.f; f[2] = 0.f; 56572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams f[3] = -1.f; f[4] = 5.f; f[5] = -1.f; 57572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams f[6] = 0.f; f[7] = -1.f; f[8] = 0.f; 58572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams mConvolve.setCoefficients(f); 59572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams 60572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams Matrix4f m = new Matrix4f(); 61572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams m.set(1, 0, 0.2f); 62572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams m.set(1, 1, 0.9f); 63572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams m.set(1, 2, 0.2f); 64572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams mMatrix.setColorMatrix(m); 65572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams 66572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams Type.Builder tb = new Type.Builder(mRS, Element.U8_4(mRS)); 67572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams tb.setX(mWidth); 68572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams tb.setY(mHeight); 69572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams Type connect = tb.create(); 70572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams 71572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams if (mUseNative) { 72a7481b21c4ad4127f54c02c7402039d068948a34Yang Ni ScriptGroup.Builder2 b = new ScriptGroup.Builder2(mRS); 73a7481b21c4ad4127f54c02c7402039d068948a34Yang Ni ScriptGroup.Input in = b.addInput(); 74a7481b21c4ad4127f54c02c7402039d068948a34Yang Ni ScriptGroup.Closure c; 75a7481b21c4ad4127f54c02c7402039d068948a34Yang Ni 76a7481b21c4ad4127f54c02c7402039d068948a34Yang Ni c = b.addKernel(mConvolve.getKernelID(), 77a7481b21c4ad4127f54c02c7402039d068948a34Yang Ni connect, 78a7481b21c4ad4127f54c02c7402039d068948a34Yang Ni new ScriptGroup.Binding(mConvolve.getFieldID_Input(), in)); 79a7481b21c4ad4127f54c02c7402039d068948a34Yang Ni 80a7481b21c4ad4127f54c02c7402039d068948a34Yang Ni c = b.addKernel(mMatrix.getKernelID(), 81a7481b21c4ad4127f54c02c7402039d068948a34Yang Ni mOutPixelsAllocation.getType(), 82a7481b21c4ad4127f54c02c7402039d068948a34Yang Ni c.getReturn()); 83a7481b21c4ad4127f54c02c7402039d068948a34Yang Ni mGroup = b.create("group", c.getReturn()); 84572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams } else { 85572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams mScratchPixelsAllocation1 = Allocation.createTyped(mRS, connect); 86572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams } 87572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams } 88572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams 89572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams public void runTest() { 90572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams mConvolve.setInput(mInPixelsAllocation); 91572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams if (mUseNative) { 92a7481b21c4ad4127f54c02c7402039d068948a34Yang Ni mOutPixelsAllocation = (Allocation)mGroup.execute(mInPixelsAllocation)[0]; 93572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams } else { 94572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams mConvolve.forEach(mScratchPixelsAllocation1); 95572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams mMatrix.forEach(mScratchPixelsAllocation1, mOutPixelsAllocation); 96572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams } 97572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams } 98572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams 99572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams} 100