stitcherGui2d.py

import twodstitcher
import numpy as np
import os
import tkinter
from tkinter import filedialog
from twodstitcher import TwoDStitcher
import _thread

class StitcherGUI:

	fileList = None;
	columnList = None;
	finishedFiles = None;
	readyToTerminate = False;
	progress = 0;

	def shorten_filename(self,filename):
		f = os.path.split(filename)[1]
		return "%s~%s" % (f[:3], f[-16:]) if len(f) > 19 else f

	def __init__(self, master):
		self.master = master
		master.title("Stitcher")
		self.outputFile = None
		self.browse_button = tkinter.Button(master, text="Select Files", command=self.browseFiles)
		self.browse_button.grid(row=0)

		self.save_file = tkinter.Button(master, text="Output File", command=self.saveFile)
		self.save_file.grid(row=1)


		self.file_label_text = tkinter.StringVar()
		self.file_label_text.set(self.outputFile)
		self.file_label = tkinter.Label(master, textvariable=self.file_label_text)
		self.label = tkinter.Label(master, text="Output File:")
		self.label.grid(row=2, column=0)
		self.file_label.grid(row=2, column=1)


		self.stitch = tkinter.Button(master, text="Stitch", command=self.stitch)
		self.stitch.grid(row=3, column=1)

		self.progressText = tkinter.IntVar()
		self.progressText.set(self.progress)
		
		tkinter.Label(master, text="Progress :").grid(row=4,column=0)
		self.progress_label = tkinter.Label(master, textvariable=self.progressText)
		self.progress_label.grid(row=4, column=1)
		
	def browseFiles(self):
		# Tk().withdraw()
		files = filedialog.askopenfilenames(filetypes = (("JPEG", "*.jpeg"),("JPEG", "*.jpg")
														 ,("PNG", "*.png")
														 ,("All files", "*.*") ))
		parentName = os.path.split(os.path.dirname(files[0]))[1]
		self.outputFile = os.path.join(os.path.dirname(files[0]),parentName + ".tiff")
		self.file_label_text.set(parentName + ".tiff")
		self.fileList = root.tk.splitlist(files)


	def saveFile(self):
		# Tk().withdraw()
		self.outputFile = filedialog.asksaveasfilename(filetypes=[("TIFF", ".tiff")], title="Select Output File")
		self.file_label_text.set(self.shorten_filename(self.outputFile))

	def stitch(self):
		stitcherHandler = TwoDStitcher()
		print(self.fileList)
		print(self.outputFile);
		if(self.outputFile is None):
			exit()

		columnArray = []
		for file in self.fileList:
			filename = os.path.basename(file)
			splitString = filename.split("_row")
			columnInfo = int(splitString[0][4:])
			if(len(columnArray) < columnInfo):
				columnArray.append([]);
			columnArray[columnInfo-1].append(file)

		self.columnList = columnArray;

		firstItem = self.columnList.pop(0)
		parentName = os.path.split(os.path.dirname(self.fileList[0]))[1]
		outputFile = os.path.join(os.path.dirname(self.fileList[0]),parentName + str(len(self.columnList)) + ".tiff")
		self.finishedFiles = []
		self.finishedFiles.append(outputFile)

		_thread.start_new_thread(stitcherHandler.stitchFileList, (firstItem, outputFile, "vertical",self.progressCallback))

	def popNextJob(self):
		stitcherHandler = TwoDStitcher()
		if(self.readyToTerminate):
			exit()

		if(len(self.columnList) == 0):
			print(self.finishedFiles)
			self.readyToTerminate = True
			_thread.start_new_thread(stitcherHandler.stitchFileList, (self.finishedFiles, self.outputFile, "horizontal",self.progressCallback))
		else:
			firstItem = self.columnList.pop(0)
			parentName = os.path.split(os.path.dirname(self.fileList[0]))[1]
			outputFile = os.path.join(os.path.dirname(self.fileList[0]),parentName + str(len(self.columnList)) + ".tiff")
			self.finishedFiles.append(outputFile);

			_thread.start_new_thread(stitcherHandler.stitchFileList, (firstItem, outputFile, "vertical",self.progressCallback))


	def progressCallback(self, status, progress):
		self.progressText.set(progress);
		if(progress == 100):
			self.popNextJob();


root = tkinter.Tk()
root.geometry('{}x{}'.format(400, 200))
my_gui = StitcherGUI(root)
root.mainloop()