1 {-# LANGUAGE FlexibleInstances #-}
15 import Data.Array.Repa (
28 import Data.Array.Repa.IO.Vector (readVectorFromTextFile)
29 import System.FilePath ()
30 import Test.QuickCheck (Arbitrary(..), Gen, choose, vectorOf)
33 import ScaleFactor (ScaleFactor)
36 type Values sh = Array sh Double
37 type Values1D = Values DIM1
38 type Values2D = Values DIM2
39 type Values3D = Values DIM3
42 instance Arbitrary Values3D where
44 -- I declare not to care about empty lists.
45 x_dim <- choose (1, 27)
46 y_dim <- choose (1, 27)
47 z_dim <- choose (1, 27)
48 elements <- vectorOf (x_dim * y_dim * z_dim) (arbitrary :: Gen Double)
49 let new_shape = (Z :. x_dim :. y_dim :. z_dim)
50 let three_d = Data.Array.Repa.fromList new_shape elements
54 read_values_1d :: FilePath -> IO Values1D
55 read_values_1d = readVectorFromTextFile
58 read_values_3d :: DIM3 -> FilePath -> IO Values3D
59 read_values_3d sh path = do
60 one_d <- read_values_1d path
61 return $ reshape sh one_d
64 empty3d = Data.Array.Repa.fromList (Z :. 0 :. 0 :. 0) []
67 dims :: Values3D -> (Int, Int, Int)
69 let (Z :. x :. y :. z) = extent v3d
74 idx :: Values3D -> Int -> Int -> Int -> Double
79 shape = (Z :. i :. j :. k)
82 zoom_shape :: ScaleFactor -> DIM3 -> DIM3
83 zoom_shape (sfx, sfy, sfz) sh =
84 let (Z :. x :. y :. z) = sh