From 6db078cf4696289bb40169398ebe5f3b1f5e87bb Mon Sep 17 00:00:00 2001
From: Paweł Redman <pawel.redman@gmail.com>
Date: Thu, 2 May 2019 18:30:25 +0200
Subject: Transmissivity

---
 src/ui_widgets.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

(limited to 'src/ui_widgets.py')

diff --git a/src/ui_widgets.py b/src/ui_widgets.py
index 48282c4..636f2ed 100644
--- a/src/ui_widgets.py
+++ b/src/ui_widgets.py
@@ -29,8 +29,11 @@ class EllipseWidget(QWidget):
 		self.ellipse = None
 		self.is_used = True
 
+		self.setSizePolicy(QSizePolicy(QSizePolicy.MinimumExpanding, \
+		                               QSizePolicy.MinimumExpanding))
+
 	def minimumSizeHint(self):
-		return QSize(150, 150)
+		return QSize(100, 100)
 
 	def paintEvent(self, event):
 		P = QPainter(self)
@@ -68,6 +71,10 @@ class EllipseWidget(QWidget):
 		if self.state is None:
 			return
 
+		if np.isnan(self.ellipse.a) or np.isnan(self.ellipse.b) \
+		   or np.isnan(self.ellipse.alpha) or np.isnan(self.ellipse.theta):
+			return
+
 		P.resetTransform()
 		r *= 0.88
 		P.translate(cx, cy)
-- 
cgit