[med-svn] [python-mne] 255/376: ENH : can crop SourceEstimate

[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 49b0fe03cd3d956bcd18e6ef447cebcc30d91648
Author: Alexandre Gramfort <alexandre.gramfort at inria.fr>
Date:   Tue May 17 12:27:29 2011 -0400

    ENH : can crop SourceEstimate
---
 mne/minimum_norm/inverse.py        |  2 +-
 mne/minimum_norm/time_frequency.py |  2 +-
 mne/source_estimate.py             | 10 ++++++++++
 3 files changed, 12 insertions(+), 2 deletions(-)

diff --git a/mne/minimum_norm/inverse.py b/mne/minimum_norm/inverse.py
index 2e99a5e..375f23e 100755
--- a/mne/minimum_norm/inverse.py
+++ b/mne/minimum_norm/inverse.py
@@ -20,7 +20,7 @@ from ..cov import read_cov
 from ..source_space import read_source_spaces_from_tree, find_source_space_hemi
 from ..forward import _block_diag
 from ..transforms import invert_transform, transform_source_space_to
-from ..stc import SourceEstimate
+from ..source_estimate import SourceEstimate
 
 
 def read_inverse_operator(fname):
diff --git a/mne/minimum_norm/time_frequency.py b/mne/minimum_norm/time_frequency.py
index d98092e..9d5b5c5 100644
--- a/mne/minimum_norm/time_frequency.py
+++ b/mne/minimum_norm/time_frequency.py
@@ -6,7 +6,7 @@ import numpy as np
 from scipy import linalg
 
 from ..fiff.constants import FIFF
-from ..stc import SourceEstimate
+from ..source_estimate import SourceEstimate
 from ..time_frequency.tfr import cwt, morlet
 from .inverse import combine_xyz, prepare_inverse_operator
 
diff --git a/mne/source_estimate.py b/mne/source_estimate.py
index 671dea6..397c1d8 100755
--- a/mne/source_estimate.py
+++ b/mne/source_estimate.py
@@ -155,6 +155,16 @@ class SourceEstimate(object):
         s += ", data size : %s x %s" % self.data.shape
         return "SourceEstimate (%s)" % s
 
+    def crop(self, tmin=None, tmax=None):
+        mask = np.ones(len(self.times), dtype=np.bool)
+        if tmin is not None:
+            mask = mask & (self.times >= tmin)
+        if tmax is not None:
+            mask = mask & (self.times <= tmax)
+        self.times = self.times[mask]
+        self.data = self.data[:, mask]
+        self.tmin = self.times[0]
+
 
 ###############################################################################
 # Morphing

-- 
Alioth's /usr/local/bin/git-commit-notice on /srv/git.debian.org/git/debian-med/python-mne.git