Home | History | Annotate | Download | only in renderscript 
      1 /*
      2  * Copyright (C) 2012 The Android Open Source Project
      3  *
      4  * Licensed under the Apache License, Version 2.0 (the "License");
      5  * you may not use this file except in compliance with the License.
      6  * You may obtain a copy of the License at
      7  *
      8  *      http://www.apache.org/licenses/LICENSE-2.0
      9  *
     10  * Unless required by applicable law or agreed to in writing, software
     11  * distributed under the License is distributed on an "AS IS" BASIS,
     12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
     13  * See the License for the specific language governing permissions and
     14  * limitations under the License.
     15  */
     16 
     17 package android.renderscript;
     18 
     19 import android.util.Log;
     20 
     21 /**
     22  * Intrinsic for applying a 3x3 convolve to an allocation.
     23  *
     24  **/
     25 public final class ScriptIntrinsicConvolve3x3 extends ScriptIntrinsic {
     26     private final float[] mValues = new float[9];
     27     private Allocation mInput;
     28 
     29     private ScriptIntrinsicConvolve3x3(int id, RenderScript rs) {
     30         super(id, rs);
     31     }
     32 
     33     /**
     34      * Supported elements types are {@link Element#U8}, {@link
     35      * Element#U8_2}, {@link Element#U8_3}, {@link Element#U8_4},
     36      * {@link Element#F32}, {@link Element#F32_2}, {@link
     37      * Element#F32_3}, and {@link Element#F32_4}
     38      *
     39      * The default coefficients are.
     40      *
     41      * <code>
     42      * <p> [ 0,  0,  0 ]
     43      * <p> [ 0,  1,  0 ]
     44      * <p> [ 0,  0,  0 ]
     45      * </code>
     46      *
     47      * @param rs The RenderScript context
     48      * @param e Element type for intputs and outputs
     49      *
     50      * @return ScriptIntrinsicConvolve3x3
     51      */
     52     public static ScriptIntrinsicConvolve3x3 create(RenderScript rs, Element e) {
     53         float f[] = { 0, 0, 0, 0, 1, 0, 0, 0, 0};
     54         if (!e.isCompatible(Element.U8(rs)) &&
     55             !e.isCompatible(Element.U8_2(rs)) &&
     56             !e.isCompatible(Element.U8_3(rs)) &&
     57             !e.isCompatible(Element.U8_4(rs)) &&
     58             !e.isCompatible(Element.F32(rs)) &&
     59             !e.isCompatible(Element.F32_2(rs)) &&
     60             !e.isCompatible(Element.F32_3(rs)) &&
     61             !e.isCompatible(Element.F32_4(rs))) {
     62             throw new RSIllegalArgumentException("Unsuported element type.");
     63         }
     64         int id = rs.nScriptIntrinsicCreate(1, e.getID(rs));
     65         ScriptIntrinsicConvolve3x3 si = new ScriptIntrinsicConvolve3x3(id, rs);
     66         si.setCoefficients(f);
     67         return si;
     68 
     69     }
     70 
     71     /**
     72      * Set the input of the blur.
     73      * Must match the element type supplied during create.
     74      *
     75      * @param ain The input allocation.
     76      */
     77     public void setInput(Allocation ain) {
     78         mInput = ain;
     79         setVar(1, ain);
     80     }
     81 
     82     /**
     83      * Set the coefficients for the convolve.
     84      *
     85      * The convolve layout is
     86      * <code>
     87      * <p> [ 0,  1,  2 ]
     88      * <p> [ 3,  4,  5 ]
     89      * <p> [ 6,  7,  8 ]
     90      * </code>
     91      *
     92      * @param v The array of coefficients to set
     93      */
     94     public void setCoefficients(float v[]) {
     95         FieldPacker fp = new FieldPacker(9*4);
     96         for (int ct=0; ct < mValues.length; ct++) {
     97             mValues[ct] = v[ct];
     98             fp.addF32(mValues[ct]);
     99         }
    100         setVar(0, fp);
    101     }
    102 
    103     /**
    104      * Apply the filter to the input and save to the specified
    105      * allocation.
    106      *
    107      * @param aout Output allocation. Must match creation element
    108      *             type.
    109      */
    110     public void forEach(Allocation aout) {
    111         forEach(0, null, aout, null);
    112     }
    113 
    114     /**
    115      * Get a KernelID for this intrinsic kernel.
    116      *
    117      * @return Script.KernelID The KernelID object.
    118      */
    119     public Script.KernelID getKernelID() {
    120         return createKernelID(0, 2, null, null);
    121     }
    122 
    123     /**
    124      * Get a FieldID for the input field of this intrinsic.
    125      *
    126      * @return Script.FieldID The FieldID object.
    127      */
    128     public Script.FieldID getFieldID_Input() {
    129         return createFieldID(1, null);
    130     }
    131 
    132 }
    133 
    134