where
import Data.Array.Repa (
- Array,
+ Array, U,
Z(..),
(:.)(..),
DIM2,
DIM3,
extent,
- fromList,
+ fromListUnboxed,
unsafeIndex,
)
import ScaleFactor (ScaleFactor)
-type Values sh = Array sh Double
-type Values2D = Values DIM2
-type Values3D = Values DIM3
+type Values sh = Array U sh Double
+type Values2D = Values DIM2
+type Values3D = Values DIM3
instance Arbitrary Values3D where
z_dim <- choose (1, 27)
elements <- vectorOf (x_dim * y_dim * z_dim) (arbitrary :: Gen Double)
let new_shape = (Z :. x_dim :. y_dim :. z_dim)
- let three_d = Data.Array.Repa.fromList new_shape elements
+ let three_d = Data.Array.Repa.fromListUnboxed new_shape elements
return three_d
empty3d :: Values3D
-empty3d = Data.Array.Repa.fromList (Z :. 0 :. 0 :. 0) []
+empty3d = Data.Array.Repa.fromListUnboxed (Z :. 0 :. 0 :. 0) []
dims :: Values3D -> (Int, Int, Int)