kicad: Extend query API

1 files changed, 40 insertions, 0 deletions

diff --git a/gerbonara/cad/kicad/pcb.py b/gerbonara/cad/kicad/pcb.py
index 8a90a7e..22ad23e 100644
--- a/gerbonara/cad/kicad/pcb.py
+++ b/gerbonara/cad/kicad/pcb.py
@@ -5,6 +5,7 @@ Library for handling KiCad's PCB files (`*.kicad_mod`).
 from pathlib import Path
 from dataclasses import field
 from itertools import chain
+import re
 import fnmatch
 
 from .sexp import *
@@ -23,6 +24,12 @@ from ...newstroke import Newstroke
 from ...utils import MM
 
 
+def match_filter(f, value):
+    if isinstance(f, str) and re.fullmatch(f, value):
+        return True
+    return value in f
+
+
 @sexp_type('general')
 class GeneralSection:
     thickness: Named(float) = 1.60
@@ -234,6 +241,39 @@ class Board:
     original_filename: str = None
     _bounding_box: tuple = None
 
+
+    def __after_parse__(self, parent):
+        self.properties = {prop.key: prop.value for prop in self.properties}
+
+        for fp in self.footprints:
+            fp.board = self
+
+    def __before_sexp__(self):
+        self.properties = [Property(key, value) for key, value in self.properties.items()]
+
+    def find_pads(self, net=None):
+        for fp in self.footprints:
+            for pad in fp.pads:
+                if net and not match_filter(net, pad.net.name):
+                    continue
+                yield pad
+
+    def find_footprints(self, value=None, reference=None, name=None, net=None, sheetname=None, sheetfile=None):
+        for fp in self.footprints:
+            if name and not match_filter(name, fp.name):
+                continue
+            if value and not match_filter(value, fp.properties.get('value', '')):
+                continue
+            if reference and not match_filter(reference, fp.properties.get('reference', '')):
+                continue
+            if net and not any(match_filter(net, pad.net.name) for pad in fp.pads):
+                continue
+            if sheetname and not match_filter(sheetname, fp.sheetname):
+                continue
+            if sheetfile and not match_filter(sheetfile, fp.sheetfile):
+                continue
+            yield fp
+
     @property
     def version(self):
         return self._version