# emacs: -*- mode: python; py-indent-offset: 4; indent-tabs-mode: nil -*-# vi: set ft=python sts=4 ts=4 sw=4 et:## Copyright 2023 The NiPreps Developers <nipreps@gmail.com>## Licensed under the Apache License, Version 2.0 (the "License");# you may not use this file except in compliance with the License.# You may obtain a copy of the License at## http://www.apache.org/licenses/LICENSE-2.0## Unless required by applicable law or agreed to in writing, software# distributed under the License is distributed on an "AS IS" BASIS,# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.# See the License for the specific language governing permissions and# limitations under the License.## We support and encourage derived works from this project, please read# about our expectations at## https://www.nipreps.org/community/licensing/## STATEMENT OF CHANGES: This file was ported carrying over full git history from niworkflows,# another NiPreps project licensed under the Apache-2.0 terms, and has been changed since.# The original file this work derives from is found at:# https://github.com/nipreps/niworkflows/blob/fa273d004c362d9562616253180e95694f07be3b/# niworkflows/utils/images.py"""Tooling to manipulate n-dimensional images."""importnibabelasnbimportnumpyasnp
[docs]defrotation2canonical(img):"""Calculate the rotation w.r.t. cardinal axes of input image."""img=nb.as_closest_canonical(img)newaff=np.diag(img.header.get_zooms()[:3])r=newaff@np.linalg.pinv(img.affine[:3,:3])ifnp.allclose(r,np.eye(3)):returnNonereturnr
[docs]defrotate_affine(img,rot=None):"""Rewrite the affine of a spatial image."""ifrotisNone:returnimgimg=nb.as_closest_canonical(img)affine=np.eye(4)affine[:3]=rot@img.affine[:3]returnimg.__class__(img.dataobj,affine,img.header)