Ticket #44: replicateMPrim

1 patch for repository http://code.haskell.org/vector:

Fri Sep 17 08:12:01 CEST 2010  Daniel Peebles <pumpkingod@gmail.com>
  * Specialize replicateM for primitive Monad instances

New patches:

[Specialize replicateM for primitive Monad instances
Daniel Peebles <pumpkingod@gmail.com>**20100917061201
 Ignore-this: fd529ef86eea92198138ac42f0f54bf
] {
hunk ./Data/Vector/Generic.hs 598
 -- | /O(n)/ Execute the monadic action the given number of times and store the
 -- results in a vector.
 replicateM :: (Monad m, Vector v a) => Int -> m a -> m (v a)
--- FIXME: specialise for ST and IO?
 {-# INLINE replicateM #-}
 replicateM n m = fromListN n `Monad.liftM` Monad.replicateM n m
 
hunk ./Data/Vector/Generic.hs 601
+{-# RULES
+"replicateM(IO)" forall n (m :: IO a).
+  replicateM n m = replicateMPrim n m
+
+"replicateM(ST)" forall n (m :: forall s. ST s a).
+  replicateM n m = replicateMPrim n m
+  #-}
+
+-- A more efficient replicateM that doesn't go through lists, for primitive Monads.
+-- Not exported, as the only way of getting it is through the above rewrite rules.
+-- From http://haskell.org/haskellwiki/Numeric_Haskell:_A_Vector_Tutorial
+{-# INLINE replicateMPrim #-}
+replicateMPrim :: (PrimMonad m, Vector v a) => Int -> m a -> m (v a)
+replicateMPrim n m = do
+    v  <- M.new n
+    fill v 0
+    unsafeFreeze v
+  where
+    fill v i
+        | i < n = do
+            x <- m
+            M.unsafeWrite v i x
+            fill v (i+1)
+        | otherwise = return ()
+
 -- | Execute the monadic action and freeze the resulting vector.
 --
 -- @
}

Context:

[Fix bad bug in ptrToOffset
Roman Leshchinskiy <rl@cse.unsw.edu.au>**20100821050043
 Ignore-this: 79f397bb354deaef91029c0c6a045d5b
] 
[Add function for safe indexing 
Khudyakov Alexey <alexey.skladnoy@gmail.com>**20100727212532
 Ignore-this: b49d0d25d8dc2b38cc3430e990a5ea9f
] 
[TAG 0.6.0.2
Roman Leshchinskiy <rl@cse.unsw.edu.au>**20100615030825
 Ignore-this: eb4bbce255ad4cbdef645655454ed681
] 
[Changelog
Roman Leshchinskiy <rl@cse.unsw.edu.au>**20100615030335
 Ignore-this: de9af15a7972cb436574f2ba962b6c81
] 
[Eta-expand create to work around GHC bug 4120
Roman Leshchinskiy <rl@cse.unsw.edu.au>**20100615030308
 Ignore-this: 4c3875d889a56217d147b15145e595c7
] 
[Add freeze
Roman Leshchinskiy <rl@cse.unsw.edu.au>**20100524075434
 Ignore-this: 288ab46bd96aed8968437b245f22ad8d
] 
[Add Mutable.clone
Roman Leshchinskiy <rl@cse.unsw.edu.au>**20100524075045
 Ignore-this: f6710074f85671b32d2b2002bcbc1ebb
] 
[Fix comments
Roman Leshchinskiy <rl@cse.unsw.edu.au>**20100524073956
 Ignore-this: 8935cfcfba5c13e7274954f326e8c630
] 
[Add unsafeThaw
Roman Leshchinskiy <rl@cse.unsw.edu.au>**20100524072249
 Ignore-this: 13c1cd3221c95355e35213c285538618
] 
[Add basicUnsafeThaw
Roman Leshchinskiy <rl@cse.unsw.edu.au>**20100524071700
 Ignore-this: 790b47cc41ea0df223f9a6b8ef549661
] 
[Require primitive >= 0.3.1
Roman Leshchinskiy <rl@cse.unsw.edu.au>**20100524071649
 Ignore-this: d0328714d4332f81e98d2c6c7688de71
] 
[Comment
Roman Leshchinskiy <rl@cse.unsw.edu.au>**20100524060303
 Ignore-this: c8e6612a32b29bad6d6bc486175a2d63
] 
[Rename unsafeFreeze to basicUnsafeFreeze and add unsafeFreeze as a free function
Roman Leshchinskiy <rl@cse.unsw.edu.au>**20100524055727
 Ignore-this: 82af49dbedf2fddd4f1b7acc0da41845
] 
[Bump version to 0.7
Roman Leshchinskiy <rl@cse.unsw.edu.au>**20100524055716
 Ignore-this: e9d27d0a283e5ec8a9cbd7f74f40f219
] 
[Improve tridiag benchmark a bit
Roman Leshchinskiy <rl@cse.unsw.edu.au>**20100524044438
 Ignore-this: e182a804ebbee6a9c43d96be6bf4d0c5
] 
[Don't actually inspect SPEC, just seq on it
Roman Leshchinskiy <rl@cse.unsw.edu.au>**20100524044300
 Ignore-this: dbf0ab999aafdbee3a21f831a7b6751e
 
 This makes the intermediate code much simpler and might improve demand
 analysis.
] 
[Compile benchmarks with -fno-method-sharing
Roman Leshchinskiy <rl@cse.unsw.edu.au>**20100524040909
 Ignore-this: 24ff9ef3fa83bf3ea1090a7e5e86398d
] 
[Avoid last LiberateCase in D.V.Generic
Roman Leshchinskiy <rl@cse.unsw.edu.au>**20100524040316
 Ignore-this: 6d5ca63eabc1ac8fa7384d12a648357c
] 
[More bangs to avoid LiberateCase
Roman Leshchinskiy <rl@cse.unsw.edu.au>**20100524034156
 Ignore-this: 6735856042b634c37223b55207335d3e
] 
[Be explicitly strict in vector lengths more often to avoid triggering LiberateCase
Roman Leshchinskiy <rl@cse.unsw.edu.au>**20100524032222
 Ignore-this: d8b58f39e6ee245977c0fa87be23b2c8
] 
[Make D.V.Generic.stream strict in the length of the vector
Roman Leshchinskiy <rl@cse.unsw.edu.au>**20100523055128
 Ignore-this: 40b5ef525577ba879f2abb3a95ef64a5
] 
[Make enumFromTo* strict in the bounds
Roman Leshchinskiy <rl@cse.unsw.edu.au>**20100523053844
 Ignore-this: 3ff8dbe94783127ff0b0b099e49acb7b
] 
[Remove thawMany
Roman Leshchinskiy <rl@cse.unsw.edu.au>**20100518054931
 Ignore-this: 1746ab5c8155999850cb07eabc615799
] 
[Add Stream.flatten and use it to implement concat
Roman Leshchinskiy <rl@cse.unsw.edu.au>**20100518054842
 Ignore-this: 1124240c85914fc5059bc0c601ee9eef
] 
[Sanitise Mutable exports and reorder code
Roman Leshchinskiy <rl@cse.unsw.edu.au>**20100505164012
 Ignore-this: 685ea7d5bd3f77276af3fffd0a40a476
] 
[Delete comment
Roman Leshchinskiy <rl@cse.unsw.edu.au>**20100505155507
 Ignore-this: b08ae6a4f9977d1156cb10eea267f3f7
] 
[Add Monoid instances
Roman Leshchinskiy <rl@cse.unsw.edu.au>**20100505155345
 Ignore-this: 2ef598c8b8bbe780b3e5f9561ae59d92
] 
[Add convert
Roman Leshchinskiy <rl@cse.unsw.edu.au>**20100505151407
 Ignore-this: 9665dd334d44fa5d3ef57ff978c8ded6
] 
[Replace remaining occurences of basicUnsafeNewWith with basicUnsafeReplicate
Roman Leshchinskiy <rl@cse.unsw.edu.au>**20100505150308
 Ignore-this: 7129e607fa1e50481f61f378d714f152
] 
[Add Mutable.replicate and deprecate Mutable.n{newWith,unsafeNewWith}
Roman Leshchinskiy <rl@cse.unsw.edu.au>**20100505144550
 Ignore-this: 39aa15df1e0bc3bc1380a3a74443c073
] 
[Add concat and make concatMap more efficient
Roman Leshchinskiy <rl@cse.unsw.edu.au>**20100505141141
 Ignore-this: a587071dc839b44fa219d7536cb81d9a
] 
[Add thaw and thawMany
Roman Leshchinskiy <rl@cse.unsw.edu.au>**20100505141028
 Ignore-this: 14d5862d0940d3598554b07ba7652648
] 
[Export Generic.Mutable.unsafe{Take,Drop}
Roman Leshchinskiy <rl@cse.unsw.edu.au>**20100505135854
 Ignore-this: dbbd6e97ae3276cf312c1036a6f32351
] 
[Fusion rules for monadic indexing
Roman Leshchinskiy <rl@cse.unsw.edu.au>**20100503031801
 Ignore-this: 2ed629ca5b44031b3641a5716f88e763
] 
[Rearrange code
Roman Leshchinskiy <rl@cse.unsw.edu.au>**20100428015052
 Ignore-this: 93c3164d2db02de93a2e415960a4d2d7
] 
[Bump version number
Roman Leshchinskiy <rl@cse.unsw.edu.au>**20100428015037
 Ignore-this: 25626073a2bcad8c11c844975000a728
] 
[TAG 0.6.0.1
Roman Leshchinskiy <rl@cse.unsw.edu.au>**20100427053224
 Ignore-this: a04080a48ae45f144f8556bfba1b43ee
] 
Patch bundle hash:
8077eb40c3f165676c9fd4bc556e15a287b14494