Main.hs

{-# LANGUAGE TypeApplications #-}

module Main where

import qualified Data.R2Tree.Float as R

import           Control.Monad
import           Data.Foldable
import           Data.List hiding (lookup, map)
import           Prelude hiding (lookup, map)
import           System.Random.Stateful
import           Weigh



randMBR :: StatefulGen g m => g -> m R.MBR
randMBR g = do
  a <- uniformRM (0, 2 ^ (20 :: Int)) g
  b <- uniformRM (0, 2 ^ (20 :: Int)) g
  return $ R.MBR a b (a + 1) (b + 1)



main :: IO ()
main = do
  g <- newIOGenM $ mkStdGen 0
  raw <- flip zip [0 :: Int ..] <$> replicateM 16384 (randMBR g)

  mainWith $ do
    setColumns [Case, Allocated, Max, Live, GCs]
    wgroup "insert" $ do
      io "BKSS" (pure . foldr (uncurry R.insert) R.empty) raw
      io "Gut" (pure . foldr (uncurry R.insertGut) R.empty) raw