From 05d8e1787cac9852710c9d48f7dfeee3d11de9df Mon Sep 17 00:00:00 2001
From: sldesnoo-Delft <s.l.desnoo@tudelft.nl>
Date: Wed, 16 Nov 2022 10:14:41 +0100
Subject: [PATCH] Fixed setting voltage offset with +=
---
pulse_lib/segments/segment_base.py | 17 +++++++++++++++++
1 file changed, 17 insertions(+)
diff --git a/pulse_lib/segments/segment_base.py b/pulse_lib/segments/segment_base.py
index 7a1e16ca..e4e373f3 100644
--- a/pulse_lib/segments/segment_base.py
+++ b/pulse_lib/segments/segment_base.py
@@ -111,9 +111,26 @@ class segment_base():
return new_segment
+ def __iadd__(self, other):
+ '''
+ define addition operator for segment_single
+ '''
+ if isinstance(other, segment_base):
+ self.data = self.data + other.data
+
+ elif type(other) == int or type(other) == float:
+ self.data += other
+ else:
+ raise TypeError("Please add up segment_single type or a number ")
+
+ return self
+
def __sub__(self, other):
return self.__add__(other*-1)
+ def __isub__(self, other):
+ return self.__iadd__(other*-1)
+
def __mul__(self, other):
'''
muliplication operator for segment_single
--
GitLab