1227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks/* 2227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks * Copyright (C) 2012 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 Hendricks// Calculates the mean and standard deviation of the values in the input image. 18227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks// It takes in an RGBA image, but assumes that r, g, b, a are all the same values. 19227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 20227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendrickspackage androidx.media.filterpacks.numeric; 21227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 22227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricksimport android.util.Log; 23227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 24227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricksimport androidx.media.filterfw.Filter; 25227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricksimport androidx.media.filterfw.Frame; 26227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricksimport androidx.media.filterfw.FrameBuffer2D; 27227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricksimport androidx.media.filterfw.FrameType; 28227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricksimport androidx.media.filterfw.FrameValue; 29227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricksimport androidx.media.filterfw.InputPort; 30227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricksimport androidx.media.filterfw.MffContext; 31227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricksimport androidx.media.filterfw.OutputPort; 32227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricksimport androidx.media.filterfw.Signature; 33227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricksimport androidx.media.filterfw.geometry.Quad; 34227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 35227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricksimport java.nio.ByteBuffer; 36227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 37227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks/** 38227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks * Get the sample mean and variance of a 2-D buffer of bytes over a given rectangle. 39227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks * TODO: Add more statistics as needed. 40227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks * TODO: Check if crop rectangle is necessary to be included in this filter. 41227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks */ 42227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendrickspublic class StatsFilter extends Filter { 43227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 44227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks private static final int MEAN_INDEX = 0; 45227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks private static final int STDEV_INDEX = 1; 46227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 47227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks private final float[] mStats = new float[2]; 48227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 49227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks private Quad mCropRect = Quad.fromRect(0f, 0f, 1f, 1f); 50227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks private static final String TAG = "StatsFilter"; 51227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks private static boolean mLogVerbose = Log.isLoggable(TAG, Log.VERBOSE); 52227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 53227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks /** 54227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks * @param context 55227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks * @param name 56227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks */ 57227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks public StatsFilter(MffContext context, String name) { 58227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks super(context, name); 59227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks } 60227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 61227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks @Override 62227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks public Signature getSignature() { 63227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks FrameType inputFrame = FrameType.buffer2D(FrameType.ELEMENT_INT8); 64227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks FrameType floatT = FrameType.single(float.class); 65227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks return new Signature() 66227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks .addInputPort("buffer", Signature.PORT_REQUIRED, inputFrame) 67227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks .addInputPort("cropRect", Signature.PORT_OPTIONAL, FrameType.single(Quad.class)) 68227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks .addOutputPort("mean", Signature.PORT_REQUIRED, floatT) 69227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks .addOutputPort("stdev", Signature.PORT_REQUIRED, floatT) 70227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks .disallowOtherPorts(); 71227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks } 72227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 73227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks @Override 74227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks public void onInputPortOpen(InputPort port) { 75227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks if (port.getName().equals("cropRect")) { 76227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks port.bindToFieldNamed("mCropRect"); 77227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks port.setAutoPullEnabled(true); 78227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks } 79227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks } 80227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 81227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks private void calcMeanAndStd(ByteBuffer pixelBuffer, int width, int height, Quad quad) { 82227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks // Native 83227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks pixelBuffer.rewind(); 84227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks regionscore(pixelBuffer, width, height, quad.topLeft().x, quad.topLeft().y, 85227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks quad.bottomRight().x, quad.bottomRight().y, mStats); 86227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks if (mLogVerbose) { 87227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks Log.v(TAG, "Native calc stats: Mean = " + mStats[MEAN_INDEX] + ", Stdev = " 88227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks + mStats[STDEV_INDEX]); 89227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks } 90227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks } 91227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 92227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks /** 93227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks * @see androidx.media.filterfw.Filter#onProcess() 94227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks */ 95227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks @Override 96227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks protected void onProcess() { 97227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks FrameBuffer2D inputFrame = getConnectedInputPort("buffer").pullFrame().asFrameImage2D(); 98227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks ByteBuffer pixelBuffer = inputFrame.lockBytes(Frame.MODE_READ); 99227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 100227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks calcMeanAndStd(pixelBuffer, inputFrame.getWidth(), inputFrame.getHeight(), mCropRect); 101227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks inputFrame.unlock(); 102227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 103227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks OutputPort outPort = getConnectedOutputPort("mean"); 104227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks FrameValue outFrame = outPort.fetchAvailableFrame(null).asFrameValue(); 105227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks outFrame.setValue(mStats[MEAN_INDEX]); 106227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks outPort.pushFrame(outFrame); 107227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 108227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks OutputPort outPortStdev = getConnectedOutputPort("stdev"); 109227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks FrameValue outFrameStdev = outPortStdev.fetchAvailableFrame(null).asFrameValue(); 110227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks outFrameStdev.setValue(mStats[STDEV_INDEX]); 111227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks outPortStdev.pushFrame(outFrameStdev); 112227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks } 113227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 114227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks private native void regionscore(ByteBuffer imageBuffer, int width, int height, float left, 115227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks float top, float right, float bottom, float[] statsArray); 116227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 117227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks static { 118227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks System.loadLibrary("smartcamera_jni"); 119227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks } 120227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks} 121