pydaq/test_reward.py at master · codedragon/pydaq · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
from PyDAQmx import *
import numpy as np
import time
import pydaq

# A little test function to make sure reward is hooked up correctly. Reward
# as used in goBananas is in pydaq.py


reward = pydaq.GiveReward()
for i in range(3):
    reward.pumpOut()
    time.sleep(0.2)
reward.close()

# pumpTaskHandle = TaskHandle()
#
# try:
#     DAQmxCreateTask("", byref(pumpTaskHandle))
#     DAQmxCreateDOChan(pumpTaskHandle, "Dev1/port0/line0", "", DAQmx_Val_ChanPerLine)
#
#     DAQmxStartTask(pumpTaskHandle)
#
#     pumpWrite = np.zeros(1, dtype=numpy.uint8)
#     pumpWrite[0] = 1
#     DAQmxWriteDigitalLines(pumpTaskHandle, 1, False, DAQmx_Val_WaitInfinitely, DAQmx_Val_GroupByChannel,
#                            pumpWrite, None, None)
#     pumpWrite[0] = 0
#     time.sleep(.05)
#     DAQmxWriteDigitalLines(pumpTaskHandle, 1, False, DAQmx_Val_WaitInfinitely, DAQmx_Val_GroupByChannel,
#                            pumpWrite, None, None)
#     print "sent reward impulse"
#
# except DAQError as err:
#     print "DAQmx Error: %s"%err
#
# finally:
#     if pumpTaskHandle:
#         DAQmxStopTask(pumpTaskHandle)
#         DAQmxClearTask(pumpTaskHandle)

# Stuff for writing to file on other machine?
#pEncode = np.zeros(1, dtype=numpy.uint32)
#pEncodeTaskHandle = TaskHandle()
#
#try:
#    DAQmxCreateTask("",byref(pEncodeTaskHandle))
#    DAQmxCreateDOChan( pEncodeTaskHandle, "Dev1/port2","",DAQmx_Val_ChanForAllLines)
#
#    DAQmxStartTask(pEncodeTaskHandle)
#pEncode[0] = event
#DAQmxWriteDigitalU32(pEncodeTaskHandle, 1, 0, 10.0,
#                     DAQmx_Val_GroupByChannel, pEncode, byref(int32()),
#                     None)