X-Git-Url: http://gitweb.michael.orlitzky.com/?p=spline3.git;a=blobdiff_plain;f=src%2FValues.hs;h=7fc577529f9ccdd2ce0fe1fc85cf14d8744e1e21;hp=a4720252bbe626ae70a52913822ed318665ffc08;hb=fd3d394c27e3a90de8238b98bd112c4fe3468ee0;hpb=3f7331f579118687cd73b977ce6aa7d401f88a09 diff --git a/src/Values.hs b/src/Values.hs index a472025..7fc5775 100644 --- a/src/Values.hs +++ b/src/Values.hs @@ -7,61 +7,33 @@ module Values ( dims, empty3d, idx, - read_values_3d, zoom_shape ) where import Data.Array.Repa ( - Array, + Array, U, Z(..), (:.)(..), - DIM1, DIM2, DIM3, extent, - fromList, + fromListUnboxed, unsafeIndex, - reshape ) -import Data.Array.Repa.IO.Vector (readVectorFromTextFile) import System.FilePath () -import Test.QuickCheck (Arbitrary(..), Gen, choose, vectorOf) - import ScaleFactor (ScaleFactor) -type Values sh = Array sh Double -type Values1D = Values DIM1 -type Values2D = Values DIM2 -type Values3D = Values DIM3 - - -instance Arbitrary Values3D where - arbitrary = do - -- I declare not to care about empty lists. - x_dim <- choose (1, 27) - y_dim <- choose (1, 27) - 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 - return three_d - - -read_values_1d :: FilePath -> IO Values1D -read_values_1d = readVectorFromTextFile - +type Values sh = Array U sh Double +type Values2D = Values DIM2 +type Values3D = Values DIM3 -read_values_3d :: DIM3 -> FilePath -> IO Values3D -read_values_3d sh path = do - one_d <- read_values_1d path - return $ reshape sh one_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)