Keisuke Sehara
/
STM32_Whisking
fast-feedback virtual target task code on STM Nucleo
trial.cpp@27:b31ea8d74f9e, 2018-07-21 (annotated)
- Committer:
- gwappa
- Date:
- Sat Jul 21 00:36:38 2018 +0000
- Revision:
- 27:b31ea8d74f9e
- Parent:
- 26:b4421d1ee57a
- Child:
- 28:797536a42b9f
add scheduler in order to fix the zero-latency bug
Who changed what in which revision?
User | Revision | Line number | New contents of line |
---|---|---|---|
gwappa | 2:0c241937eabd | 1 | #include "trial.h" |
gwappa | 11:897ecd5413e0 | 2 | #include "rig.h" |
gwappa | 2:0c241937eabd | 3 | #include "IO.h" |
gwappa | 2:0c241937eabd | 4 | #include "random.h" |
gwappa | 2:0c241937eabd | 5 | |
gwappa | 11:897ecd5413e0 | 6 | const uint64_t MicrosecondsInSecond = 1000000; |
gwappa | 11:897ecd5413e0 | 7 | |
gwappa | 20:4c06d3041337 | 8 | inline uint64_t calc_blink_duration(const uint16_t& freq) { |
gwappa | 20:4c06d3041337 | 9 | return (freq == 0)? 0 : (MicrosecondsInSecond/(freq * 2)); |
gwappa | 20:4c06d3041337 | 10 | } |
gwappa | 20:4c06d3041337 | 11 | |
gwappa | 20:4c06d3041337 | 12 | void TrialFlag::clear() { |
gwappa | 19:50663f8815b8 | 13 | cued = false; |
gwappa | 19:50663f8815b8 | 14 | responded = false; |
gwappa | 19:50663f8815b8 | 15 | reset = false; |
gwappa | 19:50663f8815b8 | 16 | } |
gwappa | 19:50663f8815b8 | 17 | |
gwappa | 20:4c06d3041337 | 18 | void TrialFlag::writeToSerial() { |
gwappa | 19:50663f8815b8 | 19 | if (reset) { |
gwappa | 19:50663f8815b8 | 20 | IO::write("reset"); |
gwappa | 19:50663f8815b8 | 21 | } else if (cued) { |
gwappa | 19:50663f8815b8 | 22 | IO::write(responded? "hit":"miss"); |
gwappa | 19:50663f8815b8 | 23 | } else { |
gwappa | 19:50663f8815b8 | 24 | IO::write(responded? "catch":"reject"); |
gwappa | 19:50663f8815b8 | 25 | } |
gwappa | 19:50663f8815b8 | 26 | } |
gwappa | 19:50663f8815b8 | 27 | |
gwappa | 2:0c241937eabd | 28 | void Trial::reset(const Task& task){ |
gwappa | 2:0c241937eabd | 29 | // reset score |
gwappa | 20:4c06d3041337 | 30 | flag.clear(); |
gwappa | 2:0c241937eabd | 31 | |
gwappa | 2:0c241937eabd | 32 | // set delay duration |
gwappa | 5:849446d19406 | 33 | delay_dur_ms = task.delay_min_ms.value |
gwappa | 5:849446d19406 | 34 | + random::exponential(task.delay_var_ms.value, task.delay_var_ms.value*3); |
gwappa | 11:897ecd5413e0 | 35 | |
gwappa | 24:e236faf66935 | 36 | // reset cues |
gwappa | 20:4c06d3041337 | 37 | audioOut.setEnabled(false); |
gwappa | 15:20f7f737c256 | 38 | visualOut.setEnabled(false); |
gwappa | 16:33c17c62840e | 39 | vis_dur_us = 0; |
gwappa | 16:33c17c62840e | 40 | vis_onset_us = 0; |
gwappa | 15:20f7f737c256 | 41 | |
gwappa | 24:e236faf66935 | 42 | // configure reward |
gwappa | 24:e236faf66935 | 43 | rewardOut.setOnset(0); |
gwappa | 24:e236faf66935 | 44 | rewardOut.setDuration(ms_to_us(task.reward_dur_ms.value)); |
gwappa | 24:e236faf66935 | 45 | |
gwappa | 24:e236faf66935 | 46 | // configure cued state duration + visual stimulus position (if any) |
gwappa | 15:20f7f737c256 | 47 | switch (task.mode.value) { |
gwappa | 15:20f7f737c256 | 48 | case Pair: |
gwappa | 15:20f7f737c256 | 49 | case Report: |
gwappa | 15:20f7f737c256 | 50 | case Associate: |
gwappa | 24:e236faf66935 | 51 | visualOut.setBlinkDuration( calc_blink_duration(task.vis_blink_hz.value) ); |
gwappa | 24:e236faf66935 | 52 | assignCues(task, true); |
gwappa | 24:e236faf66935 | 53 | break; |
gwappa | 16:33c17c62840e | 54 | case Motion: |
gwappa | 16:33c17c62840e | 55 | case MotionAlt: |
gwappa | 24:e236faf66935 | 56 | assignCues(task, false); |
gwappa | 24:e236faf66935 | 57 | break; |
gwappa | 24:e236faf66935 | 58 | } |
gwappa | 24:e236faf66935 | 59 | |
gwappa | 24:e236faf66935 | 60 | // assign Auditory cues (if any) |
gwappa | 24:e236faf66935 | 61 | switch (task.mode.value) { |
gwappa | 24:e236faf66935 | 62 | case Pair: |
gwappa | 24:e236faf66935 | 63 | case Report: |
gwappa | 24:e236faf66935 | 64 | break; |
gwappa | 24:e236faf66935 | 65 | case Associate: |
gwappa | 24:e236faf66935 | 66 | case Motion: |
gwappa | 24:e236faf66935 | 67 | case MotionAlt: |
gwappa | 24:e236faf66935 | 68 | audioOut.setEnabled(true); |
gwappa | 24:e236faf66935 | 69 | audioOut.setOnset(0); |
gwappa | 24:e236faf66935 | 70 | audioOut.setDuration(cued_dur_us); |
gwappa | 24:e236faf66935 | 71 | audioOut.setBlinkDuration( calc_blink_duration(task.aud_tick_hz.value) ); |
gwappa | 15:20f7f737c256 | 72 | break; |
gwappa | 13:8ea85a33e37a | 73 | } |
gwappa | 2:0c241937eabd | 74 | |
gwappa | 3:991c6d5ce19d | 75 | // reset interrupt events |
gwappa | 26:b4421d1ee57a | 76 | lickIn.reset(task.lick_debounce_ms.value, true); |
gwappa | 26:b4421d1ee57a | 77 | whiskIn.reset(task.whisk_debounce_ms.value, true); |
gwappa | 26:b4421d1ee57a | 78 | gateIn.reset(-1, true); |
gwappa | 2:0c241937eabd | 79 | } |
gwappa | 2:0c241937eabd | 80 | |
gwappa | 24:e236faf66935 | 81 | void Trial::assignCues(const Task& task, const bool& assignVisual) |
gwappa | 13:8ea85a33e37a | 82 | { |
gwappa | 24:e236faf66935 | 83 | if (!assignVisual) { |
gwappa | 24:e236faf66935 | 84 | visualOut.setEnabled(false); |
gwappa | 27:b31ea8d74f9e | 85 | visualOut.direct(true); |
gwappa | 24:e236faf66935 | 86 | vis_dur_us = 0; |
gwappa | 24:e236faf66935 | 87 | vis_onset_us = 0; |
gwappa | 24:e236faf66935 | 88 | |
gwappa | 24:e236faf66935 | 89 | cued_dur_us = ms_to_us(task.aud_dur_ms.value); |
gwappa | 24:e236faf66935 | 90 | return; |
gwappa | 24:e236faf66935 | 91 | } |
gwappa | 24:e236faf66935 | 92 | |
gwappa | 13:8ea85a33e37a | 93 | const uint32_t phasedur_us = ms_to_us(task.aud_dur_ms.value); |
gwappa | 15:20f7f737c256 | 94 | const uint32_t minonset_us = ms_to_us(task.pre_min_ms.value); |
gwappa | 13:8ea85a33e37a | 95 | const uint32_t respdur_us = ms_to_us(task.resp_dur_ms.value); |
gwappa | 14:af3adf5d5ddf | 96 | const uint32_t avgdur_us = ms_to_us(task.vis_avg_ms.value); |
gwappa | 13:8ea85a33e37a | 97 | const uint32_t mindur_us = ms_to_us(task.vis_min_ms.value); |
gwappa | 15:20f7f737c256 | 98 | const double failure_perc = (task.vis_fail_perc.value > 0)? ((double)(task.vis_fail_perc.value)):1e-6; |
gwappa | 13:8ea85a33e37a | 99 | |
gwappa | 20:4c06d3041337 | 100 | const double onset_vrng = (double)(phasedur_us - minonset_us - respdur_us - mindur_us); |
gwappa | 20:4c06d3041337 | 101 | const double logdecay = ::log(100.0) - ::log(failure_perc); |
gwappa | 20:4c06d3041337 | 102 | const double ftau = onset_vrng/logdecay; |
gwappa | 14:af3adf5d5ddf | 103 | const uint32_t tau = (uint32_t)(ftau + 0.5); |
gwappa | 20:4c06d3041337 | 104 | const uint32_t onset_us = random::exponential(tau, onset_vrng+10) + minonset_us; |
gwappa | 20:4c06d3041337 | 105 | const uint32_t maxonset_us = onset_vrng + minonset_us; |
gwappa | 13:8ea85a33e37a | 106 | |
gwappa | 20:4c06d3041337 | 107 | if ((onset_us > 0) && (onset_us < maxonset_us)) { |
gwappa | 13:8ea85a33e37a | 108 | // generate duration |
gwappa | 20:4c06d3041337 | 109 | const uint32_t dur_vrng = maxonset_us - onset_us; |
gwappa | 20:4c06d3041337 | 110 | const uint32_t dur_us = random::exponential(avgdur_us - mindur_us, dur_vrng) + mindur_us; |
gwappa | 13:8ea85a33e37a | 111 | |
gwappa | 15:20f7f737c256 | 112 | visualOut.setEnabled(true); |
gwappa | 17:0b241aa1f5b6 | 113 | visualOut.setDuration(dur_us); |
gwappa | 17:0b241aa1f5b6 | 114 | visualOut.setOnset(onset_us); |
gwappa | 13:8ea85a33e37a | 115 | |
gwappa | 17:0b241aa1f5b6 | 116 | vis_dur_us = dur_us; |
gwappa | 17:0b241aa1f5b6 | 117 | vis_onset_us = onset_us; |
gwappa | 24:e236faf66935 | 118 | |
gwappa | 24:e236faf66935 | 119 | cued_dur_us = onset_us + dur_us + respdur_us; |
gwappa | 24:e236faf66935 | 120 | |
gwappa | 13:8ea85a33e37a | 121 | } else { |
gwappa | 15:20f7f737c256 | 122 | // no pulse |
gwappa | 15:20f7f737c256 | 123 | visualOut.setEnabled(false); |
gwappa | 16:33c17c62840e | 124 | vis_dur_us = 0; |
gwappa | 16:33c17c62840e | 125 | vis_onset_us = 0; |
gwappa | 24:e236faf66935 | 126 | |
gwappa | 24:e236faf66935 | 127 | cued_dur_us = phasedur_us; |
gwappa | 13:8ea85a33e37a | 128 | } |
gwappa | 13:8ea85a33e37a | 129 | } |
gwappa | 13:8ea85a33e37a | 130 | |
gwappa | 11:897ecd5413e0 | 131 | void Trial::markTrialStart() { |
gwappa | 26:b4421d1ee57a | 132 | starting = TrialTimer.read_ms(); |
gwappa | 12:06ea96546af1 | 133 | trialStart.run(); |
gwappa | 26:b4421d1ee57a | 134 | TrialTimer.start(); |
gwappa | 11:897ecd5413e0 | 135 | } |
gwappa | 11:897ecd5413e0 | 136 | |
gwappa | 11:897ecd5413e0 | 137 | void Trial::markEndOfWait() { |
gwappa | 26:b4421d1ee57a | 138 | cuestarting = TrialTimer.read_ms(); |
gwappa | 11:897ecd5413e0 | 139 | waiting = cuestarting - starting; |
gwappa | 11:897ecd5413e0 | 140 | } |
gwappa | 11:897ecd5413e0 | 141 | |
gwappa | 11:897ecd5413e0 | 142 | void Trial::markTrialEnd() { |
gwappa | 12:06ea96546af1 | 143 | trialEnd.run(); |
gwappa | 11:897ecd5413e0 | 144 | trialEnd.wait(); |
gwappa | 11:897ecd5413e0 | 145 | writeToSerial(); |
gwappa | 26:b4421d1ee57a | 146 | TrialTimer.stop(); |
gwappa | 26:b4421d1ee57a | 147 | TrialTimer.reset(); |
gwappa | 27:b31ea8d74f9e | 148 | visualOut.direct(false); // if any |
gwappa | 11:897ecd5413e0 | 149 | } |
gwappa | 11:897ecd5413e0 | 150 | |
gwappa | 2:0c241937eabd | 151 | void Trial::writeToSerial() { |
gwappa | 19:50663f8815b8 | 152 | IO::write("%c",IO::RESULT_HEADER); |
gwappa | 19:50663f8815b8 | 153 | flag.writeToSerial(); |
gwappa | 2:0c241937eabd | 154 | |
gwappa | 19:50663f8815b8 | 155 | IO::write(";wait%u", waiting); |
gwappa | 13:8ea85a33e37a | 156 | |
gwappa | 20:4c06d3041337 | 157 | if (visualOut.isEnabled() && (!flag.reset)) { |
gwappa | 23:61af0317e404 | 158 | IO::write(";visual(onset%s,duration%s)", |
gwappa | 23:61af0317e404 | 159 | uint64_to_str(vis_onset_us).c_str(), |
gwappa | 23:61af0317e404 | 160 | uint64_to_str(vis_dur_us).c_str()); |
gwappa | 13:8ea85a33e37a | 161 | } |
gwappa | 13:8ea85a33e37a | 162 | |
gwappa | 19:50663f8815b8 | 163 | trialtime_t zero = (flag.reset)? starting : cuestarting; |
gwappa | 3:991c6d5ce19d | 164 | IO::write(";whisk"); |
gwappa | 26:b4421d1ee57a | 165 | whiskIn.writeToSerial(zero); |
gwappa | 26:b4421d1ee57a | 166 | IO::write(";gate"); |
gwappa | 26:b4421d1ee57a | 167 | gateIn.writeToSerial(zero); |
gwappa | 3:991c6d5ce19d | 168 | IO::write(";lick"); |
gwappa | 26:b4421d1ee57a | 169 | lickIn.writeToSerial(zero); |
gwappa | 4:fcf597f82632 | 170 | IO::write(";\r\n"); |
gwappa | 2:0c241937eabd | 171 | } |