X-Git-Url: http://gitweb.michael.orlitzky.com/?a=blobdiff_plain;f=src%2FValues.hs;h=b5dc15a9c6299bb0ec74511a8e561d6c0bdcf09a;hb=d910ddbbb9e6b9c8bf8169ce30c42660ad8746d4;hp=8e40212eda252af463b231782b9f424ce4ba48ad;hpb=83e7315396684663d9d9893ec798381682bb1f9b;p=spline3.git diff --git a/src/Values.hs b/src/Values.hs index 8e40212..b5dc15a 100644 --- a/src/Values.hs +++ b/src/Values.hs @@ -12,13 +12,13 @@ module Values ( where import Data.Array.Repa ( - Array, + Array, U, Z(..), (:.)(..), DIM2, DIM3, extent, - fromList, + fromListUnboxed, unsafeIndex, ) @@ -29,9 +29,9 @@ import Test.QuickCheck (Arbitrary(..), Gen, choose, vectorOf) 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 @@ -42,12 +42,12 @@ 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)