import Data.Word
import Data.Bits
-import Data.Array.Repa as R
-import Data.Array.Repa.Repr.Unboxed as R
-import Data.Array.Repa.IO.Binary as R
+import Data.Array.Repa as R
+import Data.Array.Repa.Eval as R (now)
+import Data.Array.Repa.Repr.Unboxed as R
+import Data.Array.Repa.IO.Binary as R
import Data.Array.Repa.Algorithms.ColorRamp as R
-import Data.Array.Repa.IO.BMP as R (writeImageToBMP)
+import Data.Array.Repa.Operators.Traversal as R (unsafeTraverse)
+import Data.Array.Repa.IO.BMP as R (writeImageToBMP)
import Values
read_word16s :: FilePath -> IO RawData3D
read_word16s path = do
arr <- R.readArrayFromStorableFile path mri_shape
- arr' <- now $ R.copy arr
- return arr'
+ now $ R.copyS arr
+
bracket :: Double -> Word16
{-# INLINE swap_bytes #-}
swap_bytes :: (Shape sh, Repr r Word16) => Array r sh Word16
-> Array D sh Word16
-swap_bytes arr =
- R.map flip16 arr
+swap_bytes =
+ R.map flip16
bracket_array :: Shape sh => Values sh -> Array D sh Word16
-bracket_array arr =
- R.map bracket arr
+bracket_array =
+ R.map bracket
{-# INLINE round_array #-}
round_array :: Shape sh => Values sh -> Array D sh Word16
-round_array arr =
- R.map round arr
+round_array =
+ R.map round
flip_y :: Repr r Word16 => Array r DIM3 Word16 -> Array D DIM3 Word16
values_to_colors :: (Shape sh) => (Values sh) -> (ColorData sh)
values_to_colors arr =
- R.compute $ R.map (truncate_rgb . ramp_it) arr
+ R.computeS $ R.map (truncate_rgb . ramp_it) arr
where
ramp_it :: Double -> (Double, Double, Double)
ramp_it x =
R.writeImageToBMP path colors
where
arr_bracketed = bracket_array v3d
- colors = values_to_colors $ R.compute $ R.map fromIntegral arr_bracketed
+ colors = values_to_colors $ R.computeS $ R.map fromIntegral arr_bracketed