Ticket #2387: Unboxed.hs

{-# LANGUAGE MagicHash, UnboxedTuples #-}

module Main (main) where

import Control.Monad.ST

import GHC.ST
import GHC.Prim
import GHC.Base

import System.Environment

whileLoop :: Int# -> State# s -> (# State# s, Int# #)
whileLoop = go 0#
 where
 go :: Int# -> Int# -> State# s -> (# State# s, Int# #)
 go n k s
   | k ==# 0#  = (# s, n #)
   | otherwise = go (n +# 1#) (k -# 1#) s
{-# INLINE whileLoop #-}

mainLoop :: Int# -> Int# -> State# s -> (# State# s, Int #)
mainLoop k n s = case whileLoop 40# s of { (# s', k' #) ->
                 case k <=# k' of
                   False ->
                     case n of
                       0# -> (# s', I# k #)
                       _ -> mainLoop k (n -# 1#) s'
                   True ->
                     case n of
                       0# -> (# s', I# k' #)
                       _ -> mainLoop k' (n -# 1#) s' }

mlWrapper :: Int -> Int -> ST s Int
mlWrapper (I# k) (I# n) = ST (mainLoop k n)

main = print =<< stToIO . mlWrapper 0 . read . head =<< getArgs