Allow iterating over Julia objects from within GAP

CHANGES.md

@@ -1,5 +1,9 @@
 # Changes in GAP.jl
 
+## Version 0.9.5-DEV (released YYYY-MM-DD)
+
+- Allow iterating over Julia objects from within GAP
+
 ## Version 0.9.4 (released 2023-03-04)
 
 - Tweak @gapattribute to not require `import GAP: @gapwrap`

pkg/JuliaInterface/gap/adapter.gi

@@ -84,6 +84,44 @@ InstallOtherMethod( \.\:\=,
     end );
 
 
+#############################################################################
+##
+##  Iterating over Julia objects
+##
+BindGlobal("NextIterator_Julia", function(it)
+  local res;
+  res := it!.state[1];
+  it!.state := Julia.iterate(it!.obj, it!.state[2]);
+  return res;
+end);
+
+BindGlobal("IsDoneIterator_Julia", function(it)
+  return Julia.isnothing(it!.state);
+end);
+
+BindGlobal("ShallowCopy_Julia", function(it)
+  return rec(NextIterator := it!.NextIterator,
+    IsDoneIterator := it!.IsDoneIterator,
+    ShallowCopy := it!.ShallowCopy,
+    state := Julia.Base.deepcopy(it!.state),
+    obj := it!.obj,
+    );
+end);
+
+InstallOtherMethod( Iterator,
+    "for a Julia object",
+    [ IsJuliaObject ],
+function(obj)
+  return IteratorByFunctions(rec(
+    NextIterator := NextIterator_Julia,
+    IsDoneIterator := IsDoneIterator_Julia,
+    ShallowCopy := ShallowCopy_Julia,
+    state := Julia.iterate(obj),
+    obj := obj,
+  ));
+end);
+
+
 #############################################################################
 ##
 ##  Create random numbers via Julia random number generators.

pkg/JuliaInterface/tst/adapter.tst

@@ -89,6 +89,10 @@ gap> TestBinOp({a,b} -> a = b);
 #
 gap> l := GAPToJulia([1, 2, 3]);
 <Julia: Any[1, 2, 3]>
+gap> for i in l do Display(i); od;
+1
+2
+3
 gap> l[1];
 1
 gap> l[2] := false;
@@ -101,6 +105,11 @@ gap> l;
 #
 gap> m:=JuliaEvalString("[1 2; 3 4]");
 <Julia: [1 2; 3 4]>
+gap> for i in m do Display(i); od;
+1
+3
+2
+4
 gap> m[3];
 2
 gap> m[1,2];

pkg/JuliaInterface/tst/convert.tst

@@ -222,5 +222,12 @@ gap> dict:= GAPToJulia( rec( juliafunc:= Julia.Base.map,
 gap> JuliaToGAP( IsRecord, dict );
 rec( juliafunc := <Julia: map> )
 
+# iterating over dict gives key-value pairs
+gap> dict:= GAPToJulia( rec( a := 1, b := 2 ) );
+<Julia: Dict{Symbol, Any}(:a => 1, :b => 2)>
+gap> for i in dict do Display(i); od;
+Pair{Symbol, Any}(:a, 1)
+Pair{Symbol, Any}(:b, 2)
+
 ##
 gap> STOP_TEST( "convert.tst", 1 );