roundVector

4 files changed, 22 insertions, 5 deletions

diff --git a/packages/base/src/C/vector-aux.c b/packages/base/src/C/vector-aux.c
index b176089..efb82af 100644
--- a/packages/base/src/C/vector-aux.c
+++ b/packages/base/src/C/vector-aux.c
@@ -783,3 +783,13 @@ int sort_values(KDVEC(v),DVEC(r)) {
     OK
 }
 
+////////////////////////////////////////////////////////////////////////////////
+
+int round_vector(KDVEC(v),DVEC(r)) {
+    int k;
+    for(k=0; k<vn; k++) {
+        rp[k] = round(vp[k]);
+    }
+    OK
+}
+
diff --git a/packages/base/src/Data/Packed/Internal/Numeric.hs b/packages/base/src/Data/Packed/Internal/Numeric.hs
index 56ecdfe..9adc023 100644
--- a/packages/base/src/Data/Packed/Internal/Numeric.hs
+++ b/packages/base/src/Data/Packed/Internal/Numeric.hs
@@ -36,9 +36,8 @@ module Data.Packed.Internal.Numeric (
     Convert(..),
     Complexable(),
     RealElement(),
-
+    roundVector,
     RealOf, ComplexOf, SingleOf, DoubleOf,
-
     IndexOf,
     module Data.Complex
 ) where
diff --git a/packages/base/src/Data/Packed/Numeric.hs b/packages/base/src/Data/Packed/Numeric.hs
index cc26d7a..ab9e8a2 100644
--- a/packages/base/src/Data/Packed/Numeric.hs
+++ b/packages/base/src/Data/Packed/Numeric.hs
@@ -57,9 +57,8 @@ module Data.Packed.Numeric (
     Convert(..),
     Complexable(),
     RealElement(),
-
     RealOf, ComplexOf, SingleOf, DoubleOf,
-
+    roundVector,
     IndexOf,
     module Data.Complex,
     -- * IO
diff --git a/packages/base/src/Numeric/Vectorized.hs b/packages/base/src/Numeric/Vectorized.hs
index 82504a9..936daba 100644
--- a/packages/base/src/Numeric/Vectorized.hs
+++ b/packages/base/src/Numeric/Vectorized.hs
@@ -19,7 +19,7 @@ module Numeric.Vectorized (
     FunCodeVV(..), vectorZipR, vectorZipC, vectorZipF, vectorZipQ,
     vectorScan, saveMatrix,
     Seed, RandDist(..), randomVector,
-    sortVector
+    sortVector, roundVector
 ) where
 
 import Data.Packed.Internal.Common
@@ -354,3 +354,12 @@ sortVector v = unsafePerformIO $ do
 
 foreign import ccall unsafe "sort_values" c_sort_values :: TVV
 
+--------------------------------------------------------------------------------
+
+roundVector v = unsafePerformIO $ do
+    r <- createVector (dim v)
+    app2 c_round_vector vec v vec r "roundVector"
+    return r
+
+foreign import ccall unsafe "round_vector" c_round_vector :: TVV
+