MethodPool: removed demo call stuff - added printGCStats instead;

[mate.git] / Mate / GarbageAlloc.hs

diff --git a/Mate/GarbageAlloc.hs b/Mate/GarbageAlloc.hs
index f5b109385d14139761a6debfb19b6f0824664ae1..dc806b7418a99a90ab10fe3f5f1a77baacf4ca3f 100644 (file)
--- a/Mate/GarbageAlloc.hs
+++ b/Mate/GarbageAlloc.hs
@@ -1,19 +1,62 @@
-module Mate.GarbageAlloc where
+{-# LANGUAGE ForeignFunctionInterface #-}
+{-# LANGUAGE CPP #-}
+#include "debug.h"
+module Mate.GarbageAlloc(
+    mallocClassData,
+    mallocString,
+    mallocObject,
+    getHeapMemory,
+    printMemoryUsage,
+    mallocStringVM,
+    mallocObjectVM,
+    printGCStats)  where
 
 import Foreign
 import Foreign.C
 
+import Mate.GC.Boehm
+
+--import Text.Printf
+import Mate.Debug
+
 -- unified place for allocating Memory
 -- TODO: implement GC stuff ;-)
 
 mallocClassData :: Int -> IO (Ptr a)
-mallocClassData = mallocBytes
+mallocClassData size = do
+  printfStr "mallocClassData: %d\n" size
+  mem <- mallocBytes size
+  addRootGC mem (plusPtr mem size)
+  return mem
 
 mallocString :: Int -> IO (Ptr a)
-mallocString = mallocBytes
+mallocString size = do
+  printfStr "mallocString: %d\n" size
+  mallocBytesGC size
 
-foreign export ccall mallocObject :: Int -> IO CUInt
-mallocObject :: Int -> IO CUInt
+mallocStringVM :: Int -> IO (Ptr a)
+mallocStringVM = mallocBytes
+
+foreign export ccall mallocObject :: Int -> IO CPtrdiff
+mallocObject :: Int -> IO CPtrdiff
 mallocObject size = do
+  ptr <- mallocBytesGC size
+  printfStr "mallocObject: %d\n" size
+  return $ fromIntegral $ ptrToIntPtr ptr
+
+mallocObjectVM :: Int -> IO CPtrdiff
+mallocObjectVM size = do
   ptr <- mallocBytes size
+  printfStr "mallocObject VM: %d\n" size
   return $ fromIntegral $ ptrToIntPtr ptr
+
+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"