Add methods to token class in addons (danmar#4320)

pfultz2 · web-flow · commit 754250cd5770 · 2022-07-30T11:09:22.000+02:00
diff --git a/addons/cppcheckdata.py b/addons/cppcheckdata.py
@@ -437,6 +437,45 @@ def isUnaryOp(self, op):
     def isBinaryOp(self):
         return self.astOperand1 and self.astOperand2
 
+    def forward(self, end=None):
+        token = self
+        while token and token != end:
+            yield token
+            token = token.next
+
+    def backward(self, start=None):
+        token = self
+        while token and token != start:
+            yield token
+            token = token.previous
+
+    def astParents(self):
+        token = self
+        while token and token.astParent:
+            token = token.astParent
+            yield token
+
+    def astTop(self):
+        top = None
+        for parent in self.astParents():
+            top = parent
+        return top
+
+    def tokAt(self, n):
+        tl = self.forward()
+        if n < 0:
+            tl = self.backward()
+            n = -n
+        for i, t in enumerate(tl):
+            if i == n:
+                return t
+
+    def linkAt(self, n):
+        token = self.tokAt(n)
+        if token:
+            return token.link
+        return None
+
 class Scope:
     """
     Scope. Information about global scope, function scopes, class scopes, inner scopes, etc.
