import pygame, easygui, sys, random

try:
	import openpyxl
except ImportError:
	print "Please install openpyxl. To do that use pip install openpyxl"
	raise SystemExit

def main():
	#All font and text
	font = pygame.font.SysFont("times", 30)
	choiceALabel = font.render("Choice A", 1, (255,255,255))
	choiceBLabel = font.render("Choice B", 1, (255,255,255))
	submitLabel = font.render("Pick one ->", 1, (255,255,255))
	#End of font and text
	
	attributesList = []
	levelsDict = {}
	attributesSurfaces = []
	levelsSurfaces = []
	excelInformation = {}
	trial = 0
	numTrials = 0
	attributesPerPage = 0
	try:
		attributesFile = open("surveyinfo.txt", "r")
		fileContents = attributesFile.read().split("\n")
		#Checking to see if the file contains the magic number
		if fileContents[0] != "040":
			easygui.msgbox(msg="Survey Error. File not setup properly. Please use the conjointSDT.py and export to Jacob-formatted TXT.",title="Error")
			return
		fileContentsNoMagicNumber = fileContents[1:]
		#Iterate everything besides the magic number to build the attributes and levels
		for i in range(len(fileContentsNoMagicNumber)):
			if i != len(fileContentsNoMagicNumber) - 1:
				attributesList.append(fileContentsNoMagicNumber[i].split(":")[0])
				levelsDict[attributesList[i]] = fileContentsNoMagicNumber[i].split(":")[1].split(",")
			else:
				numTrials = int(fileContentsNoMagicNumber[i])
	#If the file doesn't exist then exit the program
	except IOError:
		easygui.msgbox(msg="Survey Error. Missing file \"surveyinfo.txt\". Please use conjointSDT.py and export to Jacob-formatted TXT.",title="Error")
		return
	
	if len(attributesList) > 5:
		attributesPerPage = 5
	else:
		attributesPerPage = len(attributesList)
		
	#print attributesPerPage

	clock = pygame.time.Clock()

	screen = pygame.display.set_mode((640,480), pygame.NOFRAME)
	
	#Here we set a unique userID
	userID = 1
	try:
		#Here we read from the file if we can and get the unique ID of the user
		userIDFile = open("id.txt", "r")
		userID = int(userIDFile.read())
		userIDFile.close()
	except IOError:
		#If the file doesn't exist then we pass by the exception because it's not terrible
		pass
	
	#Here we replace the contents of the old file with a new userID	
	userIDFile = open("id.txt", "w")
	userIDFile.write(str(userID+1))
	userIDFile.close()
	#End of unique userID
	
	#Early questions here
	age = int(easygui.enterbox("Please enter your age."))
	gender = easygui.buttonbox(msg="Are you male or female?", choices=('Male', 'Female'), cancel_choice='Not Specified')
	print gender
	#End of early questions
	
	running = True
	createPage = True
	aFirst = True
	bFirst = False
	while running:
		if createPage:
			isAFirst = random.randint(0,1)
			if isAFirst:
				aFirst = True
				bFirst = False
			else:
				aFirst = False
				bFirst = True
			chosenAttributes = random.sample(attributesList, attributesPerPage)
			chosenLevelsForA = []
			chosenLevelsForB = []
			for i in range(attributesPerPage):
				chosenLevelsForA.append(levelsDict[chosenAttributes[i]][random.randint(0,len(levelsDict[chosenAttributes[i]])) - 1])
				chosenLevelsForB.append(levelsDict[chosenAttributes[i]][random.randint(0,len(levelsDict[chosenAttributes[i]])) - 1])
			attributesSurfaces = []
			levelsSurfacesForA = []
			levelsSurfacesForB = []
			for i in range(attributesPerPage):
				attributesSurfaces.append(font.render(chosenAttributes[i], 1, (255,255,255)))
				levelsSurfacesForA.append(font.render(chosenLevelsForA[i], 1, (255,255,255)))
				levelsSurfacesForB.append(font.render(chosenLevelsForB[i], 1, (255,255,255)))
			createPage = False
		for event in pygame.event.get():
			if event.type == pygame.QUIT:
				running = False
			if event.type == pygame.MOUSEBUTTONUP:
				mousePos = pygame.mouse.get_pos()
				#Getting the click for the radio buttons
				outcome = 0
				clickedOnButton = False
				if mousePos[0] >= 350 and mousePos[0] <= 370 and mousePos[1] >= 445 and mousePos[1] <= 465: #First button
					clickedOnButton = True
					if aFirst == True:
						outcome = 1
					else:
						outcome = 2
				elif mousePos[0] >= 535 and mousePos[0] <= 555 and mousePos[1] >= 445 and mousePos[1] <= 465: #Second button
					clickedOnButton = True
					if aFirst == True:
						outcome = 3
					else:
						outcome = 4
				
				excelInformation["Trial " + str(trial + 1)] = {}
				for i in range(len(attributesList)):
					excelInformation["Trial " + str(trial + 1)][attributesList[i]] = ""
				excelInformation["Trial " + str(trial + 1)]["Option Chosen"] = ""
				excelInformation["Trial " + str(trial + 1)]["Neither?"] = ""
				excelInformation["Trial " + str(trial + 1)]["Age"] = age
				excelInformation["Trial " + str(trial + 1)]["Gender"] = gender
					
				#print "Outcome: " + str(outcome)
				
				if outcome == 1:#Picked B when B was first
					for i in range(len(chosenAttributes)):
						#print chosenAttributes[i] + ":" + chosenLevelsForA[i]#Wrong but it works
						excelInformation["Trial " + str(trial + 1)][chosenAttributes[i]] = chosenLevelsForA[i]
					keepOption = easygui.ynbox("You have chosen Option B. If given the option between option B and no option would you still choose option B?", "Confirmation Screen", ("Yes", "No"))
					excelInformation["Trial " + str(trial + 1)]["Option Chosen"] = "B"
					if keepOption:
						excelInformation["Trial " + str(trial + 1)]["Neither?"] = False
					else:
						excelInformation["Trial " + str(trial + 1)]["Neither?"] = True
				elif outcome == 2:#Picked A when A was first
					for i in range(len(chosenAttributes)):
						#print chosenAttributes[i] + ":" + chosenLevelsForA[i]
						excelInformation["Trial " + str(trial + 1)][chosenAttributes[i]] = chosenLevelsForA[i]
					keepOption = easygui.ynbox("You have chosen Option A. If given the option between option A and no option would you still choose option A?", "Confirmation Screen", ("Yes", "No"))
					excelInformation["Trial " + str(trial + 1)]["Option Chosen"] = "A"
					if keepOption:
						excelInformation["Trial " + str(trial + 1)]["Neither?"] = False
					else:
						excelInformation["Trial " + str(trial + 1)]["Neither?"] = True
				elif outcome == 3:#Picked A when B was first
					for i in range(len(chosenAttributes)):
						#print chosenAttributes[i] + ":" + chosenLevelsForB[i]#Wrong but it works
						excelInformation["Trial " + str(trial + 1)][chosenAttributes[i]] = chosenLevelsForB[i]
					keepOption = easygui.ynbox("You have chosen Option A. If given the option between option A and no option would you still choose option A?", "Confirmation Screen", ("Yes", "No"))
					excelInformation["Trial " + str(trial + 1)]["Option Chosen"] = "A"
					if keepOption:
						excelInformation["Trial " + str(trial + 1)]["Neither?"] = False
					else:
						excelInformation["Trial " + str(trial + 1)]["Neither?"] = True
				elif outcome == 4:#Picked B when A was first
					for i in range(len(chosenAttributes)):
						#print chosenAttributes[i] + ":" + chosenLevelsForB[i]
						excelInformation["Trial " + str(trial + 1)][chosenAttributes[i]] = chosenLevelsForB[i]
					keepOption = easygui.ynbox("You have chosen Option B. If given the option between option B and no option would you still choose option B?", "Confirmation Screen", ("Yes", "No"))
					excelInformation["Trial " + str(trial + 1)]["Option Chosen"] = "B"
					if keepOption:
						excelInformation["Trial " + str(trial + 1)]["Neither?"] = False
					else:
						excelInformation["Trial " + str(trial + 1)]["Neither?"] = True
				if clickedOnButton:
					trial += 1
					createPage = True
					if trial == numTrials:
						running = False
						
						try:
							wb = openpyxl.load_workbook('survey_results.xlsx')
							ws = wb.worksheets[0]
						except IOError:
							wb = openpyxl.Workbook()
							ws = wb.worksheets[0]
							ws.cell(row = 1, column = 1).value = "User #"
							ws.cell(row = 1, column = 2).value = "Trial #"
							for i in range(len(attributesList)):
								ws.cell(row = 1, column = 3 + i).value = attributesList[i]
							ws.cell(row = 1, column = 3 + len(attributesList)).value = "A/B"
							ws.cell(row = 1, column = 4 + len(attributesList)).value = "Neither?"
							ws.cell(row = 1, column = 5 + len(attributesList)).value = "Age"
							ws.cell(row = 1, column = 6 + len(attributesList)).value = "Gender"
						
						for i in range(numTrials):
							#print userID
							ws.cell(row = 2 + i + numTrials*(userID-1), column = 1).value = userID
							ws.cell(row = 2 + i + numTrials*(userID-1), column = 2).value = i + 1
							for j in range(len(attributesList)):
								try:
									ws.cell(row = 2 + i + numTrials*(userID-1), column = 3 + j).value = int(excelInformation["Trial " + str(i + 1)][attributesList[j]])
								except ValueError:
									ws.cell(row = 2 + i + numTrials*(userID-1), column = 3 + j).value = excelInformation["Trial " + str(i + 1)][attributesList[j]]
							ws.cell(row = 2 + i + numTrials*(userID-1), column = 3 + len(attributesList)).value = excelInformation["Trial " + str(i + 1)]["Option Chosen"]
							ws.cell(row = 2 + i + numTrials*(userID-1), column = 4 + len(attributesList)).value = excelInformation["Trial " + str(i + 1)]["Neither?"]
							ws.cell(row = 2 + i + numTrials*(userID-1), column = 5 + len(attributesList)).value = excelInformation["Trial " + str(i + 1)]["Age"]
							ws.cell(row = 2 + i + numTrials*(userID-1), column = 6 + len(attributesList)).value = excelInformation["Trial " + str(i + 1)]["Gender"]
							
						wb.save("survey_results.xlsx")
						
						#print excelInformation
					
		#Drawing the UI
		screen.fill((211,211,211))
		pygame.draw.line(screen, (0,0,0), (453, 0), (453, 480))
		pygame.draw.line(screen, (0,0,0), (266, 0), (266, 480))
		pygame.draw.line(screen, (0,0,0), (0, 72), (640, 72))
		pygame.draw.line(screen, (0,0,0), (0, 144), (640, 144))
		pygame.draw.line(screen, (0,0,0), (0, 216), (640, 216))
		pygame.draw.line(screen, (0,0,0), (0, 288), (640, 288))
		pygame.draw.line(screen, (0,0,0), (0, 360), (640, 360))
		pygame.draw.line(screen, (0,0,0), (0, 432), (640, 432))
		pygame.draw.ellipse(screen, (255,255,255), (350,445,20,20))
		pygame.draw.ellipse(screen, (0,0,0), (349,444,22,22), 1)
		pygame.draw.ellipse(screen, (255,255,255), (535,445,20,20))
		pygame.draw.ellipse(screen, (0,0,0), (534,444,22,22), 1)
		screen.blit(choiceALabel, (303 + int(aFirst)*187, 20))
		screen.blit(choiceBLabel, (303 + int(bFirst)*187, 20))
		screen.blit(submitLabel, (screen.get_width()/5 - submitLabel.get_width()/2, screen.get_height() - 40))
		
		#print chosenAttributes
		#print chosenLevels
		
		for i in range(0, attributesPerPage):
			screen.blit(attributesSurfaces[i], (5, 90 + 72*i))
			screen.blit(levelsSurfacesForA[i], (360 - levelsSurfacesForA[i].get_width()/2, 90 + 72*i))
			screen.blit(levelsSurfacesForB[i], (545 - levelsSurfacesForB[i].get_width()/2, 90 + 72*i))
	
		pygame.display.update()
		
		msElapsed = clock.tick(60)

if __name__ == "__main__":
	pygame.init()
	main()
	pygame.quit()