Home | History | Annotate | Download | only in filterfw 
      1 /*
      2  * Copyright (C) 2011 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 androidx.media.filterpacks.image;
     18 
     19 import androidx.media.filterfw.Filter;
     20 import androidx.media.filterfw.FrameImage2D;
     21 import androidx.media.filterfw.FrameType;
     22 import androidx.media.filterfw.ImageShader;
     23 import androidx.media.filterfw.InputPort;
     24 import androidx.media.filterfw.MffContext;
     25 import androidx.media.filterfw.OutputPort;
     26 import androidx.media.filterfw.Signature;
     27 
     28 public class BrightnessFilter extends Filter {
     29 
     30     private float mBrightness = 1.0f;
     31     private ImageShader mShader;
     32 
     33     private static final String mBrightnessShader =
     34             "precision mediump float;\n" +
     35             "uniform sampler2D tex_sampler_0;\n" +
     36             "uniform float brightness;\n" +
     37             "varying vec2 v_texcoord;\n" +
     38             "void main() {\n" +
     39             "  vec4 color = texture2D(tex_sampler_0, v_texcoord);\n" +
     40             "  if (brightness < 0.5) {\n" +
     41             "    gl_FragColor = color * (2.0 * brightness);\n" +
     42             "  } else {\n" +
     43             "    vec4 diff = 1.0 - color;\n" +
     44             "    gl_FragColor = color + diff * (2.0 * (brightness - 0.5));\n" +
     45             "  }\n" +
     46             "}\n";
     47 
     48 
     49     public BrightnessFilter(MffContext context, String name) {
     50         super(context, name);
     51     }
     52 
     53     @Override
     54     public Signature getSignature() {
     55         FrameType imageIn = FrameType.image2D(FrameType.ELEMENT_RGBA8888, FrameType.READ_GPU);
     56         FrameType imageOut = FrameType.image2D(FrameType.ELEMENT_RGBA8888, FrameType.WRITE_GPU);
     57         return new Signature()
     58             .addInputPort("image", Signature.PORT_REQUIRED, imageIn)
     59             .addInputPort("brightness", Signature.PORT_OPTIONAL, FrameType.single(float.class))
     60             .addOutputPort("image", Signature.PORT_REQUIRED, imageOut)
     61             .disallowOtherPorts();
     62     }
     63 
     64     @Override
     65     public void onInputPortOpen(InputPort port) {
     66         if (port.getName().equals("brightness")) {
     67             port.bindToFieldNamed("mBrightness");
     68             port.setAutoPullEnabled(true);
     69         }
     70     }
     71 
     72     @Override
     73     protected void onPrepare() {
     74         mShader = new ImageShader(mBrightnessShader);
     75     }
     76 
     77     @Override
     78     protected void onProcess() {
     79         OutputPort outPort = getConnectedOutputPort("image");
     80         FrameImage2D inputImage = getConnectedInputPort("image").pullFrame().asFrameImage2D();
     81         int[] dim = inputImage.getDimensions();
     82         FrameImage2D outputImage = outPort.fetchAvailableFrame(dim).asFrameImage2D();
     83         mShader.setUniformValue("brightness", mBrightness);
     84         mShader.process(inputImage, outputImage);
     85         outPort.pushFrame(outputImage);
     86     }
     87 }
     88 
     89