{-# LANGUAGE CPP, MagicHash, Rank2Types, TypeFamilies, BangPatterns, TypeOperators #-}
module Data.Double.Conversion.Internal.Text
(
convert
) where
import Control.Monad (when)
#if MIN_VERSION_base(4,4,0)
import Control.Monad.ST.Unsafe (unsafeIOToST)
#else
import Control.Monad.ST (unsafeIOToST)
#endif
import Control.Monad.ST (ST, runST)
import Data.Double.Conversion.Internal.FFI (ForeignFloating)
import qualified Data.Text.Array as A
import Data.Text.Internal (Text(Text))
import Foreign.C.Types (CDouble, CFloat, CInt)
import GHC.Prim (MutableByteArray#)
convert :: (RealFloat a, RealFloat b, b ~ ForeignFloating a) => String -> CInt
-> (forall s. b -> MutableByteArray# s -> IO CInt)
-> a -> Text
{-# SPECIALIZE convert :: String -> CInt -> (forall s. CDouble -> MutableByteArray# s -> IO CInt) -> Double -> Text #-}
{-# SPECIALIZE convert :: String -> CInt -> (forall s. CFloat -> MutableByteArray# s -> IO CInt) -> Float -> Text #-}
{-# INLINABLE convert #-}
convert :: forall a b.
(RealFloat a, RealFloat b, b ~ ForeignFloating a) =>
String
-> CInt
-> (forall s. b -> MutableByteArray# s -> IO CInt)
-> a
-> Text
convert String
func CInt
len forall s. b -> MutableByteArray# s -> IO CInt
act a
val = forall a. (forall s. ST s a) -> a
runST forall s. ST s Text
go
where
go :: (forall s. ST s Text)
go :: forall s. ST s Text
go = do
MArray s
buf <- forall s. Int -> ST s (MArray s)
A.new (forall a b. (Integral a, Num b) => a -> b
fromIntegral CInt
len)
#if MIN_VERSION_text(2,0,0)
let !(A.MutableByteArray ma) = buf
#else
let ma :: MutableByteArray# s
ma = forall s. MArray s -> MutableByteArray# s
A.maBA MArray s
buf
#endif
CInt
size <- forall a s. IO a -> ST s a
unsafeIOToST forall a b. (a -> b) -> a -> b
$ forall s. b -> MutableByteArray# s -> IO CInt
act (forall a b. (Real a, Fractional b) => a -> b
realToFrac a
val) MutableByteArray# s
ma
forall (f :: * -> *). Applicative f => Bool -> f () -> f ()
when (CInt
size forall a. Eq a => a -> a -> Bool
== -CInt
1) forall b c a. (b -> c) -> (a -> b) -> a -> c
.
forall a. HasCallStack => String -> a
error forall a b. (a -> b) -> a -> b
$ String
"Data.Double.Conversion.Text." forall a. [a] -> [a] -> [a]
++ String
func forall a. [a] -> [a] -> [a]
++
String
": conversion failed (invalid precision requested)"
Array
frozen <- forall s. MArray s -> ST s Array
A.unsafeFreeze MArray s
buf
forall (m :: * -> *) a. Monad m => a -> m a
return forall a b. (a -> b) -> a -> b
$ Array -> Int -> Int -> Text
Text Array
frozen Int
0 (forall a b. (Integral a, Num b) => a -> b
fromIntegral CInt
size)