from Entity import *

class NeuralEntity(Entity):
    def __init__(self, nsfile, eid, info):
        super(NeuralEntity,self).__init__(eid, nsfile, info)

    @property
    def probe_info(self):
        return self._info['ProbeInfo']

    @property
    def source_entity_id(self):
	return self._info['SourceEntityID']

    @property
    def source_unit_id(self):
        return self._info['SourceUnitID']

    def get_data (self, index=0, count=-1):
        lib = self.file.library
        if count < 0:
            count = self.item_count
        data = lib._get_neural_data (self, index, count)
        return data

@@ -43,16 +43,268 @@
 ntity):%0A
+    %22%22%22Entity the represents timestamps of action potentials, i.e. spike times.%0A    Cutouts of the waveforms corresponding to spike data in a neural entity%0A    might be found in a separate :class:%60SegmentEntity%60 (cf. :func:%60source_entity_id%60).%0A    %22%22%22%0A
     def 
@@ -526,16 +526,238 @@
 (self):%0A
+        %22%22%22%5B**Optional**%5D Id of the source entity of this spike, if any.%0A        For example the spike waveform of the action potential corresponding%0A        to this spike might have been recoreded in a segment entity.%22%22%22%0A
 %09return 
@@ -917,16 +917,181 @@
 nt=-1):%0A
+        %22%22%22Retrieve the spike times associated with this entity. A subset%0A        of the data can be requested via the %60%60inde%60%60 and %60%60count%60%60%0A        parameters.%22%22%22%0A
         

Add basic doc to Neural entities