mri_shape :: DIM3
mri_shape = (Z :. mri_depth :. mri_height :. mri_width)
-mri_lower_threshold :: Int
+mri_lower_threshold :: Double
mri_lower_threshold = 1400
-mri_upper_threshold :: Int
+mri_upper_threshold :: Double
mri_upper_threshold = 2500
mri_slice3d :: DIM3
{-# INLINE bracket #-}
-bracket :: Int -> Int -> Int -> Word16
-bracket low high x
- | x < low = 0
- | x > high = 255
- | otherwise = truncate (r * 255)
+bracket :: Double -> Word16
+bracket x
+ | x < mri_lower_threshold = 0
+ | x > mri_upper_threshold = 255
+ | otherwise = truncate (r * 255)
where
- numerator = fromIntegral (x - low) :: Double
- denominator = fromIntegral (high - low) :: Double
+ numerator = x - mri_lower_threshold
+ denominator = mri_upper_threshold - mri_lower_threshold
r = numerator/denominator
swap_bytes arr =
R.force $ R.map flip16 arr
-bracket_array :: (Shape sh) => (RawData sh) -> (RawData sh)
+bracket_array :: (Shape sh) => (Values sh) -> (RawData sh)
bracket_array arr =
R.force $ R.map f arr
where
- f = (bracket mri_lower_threshold mri_upper_threshold) . fromIntegral
+ f = bracket
flip_y :: RawData3D -> RawData3D
z_slice :: Elt a => Int -> Array DIM3 a -> Array DIM2 a
z_slice n arr =
slice arr (Any :. n :. All :. All)
+
+
+transpose_zx :: Elt a => Array DIM3 a -> Array DIM3 a
+transpose_zx arr =
+ traverse arr
+ (\(Z :. zdim :. ydim :. xdim) -> (Z :. xdim :. ydim :. zdim))
+ (\_ -> (\(Z :. z :. y :. x) -> arr ! (Z :. x :. y :. z)))
+
+
+z_slice3 :: Elt a => Int -> Array DIM3 a -> Array DIM3 a
+z_slice3 n arr
+ | n == 0 = transpose_zx $ current R.++ next
+ | n == zdim-1 = transpose_zx $ previous R.++ current
+ | otherwise = transpose_zx $ previous R.++ current R.++ next
+ where
+ (Z :. zdim :. _ :. _) = extent 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)