X-Git-Url: http://gitweb.michael.orlitzky.com/?a=blobdiff_plain;f=src%2FValues.hs;h=8e40212eda252af463b231782b9f424ce4ba48ad;hb=985ab5303039bc69d57139092b4adf9603cc2d0c;hp=070abbbf2554ea5eec5276e5496fe1554fcaf3d3;hpb=3b111f8df1bde25f32f3d5378844cbcf34404015;p=spline3.git diff --git a/src/Values.hs b/src/Values.hs index 070abbb..8e40212 100644 --- a/src/Values.hs +++ b/src/Values.hs @@ -1,34 +1,37 @@ {-# LANGUAGE FlexibleInstances #-} -module Values +module Values ( + Values, + Values2D, + Values3D, + dims, + empty3d, + idx, + zoom_shape + ) where import Data.Array.Repa ( Array, Z(..), (:.)(..), - DIM1, DIM2, DIM3, extent, fromList, unsafeIndex, - reshape, - size ) -import Data.Array.Repa.IO.Vector (readVectorFromTextFile, - writeVectorToTextFile) import System.FilePath () import Test.QuickCheck (Arbitrary(..), Gen, choose, vectorOf) -import ScaleFactor +import ScaleFactor (ScaleFactor) -type Values1D = Array DIM1 Double -type Values2D = Array DIM2 Double -type Values3D = Array DIM3 Double +type Values sh = Array sh Double +type Values2D = Values DIM2 +type Values3D = Values DIM3 instance Arbitrary Values3D where @@ -43,22 +46,6 @@ instance Arbitrary Values3D where return three_d -read_values_1d :: FilePath -> IO Values1D -read_values_1d = readVectorFromTextFile - - -read_values_3d :: DIM3 -> FilePath -> IO Values3D -read_values_3d sh path = do - one_d <- read_values_1d path - return $ reshape sh one_d - -write_values_1d :: Values3D -> FilePath -> IO () -write_values_1d v3d path = do - let size3d = size $ extent v3d - let shape1d = (Z :. size3d) - let v1d = reshape shape1d v3d - writeVectorToTextFile v1d path - empty3d :: Values3D empty3d = Data.Array.Repa.fromList (Z :. 0 :. 0 :. 0) []