// BIOP Functions Library

/*
 * Returns the name of the parameters window, as we cannot use global variables, 
 * we just define a function that can act as a global variable
 */
function getWinTitle() {
    	win_title= toolName();
    	// If something is already open, keep it as-is.
	if(!isOpen(win_title)) {
		run("New... ", "name=["+win_title+"] type=Table");
		print("["+win_title+"]", "\\Update0:This window contains data "+win_title+" needs.");
		print("["+win_title+"]", "\\Update1:Please do not close it.");
	}
	return win_title;
}
    
/*
 * Based on an example by Wayne Rasband, we use the "getData" and "setData" functions to 
 * read and write data to and from an opened text window. This allows us to save parameters
 * for an ActionBar in a visible way for the user, instead of relying on IJ.prefs.
 */
function getData(key) {

	winTitle = getWinTitle();
	win = "["+winTitle+"]";

	selectWindow(winTitle);
	lines = split(getInfo(),'\n');
	i=0;
	done=false;
	value = "";
	while (!done && i < lines.length) {
		// The structure for the data is "key : value", so we use a regex to find the key and place ourselves after the " : "
		if(matches(lines[i], ".*"+key+".*")) {
			value = substring(lines[i], indexOf(lines[i]," : ")+3,lengthOf(lines[i]));
			done = true;
		
		} else {
			i++;
		}	
	}

	return value;
}

/* Like getData, but takes a default argument
 * and returns it if the key is not found
*/
function getDataD(key, default) {
	value = getData(key);
	if (value == "") {
		return default;
	} else {
		return value;
	}
}

/* 
 *  See Above Comment
 */
	function setData(key, value) {
    	//Open the file and parse the data
	winTitle = getWinTitle();
	win = "["+winTitle+"]";

	selectWindow(winTitle);
	lines = split(getInfo(),'\n');
	i=0;
	done=false;
	if (lines.length > 0) {
		while (!done && i < lines.length) {
			if(matches(lines[i], ".*"+key+".*")) {
				done=true;
			} else {
				i++;
			}		
		}
			print(win, "\\Update"+i+":"+key+" : "+value);
	} else { 
		// The key did not exist
		print(win, key+" : "+value);
	}
}

/*
 * Setter and getter for boolean values
 */
function setBool(key, bool) {
	if (bool) {
		setData(key, "Yes");
	} else {
		setData(key, "No");
	}
}

function getBool(key) {
	val = getData(key);
	if (val == "Yes") {
		val = true;
	} else {
		val=false;
	}
	return val;
}

/*
 * Functions to read and write from a text file to a parameters window
 * These are sued by the Save Parameters and Load Parameters Buttons
 */
function loadParameters() {
	// Get the file
	file = File.openDialog("Select Parameters File");
	
	//Get the contents
	filestr = File.openAsString(file);
	lines = split(filestr, "\n");
	
	//Open the file and parse the data
	settingName = getWinTitle();;
	
	t = "["+settingName+"]";
	
	// If something is already open, keep it as-is.
	if(!isOpen(settingName)) {
		run("New... ", "name="+t+" type=Table");
	}
	selectWindow("Tile Scan Opener Parameters");
	for (i=0; i<lines.length; i++) {
		print(t, "\\Update"+i+":"+lines[i]);
	}
}

/*
 * Same as above.
*/
function saveParameters() {
	winName = getWinTitle();
	selectWindow(winName);
	saveAs("Text", "");
}

/* 
 *  isImage lets you know whether the current file is an image. Useful below
 */
function isImage(filename) {
	extensions= newArray("lsm", "lei", "lif", "tif", "ics", "bmp", "jpg", "png", "TIF", "tiff");
	for (i=0; i<extensions.length; i++) {
		if(endsWith(filename, "."+extensions[i])) {
			return true;
		}
	}
	return false;
}


/*
 *  getImageFolder returns the current value of the 'Image Folder' key 
 *  in the parameters window. If it's not set, it calls setImageFolder below.
 */ 
function getImageFolder() {
	dir = getData("Image Folder");
	if(dir=="") {
		dir = setImageFolder("Image Folder");
	}
	return dir;
}

/* 
 * Display a getDirectory dialog box and save the value under the 
 * 'Image Folder' key in the parameters window.
*/
function setImageFolder(title) {
	dir = getDirectory(title);
	setData("Image Folder", dir);
	return dir;
}

 /*
  * getSaveFolder returns the current value of the 'Save Folder' key 
  * in the parameters window. If it's not set, it calls setSaveFolder below.
  */
function getSaveFolder() {
	dir = getData("Save Folder");
	if(dir=="") {
		dir = setSaveFolder();
	}
	return dir;
}

/*
 * Sets the Save folder as a subfolder of the Image Folder
 * 'Save Folder' key in the parameters window.
 */
function setSaveFolder() {
	dir = getImageFolder();
	saveFolder = dir+"Processed"+File.separator;
	setData("Save Folder", saveFolder);
	File.makeDirectory(saveFolder);
	return dir;
}

/*
 * By using isImage above, this function counts how many images are currently in the selected
 * image folder (The folder is defined in the parameters window 
 */ 
function getNumberImages() {
	dir = getImageFolder();
	file = getFileList(dir);
	n=0;
	for (i=0; i<file.length;i++) {
		if (isImage(file[i])) {
			n++;
		}
	}
	return n;
}

/*
 * By using isImage and getNumberImages, we can now open the ith image from a folder easily
 * This is useful when running a batch on a folder
*/
function openImageN(n) {
	nFiles=-1;
	dir = getImageFolder();
	file = getFileList(dir);
	nI = getNumberImages();
	for (i=0; i<file.length; i++) {
		if(isImage(file[i])) {
			 nFiles++;
			 if (nFiles==n) {
				open(dir+file[i]);
				//Check if the image has a ROI set and open it
				openRoiSet(file[i]);
			}
		}
	}
}

/*
 * Mainly used for the selectImageDialog function, 
 * this function returns a list of image names from
 * the current image folder. Again using isImage
 */
function getImagesList() {
	dir = getImageFolder();
	
	list = getFileList(dir);

	images = newArray(list.length);
	k=0;
	// Check things in the list
	for (i=0; i<list.length; i++) {
		if(isImage(list[i])) {
			images[k] = list[i];
			k++;
		}
	}
	images = Array.trim(images,k);
	return images;
	
}

/* 
 * Simple function to check the presence of a ROI set
 * The macros here use a function called getRoiDir to get which is the folder that should contain the ROI sets
 * The ROI set must have EXACTLY the same name as the filename and end in '.zip'
*/
function hasRoiSet(file) {
	
	roiDir = getRoiFolder("Open");
	file = getFileNameNoExt(file);
	
	if (File.exists(roiDir+file+".zip")) {
		return true;
	} else {
		return false;

}

/*
 * openRoiSet simply opens the ROIs associated with the image 'file' if it exists
 */ 
function openRoiSet(file) {
	if (hasRoiSet(file)) {
		roiDir = getRoiFolder("Open");
		//Load ROI. set
		file = getFileNameNoExt(file);
		roiManager("reset");
		roiManager("Open", roiDir+file+".zip")
		roiManager("Show All");
	}
}

/*
 * returns the directory where the ROIs are stored, a subfolder of the image folder.
 */
function getRoiFolder(mode) {
	// Feel free to rename it if you like that sort of thing.
	dirName = "ROI Sets";
	if (mode == "Open") {
		dir = getImageFolder();
	} else {
		dir = getSaveFolder();
	}
	roiDir = dir+dirName+File.separator;
	File.makeDirectory(dir);
	File.makeDirectory(roiDir);
	
	return roiDir;
}


/*
 * Saves the ROIs of the current image
 */ 
function saveRois(mode) {
	name = getTitle();
	roiDir = getRoiFolder(mode);
	// If image has an extension, remove it.
	name = getFileNameNoExt(name);
	nR = roiManager("Count");

	// if there are ROIs save them
	if (nR > 0) {
		//Save Roi Set
		File.makeDirectory(roiDir);
		roiManager("Save", roiDir+name+".zip");
		print("ROI Set Saved for image "+name);
	}
}

/*
 * Saves the current image as a TIFF in the currentImage Folder
 */ 
function saveCurrentImage() {
	name = getTitle();
	print(name);
	dir = getSaveFolder();
	print(dir);
	File.makeDirectory(dir);
	name = getFileNameNoExt(name);
	saveAs("TIFF", dir+name+".tif");
}

/*
 *  Returns the file name witout the extension
 */ 
function getFileNameNoExt(file) {
		// Get the file name without the extension, regex
	if (matches(file,".+\\.\\w{3,4}")) {
		file = substring(file,0,lastIndexOf(file,"."));
	}
	return file;
}