| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102 |
- """
- Class for reading/writing data in a raw binary interleaved compact file.
- Sampling rate, units, number of channel and dtype must be externally known.
- This generic format is quite widely used in old acquisition systems and is quite universal
- for sharing data.
- Supported : Read/Write
- Author: sgarcia
- """
- import os
- import numpy as np
- import quantities as pq
- from neo.io.baseio import BaseIO
- from neo.core import Segment, AnalogSignal
- from neo.io.basefromrawio import BaseFromRaw
- from neo.rawio.rawbinarysignalrawio import RawBinarySignalRawIO
- class RawBinarySignalIO(RawBinarySignalRawIO, BaseFromRaw):
- """
- Class for reading/writing data in a raw binary interleaved compact file.
- **Important release note**
- Since the version neo 0.6.0 and the neo.rawio API,
- argmuents of the IO (dtype, nb_channel, sampling_rate) must be
- given at the __init__ and not at read_segment() because there is
- no read_segment() in neo.rawio classes.
- So now the usage is:
- >>>>r = io.RawBinarySignalIO(filename='file.raw', dtype='int16',
- nb_channel=16, sampling_rate=10000.)
- """
- _prefered_signal_group_mode = 'group-by-same-units'
- _default_group_mode_have_change_in_0_9 = True
- is_readable = True
- is_writable = True
- supported_objects = [Segment, AnalogSignal]
- readable_objects = [Segment]
- writeable_objects = [Segment]
- def __init__(self, filename, dtype='int16', sampling_rate=10000.,
- nb_channel=2, signal_gain=1., signal_offset=0., bytesoffset=0):
- RawBinarySignalRawIO.__init__(self, filename=filename, dtype=dtype,
- sampling_rate=sampling_rate, nb_channel=nb_channel,
- signal_gain=signal_gain,
- signal_offset=signal_offset, bytesoffset=bytesoffset)
- BaseFromRaw.__init__(self, filename)
- def write_segment(self, segment):
- """
- **Arguments**
- segment : the segment to write. Only analog signals will be written.
- Support only 2 cases:
- * segment.analogsignals have one 2D AnalogSignal
- * segment.analogsignals have several 1D AnalogSignal with
- same length/sampling_rate/dtype
- """
- if self.bytesoffset:
- raise NotImplementedError('bytesoffset values other than 0 ' +
- 'not supported')
- anasigs = segment.analogsignals
- assert len(anasigs) > 0, 'No AnalogSignal'
- anasig0 = anasigs[0]
- if len(anasigs) == 1 and anasig0.ndim == 2:
- numpy_sigs = anasig0.magnitude
- else:
- assert anasig0.ndim == 1 or (anasig0.ndim == 2 and anasig0.shape[1] == 1)
- # all AnaologSignal from Segment must have the same length/sampling_rate/dtype
- for anasig in anasigs[1:]:
- assert anasig.shape == anasig0.shape
- assert anasig.sampling_rate == anasig0.sampling_rate
- assert anasig.dtype == anasig0.dtype
- numpy_sigs = np.empty((anasig0.size, len(anasigs)))
- for i, anasig in enumerate(anasigs):
- numpy_sigs[:, i] = anasig.magnitude.flatten()
- numpy_sigs -= self.signal_offset
- numpy_sigs /= self.signal_gain
- numpy_sigs = numpy_sigs.astype(self.dtype)
- with open(self.filename, 'wb') as f:
- f.write(numpy_sigs.tobytes())
|
