130ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni/* 230ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni * Copyright (C) 2011 The Android Open Source Project 330ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni * 430ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni * Licensed under the Apache License, Version 2.0 (the "License"); 530ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni * you may not use this file except in compliance with the License. 630ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni * You may obtain a copy of the License at 730ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni * 830ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni * http://www.apache.org/licenses/LICENSE-2.0 930ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni * 1030ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni * Unless required by applicable law or agreed to in writing, software 1130ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni * distributed under the License is distributed on an "AS IS" BASIS, 1230ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 1330ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni * See the License for the specific language governing permissions and 1430ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni * limitations under the License. 1530ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni */ 1630ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni 1730ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni 1830ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceronipackage android.filterpacks.imageproc; 1930ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni 2030ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroniimport android.filterfw.core.Filter; 21692c3bf7a1c7e2d5b1066cf7a1b057e993742c87Marius Rennimport android.filterfw.core.FilterContext; 2230ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroniimport android.filterfw.core.Frame; 2330ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroniimport android.filterfw.core.FrameFormat; 2430ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroniimport android.filterfw.core.GLFrame; 2530ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroniimport android.filterfw.core.NativeProgram; 2630ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroniimport android.filterfw.core.NativeFrame; 2730ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroniimport android.filterfw.core.Program; 2830ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroniimport android.filterfw.core.ShaderProgram; 2930ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroniimport android.filterfw.geometry.Quad; 30c0017fd82acec5d0427306ea5f536c3d78854f95Marius Rennimport android.filterfw.format.ImageFormat; 3121d0ac7403b836e32e2bdbdc8dc98f42b2dfa4e5Marius Rennimport android.filterfw.format.ObjectFormat; 3221d0ac7403b836e32e2bdbdc8dc98f42b2dfa4e5Marius Renn 3330ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroniimport android.opengl.GLES20; 3430ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni 35a3bfbe5389c6146abe318a7add3fa688d69bc01bEino-Ville Talvala/** 36a3bfbe5389c6146abe318a7add3fa688d69bc01bEino-Ville Talvala * @hide 37a3bfbe5389c6146abe318a7add3fa688d69bc01bEino-Ville Talvala */ 3830ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceronipublic class DrawOverlayFilter extends Filter { 3930ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni 4030ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni private ShaderProgram mProgram; 4130ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni 4230ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni public DrawOverlayFilter(String name) { 4330ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni super(name); 4430ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni } 4530ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni 4633d107bf315c9cc01a02a7a4a2c10a01f62e8c85Wei Hua @Override 4721d0ac7403b836e32e2bdbdc8dc98f42b2dfa4e5Marius Renn public void setupPorts() { 4821d0ac7403b836e32e2bdbdc8dc98f42b2dfa4e5Marius Renn FrameFormat imageFormatMask = ImageFormat.create(ImageFormat.COLORSPACE_RGBA, 4921d0ac7403b836e32e2bdbdc8dc98f42b2dfa4e5Marius Renn FrameFormat.TARGET_GPU); 5021d0ac7403b836e32e2bdbdc8dc98f42b2dfa4e5Marius Renn addMaskedInputPort("source", imageFormatMask); 5121d0ac7403b836e32e2bdbdc8dc98f42b2dfa4e5Marius Renn addMaskedInputPort("overlay", imageFormatMask); 52bf4aaebc555cfb1e49ee411e3477203749fe6a11Marius Renn addMaskedInputPort("box", ObjectFormat.fromClass(Quad.class, FrameFormat.TARGET_SIMPLE)); 5321d0ac7403b836e32e2bdbdc8dc98f42b2dfa4e5Marius Renn addOutputBasedOnInput("image", "source"); 5430ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni } 5530ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni 5633d107bf315c9cc01a02a7a4a2c10a01f62e8c85Wei Hua @Override 5721d0ac7403b836e32e2bdbdc8dc98f42b2dfa4e5Marius Renn public FrameFormat getOutputFormat(String portName, FrameFormat inputFormat) { 5821d0ac7403b836e32e2bdbdc8dc98f42b2dfa4e5Marius Renn return inputFormat; 5930ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni } 6030ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni 6133d107bf315c9cc01a02a7a4a2c10a01f62e8c85Wei Hua @Override 62511360e61650864ea22a171159efe073c80d0cdbMarius Renn public void prepare(FilterContext context) { 63511360e61650864ea22a171159efe073c80d0cdbMarius Renn mProgram = ShaderProgram.createIdentity(context); 6430ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni } 6530ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni 6633d107bf315c9cc01a02a7a4a2c10a01f62e8c85Wei Hua @Override 6721d0ac7403b836e32e2bdbdc8dc98f42b2dfa4e5Marius Renn public void process(FilterContext env) { 6830ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni // Get input frame 6921d0ac7403b836e32e2bdbdc8dc98f42b2dfa4e5Marius Renn Frame sourceFrame = pullInput("source"); 7021d0ac7403b836e32e2bdbdc8dc98f42b2dfa4e5Marius Renn Frame overlayFrame = pullInput("overlay"); 7121d0ac7403b836e32e2bdbdc8dc98f42b2dfa4e5Marius Renn Frame boxFrame = pullInput("box"); 7230ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni 7330ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni // Get the box 7430ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni Quad box = (Quad)boxFrame.getObjectValue(); 7530ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni box = box.translated(1.0f, 1.0f).scaled(2.0f); 7630ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni 7730ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni mProgram.setTargetRegion(box); 7830ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni 7930ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni // Create output frame with copy of input 80c0017fd82acec5d0427306ea5f536c3d78854f95Marius Renn Frame output = env.getFrameManager().newFrame(sourceFrame.getFormat()); 8130ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni output.setDataFromFrame(sourceFrame); 8230ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni 8330ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni // Draw onto output 8430ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni mProgram.process(overlayFrame, output); 8530ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni 8630ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni // Push output 8721d0ac7403b836e32e2bdbdc8dc98f42b2dfa4e5Marius Renn pushOutput("image", output); 8830ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni 8930ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni // Release pushed frame 9030ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni output.release(); 9130ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni } 9230ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni} 93