import pygame, easygui, sys, random

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\".",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 = 0
	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
	##firstName = easygui.enterbox("Please enter your first name.")
	##lastName = easygui.enterbox("Please enter your last name.")
	##age = int(easygui.enterbox("Please enter your age."))
	#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
				if mousePos[0] >= 350 and mousePos[0] <= 370 and mousePos[1] >= 445 and mousePos[1] <= 465: #First button
					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
					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?"] = ""
					
				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
				trial += 1
				createPage = True
				if trial == numTrials:
					running = False
					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()