From 2b798ba44e50f3cc95935791d5e4ab8122d490d1 Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Thu, 2 Jan 2020 16:08:48 +0100
Subject: [PATCH] Change interface of eigenvalues_3d
---
src/pystencils_reco/vesselness.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/pystencils_reco/vesselness.py b/src/pystencils_reco/vesselness.py
index 1260957..2afaf38 100644
--- a/src/pystencils_reco/vesselness.py
+++ b/src/pystencils_reco/vesselness.py
@@ -29,7 +29,7 @@ use_complex_sqrt = ReplaceOptim(
@crazy
-def eigenvalues_3d(eigenvaluefield: {'index_dimensions': 1}, xx, xy, xz, yy, yz, zz):
+def eigenvalues_3d(eig1, eig2, eig3, xx, xy, xz, yy, yz, zz):
H = sympy.Matrix([[xx.center, xy.center, xz.center],
[xy.center, yy.center, yz.center],
@@ -39,7 +39,7 @@ def eigenvalues_3d(eigenvaluefield: {'index_dimensions': 1}, xx, xy, xz, yy, yz,
eigenvalues = list(H.eigenvals())
assignments = pystencils.AssignmentCollection({
- eigenvaluefield.center(i): sympy.re(eigenvalues[i]) for i in range(3)
+ [eig1.center, eig2.center, eig3.center][i]: sympy.re(eigenvalues[i]) for i in range(3)
})
class complex_symbol_generator():
--
GitLab