/**
Marvin Project <2007-2009>

Initial version by:

Danilo Rosetto Munoz
Fabio Andrijauskas
Gabriel Ambrosio Archanjo

site: http://marvinproject.sourceforge.net

GPL
Copyright (C) <2007>  

This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 2 of the License, or
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 General Public License for more details.

You should have received a copy of the GNU General Public License along
with this program; if not, write to the Free Software Foundation, Inc.,
51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
*/

package org.marvinproject.image.statistical.median;

import marvin.gui.MarvinFilterWindow;
import marvin.image.MarvinImage;
import marvin.image.MarvinImageMask;
import marvin.performance.MarvinPerformanceMeter;
import marvin.plugin.MarvinAbstractImagePlugin;
import marvin.util.MarvinAttributes;

/** 
 * @author Fabio Andrijauskas
 */
public class Median extends MarvinAbstractImagePlugin {

	MarvinAttributes attributes;
	MarvinPerformanceMeter performanceMeter;

	public void load() {
		attributes = getAttributes();
		attributes.set("size", 3);
		//attributes.set("shift", 0);
		//attributes.set("circlesDistance", 0);
		performanceMeter = new MarvinPerformanceMeter();
	}

	public void process
	(
		MarvinImage a_imageIn, 
		MarvinImage a_imageOut,
		MarvinAttributes a_attributesOut,
		MarvinImageMask a_mask, 
		boolean a_previewMode
	)
	{
		performanceMeter.start("Median Filter");
		
		
		int l_size = (Integer)attributes.get("size");
		int l_totalRed = 0;
		int l_totalGreen = 0;
		int l_totalBlue = 0;
		int qtd = 0;
		int tmpx = 0;
		int tmpy = 0;
		int width = a_imageIn.getWidth();
		int height = a_imageIn.getHeight();
		
		performanceMeter.enableProgressBar("Median Filter" ,a_imageIn.getWidth() *  a_imageIn.getHeight() );
		performanceMeter.startEvent("Median Filter");
		
		boolean[][] l_arrMask = a_mask.getMaskArray();
		
		for (int x = 0; x < width; x++) {
			for (int y = 0; y < height; y++) {
				if(l_arrMask != null && !l_arrMask[x][y]){
					continue;
				}

					tmpx  = x - l_size;
					tmpy  = y - l_size;
					
					if(tmpx < 0)
						tmpx = 0;
					if(tmpy < 0)
						tmpy = 0;
					
					int finalX =  x + l_size;
					
					int finalY = y + l_size;
					
					if(finalX > width)
						finalX = width;
					
					if(finalY > height)
						finalY = height;
					
					for (int xm = tmpx; xm < finalX ; xm++) {
						for (int ym = tmpy; ym <  y + l_size ; ym++) {
							
							if(xm >= 0 && xm < width && ym >= 0 && ym < height  )
							{
								int rgb = a_imageIn.getIntColor(xm, ym);
								l_totalRed += (int)(rgb & 0x00FF0000) >>> 16;
								l_totalGreen += (int) ((rgb & 0x0000FF00) >>> 8);
								l_totalBlue += (int) (rgb & 0x000000FF);
								qtd++;
							}
						}
					}

					a_imageOut.setIntColor(x, y, l_totalRed / qtd, l_totalGreen / qtd, l_totalBlue / qtd);

					l_totalRed = 0;
					l_totalGreen = 0;
					l_totalBlue = 0;
					qtd = 0;
			}
			performanceMeter.incProgressBar(height);
			performanceMeter.stepsFinished(height);
		}
		performanceMeter.finishEvent();
		performanceMeter.finish();
	}

	public void show() {

		MarvinFilterWindow l_filterWindow = new MarvinFilterWindow("Avarage Filter", 420,350, getImagePanel(), this);
		l_filterWindow.addLabel("lblWidth", "Size:");
		l_filterWindow.addTextField("txtSize", "size", attributes);
		l_filterWindow.newComponentRow();
		l_filterWindow.setVisible(true);

	}

}