feat: add FileIO support (#14)

Author: johnnychen94

Project.toml

@@ -5,11 +5,13 @@ version = "0.1.0"
 
 [deps]
 CEnum = "fa961155-64e5-5f13-b03f-caf6b980ea82"
+FileIO = "5789e2e9-d7fb-5bc7-8068-2c6fae9b9549"
 ImageCore = "a09fc81d-aa75-5fe9-8630-4744c3626534"
 JpegTurbo_jll = "aacddb02-875f-59d6-b918-886e6ef4fbf8"
 
 [compat]
 CEnum = "0.3, 0.4"
+FileIO = "1"
 ImageCore = "0.8, 0.9"
 JpegTurbo_jll = "~2.1"
 julia = "1.6"

src/JpegTurbo.jl

@@ -10,6 +10,8 @@ include("common.jl")
 include("encode.jl")
 include("decode.jl")
 
+include("fileio.jl")
+
 export jpeg_encode, jpeg_decode
 
 end

src/fileio.jl

@@ -0,0 +1,17 @@
+using FileIO
+
+function fileio_load(f::File{format"JPEG"}; kwargs...)
+    open(f.filename, "r") do io
+        jpeg_decode(io; kwargs...)
+    end
+end
+fileio_load(io::Stream{format"JPEG"}; kwargs...) = jpeg_decode(read(io); kwargs...)
+
+function fileio_save(f::File{format"JPEG"}, img::AbstractArray; kwargs...)
+    open(f.filename, "w") do io
+        jpeg_encode(io, img; kwargs...)
+    end
+end
+function fileio_save(io::Stream{format"JPEG"}, img::AbstractArray; kwargs...)
+    jpeg_encode(io.io, img; kwargs...)
+end

test/Project.toml

@@ -1,6 +1,7 @@
 [deps]
 Aqua = "4c88cf16-eb10-579e-8560-4a9242c79595"
 Documenter = "e30172f5-a6a5-5a46-863b-614d45cd2de4"
+FileIO = "5789e2e9-d7fb-5bc7-8068-2c6fae9b9549"
 ImageCore = "a09fc81d-aa75-5fe9-8630-4744c3626534"
 ImageIO = "82e4d734-157c-48bb-816b-45c225c6df19"
 ImageQualityIndexes = "2996bd0c-7a13-11e9-2da2-2f5ce47296a9"

test/fileio.jl

@@ -0,0 +1,17 @@
+@testset "FileIO" begin
+    img = testimage("cameraman")
+    ref = jpeg_decode(jpeg_encode(img))
+    tmpfile = File{format"JPEG"}(joinpath(tmpdir, "tmp.jpg"))
+
+    JpegTurbo.fileio_save(tmpfile, img)
+    data = JpegTurbo.fileio_load(tmpfile)
+    @test data == ref
+
+    open(tmpfile, "w") do s
+        JpegTurbo.fileio_save(s, img)
+    end
+    data = open(tmpfile) do s
+        JpegTurbo.fileio_load(s)
+    end
+    @test data == ref
+end

test/runtests.jl

@@ -51,4 +51,7 @@ const tmpdir = tempdir()
     else
         @info "Multi-threads test: skipped"
     end
+
+    # TODO(johnnychen94): enable after JpegTurbo is registered in FileIO
+    # include("fileio.jl")
 end