def register_FA_same_subj_diff_sessions(dname_grid,dname_shell):
print('create temporary directory')
tmp_dir='/tmp'
print('load dicom data')
data_gr,affine_gr,bvals_gr,gradients_gr=dp.load_dcm_dir(dname_grid)
data_sh,affine_sh,bvals_sh,gradients_sh=dp.load_dcm_dir(dname_shell)
print('save DWI reference as nifti')
tmp_grid=os.path.join(tmp_dir,os.path.basename(dname_grid)+'_ref.nii')
tmp_shell=os.path.join(tmp_dir,os.path.basename(dname_shell)+'_ref.nii')
ni.save(ni.Nifti1Image(data_gr[...,0],affine_gr),tmp_grid)
ni.save(ni.Nifti1Image(data_sh[...,0],affine_sh),tmp_shell)
print('prepare filenames for haircut (bet)')
tmp_grid_bet=os.path.join(os.path.dirname(tmp_grid),\
os.path.splitext(os.path.basename(dname_grid))[0]+\
'_ref_bet.nii.gz')
tmp_shell_bet=os.path.join(os.path.dirname(tmp_shell),\
os.path.splitext(os.path.basename(dname_shell))[0]+\
'_ref_bet.nii.gz')
print('bet is running')
haircut_dwi_reference(tmp_grid,tmp_grid_bet)
haircut_dwi_reference(tmp_shell,tmp_shell_bet)
print('load nii.gz reference (s0) volumes')
img_gr_bet=ni.load(tmp_grid_bet)
img_sh_bet=ni.load(tmp_shell_bet)
print('register the shell reference to the grid reference')
source=img_sh_bet
target=img_gr_bet
T=dp.volume_register(source,target,similarity,\
interp,subsampling,search,optimizer)
print('apply the inverse of the transformation matrix')
dp.volume_transform(source, T.inv(), reference=target)
#ni.save(sourceT,'/tmp/result.nii.gz')
print('calculate FA for grid and shell data')
FA_grid=dp.Tensor( data_gr,bvals_gr,gradients_gr,thresh=50).FA
FA_shell=dp.Tensor(data_sh,bvals_sh,gradients_sh,thresh=50).FA
print('create an FA nibabel image for shell')
FA_shell_img=ni.Nifti1Image(FA_shell,affine_sh)
print('transform FA_shell')
FA_shell_imgT=dp.volume_transform(FA_shell_img,T.inv(),reference=target)
return ni.Nifti1Image(FA_grid,affine_gr),FA_shell_imgT
def test_registration():
# S012='/tmp/compare_12_with_32_Verio_directly/18620_0004.nii_S0.nii.gz'
# S032='/tmp/compare_12_with_32_Verio_directly/18620_0006.nii_S0.nii.gz'
S012T='/tmp/compare_12_with_32_Verio_directly/S0_reg.nii.gz'
MP='/tmp/compare_12_with_32_Verio_directly/MPRAGE.nii'
D114=resources.get_paths('DTI STEAM 114 Trio')[2]
data,affine,bvals,gradients=dp.load_dcm_dir(D114)
D114i=ni.Nifti1Image(data[...,0],affine)
D101=resources.get_paths('DSI STEAM 101 Trio')[2]
data,affine,bvals,gradients=dp.load_dcm_dir(D101)
D101i=ni.Nifti1Image(data[...,0],affine)
ni.save(D101i,'/tmp/compare_12_with_32_Verio_directly/S0_101_reg.nii.gz')
#source=ni.load(S012)
source=D114i
#target=D101i
#target=ni.load(S032)
target=ni.load(MP)
target._data=np.squeeze(target._data)
#target._affine= np.dot(np.diag([-1, -1, 1, 1]), target._affine)
similarity='cr'
interp = 'tri'
subsampling=None
search='affine'
optimizer= 'powell'
T=dp.volume_register(source,target,similarity,\
interp,subsampling,search,optimizer)
print('Transformation matrix')
print(T.inv())
sourceT=dp.volume_transform(source,T.inv(),reference=target,interp_order=0)
sourceTd=sourceT.get_data()
sourceTd[sourceTd<0]=0
sourceT._data=sourceTd
ni.save(sourceT,S012T)
sourced=source.get_data()
targetd=target.get_data()
sourceTd=sourceT.get_data()
print 'source info',sourced.min(), sourced.max()
print 'target info',targetd.min(), targetd.max()
print 'sourceT info',sourceTd.min(), sourceTd.max()
import numpy as np
import dipy as dp
import nibabel as ni
dname = '/home/eg01/Data_Backup/Data/Eleftherios/CBU090133_METHODS/20090227_145404/Series_003_CBU_DTI_64D_iso_1000'
#dname = '/home/eg01/Data_Backup/Data/Frank_Eleftherios/frank/20100511_m030y_cbu100624/08_ep2d_advdiff_101dir_DSI'
data,affine,bvals,gradients=dp.load_dcm_dir(dname)
'''
rot=np.array([[1,0,0,0],
[0,np.cos(np.pi/2),-np.sin(np.pi/2),0],
[0,np.sin(np.pi/2), np.cos(np.pi/2),0],
[0,0,0,1]])
from scipy.ndimage import affine_transform as aff
naffine=np.dot(affine,rot)
'''
data[:,:,:,1]
source=ni.Nifti1Image(data[:,:,:,1],affine)
target=ni.Nifti1Image(data[:,:,:,0],affine)
#similarity 'cc', 'cr', 'crl1', 'mi', je', 'ce', 'nmi', 'smi'. 'cr'
similarity='cr'
#interp 'pv', 'tri'
interp = 'tri'
import numpy as np
import dipy as dp
import nibabel as ni
dname = '/home/eg01/Data_Backup/Data/Eleftherios/CBU090133_METHODS/20090227_145404/Series_003_CBU_DTI_64D_iso_1000'
#dname = '/home/eg01/Data_Backup/Data/Frank_Eleftherios/frank/20100511_m030y_cbu100624/08_ep2d_advdiff_101dir_DSI'
data, affine, bvals, gradients = dp.load_dcm_dir(dname)
'''
rot=np.array([[1,0,0,0],
[0,np.cos(np.pi/2),-np.sin(np.pi/2),0],
[0,np.sin(np.pi/2), np.cos(np.pi/2),0],
[0,0,0,1]])
from scipy.ndimage import affine_transform as aff
naffine=np.dot(affine,rot)
'''
data[:, :, :, 1]
source = ni.Nifti1Image(data[:, :, :, 1], affine)
target = ni.Nifti1Image(data[:, :, :, 0], affine)
#similarity 'cc', 'cr', 'crl1', 'mi', je', 'ce', 'nmi', 'smi'. 'cr'
similarity = 'cr'
#interp 'pv', 'tri'
interp = 'tri'
#subsampling None or sequence (3,)
