private void doRGBProjection(ImageStack stack) {
ImageStack[] channels = ChannelSplitter.splitRGB(stack, true);
ImagePlus red = new ImagePlus("Red", channels[0]);
ImagePlus green = new ImagePlus("Green", channels[1]);
ImagePlus blue = new ImagePlus("Blue", channels[2]);
imp.unlock();
ImagePlus saveImp = imp;
imp = red;
color = "(red)";
doProjection();
ImagePlus red2 = projImage;
imp = green;
color = "(green)";
doProjection();
ImagePlus green2 = projImage;
imp = blue;
color = "(blue)";
doProjection();
ImagePlus blue2 = projImage;
int w = red2.getWidth(), h = red2.getHeight(), d = red2.getStackSize();
if (method == SD_METHOD) {
ImageProcessor r = red2.getProcessor();
ImageProcessor g = green2.getProcessor();
ImageProcessor b = blue2.getProcessor();
double max = 0;
double rmax = r.getStatistics().max;
if (rmax > max) max = rmax;
double gmax = g.getStatistics().max;
if (gmax > max) max = gmax;
double bmax = b.getStatistics().max;
if (bmax > max) max = bmax;
double scale = 255 / max;
r.multiply(scale);
g.multiply(scale);
b.multiply(scale);
red2.setProcessor(r.convertToByte(false));
green2.setProcessor(g.convertToByte(false));
blue2.setProcessor(b.convertToByte(false));
}
RGBStackMerge merge = new RGBStackMerge();
ImageStack stack2 =
merge.mergeStacks(w, h, d, red2.getStack(), green2.getStack(), blue2.getStack(), true);
imp = saveImp;
projImage = new ImagePlus(makeTitle(), stack2);
}
public void doHyperStackProjection(boolean allTimeFrames) {
int start = startSlice;
int stop = stopSlice;
int firstFrame = 1;
int lastFrame = imp.getNFrames();
if (!allTimeFrames) firstFrame = lastFrame = imp.getFrame();
ImageStack stack = new ImageStack(imp.getWidth(), imp.getHeight());
int channels = imp.getNChannels();
int slices = imp.getNSlices();
if (slices == 1) {
slices = imp.getNFrames();
firstFrame = lastFrame = 1;
}
int frames = lastFrame - firstFrame + 1;
increment = channels;
boolean rgb = imp.getBitDepth() == 24;
for (int frame = firstFrame; frame <= lastFrame; frame++) {
for (int channel = 1; channel <= channels; channel++) {
startSlice = (frame - 1) * channels * slices + (start - 1) * channels + channel;
stopSlice = (frame - 1) * channels * slices + (stop - 1) * channels + channel;
if (rgb) doHSRGBProjection(imp);
else doProjection();
stack.addSlice(null, projImage.getProcessor());
}
}
projImage = new ImagePlus(makeTitle(), stack);
projImage.setDimensions(channels, 1, frames);
if (channels > 1) {
projImage = new CompositeImage(projImage, 0);
((CompositeImage) projImage).copyLuts(imp);
if (method == SUM_METHOD || method == SD_METHOD)
((CompositeImage) projImage).resetDisplayRanges();
}
if (frames > 1) projImage.setOpenAsHyperStack(true);
Overlay overlay = imp.getOverlay();
if (overlay != null) {
startSlice = start;
stopSlice = stop;
if (imp.getType() == ImagePlus.COLOR_RGB)
projImage.setOverlay(projectRGBHyperStackRois(overlay));
else projImage.setOverlay(projectHyperStackRois(overlay));
}
IJ.showProgress(1, 1);
}
public void run(String arg) {
imp = IJ.getImage();
int stackSize = imp.getStackSize();
if (imp == null) {
IJ.noImage();
return;
}
// Make sure input image is a stack.
if (stackSize == 1) {
IJ.error("Z Project", "Stack required");
return;
}
// Check for inverting LUT.
if (imp.getProcessor().isInvertedLut()) {
if (!IJ.showMessageWithCancel("ZProjection", lutMessage)) return;
}
// Set default bounds.
int channels = imp.getNChannels();
int frames = imp.getNFrames();
int slices = imp.getNSlices();
isHyperstack =
imp.isHyperStack()
|| (ij.macro.Interpreter.isBatchMode()
&& ((frames > 1 && frames < stackSize) || (slices > 1 && slices < stackSize)));
boolean simpleComposite = channels == stackSize;
if (simpleComposite) isHyperstack = false;
startSlice = 1;
if (isHyperstack) {
int nSlices = imp.getNSlices();
if (nSlices > 1) stopSlice = nSlices;
else stopSlice = imp.getNFrames();
} else stopSlice = stackSize;
// Build control dialog
GenericDialog gd = buildControlDialog(startSlice, stopSlice);
gd.showDialog();
if (gd.wasCanceled()) return;
if (!imp.lock()) return; // exit if in use
long tstart = System.currentTimeMillis();
setStartSlice((int) gd.getNextNumber());
setStopSlice((int) gd.getNextNumber());
method = gd.getNextChoiceIndex();
Prefs.set(METHOD_KEY, method);
if (isHyperstack) {
allTimeFrames = imp.getNFrames() > 1 && imp.getNSlices() > 1 ? gd.getNextBoolean() : false;
doHyperStackProjection(allTimeFrames);
} else if (imp.getType() == ImagePlus.COLOR_RGB) doRGBProjection(true);
else doProjection(true);
if (arg.equals("") && projImage != null) {
long tstop = System.currentTimeMillis();
projImage.setCalibration(imp.getCalibration());
if (simpleComposite) IJ.run(projImage, "Grays", "");
projImage.show("ZProjector: " + IJ.d2s((tstop - tstart) / 1000.0, 2) + " seconds");
}
imp.unlock();
IJ.register(ZProjector.class);
return;
}
/**
* Performs actual projection using specified method. If handleOverlay, adds stack overlay
* elements from startSlice to stopSlice to projection
*/
public void doProjection(boolean handleOverlay) {
doProjection();
Overlay overlay = imp.getOverlay();
if (handleOverlay && overlay != null) projImage.setOverlay(projectStackRois(overlay));
}
