X-Git-Url: http://gitweb.michael.orlitzky.com/?a=blobdiff_plain;f=src%2FMain.hs;h=c1456f4a94c968ac47d3c764b7bdbc3da9b26245;hb=3544d15ebbd0176c9aac2fd0e0e94468abc56879;hp=cfd17247b27de3c146096fd04acf503d1fecb329;hpb=429558a71324dc10606d05497117b0222b959adc;p=spline3.git diff --git a/src/Main.hs b/src/Main.hs index cfd1724..c1456f4 100644 --- a/src/Main.hs +++ b/src/Main.hs @@ -34,9 +34,9 @@ main3d = do arr <- read_word16s in_file let arr' = swap_bytes arr let arrMRI = R.reshape mri_shape arr' - let dbl_data = R.compute $ R.map fromIntegral arrMRI + let dbl_data = R.computeS $ R.map fromIntegral arrMRI let output = zoom dbl_data zoom_factor - let word16_output = R.compute $ round_array output + let word16_output = R.computeS $ round_array output write_word16s out_file word16_output @@ -47,13 +47,13 @@ main2d = do let zoom_factor = (1, scale, scale) let out_file = "output.bmp" arr <- read_word16s in_file - let arrSlice = R.computeUnboxed $ z_slice 50 $ flip_x $ flip_y $ swap_bytes arr + let arrSlice = R.computeUnboxedS $ z_slice 50 $ flip_x $ flip_y $ swap_bytes arr let arrSlice' = R.reshape mri_slice3d arrSlice -- If zoom isn't being inlined we need to extract the slice before hand, -- and convert it to the require formed. - let dbl_data = R.compute $ R.map fromIntegral arrSlice' + let dbl_data = R.computeS $ R.map fromIntegral arrSlice' let output = zoom dbl_data zoom_factor - let arrSlice0 = R.computeUnboxed $ z_slice 0 output + let arrSlice0 = R.computeUnboxedS $ z_slice 0 output write_values_slice_to_bitmap arrSlice0 out_file