ODF actor implemented with Ray Tracing

.codespellrc

@@ -2,4 +2,4 @@
 skip = .git,*.pdf,*.svg,*.bib,*.html
 # ue,lod,ans,lastr,numer,lamda,PrIs -- variable names
 # Hart,Flagg -- name
-ignore-words-list = IST,nd,te,ue,lod,hart,ans,lastr,numer,lamda,flagg,pris,lod,IST,tese
+ignore-words-list = IST,nd,te,ue,lod,hart,ans,lastr,numer,lamda,flagg,pris,lod,IST,tese,shs

fury/actor.py

@@ -7,6 +7,7 @@
 import numpy as np
 
 from fury import layout as lyt
+from fury.actors.odf import sh_odf
 from fury.actors.odf_slicer import OdfSlicerActor
 from fury.actors.peak import PeakActor
 from fury.actors.tensor import (
@@ -4019,3 +4020,67 @@ def uncertainty_cone(
     angles = main_dir_uncertainty(evals, evecs, signal, sigma, b_matrix)
 
     return double_cone(centers, evecs, angles, colors, scales, opacity)
+
+
+def odf(centers, coeffs, sh_basis="descoteaux", scales=1.0, opacity=1.0):
+    """
+    FURY actor for visualizing Orientation Distribution Functions (ODFs) given
+    an array of Spherical Harmonics (SH) coefficients.
+
+    Parameters
+    ----------
+    centers : ndarray(N, 3)
+        ODFs positions.
+    coeffs : (N, M) or (N, 6) or (N, 15) or (N, 28) or (N, 45) or (N, 66) or
+        (N, 91) ndarray.
+        Corresponding SH coefficients for the ODFs.
+    sh_basis: str, optional
+        Type of basis (descoteaux, tournier)
+        'descoteaux' for the default ``descoteaux07`` DIPY basis.
+        'tournier' for the default ``tournier07`` DIPY basis.
+    scales : float or ndarray (N, ), optional
+        ODFs size.
+    opacity : float, optional
+        Takes values from 0 (fully transparent) to 1 (opaque).
+
+    Returns
+    -------
+    odf: Actor
+
+    """
+
+    if not isinstance(centers, np.ndarray):
+        centers = np.array(centers)
+    if centers.ndim == 1:
+        centers = np.array([centers])
+
+    if not isinstance(coeffs, np.ndarray):
+        coeffs = np.array(coeffs)
+    if coeffs.ndim != 2:
+        if coeffs.ndim == 1:
+            coeffs = np.array([coeffs])
+        else:
+            raise ValueError("coeffs should be a 2D array.")
+    if coeffs.shape[0] != centers.shape[0]:
+        raise ValueError(
+            "number of odf glyphs defined does not match with number of centers"
+        )
+
+    coeffs_given = coeffs.shape[-1]
+    degree = int((np.sqrt(8 * coeffs_given + 1) - 3) / 2)
+    if degree % 2 != 0:
+        degree -= 1
+    coeffs = coeffs[:, : int(((degree + 1) * (degree + 2)) / 2)]
+    if not isinstance(scales, np.ndarray):
+        scales = np.array(scales)
+    if scales.size == 1:
+        scales = np.repeat(scales, centers.shape[0])
+    elif scales.size != centers.shape[0]:
+        scales = np.concatenate(
+            (scales, np.ones(centers.shape[0] - scales.shape[0])), axis=None
+        )
+
+    total = np.sum(abs(coeffs), axis=1)
+    coeffs = np.dot(np.diag(1 / total * scales), coeffs) * 1.7
+
+    return sh_odf(centers, coeffs, degree, sh_basis, scales, opacity)