{-# LANGUAGE ForeignFunctionInterface #-}
-{-# LANGUAGE CPP #-}
-#include "debug.h"
-module Mate.GarbageAlloc where
+module Mate.GarbageAlloc(
+ mallocClassData,
+ mallocStringGC,
+ mallocObjectGC,
+ getHeapMemory,
+ printMemoryUsage,
+ printGCStats,
+ mallocObjectUnmanaged,
+ mallocStringUnmanaged) where
import Foreign
import Foreign.C
-#ifdef DEBUG
-import Text.Printf
-#endif
+import Mate.GC.Boehm
+
import Mate.Debug
-- unified place for allocating Memory
mallocClassData :: Int -> IO (Ptr a)
mallocClassData size = do
- printfStr "mallocClassData: %d\n" size
- mallocBytes size
+ printfStr $ printf "mallocClassData: %d\n" size
+ mem <- mallocBytes size
+ addRootGC mem (plusPtr mem size)
+ return mem
-mallocString :: Int -> IO (Ptr a)
-mallocString size = do
- printfStr "mallocString: %d\n" size
- mallocBytes size
+mallocStringGC :: Int -> IO (Ptr a)
+mallocStringGC size = do
+ printfStr $ printf "mallocString: %d\n" size
+ mallocBytesGC size
-foreign export ccall mallocObject :: Int -> IO CUInt
-mallocObject :: Int -> IO CUInt
-mallocObject size = do
+foreign export ccall mallocObjectGC :: Int -> IO CPtrdiff
+mallocObjectGC :: Int -> IO CPtrdiff
+mallocObjectGC size = do
+ ptr <- mallocBytesGC size
+ printfStr $ printf "mallocObject: %d\n" size
+ return $ fromIntegral $ ptrToIntPtr ptr
+
+mallocObjectUnmanaged :: Int -> IO CPtrdiff
+mallocObjectUnmanaged size = do
ptr <- mallocBytes size
- printfStr "mallocObject: %d\n" size
+ printfStr $ printf "mallocObjectUnmanged: %d\n" size
return $ fromIntegral $ ptrToIntPtr ptr
+
+mallocStringUnmanaged :: Int -> IO (Ptr a)
+mallocStringUnmanaged size = do
+ printfStr $ printf "mallocStringUnamaged: %d\n" size
+ mallocBytes size
+
+
+getHeapMemory :: IO Int
+getHeapMemory = getHeapSizeGC
+
+foreign export ccall printMemoryUsage :: IO ()
+printMemoryUsage :: IO ()
+printMemoryUsage = getHeapMemory >>= print
+
+foreign export ccall printGCStats :: IO ()
+printGCStats :: IO ()
+printGCStats = putStrLn "Should print GC Stats"