Gray8VertVar.java

/*
 * Gray8VertVar.java
 *
 * Created on February 3, 2008, 4:32, PM
 *
 * To change this template, choose Tools | Template Manager
 * and open the template in the editor.
 *
 * Copyright 2008 by Jon A. Webb
 *     This program is free software: you can redistribute it and/or modify
 *    it under the terms of the GNU Lesser General Public License as published by
 *    the Free Software Foundation, either version 3 of the License, or
 *    (at your option) any later version.
 *
 *    This program is distributed in the hope that it will be useful,
 *    but WITHOUT ANY WARRANTY; without even the implied warranty of
 *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *    GNU Lesser General Public License for more details.
 *
 *    You should have received a copy of the Lesser GNU General Public License
 *    along with this program.  If not, see <http://www.gnu.org/licenses/>.
 *
 */

package jjil.algorithm;
import jjil.core.Error;
import jjil.core.Gray16Image;
import jjil.core.Gray8Image;
import jjil.core.Image;
import jjil.core.PipelineStage;

/**
 * Computes the variance of pixels vertically distributed around
 * the current pixel.
 * @author webb
 */
public class Gray8VertVar extends PipelineStage {
	/**
	 * The window size -- pixels within nWindow of the current
	 * pixel are included in the window.
	 */
	int nWindow;
	/**
	 * The output image.
	 */
	Gray16Image g16 = null;
    
    /**
     * Creates a new instance of Gray8VertVar
     * @param nWindow height of window to calculate variance over.
     */
    public Gray8VertVar(int nWindow) {
    	this.nWindow = nWindow;
    }
    
    /** Compute the vertical variance of pixels within nWindow
     * of the current pixel.
     * @param image the input Gray8Image
     * @throws jjil.core.Error if image is not a Gray8Image
     */
    public void push(Image image) throws jjil.core.Error {
        if (!(image instanceof Gray8Image)) {
            throw new Error(
    				Error.PACKAGE.ALGORITHM,
    				ErrorCodes.IMAGE_NOT_GRAY8IMAGE,
    				image.toString(),
    				null,
    				null);
        }
        if (this.g16 == null || 
        	this.g16.getWidth() != image.getWidth() ||
        	this.g16.getHeight() != image.getHeight()) {
        	this.g16 = new Gray16Image(
        			image.getWidth(), 
        			image.getHeight());
        }
        Gray8Image input = (Gray8Image) image;
        byte[] bIn = input.getData();
        int cHeight = input.getHeight();
        int cWidth = input.getWidth();
        short[] sOut = g16.getData();
        for (int i=0; i<cWidth; i++) {
	        int nSum = 0;
	        int nSumSq = 0;
	        int nCount = 0;
	        for (int j=0; j<nWindow; j++) {
	        	nSum += bIn[j*cWidth + i];
	        	nSumSq += bIn[j*cWidth + i] * bIn[j*cWidth + i];
	        	nCount ++;
	        }
            for (int j=0; j<cHeight; j++) {
            	if (j + this.nWindow < cHeight) {
            		nSum += bIn[(j + this.nWindow)*cWidth + i];
            		nSumSq += bIn[(j + this.nWindow)*cWidth + i] *
            			bIn[(j + this.nWindow)*cWidth + i];
            		nCount ++;
            	}
            	if (j >= this.nWindow) {
            		nSum -= bIn[(j - this.nWindow)*cWidth + i];
            		nSumSq -= bIn[(j - this.nWindow)*cWidth + i] *
            			bIn[(j - this.nWindow)*cWidth + i];
            		nCount --;
            	}
            	short nVar = (short)
            		Math.min(Short.MAX_VALUE, 
            			(nSumSq - nSum * nSum / nCount) / (nCount - 1)); 
            	sOut[j*cWidth + i] = nVar;
            }
        }
        super.setOutput(g16);
    }
    
}