From f3ea8cf523d81005a5a7cdb36d4184e4e93fc01e Mon Sep 17 00:00:00 2001
From: Robert Smith <robert.smith@florey.edu.au>
Date: Mon, 5 Aug 2024 15:56:09 +1000
Subject: [PATCH] dwi2tensor: Preserve dw_scheme for -predicted_signal

Preserving the diffusion gradient scheme in the image header for the output of the -predicted_signal option is preferable for enabling visualisation of such data using the MRView ODF Tool in Dixel mode.
---
 cmd/dwi2tensor.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/cmd/dwi2tensor.cpp b/cmd/dwi2tensor.cpp
index c9ced7dc97..c814732552 100644
--- a/cmd/dwi2tensor.cpp
+++ b/cmd/dwi2tensor.cpp
@@ -220,7 +220,6 @@ void run ()
   Header header (dwi);
   header.datatype() = DataType::Float32;
   header.ndim() = 4;
-  DWI::stash_DW_scheme (header, grad);
   PhaseEncoding::clear_scheme (header);
 
   Image<value_type> predict;
@@ -228,6 +227,7 @@ void run ()
   if (opt.size())
     predict = Image<value_type>::create (opt[0][0], header);
 
+  DWI::stash_DW_scheme (header, grad);
   header.size(3) = 6;
   auto dt = Image<value_type>::create (argument[1], header);