From 2803d1bc70cb3286c2b718c9f07afcd56bf139bc Mon Sep 17 00:00:00 2001
From: Sean Leavey <gpg@attackllama.com>
Date: Fri, 1 Nov 2019 21:56:31 +0100
Subject: [PATCH] Add component position convenience method

---
 finesse/tracing/tracesolution.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/finesse/tracing/tracesolution.py b/finesse/tracing/tracesolution.py
index 1c8a7985..10575041 100644
--- a/finesse/tracing/tracesolution.py
+++ b/finesse/tracing/tracesolution.py
@@ -155,6 +155,12 @@ class BaseTraceSolution(metaclass=abc.ABCMeta):
             qys[node] = qy
         return qys
 
+    @property
+    def component_positions(self):
+        return [(comp, self._data[comp]['z'])
+                for comp in self._data['components']
+                if not self._data[comp]['is_space']]
+
     @abc.abstractmethod
     def plot(self, filename=None, show=True, w_scale=SI.MILLI, markers=None):
         raise NotImplementedError
-- 
GitLab