# @int (Label = "Number of categories") N_category
'''
This script can be used to manually classify full images from a stack into N user-defined categories.
A first window pops up to request the number of categories.
A second window follows asking for the name to use for those categories.
Finally a third window will show up with one button per category.
Clicking on the button will generate a new entry in a table with the image name and the category.
It will also skip to the next slice for stacks.

TO DO : Add measurement possibility ? The addValue was not working so well in this case. Duplicate to another code to try with the result table
'''
from ij	            import IJ,WindowManager
from ij.measure 	import ResultsTable
from ij.gui		    import GenericDialog, NonBlockingGenericDialog
from java.awt.event import ActionListener
from java.awt 		import GridLayout,Button


class ButtonAction(ActionListener): # extends action listener 
	'''The function actionPerformed contains code executed upon click of the associated button(s)'''
	
	def __init__(self):
		# use mother class constructor
		ActionListener.__init__(self)
				
	
	def actionPerformed(self,event):
		'''Called when associated buttons are clicked'''

		imp = IJ.getImage() # get current image
		infos = imp.getOriginalFileInfo()
		
		# Recover image name
		if imp.getStackSize()==1: 
			filename = infos.fileName
		else:
			Stack = imp.getStack()
			filename = Stack.getSliceLabel(imp.currentSlice)
			
			if filename is None: # the slice label can be empty sometimes
				filename = 'Slice' + str(imp.currentSlice)	
			'''		
			else : 
				filename = SliceName.split('\n',1)[0] # can be useful when ImagesToStack was used
			'''		
		# Fill the result table
		Table.incrementCounter() # Add one additional row before filling it
		Table.addValue("Folder", infos.directory)
		Table.addValue("Image", filename)	
		
		for i, checkbox in enumerate( WinButton.getCheckboxes() ):
			Table.addValue(listCategories[i], checkbox.getState() ) # getNextBoolean keeps growing its index, I should check only index 0 to N
			
		Table.show("Classification (multi)") # Update table	  
		#Table.updateResults() # only for result table but then addValue does not work !
		
		# Go to next slice
		if imp.getStackSize() != 1 and imp.currentSlice != imp.getStackSize(): # if We have a stack and the current slice is not the last slice
			imp.setSlice(imp.currentSlice+1)
			imp.updateStatusbarValue() # update Z and pixel value (called by next slice so we should do it too ?)
		
		# Bring back the focus to the button window (otherwise the table is in the front)
		WindowManager.setWindow(WinButton)
		
		
		
############### GUI - CATEGORY DIALOG - collect N classes names (N define at first line)  #############
Win = GenericDialog("Categories names")

# Add N string field to get class names
for i in range(N_category):
	Win.addStringField("Category: ","Category_"+str(i))
	Win.addMessage("") # skip one line
	
Win.showDialog()


################# After OK clicking ###########
# Recover fields from the formular
if (Win.wasOKed()): 
	
	# Initialise result table that will contain slice name and category
	Table = ResultsTable()

	# Initialise GUI with category buttons
	WinButton = NonBlockingGenericDialog("Tick categories, then Add")
	
	listCategories = []
	# Loop over categories
	for i in range(N_category):
		
		# Recover the category name
		category = Win.getNextString()
		listCategories.append(category)
		
		# Add a button to the gui for this category
		WinButton.addCheckbox(category, False)

	# Create a Button
	ButtonAdd = Button("Add")
		
	# Bind action to Button
	ButtonAdd.addActionListener(ButtonAction())
	
	# Add button to window
	WinButton.add(ButtonAdd)
	WinButton.hideCancelButton()
	
	# initialise Layout
	Layout = GridLayout(N_category+1,1) # +1 for OK/Cancel (1 extra row)
	WinButton.setLayout(Layout)
	
	WinButton.showDialog()