Method IterateMap

gapil/compiler/map.go:494–509

IterateMap emits a map iteration calling cb for each element in the map where: i is a pointer to the sequential index of the element starting from 0 of type idxTy. k is a pointer to the element key. v is a pointer to the element value.

(s *S, mapPtr *codegen.Value, idxTy semantic.Type, cb func(i, k, v *codegen.Value))

// k is a pointer to the element key.
// v is a pointer to the element value.
func (c *C) IterateMap(s *S, mapPtr *codegen.Value, idxTy semantic.Type, cb func(i, k, v *codegen.Value)) {
	capacity := mapPtr.Index(0, MapCapacity).Load()
	elPtr := mapPtr.Index(0, MapElements).Load()
	iTy := c.T.Target(idxTy)
	i := s.LocalInit("i", s.Scalar(0).Cast(iTy))
	s.ForN(capacity.Cast(iTy), func(s *S, it *codegen.Value) *codegen.Value {
		used := elPtr.Index(it, "used").Load()
		s.If(s.Equal(used, s.Scalar(mapElementFull)), func(s *S) {
			k := elPtr.Index(it, "k")
			v := elPtr.Index(it, "v")
			cb(i, k, v)
			i.Store(s.Add(i.Load(), s.Scalar(1).Cast(iTy)))
		})
		return nil
	})
}