[med-svn] [python-mne] 21/376: writing events

[Yaroslav Halchenko]

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 <alexandre.gramfort at inria.fr>
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