Network/HTTP/Lucu/Format.hs

   1 -- 本當にこんなものを自分で書く必要があったのだらうか。Printf は重いの
   2 -- で駄目だが、それ以外のモジュールを探しても見付からなかった。
   3
   4 module Network.HTTP.Lucu.Format
   5     ( fmtInt
   6
   7     , fmtDec
   8     , fmtHex
   9     )
  10     where
  11
  12
  13 fmtInt :: Int -> Bool -> Int -> Char -> Bool -> Int -> String
  14 fmtInt base upperCase minWidth pad forceSign n
  15     = base `seq` minWidth `seq` pad `seq` forceSign `seq` n `seq`
  16       let raw     = reverse $! fmt' (abs n)
  17           sign    = if forceSign || n < 0 then
  18                         if n < 0 then "-" else "+"
  19                     else
  20                         ""
  21           padded  = padStr (minWidth - length sign) pad raw
  22       in
  23         sign ++ padded
  24     where
  25       fmt' :: Int -> String
  26       fmt' n
  27           | n < base  = (intToChar upperCase n) : []
  28           | otherwise = (intToChar upperCase $! n `mod` base) : fmt' (n `div` base)
  29
  30
  31 fmtDec :: Int -> Int -> String
  32 fmtDec minWidth n
  33     | minWidth == 2 = fmtDec2 n -- optimization
  34     | minWidth == 3 = fmtDec3 n -- optimization
  35     | minWidth == 4 = fmtDec4 n -- optimization
  36     | otherwise     = fmtInt 10 undefined minWidth '0' False n
  37 {-# INLINE fmtDec #-}
  38
  39
  40 fmtDec2 :: Int -> String
  41 fmtDec2 n
  42     | n < 0 || n >= 100 = fmtInt 10 undefined 2 '0' False n -- fallback
  43     | n < 10            =   '0'
  44                           : intToChar undefined n
  45                           : []
  46     | otherwise         =   intToChar undefined (n `div` 10)
  47                           : intToChar undefined (n `mod` 10)
  48                           : []
  49
  50
  51 fmtDec3 :: Int -> String
  52 fmtDec3 n
  53     | n < 0 || n >= 1000 = fmtInt 10 undefined 3 '0' False n -- fallback
  54     | n < 10             = '0' : '0'
  55                            : intToChar undefined n
  56                            : []
  57     | n < 100            = '0'
  58                            : intToChar undefined ((n `div` 10) `mod` 10)
  59                            : intToChar undefined ( n           `mod` 10)
  60                            : []
  61     | otherwise          =   intToChar undefined ((n `div` 100) `mod` 10)
  62                            : intToChar undefined ((n `div`  10) `mod` 10)
  63                            : intToChar undefined ( n            `mod` 10)
  64                            : []
  65
  66
  67 fmtDec4 :: Int -> String
  68 fmtDec4 n
  69     | n < 0 || n >= 10000 = fmtInt 10 undefined 4 '0' False n -- fallback
  70     | n < 10              =   '0' : '0' : '0'
  71                             : intToChar undefined n
  72                             : []
  73     | n < 100             =   '0' : '0'
  74                             : intToChar undefined ((n `div` 10) `mod` 10)
  75                             : intToChar undefined ( n           `mod` 10)
  76                             : []
  77     | n < 1000            =   '0'
  78                             : intToChar undefined ((n `div` 100) `mod` 10)
  79                             : intToChar undefined ((n `div`  10) `mod` 10)
  80                             : intToChar undefined ( n            `mod` 10)
  81                             : []
  82     | otherwise           =   intToChar undefined ((n `div` 1000) `mod` 10)
  83                             : intToChar undefined ((n `div`  100) `mod` 10)
  84                             : intToChar undefined ((n `div`   10) `mod` 10)
  85                             : intToChar undefined ( n             `mod` 10)
  86                             : []
  87
  88
  89 fmtHex :: Bool -> Int -> Int -> String
  90 fmtHex upperCase minWidth
  91     = fmtInt 16 upperCase minWidth '0' False
  92
  93
  94 padStr :: Int -> Char -> String -> String
  95 padStr minWidth pad str
  96     = let delta = minWidth - length str
  97       in
  98         if delta > 0 then
  99             replicate delta pad ++ str
 100         else
 101             str
 102
 103
 104 intToChar :: Bool -> Int -> Char
 105 intToChar _ 0  = '0'
 106 intToChar _ 1  = '1'
 107 intToChar _ 2  = '2'
 108 intToChar _ 3  = '3'
 109 intToChar _ 4  = '4'
 110 intToChar _ 5  = '5'
 111 intToChar _ 6  = '6'
 112 intToChar _ 7  = '7'
 113 intToChar _ 8  = '8'
 114 intToChar _ 9  = '9'
 115 intToChar False 10 = 'a'
 116 intToChar True  10 = 'A'
 117 intToChar False 11 = 'b'
 118 intToChar True  11 = 'B'
 119 intToChar False 12 = 'c'
 120 intToChar True  12 = 'C'
 121 intToChar False 13 = 'd'
 122 intToChar True  13 = 'D'
 123 intToChar False 14 = 'e'
 124 intToChar True  14 = 'E'
 125 intToChar False 15 = 'f'
 126 intToChar True  15 = 'F'