check user input using vizinfo panel

[armo]

Dear all

I am running an experiment with a 3x2 design, so that I have two variables, one with three levels and the other one with two.
Before the experiment starts, the experimenter has to choose what sequence of two levels, each from one of the two variables, have to be presented, but I would like to create a mechanism that controls if the input sequence has already been chosen. Specifically, I posed these conditions:

- if the sequence of variables has not already been chosen, the execution flow keeps going on
- if the sequence of variables has already been chosen but the experiment wants to go on anyway, the output data are renamed to prevent overwriting.
- if the sequence of variables has already been chosen and the experimenter wants to choose another sequence, turns back to the vizinfo panel.

The first two conditions work, but the third doesn't. Specifically, the user can choose another sequence but when he hits again the run button nothing happens and so the experiment doesn't go on. Could anyone tell me where I did wrong in my code?


best regards,
armo

import viz
import vizinfo

class Experiment:
def __init__(self,title):
self.title = title

# create a class object with the current experiment data
current_exp = Experiment('current experiment')


class ExperimentalSetting:
def __init__(self, var1=None, var2=None):
self.var1 = var1
self.var2 = var2

class ExperimentalSession:
def __init__(self, phase=None, number=None):
self.phase = phase
self.number = number
self.settings = ExperimentalSetting()

def __repr__(self):
return self.phase + ' ' + str(self.number)


# create a class object with the current session data
current_session = ExperimentalSession()

viz.go()

def get_session_info():
global phase
global current_session
global current_RT_file
global current_ACC_file

import os
def check_existing_file(file_name, extension):
'''checks if a file already exists'''
if os.path.exists( file_name + extension ):
print 'file already exists...'
return True
else: return False

def add_suffix(f):
'''adds a suffix to a file.
Used when a file already exists'''
copy_suffix = 'Copy'
f = f + ' - ' + copy_suffix
return f

def rename_existing_file(file_name, extension):
'''if a file already exists, rename it with a suffix extension'''
if check_existing_file(file_name, extension):
checked_fileName = add_suffix(file_name) + extension
else: checked_fileName = file_name + extension
return checked_fileName

current_RT_file = None
current_ACC_file = None

isEqual = False
while not isEqual:
print 'displaying vizinfo box'
#Add the vizinfo box
infoBox = vizinfo.add('')
infoBox.scale(1.5,1.5)
infoBox.translate( 0.85, 0.95 )
infoBox.title('Experiment Info: ' + current_exp.title)

#Add the GUI elements to the box
phaseTrainingBox = infoBox.add(viz.RADIO,1,'Training')
phaseExperimentBox = infoBox.add(viz.RADIO,1,'Experiment')
variable1_A_Box = infoBox.add(viz.RADIO,2,'A')
variable1_B_Box = infoBox.add(viz.RADIO,2,'B')
variable1_C_Box = infoBox.add(viz.RADIO,2,'C')
variable2_1_Box = infoBox.add(viz.RADIO,3,'1')
variable2_2_Box = infoBox.add(viz.RADIO,3,'2')

runButton = infoBox.add(viz.BUTTON,'Run')

# waits for pressing the runbutton to go forward
yield viztask.waitButtonUp(runButton)

# get the current session settings
if phaseTrainingBox.get() == viz.DOWN:
current_session.phase = 'Training'
elif phaseExperimentBox.get() == viz.DOWN:
current_session.phase = 'Experiment'

if variable1_A_Box.get() == viz.DOWN:
current_session.settings.var1 = 'A'
elif variable1_B_Box.get() == viz.DOWN:
current_session.settings.var1 = 'B'
elif variable1_C_Box.get() == viz.DOWN:
current_session.settings.var1 = 'B'

if variable2_1_Box.get() == viz.DOWN:
current_session.settings.var2 = '1'
elif variable2_2_Box.get() == viz.DOWN:
current_session.settings.var2 = '2'

current_file_labels = [
current_session.phase,
current_session.settings.var1,
current_session.settings.var2,
]

# create two files object with the experiment RT and ACC data
RT_folder = 'data/RT/'
ACC_folder = 'data/ACC/'

current_RT_file_name = RT_folder + '_'.join(current_file_labels) + '_RT'
current_ACC_file_name = ACC_folder + '_'.join(current_file_labels) + '_ACC'

print current_RT_file_name

if check_existing_file(current_RT_file_name, '.txt') and check_existing_file(current_ACC_file_name, '.txt'):
answer = viz.ask('this sequence has already been chosen. Continue?')
if answer:
print 'chose to continue and rename file'
current_RT_file = open( rename_existing_file(current_RT_file_name, '.txt'),'w')
current_ACC_file = open( rename_existing_file(current_ACC_file_name, '.txt'),'w')
isEqual = True
else:
print 'chose not to continue'
return
else:
print 'the file names don\'t already exist'
current_RT_file = open( current_RT_file_name + '.txt','w')
current_ACC_file = open( current_ACC_file_name + '.txt','w')
break

infoBox.remove()

print 'current settings:', [x for x in current_file_labels]

def experimental_flow():
yield get_session_info()

import viztask
viztask.schedule(experimental_flow)