def writeFrameCode(self, buff):
"""Write the code that will be called every frame.
"""
#some shortcuts
store = self.params['store'].val
storeCorr = self.params['storeCorrect'].val
forceEnd = self.params['forceEndRoutine'].val
useBoxTimer = self.params['useBoxTimer'].val
#check whether we need to test for allowed keys (or just take all)
allowedKeys = self.params['allowedKeys'].val.strip()
allowedKeysIsVar = _valid_var_re.match(
str(allowedKeys)) and not allowedKeys == 'None'
if allowedKeysIsVar: #only insert this code if we think allowed keys is a variable
# if it looks like a variable, check that the variable is suitable to eval at run-time
buff.writeIndented(
"# AllowedKeys looks like a variable named `%s`\n" %
allowedKeys)
buff.writeIndented("if not '%s' in locals():\n" % allowedKeys)
buff.writeIndented(
" logging.error('AllowedKeys variable `%s` is not defined.')\n"
% allowedKeys)
buff.writeIndented(" core.quit()\n")
buff.writeIndented(
"if not type(%s) in [list, tuple, np.ndarray]:\n" %
allowedKeys)
buff.writeIndented(" if not isinstance(%s, basestring):\n" %
allowedKeys)
buff.writeIndented(
" logging.error('AllowedKeys variable `%s` is not string- or list-like.')\n"
% allowedKeys)
buff.writeIndented(" core.quit()\n")
buff.writeIndented(" elif not ',' in %s: %s = (%s,)\n" %
(allowedKeys, allowedKeys, allowedKeys))
buff.writeIndented(" else: %s = eval(%s)\n" %
(allowedKeys, allowedKeys))
keyListStr = "keyList=list(%s)" % allowedKeys # eval() at run time
#now create the string that will loop-continue if
if allowedKeys in [None, "none", "None", "", "[]", "()"]:
keyCheckStr = ""
elif not allowedKeysIsVar:
try:
keyList = eval(allowedKeys)
except:
raise CodeGenerationException(self.params["name"],
"Allowed keys list is invalid.")
if type(
keyList
) == tuple: #this means the user typed "left","right" not ["left","right"]
keyList = list(keyList)
elif isinstance(keyList, basestring): #a single string/key
keyList = [keyList]
keyCheckStr = "%s" % (repr(keyList))
# if just now starting on this frame:
buff.writeIndented("# *%(name)s* updates\n" % self.params)
#write start code
self.writeStartTestCode(
buff) #writes an if statement to determine whether to start
buff.writeIndented("%(name)s.status = STARTED\n" % self.params)
buff.writeIndented("%(name)s.clock.reset() # now t=0\n" % self.params)
if self.params['discard previous'].val:
buff.writeIndented(
"# clear %(name)s responses (in a loop - the Cedrus own function doesn't work well)\n"
% self.params)
buff.writeIndented("%(name)s.poll_for_response()\n" % self.params)
buff.writeIndented("while len(%(name)s.response_queue):\n" %
self.params)
buff.writeIndented(" %(name)s.clear_response_queue()\n" %
self.params)
buff.writeIndented(
" %(name)s.poll_for_response() #often there are more resps waiting!\n"
% self.params)
if useBoxTimer:
buff.writeIndented(
"%(name)s.reset_rt_timer() #set the response time clock to 0\n"
% self.params)
buff.setIndentLevel(-1,
relative=True) # to get out of the if statement
#test for stop (only if there was some setting for duration or stop)
if self.params['stopVal'].val not in ['', None, -1, 'None']:
self.writeStopTestCode(
buff
) # writes an if statement to determine whether to draw etc
buff.writeIndented("%(name)s.status = STOPPED\n" % self.params)
buff.setIndentLevel(-1, True)
buff.writeIndented("if %(name)s.status == STARTED:\n" % self.params)
buff.setIndentLevel(1, relative=True) #to get out of the if statement
dedentAtEnd = 1 # keep track of how far to dedent later
buff.writeIndented("theseKeys=[]\n" % self.params)
buff.writeIndented("theseRTs=[]\n" % self.params)
buff.writeIndented("# check for key presses\n" % self.params)
buff.writeIndented("%(name)s.poll_for_response()\n" % self.params)
buff.writeIndented("while len(%(name)s.response_queue):\n" %
self.params)
buff.writeIndented(" evt = %(name)s.get_next_response()\n" %
self.params)
if len(keyCheckStr):
buff.writeIndented(" if evt['key'] not in %s:\n" % keyList)
buff.writeIndented(
" continue #we don't care about this key\n")
buff.writeIndented(
" if evt['pressed']: #could be extended to examine releases too?\n"
)
buff.writeIndented(" theseKeys.append(evt['key'])\n")
if useBoxTimer:
buff.writeIndented(
" theseRTs.append(evt['time']/1000.0) #NB psychopy times are in s not ms\n"
)
else:
buff.writeIndented(
" theseRTs.append(%(name)s.clock.getTime())\n" %
self.params)
buff.writeIndented(" %(name)s.poll_for_response()\n" % self.params)
buff.writeIndented(
"%(name)s.clear_response_queue() # make sure we don't process these evts again\n"
% self.params)
#how do we store it?
if store != 'nothing' or forceEnd:
#we are going to store something
buff.writeIndented(
"if len(theseKeys) > 0: # at least one key was pressed\n")
buff.setIndentLevel(1, True)
dedentAtEnd += 1 #indent by 1
if store == 'first key': #then see if a key has already been pressed
buff.writeIndented(
"if %(name)s.keys == []: # then this was the first keypress\n"
% (self.params))
buff.setIndentLevel(1, True)
dedentAtEnd += 1 #indent by 1
buff.writeIndented(
"%(name)s.keys = theseKeys[0] # just the first key pressed\n"
% (self.params))
buff.writeIndented("%(name)s.rt = theseRTs[0]\n" % (self.params))
elif store == 'last key':
buff.writeIndented(
"%(name)s.keys = theseKeys[-1] # just the last key pressed\n"
% (self.params))
buff.writeIndented("%(name)s.rt = theseRTs[-1]\n" % (self.params))
elif store == 'all keys':
buff.writeIndented(
"%(name)s.keys.extend(theseKeys) # storing all keys\n" %
(self.params))
buff.writeIndented("%(name)s.rt.extend(theseRTs)\n" %
(self.params))
else:
print store, type(store), str(store)
if storeCorr:
buff.writeIndented("# was this 'correct'?\n" % self.params)
buff.writeIndented(
"if (%(name)s.keys == str(%(correctAns)s)) or (%(name)s.keys == %(correctAns)s):\n"
% (self.params))
buff.writeIndented(" %(name)s.corr = 1\n" % (self.params))
buff.writeIndented("else:\n")
buff.writeIndented(" %(name)s.corr = 0\n" % (self.params))
if forceEnd == True:
buff.writeIndented("# a response ends the routine\n" % self.params)
buff.writeIndented("continueRoutine = False\n")
buff.setIndentLevel(-(dedentAtEnd), relative=True)
def writeFrameCode(self, buff):
"""Write the code that will be called every frame
"""
#some shortcuts
store = self.params['store'].val
storeCorr = self.params['storeCorrect'].val
forceEnd = self.params['forceEndRoutine'].val
allowedKeys = self.params['allowedKeys'].val.strip()
buff.writeIndented("\n")
buff.writeIndented("# *%s* updates\n" % (self.params['name']))
self.writeStartTestCode(
buff) #writes an if statement to determine whether to draw etc
buff.writeIndented("%(name)s.status = STARTED\n" % (self.params))
allowedKeysIsVar = _valid_var_re.match(
str(allowedKeys)) and not allowedKeys == 'None'
if allowedKeysIsVar:
# if it looks like a variable, check that the variable is suitable to eval at run-time
buff.writeIndented(
"# AllowedKeys looks like a variable named `%s`\n" %
allowedKeys)
buff.writeIndented("if not '%s' in locals():\n" % allowedKeys)
buff.writeIndented(
" logging.error('AllowedKeys variable `%s` is not defined.')\n"
% allowedKeys)
buff.writeIndented(" core.quit()\n")
buff.writeIndented(
"if not type(%s) in [list, tuple, np.ndarray]:\n" %
allowedKeys)
buff.writeIndented(" if not isinstance(%s, basestring):\n" %
allowedKeys)
buff.writeIndented(
" logging.error('AllowedKeys variable `%s` is not string- or list-like.')\n"
% allowedKeys)
buff.writeIndented(" core.quit()\n")
buff.writeIndented(" elif not ',' in %s: %s = (%s,)\n" %
(allowedKeys, allowedKeys, allowedKeys))
buff.writeIndented(" else: %s = eval(%s)\n" %
(allowedKeys, allowedKeys))
keyListStr = "keyList=list(%s)" % allowedKeys # eval() at run time
buff.writeIndented("# keyboard checking is just starting\n")
if store != 'nothing':
buff.writeIndented("%(name)s.clock.reset() # now t=0\n" %
self.params)
if self.params['discard previous'].val:
buff.writeIndented("event.clearEvents(eventType='keyboard')\n")
buff.setIndentLevel(-1, relative=True) #to get out of the if statement
#test for stop (only if there was some setting for duration or stop)
if self.params['stopVal'].val not in ['', None, -1, 'None']:
self.writeStopTestCode(
buff) #writes an if statement to determine whether to draw etc
buff.writeIndented("%(name)s.status = STOPPED\n" % (self.params))
buff.setIndentLevel(-1,
relative=True) #to get out of the if statement
buff.writeIndented("if %(name)s.status == STARTED:\n" % (self.params))
buff.setIndentLevel(1, relative=True) #to get out of the if statement
dedentAtEnd = 1 #keep track of how far to dedent later
#do we need a list of keys? (variable case is already handled)
if allowedKeys in [None, "none", "None", "", "[]", "()"]:
keyListStr = ""
elif not allowedKeysIsVar:
try:
keyList = eval(allowedKeys)
except:
raise CodeGenerationException(self.params["name"],
"Allowed keys list is invalid.")
if type(
keyList
) == tuple: #this means the user typed "left","right" not ["left","right"]
keyList = list(keyList)
elif isinstance(keyList, basestring): #a single string/key
keyList = [keyList]
keyListStr = "keyList=%s" % (repr(keyList))
#check for keypresses
buff.writeIndented("theseKeys = event.getKeys(%s)\n" % (keyListStr))
if self.exp.settings.params['Enable Escape'].val:
buff.writeIndentedLines('\n# check for quit:')
buff.writeIndented('if "escape" in theseKeys:\n')
buff.writeIndented(' endExpNow = True\n')
#how do we store it?
if store != 'nothing' or forceEnd:
#we are going to store something
buff.writeIndented(
"if len(theseKeys) > 0: # at least one key was pressed\n")
buff.setIndentLevel(1, True)
dedentAtEnd += 1 #indent by 1
if store == 'first key': #then see if a key has already been pressed
buff.writeIndented(
"if %(name)s.keys == []: # then this was the first keypress\n"
% (self.params))
buff.setIndentLevel(1, True)
dedentAtEnd += 1 #indent by 1
buff.writeIndented(
"%(name)s.keys = theseKeys[0] # just the first key pressed\n"
% (self.params))
buff.writeIndented("%(name)s.rt = %(name)s.clock.getTime()\n" %
(self.params))
elif store == 'last key':
buff.writeIndented(
"%(name)s.keys = theseKeys[-1] # just the last key pressed\n"
% (self.params))
buff.writeIndented("%(name)s.rt = %(name)s.clock.getTime()\n" %
(self.params))
elif store == 'all keys':
buff.writeIndented(
"%(name)s.keys.extend(theseKeys) # storing all keys\n" %
(self.params))
buff.writeIndented(
"%(name)s.rt.append(%(name)s.clock.getTime())\n" %
(self.params))
if storeCorr:
buff.writeIndented("# was this 'correct'?\n" % self.params)
buff.writeIndented(
"if (%(name)s.keys == str(%(correctAns)s)) or (%(name)s.keys == %(correctAns)s):\n"
% (self.params))
buff.writeIndented(" %(name)s.corr = 1\n" % (self.params))
buff.writeIndented("else:\n")
buff.writeIndented(" %(name)s.corr = 0\n" % (self.params))
if forceEnd == True:
buff.writeIndented("# a response ends the routine\n" % self.params)
buff.writeIndented("continueRoutine = False\n")
buff.setIndentLevel(-(dedentAtEnd), relative=True)
def writeFrameCode(self, buff):
"""Write the code that will be called every frame
"""
#some shortcuts
store = self.params['store'].val
storeCorr = self.params['storeCorrect'].val
forceEnd = self.params['forceEndRoutine'].val
buff.writeIndented("\n")
buff.writeIndented("# *%s* updates\n" % (self.params['name']))
self.writeStartTestCode(
buff) #writes an if statement to determine whether to draw etc
buff.writeIndented("%(name)s.status = STARTED\n" % (self.params))
buff.writeIndented("# keyboard checking is just starting\n")
if store != 'nothing':
buff.writeIndented("%(name)s.clock.reset() # now t=0\n" %
self.params)
if self.params['discard previous'].val:
buff.writeIndented("event.clearEvents()\n")
buff.setIndentLevel(-1, relative=True) #to get out of the if statement
#test for stop (only if there was some setting for duration or stop)
if self.params['stopVal'].val not in ['', None, -1, 'None']:
self.writeStopTestCode(
buff) #writes an if statement to determine whether to draw etc
buff.writeIndented("%(name)s.status = STOPPED\n" % (self.params))
buff.setIndentLevel(-1,
relative=True) #to get out of the if statement
buff.writeIndented(
"if %(name)s.status == STARTED: # only update if being drawn\n" %
(self.params))
buff.setIndentLevel(1, relative=True) #to get out of the if statement
dedentAtEnd = 1 #keep track of how far to dedent later
#do we need a list of keys?
if self.params['allowedKeys'].val in [None, "none", "None", "", "[]"]:
keyListStr = ""
else:
try:
keyList = eval(self.params['allowedKeys'].val)
except:
raise CodeGenerationException(self.params["name"],
"Allowed keys list is invalid.")
if type(
keyList
) == tuple: #this means the user typed "left","right" not ["left","right"]
keyList = list(keyList)
elif type(keyList) in [str, unicode]: #a single string value
keyList = [keyList]
keyListStr = "keyList=%s" % (repr(keyList))
#check for keypresses
buff.writeIndented("theseKeys = event.getKeys(%s)\n" % (keyListStr))
#how do we store it?
if store != 'nothing' or forceEnd:
#we are going to store something
buff.writeIndented(
"if len(theseKeys) > 0: # at least one key was pressed\n")
buff.setIndentLevel(1, True)
dedentAtEnd += 1 #indent by 1
if store == 'first key': #then see if a key has already been pressed
buff.writeIndented(
"if %(name)s.keys == []: # then this was the first keypress\n"
% (self.params))
buff.setIndentLevel(1, True)
dedentAtEnd += 1 #indent by 1
buff.writeIndented(
"%(name)s.keys = theseKeys[0] # just the first key pressed\n"
% (self.params))
buff.writeIndented("%(name)s.rt = %(name)s.clock.getTime()\n" %
(self.params))
elif store == 'last key':
buff.writeIndented(
"%(name)s.keys = theseKeys[-1] # just the last key pressed\n"
% (self.params))
buff.writeIndented("%(name)s.rt = %(name)s.clock.getTime()\n" %
(self.params))
elif store == 'all keys':
buff.writeIndented(
"%(name)s.keys.extend(theseKeys) # storing all keys\n" %
(self.params))
buff.writeIndented(
"%(name)s.rt.append(%(name)s.clock.getTime())\n" %
(self.params))
if storeCorr:
buff.writeIndented("# was this 'correct'?\n" % self.params)
buff.writeIndented(
"if (%(name)s.keys == str(%(correctAns)s)): %(name)s.corr = 1\n"
% (self.params))
buff.writeIndented("else: %(name)s.corr=0\n" % self.params)
if forceEnd == True:
buff.writeIndented("# abort routine on response\n" % self.params)
buff.writeIndented("continueRoutine = False\n")
buff.setIndentLevel(-(dedentAtEnd), relative=True)