fix: loading rows for JDFFiles

Project.toml

@@ -1,7 +1,7 @@
 name = "JDF"
 uuid = "babc3d20-cd49-4f60-a736-a8f9c08892d3"
 authors = ["Dai ZJ <[email protected]>"]
-version = "0.5.2"
+version = "0.5.3"
 
 [deps]
 Blosc = "a74b3585-a348-5f62-a45c-50e91977d574"

build-readme.jl

@@ -7,7 +7,7 @@ upcheck()
 
 using Weave
 
-weave("../README.jmd", out_path="c:/git/JDF/", doctype="github")
+weave("../README.jmd", out_path="./", doctype="github")
 
 if false
     # debug

src/JDFFile.jl

@@ -71,17 +71,15 @@ function Base.getindex(file::JDFFile, col::Symbol)
     JDF.load(file; cols = [col])[col]
 end
 
+function Base.getindex(file::JDFFile, rows, col::String)
+    # TODO make it load from column loader for faster access
+    getfield(JDF.load(file; cols = [col]), Symbol(col))[rows, :]
+end
 
+function Base.getindex(file::JDFFile, rows, cols::AbstractVector{String})
+    JDF.load(file; cols = cols)[rows, :]
+end
 
-# function Base.getindex(file::JDFFile, rows, col::String)
-#     # TODO make it load from column loader for faster access
-#     getfield(JDF.load(file; cols = [col]), Symbol(col))[rows]
-# end
-
-# function Base.getindex(file::JDFFile, rows, cols::AbstractVector{String})
-#     JDF.load(file; cols = cols)[rows, :]
-# end
-
-# Base.view(file::JDFFile, rows, cols) = getindex(file, rows, cols)
+Base.view(file::JDFFile, rows, cols) = getindex(file, rows, cols)
 
-# getindex(file::JDFFile, rows, cols) = JDF.load(file)[rows, cols]
+getindex(file::JDFFile, rows, cols) = JDF.load(file)[rows, cols]

src/Tables.jl

@@ -41,3 +41,12 @@ Tables.istable(t::Table) = true
 function Base.getindex(t::Table, col::Symbol)
     t.columns[col]
 end
+
+function Base.getindex(t::Table, rows, col::Symbol)
+    t.columns[col][rows]
+end
+
+function Base.getindex(t::Table, rows, ::Colon)
+    # TODO probably not efficient
+    NamedTuple{names(t.columns)}([nt[rows] for nt in t.columns])
+end