package ij.plugin;
import ij.*;
import ij.gui.*;
import ij.process.*;
import ij.measure.Calibration;
import java.awt.*;
import java.util.Vector;

/** Implements the Image/HyperStacks/Reduce Dimensionality command. */
public class HyperStackReducer implements PlugIn, DialogListener {
	ImagePlus imp;
	int channels1, slices1, frames1;
	int channels2, slices2, frames2;
	double imageSize;
	static boolean keep = true;
	
	/** Default constructor */
	public HyperStackReducer() {
	}

	/** Constructs a HyperStackReducer using the specified source image. */
	public HyperStackReducer(ImagePlus imp) {
		this.imp = imp;
		this.channels1 = this.channels2 = imp.getNChannels();
		this.slices1 = this.slices2 = imp.getNSlices();
		this.frames1 = this.frames2 = imp.getNFrames();
	}

	public void run(String arg) {
		//IJ.log("HyperStackReducer-1");
		imp = IJ.getImage();
		if (!imp.isHyperStack() && imp.getNChannels()==1) {
			IJ.error("Reducer", "HyperStack required");
			return;
		}
		int width = imp.getWidth();
		int height = imp.getHeight();
		imageSize = width*height*imp.getBytesPerPixel()/(1024.0*1024.0);
		channels1 = channels2 = imp.getNChannels();
		slices1 = slices2 = imp.getNSlices();
		frames1 = frames2 = imp.getNFrames();
		int z0 = imp.getSlice();
		int t0 = imp.getFrame();
		if (!showDialog())
			return;
		//IJ.log("HyperStackReducer-2: "+keep+" "+channels2+" "+slices2+" "+frames2);
		String title2 = keep?WindowManager.getUniqueName(imp.getTitle()):imp.getTitle();
		ImagePlus imp2 = null;
		if (keep) {
			imp2 = IJ.createImage(title2, imp.getBitDepth()+"-bit", width, height, channels2*slices2*frames2);
			if (imp2==null) return;
			imp2.setDimensions(channels2, slices2, frames2);
			imp2.setCalibration(imp.getCalibration());
			imp2.setOpenAsHyperStack(true);
		} else
			imp2 = imp.createHyperStack(title2, channels2, slices2, frames2, imp.getBitDepth());
		imp2.setProperty("Info", (String)imp.getProperty("Info"));		
		imp2.setProperties(imp.getPropertiesAsArray());
		reduce(imp2);
		if (channels2>1 && channels2==imp.getNChannels() && imp.isComposite()) {
			int mode = ((CompositeImage)imp).getMode();
			imp2 = new CompositeImage(imp2, mode);
			((CompositeImage)imp2).copyLuts(imp);
		} else {
			imp2.setDisplayRange(imp.getDisplayRangeMin(), imp.getDisplayRangeMax());
			if (imp.isComposite() && ((CompositeImage)imp).getMode()==IJ.GRAYSCALE)
				IJ.run(imp2, "Grays", "");
		}
		if (imp.getWindow()==null && !keep) {
			imp.setImage(imp2);
			return;
		}
		imp2.show();
		if (z0>1 || t0>1)
			imp2.setPosition(1, z0, t0);
		//IJ.log("HyperStackReducer-4");
		if (!keep) {
			imp.changes = false;
			imp.close();
		}
	}

	public void reduce(ImagePlus imp2) {
		this.channels2 = imp2.getNChannels();
		this.slices2 = imp2.getNSlices();
		this.frames2 = imp2.getNFrames();
		int channels = imp2.getNChannels();
		int slices = imp2.getNSlices();
		int frames = imp2.getNFrames();
		int c1 = imp.getChannel();
		int z1 = imp.getSlice();
		int t1 = imp.getFrame();
		int i = 1;
		int n = channels*slices*frames;
		ImageStack stack = imp.getStack();
		ImageStack stack2 = imp2.getStack();
		for (int c=1; c<=channels; c++) {
			if (channels==1) c = c1;
			LUT lut = imp.isComposite()?((CompositeImage)imp).getChannelLut():imp.getProcessor().getLut();
			imp.setPositionWithoutUpdate(c, 1, 1);
			ImageProcessor ip = imp.getProcessor();
			double min = ip.getMin();
			double max = ip.getMax();
			for (int z=1; z<=slices; z++) {
				if (slices==1) z = z1;
				for (int t=1; t<=frames; t++) {
					if (frames==1) t = t1;
					int n1 = imp.getStackIndex(c, z, t);
					ip = stack.getProcessor(n1);
					String label = stack.getSliceLabel(n1);
					int n2 = imp2.getStackIndex(c, z, t);
					if (stack2.getPixels(n2)!=null)
						stack2.getProcessor(n2).insert(ip, 0, 0);
					else
						stack2.setPixels(ip.getPixels(), n2);
					stack2.setSliceLabel(label, n2);
				}
			}
			if (lut!=null) {
				if (imp2.isComposite())
					((CompositeImage)imp2).setChannelLut(lut);
				else
					imp2.getProcessor().setColorModel(lut);
			}
			imp2.getProcessor().setMinAndMax(min, max);
		}
		imp.setPosition(c1, z1, t1);
		imp2.resetStack();
		imp2.setPosition(1, 1, 1);
		Overlay overlay = imp.getOverlay();
		if (overlay!=null && !imp.getHideOverlay())
			imp2.setOverlay(reduce(overlay));
	}

	//Added by Marcel Boeglin 2013.11.29
	/** Returns a copy of 'overlay', limited to the dimensions of the reduced image. */
	private Overlay reduce(Overlay overlay) {
		int c1 = imp.getChannel();
		int z1 = imp.getSlice();
		int t1 = imp.getFrame();
		Overlay overlay2 = overlay.duplicate();
		if (channels2==1 && slices2==slices1 && frames2==frames1)
			overlay2.crop(c1, c1, 1, slices1, 1, frames1);
		else if (channels2==channels1 && slices2==1 && frames2==frames1)
			overlay2.crop(1, channels1, z1, z1, 1, frames1);
		else if (channels2==channels1 && slices2==slices1 && frames2==1)
			overlay2.crop(1, channels1, 1, slices1, t1, t1);
		else if (channels2==channels1 && slices2==1 && frames2==1)
			overlay2.crop(1, channels1, z1, z1, t1, t1);
		else if (channels2==1 && slices2==slices1 && frames2==1)
			overlay2.crop(c1, c1, 1, slices1, t1, t1);
		else if (channels2==1 && slices2==1 && frames2==frames1)
			overlay2.crop(c1, c1, z1, z1, 1, frames1);
		else if (channels2==1 && slices2==1 && frames2==1)
			overlay2.crop(c1, c1, z1, z1, t1, t1);
		return overlay2;
	}
    
    boolean showDialog() {
		GenericDialog gd = new GenericDialog("Reduce");
		gd.setInsets(10, 20, 5);
		gd.addMessage("Create image with:");
		gd.setInsets(0, 35, 0);
		if (channels1!=1) gd.addCheckbox("Channels ("+channels1+")", true);
		gd.setInsets(0, 35, 0);
		if (slices1!=1) gd.addCheckbox("Slices ("+slices1+")", true);
		gd.setInsets(0, 35, 0);
		if (frames1!=1) gd.addCheckbox("Frames ("+frames1+")", true);
		gd.setInsets(5, 20, 0);
		gd.addMessage(getNewDimensions()+"      ");
		gd.setInsets(15, 20, 0);
		gd.addCheckbox("Keep source", keep);
		gd.addDialogListener(this);
		gd.showDialog();
		if (gd.wasCanceled())
			return false;
		else
			return true;
	}

	public boolean dialogItemChanged(GenericDialog gd, AWTEvent e) {
		if (IJ.isMacOSX()) IJ.wait(100);
		if (channels1!=1) channels2 = gd.getNextBoolean()?channels1:1;
		if (slices1!=1) slices2 = gd.getNextBoolean()?slices1:1;
		if (frames1!=1) frames2 = gd.getNextBoolean()?frames1:1;
		keep = gd.getNextBoolean();
		if (imp!=null && imp.getWindow()!=null)
			((Label)gd.getMessage()).setText(getNewDimensions());
		return true;
	}

	String getNewDimensions() {
		String s = channels2+"x"+slices2+"x"+frames2;
		s += " ("+(int)Math.round(imageSize*channels2*slices2*frames2)+"MB)";
		return(s);
	}

}
