From b9899826df9fd025d517969a8832b56e2f94e5df Mon Sep 17 00:00:00 2001
From: Daniel Brown <daniel.d.brown@adelaide.edu.au>
Date: Wed, 21 Nov 2018 20:26:07 -0800
Subject: [PATCH] Fixing lens ABCD when setting diopters to zero

---
 pykat/components.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/pykat/components.py b/pykat/components.py
index 5ce90fc..9c3a882 100644
--- a/pykat/components.py
+++ b/pykat/components.py
@@ -1441,7 +1441,10 @@ class lens(Component):
         if self.f.value is not None:
             return ABCD.lens(self.f.value)
         else:
-            return ABCD.lens(1/self.p.value)
+            if self.p.value == 0:
+                return ABCD.lens(np.inf)
+            else:
+                return ABCD.lens(1/self.p.value)
     
     def nodeConnections(self):
         return (
-- 
GitLab