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 18package android.filterpacks.imageproc; 19 20import android.filterfw.core.Filter; 21import android.filterfw.core.FilterContext; 22import android.filterfw.core.Frame; 23import android.filterfw.core.FrameFormat; 24import android.filterfw.core.GenerateFieldPort; 25import android.filterfw.core.GLFrame; 26import android.filterfw.core.NativeProgram; 27import android.filterfw.core.NativeFrame; 28import android.filterfw.core.Program; 29import android.filterfw.core.ShaderProgram; 30import android.filterfw.geometry.Quad; 31import android.filterfw.format.ImageFormat; 32import android.filterfw.format.ObjectFormat; 33 34import android.opengl.GLES20; 35 36/** 37 * @hide 38 */ 39public class DrawRectFilter extends Filter { 40 41 @GenerateFieldPort(name = "colorRed", hasDefault = true) 42 private float mColorRed = 0.8f; 43 44 @GenerateFieldPort(name = "colorGreen", hasDefault = true) 45 private float mColorGreen = 0.8f; 46 47 @GenerateFieldPort(name = "colorBlue", hasDefault = true) 48 private float mColorBlue = 0.0f; 49 50 private final String mVertexShader = 51 "attribute vec4 aPosition;\n" + 52 "void main() {\n" + 53 " gl_Position = aPosition;\n" + 54 "}\n"; 55 56 private final String mFixedColorFragmentShader = 57 "precision mediump float;\n" + 58 "uniform vec4 color;\n" + 59 "void main() {\n" + 60 " gl_FragColor = color;\n" + 61 "}\n"; 62 63 private ShaderProgram mProgram; 64 65 66 public DrawRectFilter(String name) { 67 super(name); 68 } 69 70 @Override 71 public void setupPorts() { 72 addMaskedInputPort("image", ImageFormat.create(ImageFormat.COLORSPACE_RGBA, 73 FrameFormat.TARGET_GPU)); 74 addMaskedInputPort("box", ObjectFormat.fromClass(Quad.class, FrameFormat.TARGET_SIMPLE)); 75 addOutputBasedOnInput("image", "image"); 76 } 77 78 @Override 79 public FrameFormat getOutputFormat(String portName, FrameFormat inputFormat) { 80 return inputFormat; 81 } 82 83 @Override 84 public void prepare(FilterContext context) { 85 mProgram = new ShaderProgram(context, mVertexShader, mFixedColorFragmentShader); 86 } 87 88 @Override 89 public void process(FilterContext env) { 90 // Get input frame 91 Frame imageFrame = pullInput("image"); 92 Frame boxFrame = pullInput("box"); 93 94 // Get the box 95 Quad box = (Quad)boxFrame.getObjectValue(); 96 box = box.scaled(2.0f).translated(-1.0f, -1.0f); 97 98 // Create output frame with copy of input 99 GLFrame output = (GLFrame)env.getFrameManager().duplicateFrame(imageFrame); 100 101 // Draw onto output 102 output.focus(); 103 renderBox(box); 104 105 // Push output 106 pushOutput("image", output); 107 108 // Release pushed frame 109 output.release(); 110 } 111 112 private void renderBox(Quad box) { 113 final int FLOAT_SIZE = 4; 114 115 // Get current values 116 float[] color = {mColorRed, mColorGreen, mColorBlue, 1f}; 117 float[] vertexValues = { box.p0.x, box.p0.y, 118 box.p1.x, box.p1.y, 119 box.p3.x, box.p3.y, 120 box.p2.x, box.p2.y }; 121 122 // Set the program variables 123 mProgram.setHostValue("color", color); 124 mProgram.setAttributeValues("aPosition", vertexValues, 2); 125 mProgram.setVertexCount(4); 126 127 // Draw 128 mProgram.beginDrawing(); 129 GLES20.glLineWidth(1.0f); 130 GLES20.glDrawArrays(GLES20.GL_LINE_LOOP, 0, 4); 131 } 132} 133