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.image2; 18572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams 19572a5031a5d8602db0bec0b253428a034bd4dd59Jason Samsimport java.lang.Math; 2092499bcd00319c7676f7818efa5a5ff38e435ad7Miao Wangimport android.support.v8.renderscript.Allocation; 2192499bcd00319c7676f7818efa5a5ff38e435ad7Miao Wangimport android.support.v8.renderscript.Element; 2292499bcd00319c7676f7818efa5a5ff38e435ad7Miao Wangimport android.support.v8.renderscript.ScriptIntrinsicHistogram; 23572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams 24572a5031a5d8602db0bec0b253428a034bd4dd59Jason Samspublic class WhiteBalance extends TestBase { 25572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams private ScriptC_wbalance mScript; 2692499bcd00319c7676f7818efa5a5ff38e435ad7Miao Wang private ScriptIntrinsicHistogram mHist; 2792499bcd00319c7676f7818efa5a5ff38e435ad7Miao Wang private Allocation mSums; 28572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams 29572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams public void createTest(android.content.res.Resources res) { 30572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams mScript = new ScriptC_wbalance(mRS); 3192499bcd00319c7676f7818efa5a5ff38e435ad7Miao Wang mHist = ScriptIntrinsicHistogram.create(mRS, Element.U8_4(mRS)); 3292499bcd00319c7676f7818efa5a5ff38e435ad7Miao Wang mSums = Allocation.createSized(mRS, Element.I32_4(mRS), 256); 3392499bcd00319c7676f7818efa5a5ff38e435ad7Miao Wang mHist.setOutput(mSums); 3492499bcd00319c7676f7818efa5a5ff38e435ad7Miao Wang mScript.set_histogramValues(mSums); 35572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams } 36572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams 37572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams public void runTest() { 3892499bcd00319c7676f7818efa5a5ff38e435ad7Miao Wang mHist.forEach(mInPixelsAllocation); 39572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams mScript.invoke_prepareWhiteBalance(); 40572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams mScript.forEach_whiteBalanceKernel(mInPixelsAllocation, mOutPixelsAllocation); 41572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams } 42572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams 43572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams} 44