From: Michael Orlitzky Date: Wed, 14 Sep 2011 17:35:47 +0000 (-0400) Subject: Add some convenience functions to write chunks/slices to bitmaps. X-Git-Tag: 0.0.1~133 X-Git-Url: https://gitweb.michael.orlitzky.com/?a=commitdiff_plain;h=62b333768c3f3f5380fefeb9c5d09276543d677c;p=spline3.git Add some convenience functions to write chunks/slices to bitmaps. --- diff --git a/src/MRI.hs b/src/MRI.hs index 52be3f7..5f2ceea 100644 --- a/src/MRI.hs +++ b/src/MRI.hs @@ -6,6 +6,7 @@ import Data.Bits import Data.Array.Repa as R import Data.Array.Repa.IO.Binary as R import Data.Array.Repa.IO.ColorRamp as R +import Data.Array.Repa.IO.BMP as R (writeComponentsToBMP) import Values @@ -158,3 +159,24 @@ z_slice3 n arr previous = transpose_zx $ reshape mri_slice3d (z_slice (n-1) arr) current = transpose_zx $ reshape mri_slice3d (z_slice n arr) next = transpose_zx $ reshape mri_slice3d (z_slice (n+1) arr) + + +write_values_slice_to_bitmap :: Values2D -> FilePath -> IO () +write_values_slice_to_bitmap v3d path = + R.writeComponentsToBMP path routput goutput boutput + where + arr_bracketed = bracket_array v3d + colors = values_to_colors $ R.map fromIntegral arr_bracketed + routput = R.map (\(red, _, _) -> red) colors + goutput = R.map (\(_, green, _) -> green) colors + boutput = R.map (\(_, _, blue) -> blue) colors + + +write_values_chunk_to_bitmap :: Values3D -> FilePath -> IO () +write_values_chunk_to_bitmap v3d path + | zdim /= 3 = error "not a 3xMxN chunk" + | otherwise = write_values_slice_to_bitmap target_slice path + where + (Z :. zdim :. _ :. _) = extent v3d + target_slice = z_slice 1 v3d +