Keisuke Sehara
/
STM32_Whisking
fast-feedback virtual target task code on STM Nucleo
states.cpp
- Committer:
- gwappa
- Date:
- 2018-07-05
- Revision:
- 26:b4421d1ee57a
- Parent:
- 25:56c4b22ec034
- Child:
- 27:b31ea8d74f9e
File content as of revision 26:b4421d1ee57a:
#include "states.h" #include "rig.h" #include "automaton.h" #include "IO.h" #define STATE_IS_LOGGED #ifdef STATE_IS_LOGGED #define LOGSTATE(S) IO::info(#S); #else #define LOGSTATE(S) #endif void finalize() { trial.markTrialEnd(); } void Delay::setup() { LOGSTATE(Delay) // initialize the trial-related params trial.reset(task); // set up the timeout for the next state switch (task.mode.value) { case Pair: case MotionAlt: stateTimeout.attach_us(&automaton::jump<Delay,Cued>, ms_to_us(trial.delay_dur_ms + task.prep_dur_ms.value)); break; default: stateTimeout.attach_us(&automaton::jump<Delay,Prepare>, ms_to_us(trial.delay_dur_ms)); } trial.markTrialStart(); } void Delay::teardown() { // do nothing } void Prepare::setup() { // mostly the same with for Delay // except that the animal cannot lick freely LOGSTATE(Prepare) // configure the interrupts lickIn.attach(&automaton::jump<Prepare,Abort>); // set timeout for the next state stateTimeout.attach_us(&automaton::jump<Prepare,Cued>, ms_to_us(task.prep_dur_ms.value)); } void Prepare::teardown() { // de-register lick inhibition lickIn.detach(); } void Cued::setup() { LOGSTATE(Cued) trial.markEndOfWait(); switch (task.mode.value) { case Pair: case Report: case Associate: // wait for the visual cue to flag "cued" break; case Motion: case MotionAlt: trial.flag.cued = true; break; } // configure lick handler if (task.mode.value != MotionAlt) { // licking without a visual "go-cue" will be considered a "catch" response lickIn.attach(&automaton::jump<Cued,WithResp>); } else { // for MotionAlt } // configure gate handler // use visual feedback trigger as the "gate" response switch (task.mode.value) { case Pair: case Report: case Associate: case Motion: gateIn.attach(&Cued::gate); break; case MotionAlt: gateIn.attach(&automaton::jump<Cued,WithResp>); break; default: // do nothing break; } // start cue output visualOut.run(); audioOut.run(); // sets the timeout for the next state stateTimeout.attach_us(&automaton::jump<Cued,NoResp>, trial.cued_dur_us); } void Cued::gate() { gateIn.detach(); lickIn.attach(&automaton::jump<Cued,WithResp>); trial.flag.cued = true; // in case it has not been (as in Pair/Report/Associate) } void Cued::teardown() { lickIn.detach(); // end cue output switch (task.mode.value) { case Report: case Associate: visualOut.stop(); } switch (task.mode.value) { case Associate: case Motion: case MotionAlt: audioOut.stop(); } } void Abort::setup() { LOGSTATE(Abort) lickIn.detach(); // if any gateIn.detach(); // if any trial.markEndOfWait(); trial.flag.reset = true; stateTimeout.attach_us(&automaton::done<Abort>, ms_to_us(task.post_dur_ms.value)); } void Abort::teardown() { finalize(); } void WithResp::setup() { LOGSTATE(WithResp) trial.flag.responded = true; lickIn.detach(); // if any gateIn.detach(); // if any // open/close the valve, only when the cue was there // (for Pair, reward will be there no matter the response) if ((task.mode.value == Pair) || trial.flag.cued ) { rewardOut.start(); } stateTimeout.attach_us(&automaton::done<WithResp>, ms_to_us(task.post_dur_ms.value)); } void WithResp::teardown() { finalize(); } void NoResp::setup() { LOGSTATE(NoResp) lickIn.detach(); // if any gateIn.detach(); // if any // no reward here, except for Pair if (task.mode.value == Pair) { rewardOut.start(); } stateTimeout.attach_us(&automaton::done<NoResp>, ms_to_us(task.post_dur_ms.value)); } void NoResp::teardown() { finalize(); } void TestReward::setup() { LOGSTATE(TestReward) // open/close the valve rewardOut.setOnset(0); rewardOut.setDuration(ms_to_us(task.reward_dur_ms.value)); rewardOut.start(); stateTimeout.attach_us(&automaton::done<TestReward>, ms_to_us(task.reward_dur_ms.value+100)); } void TestReward::teardown() { // do nothing }