scratch/GC: playground for datastructures to be used in GC - probably...
[mate.git] / scratch / GC.hs
index 9781d207183332f517a3cea430aaf871d9cf6c83..c78eb506629f6922a09995f00b12f429b88660f3 100644 (file)
-{-# LANGUAGE ScopedTypeVariables #-}
-module GC 
-  ({- dont export generic versions for high performance ;-) -}) where
+{-# LANGUAGE GADTs #-}
+{-# LANGUAGE ExistentialQuantification #-}
+module GC where
 
 import Control.Monad
 
---import Data.Foldable hiding (mapM_)
-
 import Foreign.Ptr
 import Foreign.Marshal.Alloc
-import Foreign.Marshal.Array
 import Foreign.Storable
+import Foreign.C.Types
 import GHC.Int
-
-import qualified Data.Set as S
 import Text.Printf
 
-class (Eq a, Ord a) => RefObj a where
-  payload :: a -> IO IntPtr
-  refs    :: a -> IO [a]
-  marked  :: a -> IO Bool
-  mark    :: a -> IO ()
-  unmark  :: a -> IO ()
-
--- TODO hs: wtf? i am failing to create a printable class with superclass refobj??
---class RefObj a => PrintableRefObj a where 
---  printRef   :: a -> IO ()
-
-class PrintableRef a where
-  printRef :: a -> IO ()
-
-instance RefObj (Ptr a) where
-  payload     = return . ptrToIntPtr
-  refs        = unpackRefs . castPtr
-  marked      = markedRef
-  mark        = markRef (0x1::Int32)
-  unmark      = markRef (0x0::Int32)
-
-instance PrintableRef (Ptr a) where
-  printRef    = printRef'
-
-
-idOff           = 0x0
-numberOfObjsOff = 0x4
-fieldsOff = 0xC
-markedOff = 0x8
-
-unpackRefs :: Ptr Int32 -> IO [Ptr b]
-unpackRefs ptr = do  --dereference number of objs; mark field skipped via fieldsOffset
-                    numberOfObjs <- peekByteOff ptr numberOfObjsOff :: IO Int32
-                    mapM (peekElemOff (ptr `plusPtr` fieldsOff)) [0..fromIntegral $ numberOfObjs-1]
-
-markedRef :: Ptr a -> IO Bool
-markedRef ptr = liftM ((/=0) . fromIntegral) (peekByteOff ptr markedOff :: IO Int32)
-
-markRef :: Int32 -> Ptr a -> IO ()
-markRef val ptr = pokeByteOff ptr markedOff val
-
-printRef' :: Ptr a -> IO ()
-printRef' ptr = do printf "obj 0x%08x\n" =<< (peekByteOff ptr idOff :: IO Int32)
-                   printf "children 0x%08x\n" =<< (peekByteOff ptr numberOfObjsOff :: IO Int32)                  
-                   printf "marked 0x%08x\n\n" =<< (peekByteOff ptr markedOff :: IO Int32)
-
--- | Generically marks a graph (can be used to set mark bit and reset mark bit at the same time
--- using customized loopcheck and marker funcs (i.e. to set the bit check on ==1 and on ==0 otherwise)
--- Furthermore it produces a list of visited nodes (this can be all live one (or dead on respectively)
-markTree'' :: RefObj a => (a -> IO Bool) -> (a -> IO ()) -> [a] -> a -> IO [a]
-markTree'' loopcheck marker ws root = do loop <- loopcheck root
-                                         if loop then return ws else liftM (root :) continue
-    where continue = marker root >> refs root >>= foldM (markTree'' loopcheck marker) ws
-
--- | For debugging only (implements custom loop check with Data.Set!)
-traverseIO :: RefObj o => (o -> IO ()) -> o -> IO ()
-traverseIO f = void . traverseIO' f S.empty
-
-traverseIO' ::  RefObj a => (a -> IO ()) -> S.Set a -> a -> IO (S.Set a)
-traverseIO' f ws root = if S.member root ws then f root >> return ws
-                           else f root >> refs root >>= cont
-  where cont = foldM (\ws x -> do let ws' = S.insert x ws
-                                  traverseIO' f ws' x) ws'
-        ws' = S.insert root ws
-
-markTree :: RefObj a => a -> IO ()
-markTree root = marked root >>= (`unless` continue)
-  where continue = mark root >> refs root >>= mapM_  markTree
-
-printTree :: Ptr a -> IO ()
-printTree = traverseIO printRef'
-
-
-emptyObj id  = do mem <- mallocBytes 0xC
-                  pokeArray mem [id,0,0::Int32]
-                  return mem
-
-twoRefs = do mem <- mallocBytes 0x14
-             -- idOfObj; numberofObj; marked waste memory Int32
-             pokeArray mem [0::Int32,2,0]
-             obj1 <- emptyObj 1
-             obj2 <- emptyObj 2
-             pokeByteOff mem 0xC obj1
-             pokeByteOff mem 0x10 obj2
-             return mem
-
-cyclR = do mem <- mallocBytes 0x18
-           pokeArray mem [0::Int32,3,0]
-           obj1 <- emptyObj 1
-           obj2 <- emptyObj 2
-           pokeByteOff mem 0xC obj1
-           pokeByteOff mem 0x10 obj2
-           pokeByteOff mem 0x14 mem
-           return mem
-
-test objr = do twoRefs <- objr
-               putStrLn "initial:\n" 
-               printTree twoRefs
-               lifeRefs <- markTree'' marked mark [] twoRefs
-               putStrLn "life refs: \n"
-               print lifeRefs
-               --forM lifeRefs printRef'
-               putStrLn "after marking\n"
-               printTree twoRefs
-               markTree'' (liftM not . marked) unmark [] twoRefs
-               putStrLn "after unmarking\n"
-               printTree twoRefs
+class RefObject a where
+  mem  :: a -> IntPtr 
+  refs :: a -> [IntPtr]
+
+
+data RefObj = RefObj IntPtr [IntPtr] deriving Show
+
+instance RefObject RefObj where
+  mem  (RefObj mem _ ) = mem
+  refs (RefObj _ refs) = refs
+
+data Succ = forall a. (RefObject a) => Succ (a -> [a])
+
+obj2 = do buffer <- mallocBytes 4
+          pokeByteOff buffer 0 (0::Int32)
+          return buffer
+
+obj3 = do buffer <- mallocBytes 4
+          pokeByteOff buffer 0 (0::Int32)
+          return buffer
+
+obj1 f g = do buffer <- mallocBytes 12
+              pokeByteOff buffer 0 (2::Int32)
+              pokeByteOff buffer 4 f
+              pokeByteOff buffer 8 g
+              return buffer
+
+ptrToRefObj ptr = do objCount <- peek ptr :: IO Int32
+                     let objsBase = ptr `plusPtr` 4
+                     objs <- mapM ((liftM ptrToIntPtr) . peekElemOff objsBase . fromIntegral) [0..objCount-1]
+                     return $ RefObj (ptrToIntPtr ptr) objs
+
+test1 = do f <- obj2
+           g <- obj3
+           (print . ptrToIntPtr) f
+           (print . ptrToIntPtr) g
+           ptrToRefObj =<< obj1 f g
+
+traverse :: (RefObject a) => (IntPtr -> IO a) -> a -> [a] -> IO [a]
+traverse dereference x ws = do children <- mapM dereference (refs x)
+                               undefined
+
+succMem :: Ptr a -> Succ
+succMem =undefined-- Succ (\_ -> obj1
+
+
+