pykat/components.py

@@ -340,6 +340,10 @@ class AbstractMirrorComponent(Component):
         if T is not None: self.T.value = T
         if L is not None: self.L.value = L
         
+        # check energy conservation
+        if self.R.value + self.T.value + self.L.value != 1:
+            raise pkex.BasePyKatException("The specified R, T and L values (%f, %f and %f) violate energy conservation" % (self.R.value, self.T.value, self.L.value))
+    
     @property
     def z(self): return self.__z
     @property

pykat/finesse.py

@@ -81,7 +81,7 @@ from collections import namedtuple, OrderedDict
 
 from pykat.node_network import NodeNetwork
 from pykat.detectors import BaseDetector as Detector
-from pykat.components import Component
+from pykat.components import Component, mirror, beamSplitter, space
 from pykat.commands import Command, xaxis
 from pykat.SIfloat import *
 from pykat.param import Param, AttrParam
@@ -809,6 +809,21 @@ class kat(object):
 
         return tuple(items)
     
+    @property
+    def mirrors(self):
+        """Returns a collection of the mirrors defined within the kat object."""
+        return self.getAll(mirror)
+
+    @property
+    def beamSplitters(self):
+        """Returns a collection of the beam splitters defined within the kat object."""
+        return self.getAll(beamSplitter)
+
+    @property
+    def spaces(self):
+        """Returns a collection of the spaces defined within the kat object."""
+        return self.getAll(space)
+        
     def __deepcopy__(self, memo):
         """
         When deep copying a kat object we need to take into account