pyScienceMode2/example.py at main · ArseneBA/pyScienceMode2 · 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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
# Import Stimulator class
from pyScienceMode2 import Stimulator as St
# Import Channel class
from pyScienceMode2 import Channel as Ch

from time import sleep

# Create a list of channels
list_channels = []

# Create all channels possible
channel_1 = Ch.Channel(mode='Single',
                       no_channel=1,
                       amplitude=50,
                       pulse_width=100,
                       enable_low_frequency=False,
                       name='Biceps')

channel_2 = Ch.Channel()
channel_2.set_mode('Single')
channel_2.set_no_channel(2)
channel_2.set_amplitude(2)
channel_2.set_pulse_width(100)
channel_2.set_name('Triceps')

channel_3 = Ch.Channel('Doublet', 3, 50, 100)
channel_4 = Ch.Channel('Single', 4, 50, 100)
channel_5 = Ch.Channel('Triplet', 5, 50, 100)
channel_6 = Ch.Channel('Single', 6, 50, 100, True)
channel_7 = Ch.Channel('Single', 7, 50, 100)
channel_8 = Ch.Channel('Single', 8, 50, 100)

# Choose which channel will be used
list_channels.append(channel_1)
list_channels.append(channel_3)
list_channels.append(channel_5)
list_channels.append(channel_6)
list_channels.append(channel_7)
list_channels.append(channel_8)

# Create our object Stimulator
stimulator = St.Stimulator(list_channels=list_channels,
                           stimulation_interval=1000,
                           port_path='COM5',
                           inter_pulse_interval=120,
                           low_frequency_factor=1)

# Display log, communication or/and watchdog, by default True if called, to deactivate : stimulator.show_log(False)
stimulator.show_log()
# stimulator.show_com()
# stimulator.show_watchdog()

"""
Initialise the channels given.
It is possible to modify the list of channels, the stimulation interval and the low_frequency_factor
"""
stimulator.init_channel()
# stimulator.init_channel(stimulation_interval=200, list_channels=nv_list_channel, low_frequency_factor=2)

"""
Start the stimulation.
It is possible to :
- Give a time after which the stimulation will be stopped but not disconnected.
- Update the parameters of the channel by giving a new list of channels. The channel given must have been
  initialised first.
"""
stimulator.start_stimulation()
# stimulator.start_stimulation(stimulation_duration=10, upd_list_channels=nw_list_channel)

# Modify some parameters,
list_channels[2].set_amplitude(10)
list_channels[3].set_amplitude(15)

# Wait a given time in seconds
sleep(10)

# Update the parameters of the stimulation
stimulator.start_stimulation(upd_list_channels=list_channels)

# Wait a given time in seconds
sleep(5)

"""
Stop the stimulation. But does not disconnect the Pc and the Rehastim.
"""
stimulator.stop_stimulation()

"""
Restart a stimulation with the same parameter for 2 seconds.
"""
stimulator.start_stimulation(stimulation_duration=2)

"""
The method init_channel must be called to update the stimulation interval (period).
"""
stimulator.init_channel(stimulation_interval=10)
stimulator.start_stimulation(stimulation_duration=2)

"""
To disconnect the computer and the Rehastim, use the disconnect method.
"""
stimulator.disconnect()

"""
After a disconnection, init_channel must be called.
"""
stimulator.init_channel(stimulation_interval=15)
stimulator.start_stimulation(2, list_channels)
stimulator.disconnect()

"""
close_port method closes the serial port.
"""
stimulator.close_port()