16ab97682fd444586ee135912a4210417a1c8781bJason Sams/* 26ab97682fd444586ee135912a4210417a1c8781bJason Sams * Copyright (C) 2012 The Android Open Source Project 36ab97682fd444586ee135912a4210417a1c8781bJason Sams * 46ab97682fd444586ee135912a4210417a1c8781bJason Sams * Licensed under the Apache License, Version 2.0 (the "License"); 56ab97682fd444586ee135912a4210417a1c8781bJason Sams * you may not use this file except in compliance with the License. 66ab97682fd444586ee135912a4210417a1c8781bJason Sams * You may obtain a copy of the License at 76ab97682fd444586ee135912a4210417a1c8781bJason Sams * 86ab97682fd444586ee135912a4210417a1c8781bJason Sams * http://www.apache.org/licenses/LICENSE-2.0 96ab97682fd444586ee135912a4210417a1c8781bJason Sams * 106ab97682fd444586ee135912a4210417a1c8781bJason Sams * Unless required by applicable law or agreed to in writing, software 116ab97682fd444586ee135912a4210417a1c8781bJason Sams * distributed under the License is distributed on an "AS IS" BASIS, 126ab97682fd444586ee135912a4210417a1c8781bJason Sams * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 136ab97682fd444586ee135912a4210417a1c8781bJason Sams * See the License for the specific language governing permissions and 146ab97682fd444586ee135912a4210417a1c8781bJason Sams * limitations under the License. 156ab97682fd444586ee135912a4210417a1c8781bJason Sams */ 166ab97682fd444586ee135912a4210417a1c8781bJason Sams 176ab97682fd444586ee135912a4210417a1c8781bJason Samspackage android.renderscript; 186ab97682fd444586ee135912a4210417a1c8781bJason Sams 196ab97682fd444586ee135912a4210417a1c8781bJason Sams/** 2080d819033d4687507907f787d47379b7b37eae19Jason Sams * Intrinsic for applying a 3x3 convolve to an allocation. 2180d819033d4687507907f787d47379b7b37eae19Jason Sams * 226ab97682fd444586ee135912a4210417a1c8781bJason Sams **/ 2380d819033d4687507907f787d47379b7b37eae19Jason Samspublic final class ScriptIntrinsicConvolve3x3 extends ScriptIntrinsic { 2480d819033d4687507907f787d47379b7b37eae19Jason Sams private final float[] mValues = new float[9]; 2519e1086df11567be9ee3eaf99fa1fc82c65b2b27Jason Sams private Allocation mInput; 266ab97682fd444586ee135912a4210417a1c8781bJason Sams 27460a04971c494fec39ffcb38e873bb8fdd82d113Tim Murray private ScriptIntrinsicConvolve3x3(long id, RenderScript rs) { 286ab97682fd444586ee135912a4210417a1c8781bJason Sams super(id, rs); 296ab97682fd444586ee135912a4210417a1c8781bJason Sams } 306ab97682fd444586ee135912a4210417a1c8781bJason Sams 316ab97682fd444586ee135912a4210417a1c8781bJason Sams /** 32d93fdb77d84aa3f2c427034cf17e72941f3c3418Jason Sams * Supported elements types are {@link Element#U8}, {@link 33d93fdb77d84aa3f2c427034cf17e72941f3c3418Jason Sams * Element#U8_2}, {@link Element#U8_3}, {@link Element#U8_4}, 34d93fdb77d84aa3f2c427034cf17e72941f3c3418Jason Sams * {@link Element#F32}, {@link Element#F32_2}, {@link 35d93fdb77d84aa3f2c427034cf17e72941f3c3418Jason Sams * Element#F32_3}, and {@link Element#F32_4} 3680d819033d4687507907f787d47379b7b37eae19Jason Sams * 3780d819033d4687507907f787d47379b7b37eae19Jason Sams * The default coefficients are. 386ab97682fd444586ee135912a4210417a1c8781bJason Sams * 3980d819033d4687507907f787d47379b7b37eae19Jason Sams * <code> 4080d819033d4687507907f787d47379b7b37eae19Jason Sams * <p> [ 0, 0, 0 ] 4180d819033d4687507907f787d47379b7b37eae19Jason Sams * <p> [ 0, 1, 0 ] 4280d819033d4687507907f787d47379b7b37eae19Jason Sams * <p> [ 0, 0, 0 ] 4380d819033d4687507907f787d47379b7b37eae19Jason Sams * </code> 446ab97682fd444586ee135912a4210417a1c8781bJason Sams * 45c11e25c4e653124def1fb18e203b894f42106cbeTim Murray * @param rs The RenderScript context 4680d819033d4687507907f787d47379b7b37eae19Jason Sams * @param e Element type for intputs and outputs 476ab97682fd444586ee135912a4210417a1c8781bJason Sams * 486ab97682fd444586ee135912a4210417a1c8781bJason Sams * @return ScriptIntrinsicConvolve3x3 496ab97682fd444586ee135912a4210417a1c8781bJason Sams */ 506ab97682fd444586ee135912a4210417a1c8781bJason Sams public static ScriptIntrinsicConvolve3x3 create(RenderScript rs, Element e) { 5180d819033d4687507907f787d47379b7b37eae19Jason Sams float f[] = { 0, 0, 0, 0, 1, 0, 0, 0, 0}; 52048c978ae599ecc9e00d68040ae9fd8a9d46df64Jason Sams if (!e.isCompatible(Element.U8(rs)) && 53d93fdb77d84aa3f2c427034cf17e72941f3c3418Jason Sams !e.isCompatible(Element.U8_2(rs)) && 54d93fdb77d84aa3f2c427034cf17e72941f3c3418Jason Sams !e.isCompatible(Element.U8_3(rs)) && 55048c978ae599ecc9e00d68040ae9fd8a9d46df64Jason Sams !e.isCompatible(Element.U8_4(rs)) && 56048c978ae599ecc9e00d68040ae9fd8a9d46df64Jason Sams !e.isCompatible(Element.F32(rs)) && 57d93fdb77d84aa3f2c427034cf17e72941f3c3418Jason Sams !e.isCompatible(Element.F32_2(rs)) && 58d93fdb77d84aa3f2c427034cf17e72941f3c3418Jason Sams !e.isCompatible(Element.F32_3(rs)) && 59048c978ae599ecc9e00d68040ae9fd8a9d46df64Jason Sams !e.isCompatible(Element.F32_4(rs))) { 6080d819033d4687507907f787d47379b7b37eae19Jason Sams throw new RSIllegalArgumentException("Unsuported element type."); 6180d819033d4687507907f787d47379b7b37eae19Jason Sams } 62460a04971c494fec39ffcb38e873bb8fdd82d113Tim Murray long id = rs.nScriptIntrinsicCreate(1, e.getID(rs)); 6380d819033d4687507907f787d47379b7b37eae19Jason Sams ScriptIntrinsicConvolve3x3 si = new ScriptIntrinsicConvolve3x3(id, rs); 6480d819033d4687507907f787d47379b7b37eae19Jason Sams si.setCoefficients(f); 6580d819033d4687507907f787d47379b7b37eae19Jason Sams return si; 666ab97682fd444586ee135912a4210417a1c8781bJason Sams 676ab97682fd444586ee135912a4210417a1c8781bJason Sams } 686ab97682fd444586ee135912a4210417a1c8781bJason Sams 6980d819033d4687507907f787d47379b7b37eae19Jason Sams /** 7080d819033d4687507907f787d47379b7b37eae19Jason Sams * Set the input of the blur. 7180d819033d4687507907f787d47379b7b37eae19Jason Sams * Must match the element type supplied during create. 7280d819033d4687507907f787d47379b7b37eae19Jason Sams * 7380d819033d4687507907f787d47379b7b37eae19Jason Sams * @param ain The input allocation. 7480d819033d4687507907f787d47379b7b37eae19Jason Sams */ 7519e1086df11567be9ee3eaf99fa1fc82c65b2b27Jason Sams public void setInput(Allocation ain) { 7619e1086df11567be9ee3eaf99fa1fc82c65b2b27Jason Sams mInput = ain; 77e6a7886674c167b4e17f6dc72d41d5b9c604cdd1Jason Sams setVar(1, ain); 7819e1086df11567be9ee3eaf99fa1fc82c65b2b27Jason Sams } 796ab97682fd444586ee135912a4210417a1c8781bJason Sams 8080d819033d4687507907f787d47379b7b37eae19Jason Sams /** 8180d819033d4687507907f787d47379b7b37eae19Jason Sams * Set the coefficients for the convolve. 8280d819033d4687507907f787d47379b7b37eae19Jason Sams * 8380d819033d4687507907f787d47379b7b37eae19Jason Sams * The convolve layout is 8480d819033d4687507907f787d47379b7b37eae19Jason Sams * <code> 8580d819033d4687507907f787d47379b7b37eae19Jason Sams * <p> [ 0, 1, 2 ] 8680d819033d4687507907f787d47379b7b37eae19Jason Sams * <p> [ 3, 4, 5 ] 8780d819033d4687507907f787d47379b7b37eae19Jason Sams * <p> [ 6, 7, 8 ] 8880d819033d4687507907f787d47379b7b37eae19Jason Sams * </code> 8980d819033d4687507907f787d47379b7b37eae19Jason Sams * 9080d819033d4687507907f787d47379b7b37eae19Jason Sams * @param v The array of coefficients to set 9180d819033d4687507907f787d47379b7b37eae19Jason Sams */ 9280d819033d4687507907f787d47379b7b37eae19Jason Sams public void setCoefficients(float v[]) { 936ab97682fd444586ee135912a4210417a1c8781bJason Sams FieldPacker fp = new FieldPacker(9*4); 946ab97682fd444586ee135912a4210417a1c8781bJason Sams for (int ct=0; ct < mValues.length; ct++) { 956ab97682fd444586ee135912a4210417a1c8781bJason Sams mValues[ct] = v[ct]; 966ab97682fd444586ee135912a4210417a1c8781bJason Sams fp.addF32(mValues[ct]); 976ab97682fd444586ee135912a4210417a1c8781bJason Sams } 986ab97682fd444586ee135912a4210417a1c8781bJason Sams setVar(0, fp); 996ab97682fd444586ee135912a4210417a1c8781bJason Sams } 10019e1086df11567be9ee3eaf99fa1fc82c65b2b27Jason Sams 10180d819033d4687507907f787d47379b7b37eae19Jason Sams /** 10280d819033d4687507907f787d47379b7b37eae19Jason Sams * Apply the filter to the input and save to the specified 10380d819033d4687507907f787d47379b7b37eae19Jason Sams * allocation. 10480d819033d4687507907f787d47379b7b37eae19Jason Sams * 10580d819033d4687507907f787d47379b7b37eae19Jason Sams * @param aout Output allocation. Must match creation element 10680d819033d4687507907f787d47379b7b37eae19Jason Sams * type. 10780d819033d4687507907f787d47379b7b37eae19Jason Sams */ 10819e1086df11567be9ee3eaf99fa1fc82c65b2b27Jason Sams public void forEach(Allocation aout) { 109949610653fdf55dd2cb3c846047e6aa2c6d73f0dChris Wailes forEach(0, (Allocation) null, aout, null); 1106f842ac8aa37dd855fbffdc09f5491bd85ab3c9aTim Murray } 1116f842ac8aa37dd855fbffdc09f5491bd85ab3c9aTim Murray 1126f842ac8aa37dd855fbffdc09f5491bd85ab3c9aTim Murray /** 1136f842ac8aa37dd855fbffdc09f5491bd85ab3c9aTim Murray * Apply the filter to the input and save to the specified 1146f842ac8aa37dd855fbffdc09f5491bd85ab3c9aTim Murray * allocation. 1156f842ac8aa37dd855fbffdc09f5491bd85ab3c9aTim Murray * 1166f842ac8aa37dd855fbffdc09f5491bd85ab3c9aTim Murray * @param aout Output allocation. Must match creation element 1176f842ac8aa37dd855fbffdc09f5491bd85ab3c9aTim Murray * type. 1186f842ac8aa37dd855fbffdc09f5491bd85ab3c9aTim Murray * @param opt LaunchOptions for clipping 1196f842ac8aa37dd855fbffdc09f5491bd85ab3c9aTim Murray */ 1206f842ac8aa37dd855fbffdc09f5491bd85ab3c9aTim Murray public void forEach(Allocation aout, Script.LaunchOptions opt) { 12148ba506dfa591d0bcd79a088457eb95a9bd4c575Stephen Hines forEach(0, (Allocation) null, aout, null, opt); 12219e1086df11567be9ee3eaf99fa1fc82c65b2b27Jason Sams } 12319e1086df11567be9ee3eaf99fa1fc82c65b2b27Jason Sams 12408a81583c18a849e442ceeb8d7baeca743fb3be8Jason Sams /** 12508a81583c18a849e442ceeb8d7baeca743fb3be8Jason Sams * Get a KernelID for this intrinsic kernel. 12608a81583c18a849e442ceeb8d7baeca743fb3be8Jason Sams * 12708a81583c18a849e442ceeb8d7baeca743fb3be8Jason Sams * @return Script.KernelID The KernelID object. 12808a81583c18a849e442ceeb8d7baeca743fb3be8Jason Sams */ 12908a81583c18a849e442ceeb8d7baeca743fb3be8Jason Sams public Script.KernelID getKernelID() { 13008a81583c18a849e442ceeb8d7baeca743fb3be8Jason Sams return createKernelID(0, 2, null, null); 13108a81583c18a849e442ceeb8d7baeca743fb3be8Jason Sams } 13208a81583c18a849e442ceeb8d7baeca743fb3be8Jason Sams 13308a81583c18a849e442ceeb8d7baeca743fb3be8Jason Sams /** 13408a81583c18a849e442ceeb8d7baeca743fb3be8Jason Sams * Get a FieldID for the input field of this intrinsic. 13508a81583c18a849e442ceeb8d7baeca743fb3be8Jason Sams * 13608a81583c18a849e442ceeb8d7baeca743fb3be8Jason Sams * @return Script.FieldID The FieldID object. 13708a81583c18a849e442ceeb8d7baeca743fb3be8Jason Sams */ 13808a81583c18a849e442ceeb8d7baeca743fb3be8Jason Sams public Script.FieldID getFieldID_Input() { 13908a81583c18a849e442ceeb8d7baeca743fb3be8Jason Sams return createFieldID(1, null); 14008a81583c18a849e442ceeb8d7baeca743fb3be8Jason Sams } 14108a81583c18a849e442ceeb8d7baeca743fb3be8Jason Sams 1426ab97682fd444586ee135912a4210417a1c8781bJason Sams} 143