Adds new type for spike waveforms

src/SpikeSorter.jl

@@ -8,6 +8,7 @@ include("viterbi.jl")
 include("readfiles.jl")
 include("features.jl")
 include("extraction.jl")
+include("io.jl")
 import GUICheck
 if GUICheck.hasgui()
 	include("plot.jl")

src/io.jl

@@ -0,0 +1,32 @@
+import Base.write
+
+function Base.write(io::IO, waveforms::SpikeWaveforms)
+    #write a header
+    headersize = 64
+    nbytes = 0
+    npoints, nchannels, nspikes = size(waveforms.waveforms) 
+    nbytes += write(io, npoints) 
+    nbytes += write(io, nchannels)
+    nbytes += write(io, nspikes)
+    seek(io, headersize)
+
+    #write timestamps, followed by waveforms
+    nbytes += write(io, waveforms.timestamps)
+    nbytes += write(io, waveforms.waveforms)
+    nbytes
+end
+
+function Base.read(io::IO, ::Type{SpikeWaveforms})
+    #read the header
+    headersize = 64
+    npoints = read(io, Int64)
+    nchannels = read(io, Int64)
+    nspikes = read(io, Int64)
+    seek(io, headersize)
+
+    #read timestamps
+    timestamps = read(io, Float64, nspikes)
+    waveforms = read(io, Float64, (npoints, nchannels, nspikes))
+    SpikeWaveforms(waveforms, timestamps)
+end
+

src/types.jl

@@ -1,5 +1,10 @@
 import HDF5
 
+type SpikeWaveforms
+    waveforms::Array{Float64,3}
+    timestamps::Array{Float64,1}
+end
+
 type TemplateFile
 	templates::Array{Float64,3}
 	cinv::Array{Float64,2}

test/io.jl

@@ -0,0 +1,9 @@
+wf = SpikeSorter.SpikeWaveforms(randn(60,1,1024),cumsum(rand(1024)))
+fname = tempname()
+
+write(fname, wf)
+wf2 = read(fname, SpikeSorter.SpikeWaveforms)
+
+@test_approx_eq wf.timestamps wf2.timestamps
+@test_approx_eq wf.waveforms wf2.waveforms
+

test/runtests.jl

@@ -0,0 +1,4 @@
+using Base.Test
+using SpikeSorter
+
+include("io.jl")