src/Values.hs

   1 {-# LANGUAGE FlexibleInstances #-}
   2
   3 module Values (
   4   Values,
   5   Values2D,
   6   Values3D,
   7   dims,
   8   empty3d,
   9   idx,
  10   zoom_shape
  11   )
  12 where
  13
  14 import Data.Array.Repa (
  15   Array, U,
  16   Z(..),
  17   (:.)(..),
  18   DIM2,
  19   DIM3,
  20   extent,
  21   fromListUnboxed,
  22   unsafeIndex,
  23   )
  24
  25 import System.FilePath ()
  26 import Test.QuickCheck (Arbitrary(..), Gen, choose, vectorOf)
  27
  28
  29 import ScaleFactor (ScaleFactor)
  30
  31
  32 type Values sh = Array U sh Double
  33 type Values2D  = Values DIM2
  34 type Values3D  = Values DIM3
  35
  36
  37 instance Arbitrary Values3D where
  38     arbitrary = do
  39       -- I declare not to care about empty lists.
  40       x_dim <- choose (1, 27)
  41       y_dim <- choose (1, 27)
  42       z_dim <- choose (1, 27)
  43       elements <- vectorOf (x_dim * y_dim * z_dim) (arbitrary :: Gen Double)
  44       let new_shape = (Z :. x_dim :. y_dim :. z_dim)
  45       let three_d = Data.Array.Repa.fromListUnboxed new_shape elements
  46       return three_d
  47
  48
  49 empty3d :: Values3D
  50 empty3d = Data.Array.Repa.fromListUnboxed (Z :. 0 :. 0 :. 0) []
  51
  52
  53 dims :: Values3D -> (Int, Int, Int)
  54 dims v3d =
  55     let (Z :. x :. y :. z) = extent v3d
  56     in
  57       (x,y,z)
  58
  59
  60 idx :: Values3D -> Int -> Int -> Int -> Double
  61 idx v3d i j k =
  62     unsafeIndex v3d shape
  63     where
  64       shape :: DIM3
  65       shape = (Z :. i :. j :. k)
  66
  67
  68 zoom_shape :: ScaleFactor -> DIM3 -> DIM3
  69 zoom_shape (sfx, sfy, sfz) sh =
  70     let (Z :. x :. y :. z) = sh
  71         x' = x * sfx
  72         y' = y * sfy
  73         z' = z * sfz
  74     in
  75       (Z :. x' :. y' :. z')