Recording Data in all three modules! #77
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,9 +1,11 @@ | ||
| import React, { useState, useCallback } from "react"; | ||
|
|
||
| import { Select, Card, Stack, Button, ButtonGroup, Modal, TextContainer } from "@shopify/polaris"; | ||
|
|
||
| import { mockMuseEEG } from "./utils/mockMuseEEG"; | ||
|
|
||
| import { generateXTics } from "./utils/chartUtils"; | ||
|
|
||
| import * as Intro from "./components/EEGEduIntro/EEGEduIntro" | ||
| import * as Raw from "./components/EEGEduRaw/EEGEduRaw"; | ||
| import * as Spectra from "./components/EEGEduSpectra/EEGEduSpectra"; | ||
|
|
@@ -13,20 +15,26 @@ import * as translations from "./translations/en.json"; | |
| import { MuseClient } from "muse-js"; | ||
| import * as generalTranslations from "./components/translations/en"; | ||
| import { emptyChannelData } from "./components/chartOptions"; | ||
| import { saveAs } from 'file-saver'; | ||
| import { take } from "rxjs/operators"; | ||
|
|
||
| export function PageSwitcher() { | ||
|
|
||
| const [introData, setIntroData] = useState(emptyChannelData) | ||
| const [rawData, setRawData] = useState(emptyChannelData); | ||
| const [spectraData, setSpectraData] = useState(emptyChannelData); | ||
| const [bandsData, setBandsData] = useState(emptyChannelData); | ||
| const [recordPop, setRecordPop] = useState(false); | ||
|
|
||
| const [introSettings] = useState(Intro.getSettings); | ||
| const [spectraSettings, setSpectraSettings] = useState(Spectra.getSettings); | ||
| const [rawSettings, setRawSettings] = useState(Raw.getSettings); | ||
| const [bandsSettings, setBandsSettings] = useState(Bands.getSettings); | ||
|
|
||
| const [status, setStatus] = useState(generalTranslations.connect); | ||
|
|
||
| const recordPopChange = useCallback(() => setRecordPop(!recordPop), [recordPop]); | ||
|
|
||
| // module at load: | ||
| const [selected, setSelected] = useState(translations.types.intro); | ||
| const handleSelectChange = useCallback(value => { | ||
|
|
@@ -40,6 +48,9 @@ export function PageSwitcher() { | |
| if (window.subscriptionSpectra$) window.subscriptionSpectra$.unsubscribe(); | ||
| if (window.subscriptionBands$) window.subscriptionBands$.unsubscribe(); | ||
|
|
||
| // buildPipe(value); | ||
| // if the case statements are uncommented below, | ||
| // need this, but then this crashes since runs before source is ready | ||
| subscriptionSetup(value); | ||
| // eslint-disable-next-line react-hooks/exhaustive-deps | ||
| }, []); | ||
|
|
@@ -51,6 +62,24 @@ export function PageSwitcher() { | |
| { label: translations.types.bands, value: translations.types.bands } | ||
| ]; | ||
|
|
||
| function buildPipe(value) { | ||
|
There was a problem hiding this comment. I did not get the case by case pipe building working in this pull request, but I did move this to a function and have it commented out for now so we can try again in a different branch/PR There was a problem hiding this comment.
Is the assumption that this will significantly improve performance? There was a problem hiding this comment. yes assumption is it will take less memory |
||
| // switch (value) { | ||
| // case translations.types.intro: | ||
| Intro.buildPipe(introSettings); | ||
| // break; | ||
| // case translations.types.raw: | ||
| Raw.buildPipe(rawSettings); | ||
| // break; | ||
| // case translations.types.spectra: | ||
| Spectra.buildPipe(spectraSettings); | ||
| // break; | ||
| // case translations.types.bands: | ||
| Bands.buildPipe(bandsSettings); | ||
| // break; | ||
| // default: console.log('Error building pipe') | ||
| // } | ||
| } | ||
|
|
||
| function subscriptionSetup(value) { | ||
| switch (value) { | ||
| case translations.types.intro: | ||
|
|
@@ -72,6 +101,106 @@ export function PageSwitcher() { | |
| } | ||
| } | ||
|
|
||
| function saveToCSV(value) { | ||
| const numSamplesToSave = 50; | ||
|
There was a problem hiding this comment. this is what we will need to adjust depending on time desired and epoch length There was a problem hiding this comment. Sampling rate will always be 256Hz unless it is a Muse 1 in which case it will be 220Hz. So, time desired would be
There was a problem hiding this comment. it isn't the sampling rate of the EEG, it is the interval between consecutive epochs of data in neurosity pipes |
||
| console.log('Saving ' + numSamplesToSave + ' samples...'); | ||
| var localObservable$ = null; | ||
| const dataToSave = []; | ||
|
|
||
| // for each module subscribe to multicast and make header | ||
| switch (value) { | ||
| case translations.types.raw: | ||
| //take one sample to get header info | ||
| localObservable$ = window.multicastRaw$.pipe( | ||
| take(1) | ||
| ); | ||
| localObservable$.subscribe({ | ||
| next(x) { | ||
| dataToSave.push( | ||
| "Timestamp (ms),", | ||
| generateXTics(x.info.samplingRate,x.data[0].length,false).map(function(f) {return "ch0_" + f + "ms"}) + ",", | ||
|
There was a problem hiding this comment. tediously creating the headers for each csv types There was a problem hiding this comment. This was for sure the hardest part and you have done great work making it useful for the user. Laudable efforts! |
||
| generateXTics(x.info.samplingRate,x.data[0].length,false).map(function(f) {return "ch1_" + f + "ms"}) + ",", | ||
| generateXTics(x.info.samplingRate,x.data[0].length,false).map(function(f) {return "ch2_" + f + "ms"}) + ",", | ||
| generateXTics(x.info.samplingRate,x.data[0].length,false).map(function(f) {return "ch3_" + f + "ms"}) + ",", | ||
| generateXTics(x.info.samplingRate,x.data[0].length,false).map(function(f) {return "chAux_" + f + "ms"}) + ",", | ||
| "info", | ||
| "\n" | ||
| ); | ||
| } | ||
| }); | ||
| console.log('making spectra headers') | ||
|
|
||
|
|
||
| localObservable$ = window.multicastRaw$.pipe( | ||
| take(numSamplesToSave) | ||
| ); | ||
| break; | ||
| case translations.types.spectra: | ||
| //take one sample to get header info | ||
| localObservable$ = window.multicastSpectra$.pipe( | ||
|
There was a problem hiding this comment. for the first two we subscribe and take 1 sample to get some info for the header |
||
| take(1) | ||
| ); | ||
| localObservable$.subscribe({ | ||
| next(x) { | ||
| let freqs = Object.values(x.freqs); | ||
| dataToSave.push( | ||
| "Timestamp (ms),", | ||
| freqs.map(function(f) {return "ch0_" + f + "Hz"}) + ",", | ||
| freqs.map(function(f) {return "ch1_" + f + "Hz"}) + ",", | ||
| freqs.map(function(f) {return "ch2_" + f + "Hz"}) + ",", | ||
| freqs.map(function(f) {return "ch3_" + f + "Hz"}) + ",", | ||
| freqs.map(function(f) {return "chAux_" + f + "Hz"}) + ",", | ||
| freqs.map(function(f) {return "f_" + f + "Hz"}) + "," , | ||
| "info", | ||
| "\n" | ||
| ); | ||
| } | ||
| }); | ||
| console.log('making spectra headers') | ||
|
|
||
| localObservable$ = window.multicastSpectra$.pipe( | ||
| take(numSamplesToSave) | ||
| ); | ||
| break; | ||
| case translations.types.bands: | ||
| console.log('making bands headers') | ||
| dataToSave.push( | ||
| "Timestamp (ms),", | ||
| "delta0,delta1,delta2,delta3,deltaAux,", | ||
| "theta0,theta1,theta2,theta3,thetaAux,", | ||
| "alpha0,alpha1,alpha2,alpha3,alphaAux,", | ||
| "beta0,beta1,beta2,beta3,betaAux,", | ||
| "delta0,delta1,delta2,delta3,deltaAux\n" | ||
| ); | ||
| localObservable$ = window.multicastBands$.pipe( | ||
| take(numSamplesToSave) | ||
| ); | ||
| break; | ||
| default: | ||
| console.log( | ||
| "Error on save to CSV: " + value | ||
| ); | ||
| } | ||
|
|
||
| localObservable$.subscribe({ | ||
| next(x) { | ||
| dataToSave.push(Date.now() + "," + Object.values(x).join(",") + "\n"); | ||
|
There was a problem hiding this comment. push values with commas between them to file |
||
| // logging is useful for debugging -yup | ||
| // console.log(x); | ||
| }, | ||
| error(err) { console.log(err); }, | ||
| complete() { | ||
| console.log('Trying to save') | ||
| var blob = new Blob( | ||
| dataToSave, | ||
| {type: "text/plain;charset=utf-8"} | ||
| ); | ||
| saveAs(blob, value + "_Recording.csv"); | ||
|
There was a problem hiding this comment. custom names for each module |
||
| console.log('Completed'); | ||
| } | ||
| }); | ||
| } | ||
|
|
||
| async function connect() { | ||
| try { | ||
| if (window.debugWithMock) { | ||
|
|
@@ -101,17 +230,14 @@ export function PageSwitcher() { | |
| window.source$.eegReadings | ||
| ) { | ||
| console.log("Connected to data source observable"); | ||
|
|
||
| // Build the data source | ||
| console.log("Starting to build the data pipes from the data source..."); | ||
| buildPipe(selected); | ||
|
There was a problem hiding this comment. now we call this function here instead of building each one |
||
| console.log("Finished building the data pipes from the data source"); | ||
| subscriptionSetup(selected); | ||
| console.log("Finished subscribing to the data source"); | ||
|
|
||
| } | ||
| } catch (err) { | ||
| setStatus(generalTranslations.connect); | ||
|
|
@@ -168,7 +294,92 @@ export function PageSwitcher() { | |
| } | ||
| } | ||
|
|
||
| function renderRecord() { | ||
|
There was a problem hiding this comment. This is the record button and popup when it is pressed |
||
| if (selected === translations.types.intro) { | ||
| console.log('No record on intro') | ||
| return null | ||
| } else if (selected === translations.types.raw) { | ||
|
There was a problem hiding this comment. if recording raw data there is a suggested setting explained for the user |
||
| return ( | ||
| <Card title={'Record Raw Data'} sectioned> | ||
| <Card.Section> | ||
| <p> | ||
| {"If you are recording raw data it is recommended you set the "} | ||
| {"number of sampling points between epochs onsets to be equal to the epoch duration. "} | ||
| {"This will ensure that consecutive rows of your output file are not overlapping in time."} | ||
| {"It will make the live plots appear more choppy."} | ||
| </p> | ||
| </Card.Section> | ||
| <Stack> | ||
| <ButtonGroup> | ||
| <Button | ||
| onClick={() => { | ||
| saveToCSV(selected); | ||
| recordPopChange(); | ||
| }} | ||
| primary={status !== generalTranslations.connect} | ||
| disabled={status === generalTranslations.connect} | ||
| > | ||
| {'Save to CSV'} | ||
| </Button> | ||
| </ButtonGroup> | ||
| <Modal | ||
| open={recordPop} | ||
| onClose={recordPopChange} | ||
| title="Recording Data" | ||
| > | ||
| <Modal.Section> | ||
| <TextContainer> | ||
| <p> | ||
| Your data is currently recording, | ||
| once complete it will be downloaded as a .csv file | ||
| and can be opened with your favorite spreadsheet program. | ||
| Close this window once the download completes. | ||
| </p> | ||
| </TextContainer> | ||
| </Modal.Section> | ||
| </Modal> | ||
| </Stack> | ||
| </Card> | ||
| ) | ||
| } else { | ||
| return ( | ||
| <Card title={'Record Data'} sectioned> | ||
| <Stack> | ||
| <ButtonGroup> | ||
| <Button | ||
| onClick={() => { | ||
| saveToCSV(selected); | ||
| recordPopChange(); | ||
| }} | ||
| primary={status !== generalTranslations.connect} | ||
| disabled={status === generalTranslations.connect} | ||
| > | ||
| {'Save to CSV'} | ||
| </Button> | ||
| </ButtonGroup> | ||
| <Modal | ||
| open={recordPop} | ||
| onClose={recordPopChange} | ||
| title="Recording Data" | ||
| > | ||
| <Modal.Section> | ||
| <TextContainer> | ||
| <p> | ||
| Your data is currently recording, | ||
| once complete it will be downloaded as a .csv file | ||
| and can be opened with your favorite spreadsheet program. | ||
| Close this window once the download completes. | ||
| </p> | ||
| </TextContainer> | ||
| </Modal.Section> | ||
| </Modal> | ||
| </Stack> | ||
| </Card> | ||
| ) | ||
| } | ||
| } | ||
|
|
||
| return ( | ||
| <React.Fragment> | ||
| <Card sectioned> | ||
| <Stack> | ||
|
|
@@ -199,7 +410,7 @@ export function PageSwitcher() { | |
| disabled={status === generalTranslations.connect} | ||
| > | ||
| {generalTranslations.disconnect} | ||
| </Button> | ||
| </ButtonGroup> | ||
| </Stack> | ||
| </Card> | ||
|
|
@@ -213,6 +424,7 @@ export function PageSwitcher() { | |
| </Card> | ||
| {pipeSettingsDisplay()} | ||
| {renderCharts()} | ||
| {renderRecord()} | ||
| </React.Fragment> | ||
| ); | ||
| } | ||
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
for the pop up when recording (modal window)
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Nice work getting a modal window pop'ped.