blob: 21aad3f1964255df71013fbf573e895ebcb762a0 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
|
{-# LANGUAGE BangPatterns #-}
{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE UndecidableInstances #-}
import GHC.Exts.Heap
import GHC.Types.SrcLoc
import qualified GHC.Data.Strict as Strict
rsl :: RealSrcLoc
rsl = mkRealSrcLoc "Foo" 1 1
main = do
let !s1 = RealSrcLoc rsl (Strict.Just (BufPos 999222))
!s2 = RealSrcLoc rsl (Strict.Just (BufPos 999333))
!s3 = RealSrcLoc rsl (Strict.Just (BufPos 999444))
!res = combineSrcSpans (combineSrcSpans (srcLocSpan s1) (srcLocSpan s2)) (srcLocSpan s3)
cs <- unbox res
-- The output must be an empty list because we don't want to retain
-- intermediate locations in the heap.
print (filter (hasDataArg 999333) cs)
hasDataArg x (ConstrClosure _ _ dataArgs _ _ _) = any (== x) dataArgs
hasDataArg x _ = False
unbox :: a -> IO [GenClosure Box]
unbox a = loop (asBox a)
where
loop :: Box -> IO [GenClosure Box]
loop (Box b) = do
c <- getClosureData b
p <- concat <$> traverse loop (allClosures c)
return (c : p)
|
