@@ -92,24 +92,23 @@ reverseStream (Text arr off len) = Stream next (off+len-1) (maxSize len)
9292-- | /O(n)/ Convert a 'Stream Char' into a 'Text'.
9393unstream :: Stream Char -> Text
9494unstream (Stream next0 s0 len) = I.textP (P.fst a) 0 (P.snd a)
95− where
96− mlen = upperBound 4 len
97− a = A.run2 (A.unsafeNew mlen >>= (\arr -> outer arr mlen s0 0))
98− outer arr top = loop
99− where
100− loop !s !i
101− | i + 1 >= top = {-# SCC "unstream/resize" #-}
102− case next0 s of
103− Done -> return (arr, i)
104− Skip s' -> loop s' i
105− _ -> do
106− let top' = (top + 1) `shiftL` 1
107− arr' <- A.unsafeNew top'
108− A.copy arr arr' >> outer arr' top' s i
109− | otherwise = case next0 s of
110− Done -> return (arr, i)
111− Skip s' -> loop s' i
112− Yield x s' -> unsafeWrite arr i x >>= loop s'
95+ where
96+ a = A.run2 (A.unsafeNew mlen >>= \arr -> outer arr mlen s0 0)
97+ where mlen = upperBound 4 len
98+ outer arr top = loop
99+ where
100+ loop !s !i =
101+ case next0 s of
102+ Done -> return (arr, i)
103+ Skip s' -> loop s' i
104+ Yield x s'
105+ | j >= top -> {-# SCC "unstream/resize" #-} do
106+ let top' = (top + 1) `shiftL` 1
107+ arr' <- A.unsafeNew top'
108+ A.copy arr arr' >> outer arr' top' s i
109+ | otherwise -> unsafeWrite arr i x >>= loop s'
110+ where j | ord x < 0x10000 = i
111+ | otherwise = i + 1
113112{-# INLINE [0] unstream #-}
114113{-# RULES "STREAM stream/unstream fusion" forall s. stream (unstream s) = s #-}
115114