From 2a90a63d5fbc8fb57a97e97054e139f5284bf7b0 Mon Sep 17 00:00:00 2001 From: Michael Orlitzky Date: Tue, 23 Aug 2011 16:23:58 -0400 Subject: [PATCH] Write the output file in main. --- src/Main.hs | 2 +- src/Values.hs | 10 +++++++++- 2 files changed, 10 insertions(+), 2 deletions(-) diff --git a/src/Main.hs b/src/Main.hs index 7b50de9..b8edf9e 100644 --- a/src/Main.hs +++ b/src/Main.hs @@ -18,4 +18,4 @@ main = do mridata <- read_values_3d mri_shape "./data/mridata.txt" let g = make_grid 1 mridata let output = zoom g 1 - print output + write_values_1d output "output.txt" diff --git a/src/Values.hs b/src/Values.hs index bd7c4d6..a33bbba 100644 --- a/src/Values.hs +++ b/src/Values.hs @@ -14,9 +14,11 @@ import Data.Array.Repa ( fromList, index, reshape, + size ) -import Data.Array.Repa.IO.Vector (readVectorFromTextFile) +import Data.Array.Repa.IO.Vector (readVectorFromTextFile, + writeVectorToTextFile) import System.FilePath () import Test.QuickCheck (Arbitrary(..), Gen) @@ -54,6 +56,12 @@ 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) [] -- 2.43.2