1227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks/* 2227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks * Copyright (C) 2011 The Android Open Source Project 3227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks * 4227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks * Licensed under the Apache License, Version 2.0 (the "License"); 5227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks * you may not use this file except in compliance with the License. 6227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks * You may obtain a copy of the License at 7227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks * 8227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks * http://www.apache.org/licenses/LICENSE-2.0 9227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks * 10227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks * Unless required by applicable law or agreed to in writing, software 11227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks * distributed under the License is distributed on an "AS IS" BASIS, 12227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks * See the License for the specific language governing permissions and 14227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks * limitations under the License. 15227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks */ 16227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 17227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendrickspackage androidx.media.filterpacks.image; 18227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 19227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricksimport androidx.media.filterfw.Filter; 20227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricksimport androidx.media.filterfw.FrameImage2D; 21227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricksimport androidx.media.filterfw.FrameType; 22227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricksimport androidx.media.filterfw.ImageShader; 23227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricksimport androidx.media.filterfw.InputPort; 24227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricksimport androidx.media.filterfw.MffContext; 25227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricksimport androidx.media.filterfw.OutputPort; 26227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricksimport androidx.media.filterfw.Signature; 27227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 28227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendrickspublic class BrightnessFilter extends Filter { 29227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 30227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks private float mBrightness = 1.0f; 31227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks private ImageShader mShader; 32227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 33227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks private static final String mBrightnessShader = 34227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks "precision mediump float;\n" + 35227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks "uniform sampler2D tex_sampler_0;\n" + 36227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks "uniform float brightness;\n" + 37227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks "varying vec2 v_texcoord;\n" + 38227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks "void main() {\n" + 39227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks " vec4 color = texture2D(tex_sampler_0, v_texcoord);\n" + 40227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks " if (brightness < 0.5) {\n" + 41227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks " gl_FragColor = color * (2.0 * brightness);\n" + 42227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks " } else {\n" + 43227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks " vec4 diff = 1.0 - color;\n" + 44227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks " gl_FragColor = color + diff * (2.0 * (brightness - 0.5));\n" + 45227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks " }\n" + 46227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks "}\n"; 47227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 48227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 49227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks public BrightnessFilter(MffContext context, String name) { 50227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks super(context, name); 51227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks } 52227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 53227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks @Override 54227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks public Signature getSignature() { 55227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks FrameType imageIn = FrameType.image2D(FrameType.ELEMENT_RGBA8888, FrameType.READ_GPU); 56227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks FrameType imageOut = FrameType.image2D(FrameType.ELEMENT_RGBA8888, FrameType.WRITE_GPU); 57227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks return new Signature() 58227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks .addInputPort("image", Signature.PORT_REQUIRED, imageIn) 59227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks .addInputPort("brightness", Signature.PORT_OPTIONAL, FrameType.single(float.class)) 60227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks .addOutputPort("image", Signature.PORT_REQUIRED, imageOut) 61227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks .disallowOtherPorts(); 62227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks } 63227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 64227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks @Override 65227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks public void onInputPortOpen(InputPort port) { 66227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks if (port.getName().equals("brightness")) { 67227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks port.bindToFieldNamed("mBrightness"); 68227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks port.setAutoPullEnabled(true); 69227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks } 70227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks } 71227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 72227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks @Override 73227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks protected void onPrepare() { 74227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks mShader = new ImageShader(mBrightnessShader); 75227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks } 76227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 77227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks @Override 78227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks protected void onProcess() { 79227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks OutputPort outPort = getConnectedOutputPort("image"); 80227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks FrameImage2D inputImage = getConnectedInputPort("image").pullFrame().asFrameImage2D(); 81227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks int[] dim = inputImage.getDimensions(); 82227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks FrameImage2D outputImage = outPort.fetchAvailableFrame(dim).asFrameImage2D(); 83227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks mShader.setUniformValue("brightness", mBrightness); 84227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks mShader.process(inputImage, outputImage); 85227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks outPort.pushFrame(outputImage); 86227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks } 87227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks} 88227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 89