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.transform; 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 Hendricksimport androidx.media.filterfw.geometry.Quad; 28227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 29227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendrickspublic class RotateFilter extends Filter { 30227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 31227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks private Quad mSourceRect = Quad.fromRect(0f, 0f, 1f, 1f); 32227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks private float mRotateAngle = 0; 33227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks private ImageShader mShader; 34227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 35227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks public RotateFilter(MffContext context, String name) { 36227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks super(context, name); 37227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks } 38227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 39227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks @Override 40227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks public Signature getSignature() { 41227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks FrameType imageIn = FrameType.image2D(FrameType.ELEMENT_RGBA8888, FrameType.READ_GPU); 42227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks FrameType imageOut = FrameType.image2D(FrameType.ELEMENT_RGBA8888, FrameType.WRITE_GPU); 43227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks return new Signature() 44227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks .addInputPort("image", Signature.PORT_REQUIRED, imageIn) 45227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks .addInputPort("rotateAngle", Signature.PORT_REQUIRED, FrameType.single(float.class)) 46227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks .addInputPort("sourceRect", Signature.PORT_OPTIONAL, FrameType.single(Quad.class)) 47227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks .addOutputPort("image", Signature.PORT_REQUIRED, imageOut) 48227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks .disallowOtherPorts(); 49227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks } 50227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 51227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks @Override 52227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks public void onInputPortOpen(InputPort port) { 53227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks if (port.getName().equals("rotateAngle")) { 54227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks port.bindToFieldNamed("mRotateAngle"); 55227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks port.setAutoPullEnabled(true); 56227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks } else if (port.getName().equals("sourceRect")) { 57227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks port.bindToFieldNamed("mSourceRect"); 58227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks port.setAutoPullEnabled(true); 59227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks } 60227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks } 61227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 62227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks @Override 63227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks protected void onPrepare() { 64227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks mShader = ImageShader.createIdentity(); 65227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks } 66227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 67227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks @Override 68227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks protected void onProcess() { 69227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks OutputPort outPort = getConnectedOutputPort("image"); 70227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks FrameImage2D inputImage = getConnectedInputPort("image").pullFrame().asFrameImage2D(); 71227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks int[] inDims = inputImage.getDimensions(); 72227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 73227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks FrameImage2D outputImage = outPort.fetchAvailableFrame(inDims).asFrameImage2D(); 74227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks mShader.setSourceQuad(mSourceRect); 75227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks Quad targetQuad = mSourceRect.rotated((float) (mRotateAngle / 180 * Math.PI)); 76227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks mShader.setTargetQuad(targetQuad); 77227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks mShader.process(inputImage, outputImage); 78227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks outPort.pushFrame(outputImage); 79227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks } 80227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks} 81