Changeset 436 for trunk/src/signals/spikes.py

Timestamp:

09/25/09 14:02:04 (3 years ago)

Author:

apdavison

Message:

Modified the mapping of cell id to image position for plotting activity maps/movies.

Files:

• trunk/src/signals/spikes.py (modified) (8 diffs)

trunk/src/signals/spikes.py

@@ -1663 +1663 @@
 
     
-    def id2position(self, id):
+    def id2position(self, id, offset=0):
         """
         Return a position (x,y) from an id if the cells are aranged on a
-        grid of size dims, as defined in the dims attribute of the SpikeList object
+        grid of size dims, as defined in the dims attribute of the SpikeList object.
+        This assumes that cells are ordered from left to right, top to bottom,
+        and that dims specifies (height, width), i.e. if dims = (10,12), this is
+        an array with 10 rows and 12 columns, and hence has width 12 units and
+        height 10 units.
         
         Inputs:
@@ -1672 +1676 @@
         
         The 'dimensions' attribute of the SpikeList must be defined
-        
-        Examples:
-            >> spklist.id2position(2536)
-                (25, 35)
         
         See also
@@ -1683 +1683 @@
             raise Exception("Dimensions of the population are not defined ! Set spikelist.dimensions")
         if len(self.dimensions) == 1:
-            return id
+            return id-offset
         if len(self.dimensions) == 2:
-            x = id % self.dimensions[0]
-            y = numpy.floor(id/self.dimensions[0])
+            x = (id-offset) % self.dimensions[1]
+            y = self.dimensions[0] - 1 - int(numpy.floor((id-offset)/self.dimensions[1]))
             return (x,y)
 
 
-    def position2id(self, position):
+    def position2id(self, position, offset=0):
         """
         Return the id of the cell at position (x,y) if the cells are aranged on a
@@ -1701 +1701 @@
         as the position argument
         
-        Examples:
-            >> spklist.position2id((25,35))
-                2536
-        
         See also
             activity_map, activity_movie, id2position
@@ -1710 +1706 @@
         if self.dimensions is None:
             raise Exception("Dimensions of the population are not defined ! Set spikelist.dimensions")
-        assert array(position).shape == self.dimensions.shape, "position does not have the correct shape !"
+        assert len(position) == len(tuple(self.dimensions)), "position does not have the correct shape !"
         if len(self.dimensions) == 1:
-            return position
+            return position+offset
         if len(self.dimensions) == 2:
-            return position[0]*self.dimensions[0] + position[1]
+            return (self.dimensions[0] - 1 - position[1])*self.dimensions[1] + position[0] + offset
 
 
@@ -1728 +1724 @@
             float_positions - None by default, meaning that the dimensions attribute 
                               of the SpikeList is used to arange the ids on a 2D grid. 
-                              Otherwise, if the cells have flotting positions, 
+                              Otherwise, if the cells have floating positions, 
                               float_positions should be an array of size
                               (2, nb_cells) with the x (first line) and y (second line) 
@@ -1763 +1759 @@
             activity_map = numpy.zeros(self.dimensions, float)
             rates        = spklist.mean_rates()
+            id_offset = min(self.id_list())
             for count, id in enumerate(spklist.id_list()):
-                position = spklist.id2position(id)
-                activity_map[position] = rates[count]
+                x,y = spklist.id2position(id, id_offset)
+                j,i = x, self.dimensions[0] - 1 -y
+                activity_map[i,j] = rates[count]
             if not subplot or not HAVE_PYLAB or not HAVE_MATPLOTLIB:
                 return activity_map
@@ -2229 +2227 @@
                 while (t_start < t_stop):
                     activity_map = numpy.zeros(spk.dimensions)
-                    while ((time[idx] < t_start + time_bin) and (idx < max_idx)):
-                        addr = spk.id2position(pos[idx])
-                        activity_map[addr] += 1
+                    while ((time[idx] < t_start + time_bin) and (idx < max_idx)):                        
+                        x,y = spk.id2position(pos[idx])
+                        j,i = x, self.dimensions[0] - 1 -y
+                        activity_map[i,j] += 1
                         idx += 1
                     im.set_array(activity_map)