1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
|
{-# LANGUAGE CPP, MagicHash, Rank2Types, TypeFamilies, TypeOperators #-}
-- |
-- Module : Data.Double.Conversion.TextBuilder
-- Copyright : (c) 2011 MailRank, Inc.
--
-- License : BSD-style
-- Maintainer : bos@serpentine.com
-- Stability : experimental
-- Portability : GHC
--
-- Fast, efficient support for converting between double precision
-- floating point values and text.
--
module Data.Double.Conversion.Internal.TextBuilder
(
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 Data.Double.Conversion.Internal.FFI (ForeignFloating)
import qualified Data.Text.Array as A
import Data.Text.Internal.Builder (Builder, writeN)
import Foreign.C.Types (CDouble, CFloat, CInt)
import GHC.Prim (MutableByteArray#)
import Control.Monad.ST (runST)
convert :: (RealFloat a, RealFloat b, b ~ ForeignFloating a) => String -> CInt
-> (forall s. b -> MutableByteArray# s -> IO CInt)
-> a -> Builder
{-# SPECIALIZE convert :: String -> CInt -> (forall s. CDouble -> MutableByteArray# s -> IO CInt) -> Double -> Builder #-}
{-# SPECIALIZE convert :: String -> CInt -> (forall s. CFloat -> MutableByteArray# s -> IO CInt) -> Float -> Builder #-}
{-# INLINABLE convert #-}
convert func len act val = runST $ do
#if MIN_VERSION_text(2,0,0)
mTempArr@(A.MutableByteArray tempMArr) <- A.new (fromIntegral len)
#else
mTempArr@(A.MArray tempMArr) <- A.new (fromIntegral len)
#endif
size <- unsafeIOToST $ act (realToFrac val) tempMArr
tempArr <- A.unsafeFreeze mTempArr
when (size == -1) .
error $ "Data.Double.Conversion.Text." ++ func ++
": conversion failed."
#if MIN_VERSION_text(2,0,0)
return $ writeN (fromIntegral size) $ \mArr _ -> A.copyI (fromIntegral size) mArr 0 tempArr 0
#else
return $ writeN (fromIntegral size) $ \mArr _ -> A.copyI mArr 0 tempArr 0 (fromIntegral size)
#endif
|
