X-Git-Url: http://gitweb.michael.orlitzky.com/?a=blobdiff_plain;f=src%2FValues.hs;h=8e40212eda252af463b231782b9f424ce4ba48ad;hb=83e7315396684663d9d9893ec798381682bb1f9b;hp=d7ede8399391b3229a1c7a6d36f04fececd3674b;hpb=f543f3bbc870d0b77cb9b5ad652963af9a70fbf3;p=spline3.git diff --git a/src/Values.hs b/src/Values.hs index d7ede83..8e40212 100644 --- a/src/Values.hs +++ b/src/Values.hs @@ -1,33 +1,35 @@ {-# 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 Values sh = Array sh Double -type Values1D = Values DIM1 type Values2D = Values DIM2 type Values3D = Values DIM3 @@ -44,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) [] @@ -87,8 +73,3 @@ zoom_shape (sfx, sfy, sfz) sh = z' = z * sfz in (Z :. x' :. y' :. z') - - -drop_z :: DIM3 -> DIM2 -drop_z (Z :. 1 :. y :. x) = (Z :. y :. x) -drop_z _ = error "can't drop the z-dimension unless its size is 1"