demo-village-project/task_test.py at main · BrainCircuitsBehaviorLab/demo-village-project · 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
from village.custom_classes.task import Event, Output, Task


class TaskTest(Task):
    def __init__(self):
        super().__init__()

    def start(self):
        pass

    def create_trial(self):

        # load play stop
        # load play
        # load play stop stop
        # load play play stop
        # load load play stop
        # load(1) play stop
        # load(1) play
        # play stop
        # stop load play

        self.bpod.add_state(
            state_name="load",
            state_timer=1,
            state_change_conditions={Event.Tup: "play"},
            output_actions=[],
        )

        self.bpod.add_state(
            state_name="play",
            state_timer=1,
            state_change_conditions={Event.Tup: "stop"},
            output_actions=[Output.SoftCode1],
        )

        self.bpod.add_state(
            state_name="stop",
            state_timer=1,
            state_change_conditions={Event.Tup: "exit"},
            output_actions=[Output.SoftCode1],
        )

    def after_trial(self):
        pass

    def close(self):
        pass