Repa配列のランレングス符号化

Question

0と1からなる1次元のRepa配列があり、そのランレングスエンコーディングを計算したいと思います。 例：[0,0,1,1,1,0,0,0,1,0,1,1] into [2,3,3,1,1,2]などと似ています。 （私は可読性のためにリスト表現を使用しています）

理想的には、私は1のランレングスを望み、0を無視します。 だから[0,0,1,1,1,0,0,0,1,0,1,1] becomes [3,1,2]。

私は結果が（Repa）アレイであることを望みます。

Repaを使ってどうすればいいですか？ mapまたはtraverseは、一度に1つの要素しか与えられないため、使用できません。私はfoldアキュムレータのいくつかの特別な種類としようとすることができますが、それは理想的ではないようだと私はそれも（モナドの法律のため）可能ですそれを知らない。

Answer 1

私は現在、Repa関数を使用せずに配列を繰り返し、リストを返すだけです。私はBooleanの1と0の代わりに作業していますが、アルゴリズムは同じです。その後、このリストをRepaアレイに変換しています。

runLength :: Array U DIM1 Bool -> [Length] 
runLength arr = go ([], 0, False) 0 arr 
    where 
    Z :. n = extent arr 
    go :: Accumulator -> Int -> Array U DIM1 Bool -> [Length] 
    go !acc@(xs, c, b) !i !arr | i == n = if c > 0 then c:xs else xs 
           | otherwise = 
           if unsafeIndex arr (Z :. i) 
           then if b 
             then go (xs, c+1, b) (i+1) arr 
             else go (xs, 1, True) (i+1) arr 
           else if b 
             then go (c:xs, 0, False) (i+1) arr 
             else go (xs, 0, False) (i+1) arr