ActiveClass: make it iterable

Author: PMeira

dss/dss_capi_gr/IActiveClass.py

@@ -27,6 +27,12 @@ def Count(self):
     def __len__(self):
         return self._lib.ActiveClass_Get_Count()
 
+    def __iter__(self):
+        n = self.First
+        while n:
+            yield self
+            n = self.Next
+
     @property
     def First(self):
         '''(read-only) Sets first element in the active class to be the active DSS object. If object is a CktElement, ActiveCktELment also points to this element. Returns 0 if none.'''

dss/dss_capi_ir/IActiveClass.py

@@ -27,6 +27,12 @@ def Count(self):
     def __len__(self):
         return self._lib.ActiveClass_Get_Count()
 
+    def __iter__(self):
+        n = self.First
+        while n:
+            yield self
+            n = self.Next
+
     @property
     def First(self):
         '''(read-only) Sets first element in the active class to be the active DSS object. If object is a CktElement, ActiveCktELment also points to this element. Returns 0 if none.'''