Merge pull request #75 from alyst/register_rdata

Register R data format

Author: SimonDanisch

src/registry.jl

@@ -1,6 +1,20 @@
 ### Simple cases
+
+# data formats
 add_format(format"JLD", "Julia data file (HDF5)", ".jld", [:JLD])
 
+# test for RD?2 magic sequence at the beginning of R data input stream
+function detect_rdata(io)
+    seekstart(io)
+    read(io, UInt8) == UInt8('R') &&
+    read(io, UInt8) == UInt8('D') &&
+    (fmt = read(io, UInt8); fmt == UInt8('A') || fmt == UInt8('B') || fmt == UInt8('X')) &&
+    read(io, UInt8) == UInt8('2') &&
+    read(io, UInt8) == 0x0A
+end
+
+add_format(format"RData", detect_rdata, [".rda", ".RData", ".rdata"], [:RData, LOAD])
+
 # Image formats
 add_format(format"PBMBinary", b"P4", ".pbm", [:ImageMagick])
 add_format(format"PGMBinary", b"P5", ".pgm", [:Netpbm])

test/files/minimal_ascii.rda

@@ -0,0 +1,47 @@
+RDA2
+A
+2
+196866
+131840
+1026
+1
+262153
+2
+df
+787
+1
+14
+2
+1.1
+2.2
+1026
+1
+262153
+5
+names
+16
+1
+262153
+3
+num
+1026
+1
+262153
+9
+row.names
+13
+2
+NA
+-2
+1026
+1
+262153
+5
+class
+16
+1
+262153
+10
+data.frame
+254
+254

test/query.jl

@@ -309,6 +309,15 @@ context("AVI Detection") do
     q = query(joinpath(file_dir, "bees.avi"))
     @fact typeof(q) --> File{format"AVI"}
 end
+context("RDA detection") do
+    q = query(joinpath(file_dir, "minimal_ascii.rda"))
+    @fact typeof(q) --> File{format"RData"}
+    open(q) do io
+        @fact position(io) --> 0
+        @fact FileIO.detect_rdata(io) --> true
+        @fact position(io) --> 5
+    end
+end
 context("Format with function for magic bytes") do
     add_format(format"FUNCTION_FOR_MAGIC_BYTES", x -> 0x00, ".wav", [:WAV])
     del_format(format"FUNCTION_FOR_MAGIC_BYTES")