DrawOverlayFilter.java revision 21d0ac7403b836e32e2bdbdc8dc98f42b2dfa4e5
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 3530ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceronipublic class DrawOverlayFilter extends Filter { 3630ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni 3730ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni private ShaderProgram mProgram; 3830ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni 3930ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni public DrawOverlayFilter(String name) { 4030ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni super(name); 4130ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni } 4230ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni 4333d107bf315c9cc01a02a7a4a2c10a01f62e8c85Wei Hua @Override 4421d0ac7403b836e32e2bdbdc8dc98f42b2dfa4e5Marius Renn public void setupPorts() { 4521d0ac7403b836e32e2bdbdc8dc98f42b2dfa4e5Marius Renn FrameFormat imageFormatMask = ImageFormat.create(ImageFormat.COLORSPACE_RGBA, 4621d0ac7403b836e32e2bdbdc8dc98f42b2dfa4e5Marius Renn FrameFormat.TARGET_GPU); 4721d0ac7403b836e32e2bdbdc8dc98f42b2dfa4e5Marius Renn addMaskedInputPort("source", imageFormatMask); 4821d0ac7403b836e32e2bdbdc8dc98f42b2dfa4e5Marius Renn addMaskedInputPort("overlay", imageFormatMask); 4921d0ac7403b836e32e2bdbdc8dc98f42b2dfa4e5Marius Renn addMaskedInputPort("box", ObjectFormat.fromClass(Quad.class, FrameFormat.TARGET_JAVA)); 5021d0ac7403b836e32e2bdbdc8dc98f42b2dfa4e5Marius Renn addOutputBasedOnInput("image", "source"); 5130ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni } 5230ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni 5333d107bf315c9cc01a02a7a4a2c10a01f62e8c85Wei Hua @Override 5421d0ac7403b836e32e2bdbdc8dc98f42b2dfa4e5Marius Renn public FrameFormat getOutputFormat(String portName, FrameFormat inputFormat) { 5521d0ac7403b836e32e2bdbdc8dc98f42b2dfa4e5Marius Renn return inputFormat; 5630ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni } 5730ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni 5833d107bf315c9cc01a02a7a4a2c10a01f62e8c85Wei Hua @Override 59692c3bf7a1c7e2d5b1066cf7a1b057e993742c87Marius Renn public void prepare(FilterContext env) { 6030ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni mProgram = ShaderProgram.createIdentity(); 6130ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni } 6230ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni 6333d107bf315c9cc01a02a7a4a2c10a01f62e8c85Wei Hua @Override 6421d0ac7403b836e32e2bdbdc8dc98f42b2dfa4e5Marius Renn public void process(FilterContext env) { 6530ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni // Get input frame 6621d0ac7403b836e32e2bdbdc8dc98f42b2dfa4e5Marius Renn Frame sourceFrame = pullInput("source"); 6721d0ac7403b836e32e2bdbdc8dc98f42b2dfa4e5Marius Renn Frame overlayFrame = pullInput("overlay"); 6821d0ac7403b836e32e2bdbdc8dc98f42b2dfa4e5Marius Renn Frame boxFrame = pullInput("box"); 6930ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni 7030ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni // Get the box 7130ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni Quad box = (Quad)boxFrame.getObjectValue(); 7230ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni box = box.translated(1.0f, 1.0f).scaled(2.0f); 7330ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni 7430ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni mProgram.setTargetRegion(box); 7530ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni 7630ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni // Create output frame with copy of input 77c0017fd82acec5d0427306ea5f536c3d78854f95Marius Renn Frame output = env.getFrameManager().newFrame(sourceFrame.getFormat()); 7830ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni output.setDataFromFrame(sourceFrame); 7930ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni 8030ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni // Draw onto output 8130ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni mProgram.process(overlayFrame, output); 8230ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni 8330ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni // Push output 8421d0ac7403b836e32e2bdbdc8dc98f42b2dfa4e5Marius Renn pushOutput("image", output); 8530ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni 8630ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni // Release pushed frame 8730ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni output.release(); 8830ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni } 8930ab3fc173709a491c9e2e103f53fb7c0d1b96b7Rodrigo Carceroni} 90