@@ -44,6 +44,45 @@ import Data.Array
44
44
import System.Metrics.Distribution.Internal (Stats (.. ))
45
45
import System.Metrics.ThreadId
46
46
47
+ -- 64-bit machine, Int ~ Int64, do it the fast way:
48
+ #if SIZEOF_HSINT == 8
49
+
50
+ #if MIN_VERSION_base(4,17,0)
51
+ int64ToDouble :: Int64 ## -> Double ##
52
+ int64ToDouble i = int2Double## (int64ToInt## i)
53
+
54
+ intToInt64 :: Int ## -> Int64 ##
55
+ intToInt64 = intToInt64##
56
+
57
+ plusInt64 :: Int64 ## -> Int64 ## -> Int64 ##
58
+ plusInt64 = plusInt64##
59
+
60
+ readInt64Array :: MutableByteArray ## d -> Int ## -> State ## d -> (## State ## d , Int64 ## ## )
61
+ readInt64Array = readInt64Array##
62
+
63
+ writeInt64Array :: MutableByteArray ## d -> Int ## -> Int64 ## -> State ## d -> State ## d
64
+ writeInt64Array = writeInt64Array##
65
+ #else
66
+
67
+ intToInt64 :: Int ## -> Int ##
68
+ intToInt64 i = i
69
+
70
+ plusInt64 :: Int ## -> Int ## -> Int ##
71
+ plusInt64 = (+##)
72
+
73
+ readInt64Array :: MutableByteArray ## d -> Int ## -> State ## d -> (## State ## d , Int ## ## )
74
+ readInt64Array = readIntArray##
75
+
76
+ writeInt64Array :: MutableByteArray # d -> Int # -> Int # -> State # d -> State # d
77
+ writeInt64Array = writeIntArray##
78
+ #endif
79
+
80
+ #else
81
+ -- I don't know a better way on 32-bit machines...
82
+ int64ToDouble i =
83
+ case fromIntegral (I64 ## i) of (D ## d) -> d
84
+ #endif
85
+
47
86
-- | An metric for tracking events.
48
87
newtype Distribution = Distribution { unD :: Array Stripe }
49
88
@@ -90,7 +129,7 @@ newDistrib = IO $ \s ->
90
129
-- probably unecessary
91
130
case atomicWriteIntArray## mba lockPos' 0 ## s1 of { s2 ->
92
131
case countPos of { (I ## countPos') ->
93
- case writeInt64Array## mba countPos' (intToInt64## 0 ## ) s2 of { s3 ->
132
+ case writeInt64Array mba countPos' (intToInt64 0 ## ) s2 of { s3 ->
94
133
case meanPos of { (I ## meanPos') ->
95
134
case writeDoubleArray## mba meanPos' 0.0 #### s3 of { s4 ->
96
135
case sumSqDeltaPos of { (I ## sumSqDeltaPos') ->
@@ -148,17 +187,6 @@ spinUnlock mba = \s ->
148
187
case writeIntArray## mba lockPos' 0 ## s of { s2 ->
149
188
s2 }}
150
189
151
- int64ToDouble :: Int64 ## -> Double ##
152
- -- 64-bit machine, Int ~ Int64, do it the fast way:
153
- #if SIZEOF_HSINT == 8
154
- int64ToDouble i = int2Double## (int64ToInt## i)
155
- #else
156
- -- I don't know a better way on 32-bit machines...
157
- int64ToDouble i =
158
- case fromIntegral (I64 ## i) of (D ## d) -> d
159
- #endif
160
-
161
- {-# INLINE int64ToDouble #-}
162
190
163
191
-- | Add the same value to the distribution N times.
164
192
-- Mean and variance are computed according to
@@ -169,7 +197,7 @@ addN distribution (D## val) (I64## n) = IO $ \s ->
169
197
case myStripe' s of { (## s1, (Stripe (Distrib mba)) ## ) ->
170
198
case spinLock mba s1 of { s2 ->
171
199
case countPos of { (I ## countPos') ->
172
- case readInt64Array## mba countPos' s2 of { (## s3, count ## ) ->
200
+ case readInt64Array mba countPos' s2 of { (## s3, count ## ) ->
173
201
case meanPos of { (I ## meanPos') ->
174
202
case readDoubleArray## mba meanPos' s3 of { (## s4, mean ## ) ->
175
203
case sumSqDeltaPos of { (I ## sumSqDeltaPos') ->
@@ -180,11 +208,11 @@ addN distribution (D## val) (I64## n) = IO $ \s ->
180
208
case readDoubleArray## mba minPos' s6 of { (## s7, dMin ## ) ->
181
209
case maxPos of { (I ## maxPos') ->
182
210
case readDoubleArray## mba maxPos' s7 of { (## s8, dMax ## ) ->
183
- case plusInt64## count n of { count' ->
211
+ case plusInt64 count n of { count' ->
184
212
case val -#### mean of { delta ->
185
213
case mean +#### ((int64ToDouble n) *#### delta /#### (int64ToDouble count')) of { mean' ->
186
214
case sumSqDelta +#### (delta *#### (val -#### mean') *#### (int64ToDouble n)) of { sumSqDelta' ->
187
- case writeInt64Array## mba countPos' count' s8 of { s9 ->
215
+ case writeInt64Array mba countPos' count' s8 of { s9 ->
188
216
case writeDoubleArray## mba meanPos' mean' s9 of { s10 ->
189
217
case writeDoubleArray## mba sumSqDeltaPos' sumSqDelta' s10 of { s11 ->
190
218
case writeDoubleArray## mba sumPos' (dSum +#### val) s11 of { s12 ->
@@ -204,9 +232,9 @@ combine :: Distrib -> Distrib -> IO ()
204
232
combine (Distrib bMBA) (Distrib aMBA) = IO $ \ s ->
205
233
case spinLock bMBA s of { s1 ->
206
234
case countPos of { (I ## countPos') ->
207
- case readInt64Array## aMBA countPos' s1 of { (## s2, aCount ## ) ->
208
- case readInt64Array## bMBA countPos' s2 of { (## s3, bCount ## ) ->
209
- case plusInt64## aCount bCount of { count' ->
235
+ case readInt64Array aMBA countPos' s1 of { (## s2, aCount ## ) ->
236
+ case readInt64Array bMBA countPos' s2 of { (## s3, bCount ## ) ->
237
+ case plusInt64 aCount bCount of { count' ->
210
238
case meanPos of { (I ## meanPos' ) ->
211
239
case readDoubleArray## aMBA meanPos' s3 of { (## s4, aMean ## ) ->
212
240
case readDoubleArray## bMBA meanPos' s4 of { (## s5, bMean ## ) ->
@@ -226,8 +254,8 @@ combine (Distrib bMBA) (Distrib aMBA) = IO $ \s ->
226
254
)
227
255
)
228
256
) of { sumSqDelta' ->
229
- case writeInt64Array## aMBA countPos' count' s7 of { s8 ->
230
- case (case eqInt64## count' (intToInt64## 0 ## ) of { 0 ## -> mean'; _ -> 0.0 #### }) of { writeMean ->
257
+ case writeInt64Array aMBA countPos' count' s7 of { s8 ->
258
+ case (case eqInt64## count' (intToInt64 0 ## ) of { 0 ## -> mean'; _ -> 0.0 #### }) of { writeMean ->
231
259
case writeDoubleArray## aMBA meanPos' writeMean s8 of { s9 ->
232
260
case writeDoubleArray## aMBA sumSqDeltaPos' sumSqDelta' s9 of { s10 ->
233
261
case sumPos of { (I ## sumPos') ->
@@ -258,7 +286,7 @@ read distrib = do
258
286
case sumPos of { (I ## sumPos') ->
259
287
case minPos of { (I ## minPos') ->
260
288
case maxPos of { (I ## maxPos') ->
261
- case readInt64Array## mba countPos' s of { (## s1, count ## ) ->
289
+ case readInt64Array mba countPos' s of { (## s1, count ## ) ->
262
290
case readDoubleArray## mba meanPos' s1 of { (## s2, mean ## ) ->
263
291
case readDoubleArray## mba sumSqDeltaPos' s2 of { (## s3, sumSqDelta ## ) ->
264
292
case readDoubleArray## mba sumPos' s3 of { (## s4, dSum ## ) ->
0 commit comments