More concrete Gadget2 format detector (#351)

islent · web-flow · commit 740638dbad5b · 2021-10-30T17:06:48.000+08:00
diff --git a/src/registry.jl b/src/registry.jl
@@ -473,7 +473,7 @@ function detect_gadget2(io)
     seek(io, sizeof(Int32)+256)
     temp2 = read(io, Int32)
     seek(io, pos)
-    return temp1 == temp2
+    return temp1 == temp2 == 256
 end
 add_format(format"Gadget2", detect_gadget2, [".gadget2", ".Gadget2", ".GADGET2"], [:AstroIO => UUID("c85a633c-0c3f-44a2-bffe-7f9d0681b3e7")])
 
diff --git a/test/files/gassphere_littleendian.gadget2 b/test/files/gassphere_littleendian.gadget2
diff --git a/test/query.jl b/test/query.jl
@@ -549,4 +549,9 @@ let file_dir = joinpath(@__DIR__, "files"), file_path = Path(file_dir)
         @test FileIO.unknown(q) # FIXME: should be RData
         @test FileIO.detect_rdata(io)
     end
+
+    @testset "Gadget2" begin
+        q = query(joinpath(file_dir, "gassphere_littleendian.gadget2"))
+        @test typeof(q) <: File{format"Gadget2"}
+    end
 end
