From 5af5c816e9325db5185826384f68e861414b7ba3 Mon Sep 17 00:00:00 2001 From: "kai.schleicher@unibas.ch" <kai.schleicher@unibas.ch> Date: Thu, 5 Nov 2020 18:19:32 +0100 Subject: [PATCH] initial commit --- 2a_identify_MHC_positive_fibers.py | 425 +++++++++++++++++++++++++++++ 1 file changed, 425 insertions(+) create mode 100644 2a_identify_MHC_positive_fibers.py diff --git a/2a_identify_MHC_positive_fibers.py b/2a_identify_MHC_positive_fibers.py new file mode 100644 index 0000000..4d6eb6c --- /dev/null +++ b/2a_identify_MHC_positive_fibers.py @@ -0,0 +1,425 @@ + +# IJ imports +from ij import IJ, WindowManager as wm +from ij.plugin import Duplicator, RoiEnlarger, RoiScaler +from trainableSegmentation import WekaSegmentation +from de.biovoxxel.toolbox import Extended_Particle_Analyzer + +# Bio-formats imports +from loci.plugins import BF +from loci.plugins.in import ImporterOptions + +# python imports +import time +import os + +#@ String (visibility=MESSAGE, value="<html><b> Welcome to Myosoft - identify MHC positive fibers! </b></html>") msg1 +#@ File (label="Select fiber-ROIs zip-file", style="file") roi_zip +#@ File (label="Select image file", description="select your image") path_to_image +#@ File (label="Select directory for output", style="directory") output_dir +#@ String (visibility=MESSAGE, value="<html><b> channel positions in the hyperstack </b></html>") msg5 +#@ Integer (label="Fiber staining (MHC) channel number", style="slider", min=1, max=5, value=3) fiber_channel +#@ Integer (label="minimum fiber intensity (0=auto)", description="0 = automatic threshold detection", value=0) min_fiber_intensity +#@ ResultsTable rt +#@ RoiManager rm + + +def fix_ij_options(): + """put IJ into a defined state + """ + # disable inverting LUT + IJ.run("Appearance...", " menu=0 16-bit=Automatic") + # set foreground color to be white, background black + IJ.run("Colors...", "foreground=white background=black selection=red") + # black BG for binary images and pad edges when eroding + IJ.run("Options...", "black pad") + # set saving format to .txt files + IJ.run("Input/Output...", "file=.txt save_column save_row") + # ============= DON’T MOVE UPWARDS ============= + # set "Black Background" in "Binary Options" + IJ.run("Options...", "black") + # scale when converting = checked + IJ.run("Conversions...", "scale") + + +def fix_ij_dirs(path): + """use forward slashes in directory paths + + Parameters + ---------- + path : string + a directory path obtained from dialogue or script parameter + + Returns + ------- + string + a more robust path with forward slashes as separators + """ + + fixed_path = str(path).replace("\\", "/") + # fixed_path = fixed_path + "/" + + return fixed_path + + +def open_image_with_BF(path_to_file): + """ use Bio-Formats to opens the first image from an image file path + + Parameters + ---------- + path_to_file : string + path to the image file + + Returns + ------- + ImagePlus + the first imp stored in a give file + """ + options = ImporterOptions() + options.setColorMode(ImporterOptions.COLOR_MODE_GRAYSCALE) + options.setAutoscale(True) + options.setId(path_to_file) + imps = BF.openImagePlus(options) # is an array of ImagePlus + + return imps[0] + + +def fix_BF_czi_imagetitle(imp): + image_title = os.path.basename( imp.getTitle() ) + image_title = image_title.replace(".czi", "") + image_title = image_title.replace(" ", "_") + image_title = image_title.replace("_-_", "") + image_title = image_title.replace("__", "_") + image_title = image_title.replace("#", "Series") + + return image_title + + +def clear_ij_roi_manager(rm): + """delete all ROIs from the RoiManager + + Parameters + ---------- + rm : RoiManager + a reference of the IJ-RoiManager + """ + rm.runCommand('reset') + + +def get_threshold_from_method(imp, channel, method): + """returns the threshold value of chosen IJ AutoThreshold method in desired channel + + Parameters + ---------- + imp : ImagePlus + the imp from which to get the threshold value + channel : integer + the channel in which to get the treshold + method : string + the AutoThreshold method to use + + Returns + ------- + list + the upper and the lower threshold (integer values) + """ + imp.setC(channel) # starts at 1 + ip = imp.getProcessor() + ip.setAutoThreshold(method + " dark") + lower_thr = ip.getMinThreshold() + upper_thr = ip.getMaxThreshold() + ip.resetThreshold() + + return lower_thr, upper_thr + + +def measure_in_all_rois( imp, channel, rm ): + """measures in all ROIS on a given channel of imp all parameters that are set in IJ "Set Measurements" + + Parameters + ---------- + imp : ImagePlus + the imp to measure on + channel : integer + the channel to measure in. starts at 1. + rm : RoiManager + a reference of the IJ-RoiManager + """ + imp.setC(channel) + rm.runCommand(imp,"Deselect") + rm.runCommand(imp,"Measure") + + +def change_all_roi_color( rm, color ): + """change the color of all ROIs in the RoiManager + + Parameters + ---------- + rm : RoiManager + a reference of the IJ-RoiManager + color : string + the desired color. e.g. "green", "red", "yellow", "magenta" ... + """ + number_of_rois = rm.getCount() + for roi in range( number_of_rois ): + rm.select(roi) + rm.runCommand("Set Color", color) + + +def change_subset_roi_color( rm, selected_rois, color ): + """change the color of selected ROIs in the RoiManager + + Parameters + ---------- + rm : RoiManager + a reference of the IJ-RoiManager + selected_rois : array + ROIs in the RoiManager to change + color : string + the desired color. e.g. "green", "red", "yellow", "magenta" ... + """ + rm.runCommand("Deselect") + rm.setSelectedIndexes(selected_rois) + rm.runCommand("Set Color", color) + rm.runCommand("Deselect") + + +def show_all_rois_on_image(rm, imp): + """shows all ROIs in the ROiManager on imp + + Parameters + ---------- + rm : RoiManager + a reference of the IJ-RoiManager + imp : ImagePlus + the imp on which to show the ROIs + """ + imp.show() + rm.runCommand(imp,"Show All") + + +def save_all_rois(rm, target): + """save all ROIs in the RoiManager as zip to target path + + Parameters + ---------- + rm : RoiManager + a reference of the IJ-RoiManager + target : string + the path in to store the ROIs. e.g. /my-images/resulting_rois.zip + """ + rm.runCommand("Save", target) + + +def save_selected_rois( rm, selected_rois, target ): + """save selected ROIs in the RoiManager as zip to target path + + Parameters + ---------- + rm : RoiManager + a reference of the IJ-RoiManager + selected_rois : array + ROIs in the RoiManager to save + target : string + the path in to store the ROIs. e.g. /my-images/resulting_rois_subset.zip + """ + rm.runCommand("Deselect") + rm.setSelectedIndexes(selected_rois) + rm.runCommand("save selected", target) + rm.runCommand("Deselect") + + +def select_positive_fibers( imp, channel, rm, min_intensity ): + """For all ROIs in the RoiManager, select ROIs based on intensity measurement in given channel of imp. + See https://imagej.nih.gov/ij/developer/api/ij/process/ImageStatistics.html + + Parameters + ---------- + imp : ImagePlus + the imp on which to measure + channel : integer + the channel on which to measure. starts at 1 + rm : RoiManager + a reference of the IJ-RoiManager + min_intensity : integer + the selection criterion (here: intensity threshold) + + Returns + ------- + array + a selection of ROIs which passed the selection criterion (are above the threshold) + """ + imp.setC(channel) + all_rois = rm.getRoisAsArray() + selected_rois = [] + for i, roi in enumerate(all_rois): + imp.setRoi(roi) + stats = imp.getStatistics() + if stats.mean > min_intensity: + selected_rois.append(i) + + return selected_rois + + +def open_rois_from_zip( rm, path ): + """open RoiManager ROIs from zip and adds them to the RoiManager + + Parameters + ---------- + rm : RoiManager + a reference of the IJ-RoiManager + path : string + path to the ROI zip file + """ + rm.runCommand("Open", path) + + +def preset_results_column( rt, column, value): + """pre-set all rows in given column of the IJ-ResultsTable with desired value + + Parameters + ---------- + rt : ResultsTable + a reference of the IJ-ResultsTable + column : string + the desired column. will be created if it does not yet exist + value : string or float or integer + the value to be set + """ + for i in range( rt.size() ): + rt.setValue(column, i, value) + + rt.show("Results") + + +def add_results( rt, column, row, value ): + """adds a value in desired rows of a given column + + Parameters + ---------- + rt : ResultsTable + a reference of the IJ-ResultsTable + column : string + the column in which to add the values + row : array + the row numbers in which too add the values. + value : string or float or integer + the value to be set + """ + for i in range( len( row ) ): + rt.setValue(column, row[i], value) + + rt.show("Results") + + +def enhance_contrast( imp ): + """use "Auto" Contrast & Brightness settings in each channel of imp + + Parameters + ---------- + imp : ImagePlus + the imp on which to change C&B + """ + for channel in range( imp.getDimensions()[2] ): + imp.setC(channel + 1) # IJ channels start at 1 + IJ.run(imp, "Enhance Contrast", "saturated=0.35") + + +def renumber_rois(rm): + """rename all ROIs in the RoiManager according to their number + + Parameters + ---------- + rm : RoiManager + a reference of the IJ-RoiManager + """ + number_of_rois = rm.getCount() + for roi in range( number_of_rois ): + rm.rename( roi, str(roi + 1) ) + + +def setup_defined_ij(rm, rt): + """set up a clean and defined Fiji user environment + + Parameters + ---------- + rm : RoiManager + a reference of the IJ-RoiManager + rt : ResultsTable + a reference of the IJ-ResultsTable + """ + fix_ij_options() + rm.runCommand('reset') + rt.reset() + IJ.log("\\Clear") + + +execution_start_time = time.time() +setup_defined_ij(rm, rt) + +# open image using Bio-Formats +path_to_image = fix_ij_dirs(path_to_image) +raw = open_image_with_BF(path_to_image) + +# get image info +raw_image_calibration = raw.getCalibration() +raw_image_title = fix_BF_czi_imagetitle(raw) + +# take care of paths and directories +input_rois_path = fix_ij_dirs( roi_zip ) +output_dir = fix_ij_dirs(output_dir) + "/2a_identify_MHC_positive_fibers/" + +if not os.path.exists( output_dir ): + os.makedirs( output_dir ) + +# update the log for the user +IJ.log( "Now working on " + str(raw_image_title) ) +if raw_image_calibration.scaled() == False: + IJ.log("Your image is not spatially calibrated! Size measurements are only possible in [px].") +IJ.log( " -- settings used -- ") +IJ.log( "Selected fiber-ROIs zip-file = " + str(input_rois_path) ) +IJ.log( "MHC positive fiber channel = " + str(fiber_channel) ) +IJ.log( " -- settings used -- ") + +# open ROIS and show on image +open_rois_from_zip( rm, input_rois_path ) +show_all_rois_on_image( rm, raw ) + +# check for positive fibers +if min_fiber_intensity == 0: + min_fiber_intensity = get_threshold_from_method(raw, fiber_channel, "Mean")[0] + IJ.log( "automatic intensity threshold detection: True" ) + +IJ.log( "fiber intensity threshold: " + str(min_fiber_intensity) ) +change_all_roi_color(rm, "blue") +positive_fibers = select_positive_fibers( raw, fiber_channel, rm, min_fiber_intensity ) +change_subset_roi_color(rm, positive_fibers, "magenta") +save_selected_rois( rm, positive_fibers, output_dir + "mhc_positive_fiber_rois.zip") + +# measure size & shape, save +IJ.run("Set Measurements...", "area perimeter shape feret's redirect=None decimal=4") +IJ.run("Clear Results", "") +measure_in_all_rois( raw, fiber_channel, rm ) +preset_results_column( rt, "MHC Positive Fibers (magenta)", "NO" ) +add_results( rt, "MHC Positive Fibers (magenta)", positive_fibers, "YES") +rt.save(output_dir + "_mhc_positive_fibers_results.csv") + +# dress up the original image, save a overlay-png, present original to the user +rm.show() +raw.show() +show_all_rois_on_image( rm, raw ) +raw.setDisplayMode(IJ.COMPOSITE) +enhance_contrast( raw ) +IJ.run("From ROI Manager", "") # ROIs -> overlays so they show up in the saved png +qc_duplicate = raw.duplicate() +IJ.saveAs(qc_duplicate, "PNG", output_dir + raw_image_title + "_mhc_positive_fibers") +qc_duplicate.close() +wm.toFront( raw.getWindow() ) +IJ.run("Remove Overlay", "") +raw.setDisplayMode(IJ.GRAYSCALE) +show_all_rois_on_image( rm, raw ) +total_execution_time_min = (time.time() - execution_start_time) / 60.0 +IJ.log("total time in minutes: " + str(total_execution_time_min)) +IJ.log( "~~ all done ~~" ) +IJ.selectWindow("Log") +IJ.saveAs("Text", str(output_dir + raw_image_title + "_mhc_positive_fibers_Log")) \ No newline at end of file -- GitLab