This is an automated email from the git hooks/post-receive script. yoh pushed a commit to annotated tag v0.1 in repository python-mne.
commit d514a11e6257d98b593e0a80278611a6a3f6df1a Author: Alexandre Gramfort <[email protected]> Date: Thu Dec 30 15:01:13 2010 -0500 writing events --- examples/read_events.py | 19 ++++++++++++------- fiff/__init__.py | 2 +- fiff/event.py | 19 +++++++++++++++++++ fiff/tests/test_event.py | 20 ++++++++++++++++++++ 4 files changed, 52 insertions(+), 8 deletions(-) diff --git a/examples/read_events.py b/examples/read_events.py index 4781ad1..14157be 100644 --- a/examples/read_events.py +++ b/examples/read_events.py @@ -7,11 +7,16 @@ import fiff fname = 'MNE-sample-data/MEG/sample/sample_audvis_raw-eve.fif' event_list = fiff.read_events(fname) +fiff.write_events('events.fif', event_list) +event_list2 = fiff.read_events(fname) -############################################################################### -# Show MEG data -import pylab as pl -pl.plot(times, data.T) -pl.xlabel('time (ms)') -pl.ylabel('MEG data (T)') -pl.show() +from scipy import linalg +print linalg.norm(event_list - event_list2) + +# ############################################################################### +# # Show MEG data +# import pylab as pl +# pl.plot(times, data.T) +# pl.xlabel('time (ms)') +# pl.ylabel('MEG data (T)') +# pl.show() diff --git a/fiff/__init__.py b/fiff/__init__.py index bfc53a9..dce372e 100644 --- a/fiff/__init__.py +++ b/fiff/__init__.py @@ -5,7 +5,7 @@ from .open import fiff_open from .evoked import read_evoked, write_evoked from .cov import read_cov, write_cov, write_cov_file from .raw import setup_read_raw, read_raw_segment, read_raw_segment_times -from .event import read_events +from .event import read_events, write_events from .forward import read_forward_solution from .stc import read_stc from .bem_surfaces import read_bem_surfaces diff --git a/fiff/event.py b/fiff/event.py index 6bfed59..fcac9a6 100644 --- a/fiff/event.py +++ b/fiff/event.py @@ -8,6 +8,7 @@ from .constants import FIFF from .tree import dir_tree_find from .tag import read_tag from .open import fiff_open +from .write import write_int, start_block, start_file, end_block, end_file def read_events(filename): @@ -49,3 +50,21 @@ def read_events(filename): event_list = event_list.reshape(len(event_list) / 3, 3) return event_list + + +def write_events(filename, event_list): + """ + % + % mne_write_events(filename,eventlist) + % + % Write an event list into a fif file + % + """ + # Start writing... + fid = start_file(filename) + + start_block(fid, FIFF.FIFFB_MNE_EVENTS) + write_int(fid, FIFF.FIFF_MNE_EVENT_LIST, event_list) + end_block(fid, FIFF.FIFFB_MNE_EVENTS) + + end_file(fid) diff --git a/fiff/tests/test_event.py b/fiff/tests/test_event.py new file mode 100644 index 0000000..5b224fc --- /dev/null +++ b/fiff/tests/test_event.py @@ -0,0 +1,20 @@ +import os +import os.path as op + +import numpy as np +from numpy.testing import assert_array_almost_equal + +import fiff + +MNE_SAMPLE_DATASET_PATH = os.getenv('MNE_SAMPLE_DATASET_PATH') +fname = op.join(MNE_SAMPLE_DATASET_PATH, 'MEG', 'sample', + 'sample_audvis_raw-eve.fif') + + +def test_io_cov(): + """Test IO for noise covariance matrices + """ + event_list = fiff.read_events(fname) + fiff.write_events('events.fif', event_list) + event_list2 = fiff.read_events(fname) + assert_array_almost_equal(event_list, event_list2) -- Alioth's /usr/local/bin/git-commit-notice on /srv/git.debian.org/git/debian-med/python-mne.git _______________________________________________ debian-med-commit mailing list [email protected] http://lists.alioth.debian.org/cgi-bin/mailman/listinfo/debian-med-commit
