@@ -44,6 +44,52 @@ 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
+ eqInt64 :: Int64 ## -> Int64 ## -> Int ##
61
+ eqInt64 = eqInt64##
62
+
63
+ readInt64Array :: MutableByteArray ## d -> Int ## -> State ## d -> (## State ## d , Int64 ## ## )
64
+ readInt64Array = readInt64Array##
65
+
66
+ writeInt64Array :: MutableByteArray ## d -> Int ## -> Int64 ## -> State ## d -> State ## d
67
+ writeInt64Array = writeInt64Array##
68
+
69
+ #else
70
+
71
+ intToInt64 :: Int ## -> Int ##
72
+ intToInt64 i = i
73
+
74
+ plusInt64 :: Int ## -> Int ## -> Int ##
75
+ plusInt64 a b = a +## b
76
+
77
+ eqInt64 :: Int ## -> Int ## -> Int ##
78
+ eqInt64 a b = a ==## b
79
+
80
+ readInt64Array :: MutableByteArray ## d -> Int ## -> State ## d -> (## State ## d , Int ## ## )
81
+ readInt64Array = readIntArray##
82
+
83
+ writeInt64Array :: MutableByteArray ## d -> Int ## -> Int ## -> State ## d -> State ## d
84
+ writeInt64Array = writeIntArray##
85
+ #endif
86
+
87
+ #else
88
+ -- I don't know a better way on 32-bit machines...
89
+ int64ToDouble i =
90
+ case fromIntegral (I64 ## i) of (D ## d) -> d
91
+ #endif
92
+
47
93
-- | An metric for tracking events.
48
94
newtype Distribution = Distribution { unD :: Array Stripe }
49
95
@@ -90,7 +136,7 @@ newDistrib = IO $ \s ->
90
136
-- probably unecessary
91
137
case atomicWriteIntArray## mba lockPos' 0 ## s1 of { s2 ->
92
138
case countPos of { (I ## countPos') ->
93
- case writeInt64Array## mba countPos' (intToInt64## 0 ## ) s2 of { s3 ->
139
+ case writeInt64Array mba countPos' (intToInt64 0 ## ) s2 of { s3 ->
94
140
case meanPos of { (I ## meanPos') ->
95
141
case writeDoubleArray## mba meanPos' 0.0 #### s3 of { s4 ->
96
142
case sumSqDeltaPos of { (I ## sumSqDeltaPos') ->
@@ -148,17 +194,6 @@ spinUnlock mba = \s ->
148
194
case writeIntArray## mba lockPos' 0 ## s of { s2 ->
149
195
s2 }}
150
196
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
197
163
198
-- | Add the same value to the distribution N times.
164
199
-- Mean and variance are computed according to
@@ -169,7 +204,7 @@ addN distribution (D## val) (I64## n) = IO $ \s ->
169
204
case myStripe' s of { (## s1, (Stripe (Distrib mba)) ## ) ->
170
205
case spinLock mba s1 of { s2 ->
171
206
case countPos of { (I ## countPos') ->
172
- case readInt64Array## mba countPos' s2 of { (## s3, count ## ) ->
207
+ case readInt64Array mba countPos' s2 of { (## s3, count ## ) ->
173
208
case meanPos of { (I ## meanPos') ->
174
209
case readDoubleArray## mba meanPos' s3 of { (## s4, mean ## ) ->
175
210
case sumSqDeltaPos of { (I ## sumSqDeltaPos') ->
@@ -180,11 +215,11 @@ addN distribution (D## val) (I64## n) = IO $ \s ->
180
215
case readDoubleArray## mba minPos' s6 of { (## s7, dMin ## ) ->
181
216
case maxPos of { (I ## maxPos') ->
182
217
case readDoubleArray## mba maxPos' s7 of { (## s8, dMax ## ) ->
183
- case plusInt64## count n of { count' ->
218
+ case plusInt64 count n of { count' ->
184
219
case val -#### mean of { delta ->
185
220
case mean +#### ((int64ToDouble n) *#### delta /#### (int64ToDouble count')) of { mean' ->
186
221
case sumSqDelta +#### (delta *#### (val -#### mean') *#### (int64ToDouble n)) of { sumSqDelta' ->
187
- case writeInt64Array## mba countPos' count' s8 of { s9 ->
222
+ case writeInt64Array mba countPos' count' s8 of { s9 ->
188
223
case writeDoubleArray## mba meanPos' mean' s9 of { s10 ->
189
224
case writeDoubleArray## mba sumSqDeltaPos' sumSqDelta' s10 of { s11 ->
190
225
case writeDoubleArray## mba sumPos' (dSum +#### val) s11 of { s12 ->
@@ -204,9 +239,9 @@ combine :: Distrib -> Distrib -> IO ()
204
239
combine (Distrib bMBA) (Distrib aMBA) = IO $ \ s ->
205
240
case spinLock bMBA s of { s1 ->
206
241
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' ->
242
+ case readInt64Array aMBA countPos' s1 of { (## s2, aCount ## ) ->
243
+ case readInt64Array bMBA countPos' s2 of { (## s3, bCount ## ) ->
244
+ case plusInt64 aCount bCount of { count' ->
210
245
case meanPos of { (I ## meanPos' ) ->
211
246
case readDoubleArray## aMBA meanPos' s3 of { (## s4, aMean ## ) ->
212
247
case readDoubleArray## bMBA meanPos' s4 of { (## s5, bMean ## ) ->
@@ -226,8 +261,8 @@ combine (Distrib bMBA) (Distrib aMBA) = IO $ \s ->
226
261
)
227
262
)
228
263
) 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 ->
264
+ case writeInt64Array aMBA countPos' count' s7 of { s8 ->
265
+ case (case eqInt64 count' (intToInt64 0 ## ) of { 0 ## -> mean'; _ -> 0.0 #### }) of { writeMean ->
231
266
case writeDoubleArray## aMBA meanPos' writeMean s8 of { s9 ->
232
267
case writeDoubleArray## aMBA sumSqDeltaPos' sumSqDelta' s9 of { s10 ->
233
268
case sumPos of { (I ## sumPos') ->
@@ -258,7 +293,7 @@ read distrib = do
258
293
case sumPos of { (I ## sumPos') ->
259
294
case minPos of { (I ## minPos') ->
260
295
case maxPos of { (I ## maxPos') ->
261
- case readInt64Array## mba countPos' s of { (## s1, count ## ) ->
296
+ case readInt64Array mba countPos' s of { (## s1, count ## ) ->
262
297
case readDoubleArray## mba meanPos' s1 of { (## s2, mean ## ) ->
263
298
case readDoubleArray## mba sumSqDeltaPos' s2 of { (## s3, sumSqDelta ## ) ->
264
299
case readDoubleArray## mba sumPos' s3 of { (## s4, dSum ## ) ->
0 commit comments