structural-traversal-0.1.0.0: src/Data/StructuralTraversal/Indexing.hs

{-# LANGUAGE LambdaCase #-}
module Data.StructuralTraversal.Indexing where

import Data.StructuralTraversal.Class
import Control.Monad.State
import Control.Applicative

indexedTraverse :: StructuralTraversable t => (a -> [Int] -> b) -> t a -> t b
indexedTraverse f
  = flip evalState []
      . traverseUp ( modify (0:) )
                   ( modify tail ) 
                   ( \a -> f a <$> get <* modify ( \case s:st -> (s+1):st
                                                         []   -> [] ))