Changeset 446 for trunk/src/io.py

Timestamp:

04/12/10 15:37:12 (2 years ago)

Author:

apdavison

Message:

SpikeTrain.time_slice() now (again) accepts a sequence of t_start values and a sequence of t_stop values, if you wish to extract multiple slices.

Files:

• trunk/src/io.py (modified) (1 diff)

trunk/src/io.py

@@ -393 +393 @@
 
 
+class PyNNNumpyBinaryFile(FileHandler):
+    
+    def __init__(self, filename):
+        FileHandler.__init__(self, filename)
+        self.fileobj = open(self.filename, 'r')
+        
+    def read_spikes(self, params):
+        from NeuroTools.signals import spikes
+        contents = numpy.load(self.fileobj)
+        spike_data = contents['data'][:, (1,0)]
+        self.metadata = M = {}
+        for k,v in contents['metadata']:
+            M[k] = eval(v)
+        id_list = range(M['first_id'], M['last_id'])
+        t_stop = params['t_stop']
+        # really need to check the agreement between file metadata and
+        # params for all metadata items
+        return spikes.SpikeList(spike_data, id_list, t_start=0.0, t_stop=t_stop,
+                                dims=M['dimensions'])
+        
+    #def read_analogs(self, type, params):
+    #    contents = numpy.load(self.fileobj)
+    #    values, ids = contents['data'].T # need to check the shape first
+    
+
 
 class DataHandler(object):