1 {-# LANGUAGE FlexibleInstances #-}
6 import Data.Array.Repa (
20 import Data.Array.Repa.IO.Vector (readVectorFromTextFile,
21 writeVectorToTextFile)
22 import System.FilePath ()
23 import Test.QuickCheck (Arbitrary(..), Gen)
29 type Values1D = Array DIM1 Double
30 type Values2D = Array DIM2 Double
31 type Values3D = Array DIM3 Double
34 instance Arbitrary Values3D where
36 x_dim <- arbitrary :: Gen Int
37 y_dim <- arbitrary :: Gen Int
38 z_dim <- arbitrary :: Gen Int
39 one_d <- arbitrary :: Gen Values1D
40 let new_shape = (Z :. x_dim :. y_dim :. z_dim)
41 let three_d = reshape new_shape one_d
45 instance Arbitrary Values1D where
47 x <- arbitrary :: Gen [Double]
48 let shape = (Z :. (length x))
49 let one_d = Data.Array.Repa.fromList shape x
53 read_values_1d :: FilePath -> IO Values1D
54 read_values_1d path = readVectorFromTextFile path
57 read_values_3d :: DIM3 -> FilePath -> IO Values3D
58 read_values_3d sh path = do
59 one_d <- read_values_1d path
60 return $ reshape sh one_d
62 write_values_1d :: Values3D -> FilePath -> IO ()
63 write_values_1d v3d path = do
64 let size3d = size $ extent v3d
65 let shape1d = (Z :. size3d)
66 let v1d = reshape shape1d v3d
67 writeVectorToTextFile v1d path
70 empty3d = Data.Array.Repa.fromList (Z :. 0 :. 0 :. 0) []
73 dims :: Values3D -> (Int, Int, Int)
75 let (Z :. x :. y :. z) = extent v3d
80 idx :: Values3D -> Int -> Int -> Int -> Double
85 shape = (Z :. i :. j :. k)
88 zoom_shape :: ScaleFactor -> DIM3 -> DIM3
89 zoom_shape (sfx, sfy, sfz) sh =
90 let (Z :. x :. y :. z) = sh