Add helpers to figure out loop nestings from a kernel

This is a typed version of the code from this commit:
https://github.com/inducer/loopy/pull/690/commits/be5a31849c8ea16670a416c83b7b86a0cc5f71ae



Co-authored-by: Andreas Kloeckner <inform@tiker.net>