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; 20572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams 21572a5031a5d8602db0bec0b253428a034bd4dd59Jason Samsimport android.support.v8.renderscript.*; 22572a5031a5d8602db0bec0b253428a034bd4dd59Jason Samsimport android.util.Log; 23572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams 24572a5031a5d8602db0bec0b253428a034bd4dd59Jason Samspublic class ColorCube extends TestBase { 25572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams private Allocation mCube; 26572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams private ScriptC_colorcube mScript; 27572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams private ScriptIntrinsic3DLUT mIntrinsic; 28572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams private boolean mUseIntrinsic; 29572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams 30572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams public ColorCube(boolean useIntrinsic) { 31572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams mUseIntrinsic = useIntrinsic; 32572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams } 33572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams 34572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams private void initCube() { 35572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams final int sx = 32; 36572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams final int sy = 32; 37572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams final int sz = 16; 38572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams 39572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams Type.Builder tb = new Type.Builder(mRS, Element.U8_4(mRS)); 40572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams tb.setX(sx); 41572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams tb.setY(sy); 42572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams tb.setZ(sz); 43572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams Type t = tb.create(); 44572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams mCube = Allocation.createTyped(mRS, t); 45572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams 46572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams int dat[] = new int[sx * sy * sz]; 47572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams for (int z = 0; z < sz; z++) { 48572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams for (int y = 0; y < sy; y++) { 49572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams for (int x = 0; x < sx; x++ ) { 50572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams int v = 0xff000000; 51572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams v |= (0xff * x / (sx - 1)); 52572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams v |= (0xff * y / (sy - 1)) << 8; 53572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams v |= (0xff * z / (sz - 1)) << 16; 54572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams dat[z*sy*sx + y*sx + x] = v; 55572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams } 56572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams } 57572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams } 58572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams 59572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams mCube.copyFromUnchecked(dat); 60572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams } 61572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams 62572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams public void createTest(android.content.res.Resources res) { 63f1d97e536561b4731997c85873dde3b3fb721cb2Stephen Hines mScript = new ScriptC_colorcube(mRS); 64572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams mIntrinsic = ScriptIntrinsic3DLUT.create(mRS, Element.U8_4(mRS)); 65572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams 66572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams initCube(); 67572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams mScript.invoke_setCube(mCube); 68572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams mIntrinsic.setLUT(mCube); 69572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams } 70572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams 71572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams public void runTest() { 72572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams if (mUseIntrinsic) { 73572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams mIntrinsic.forEach(mInPixelsAllocation, mOutPixelsAllocation); 74572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams } else { 75572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams mScript.forEach_root(mInPixelsAllocation, mOutPixelsAllocation); 76572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams } 77572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams } 78572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams 79572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams} 80