1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
|
from itertools import zip_longest
import re
from ..cad.kicad.sexp import build_sexp
from ..cad.kicad.sexp_mapper import sexp
from ..cad.kicad.symbols import Library
from .utils import tmpfile
def test_parse(kicad_library_file):
Library.open(kicad_library_file)
def test_round_trip(kicad_library_file, tmpfile):
print('========== Stage 1 load ==========')
orig_lib = Library.open(kicad_library_file)
print('========== Stage 1 save ==========')
stage1_sexp = build_sexp(orig_lib.sexp())
print('========== Stage 2 load ==========')
reparsed_lib = Library.parse(stage1_sexp)
print('========== Stage 2 save ==========')
stage2_sexp = build_sexp(reparsed_lib.sexp())
print('========== Checks ==========')
for stage1, stage2 in zip_longest(stage1_sexp.splitlines(), stage2_sexp.splitlines()):
assert stage1 == stage2
original = re.sub(r'\(', '\n(', re.sub(r'\s+', ' ', kicad_library_file.read_text()))
original = re.sub(r'\) \)', '))', original)
original = re.sub(r'\) \)', '))', original)
original = re.sub(r'\) \)', '))', original)
original = re.sub(r'\) \)', '))', original)
tmpfile('Processed original', '.kicad_sym').write_text(original)
stage1 = re.sub(r'\(', '\n(', re.sub(r'\s+', ' ', stage1_sexp))
tmpfile('Processed stage 1 output', '.kicad_sym').write_text(stage1)
original_lines, stage1_lines = original.splitlines(), stage1.splitlines()
i, j = 0, 0
while i < len(original_lines) and j < len(stage1_lines):
original, stage1 = original_lines[i], stage1_lines[j]
if original.startswith('(version'):
i, j = i+1, j+1
continue
original, stage1 = original.strip(), stage1.strip()
if original != stage1:
if any(original.startswith(f'({foo}') for foo in ['arc', 'circle', 'rectangle', 'polyline', 'text']):
# These files have symbols with graphic primitives in non-standard order
i, j = i+1, j+1
return
# Some symbol files contain ints where floats should be.
# For instance, there is some disagreement as to whether rotation angles are ints or floats, and the spec doesn't say.
FLOAT_INT_ISSUES = ['offset', 'at', 'width', 'xy', 'start', 'mid', 'end', 'center', 'length']
if any(original.startswith(f'({name}') and stage1.startswith(f'({name}') for name in FLOAT_INT_ISSUES):
fix_floats = lambda s: re.sub(r'\.0+(\W)', r'\1', s)
original, stage1 = fix_floats(original), fix_floats(stage1)
if original.startswith('(symbol') and stage1.startswith('(symbol'):
# Re-export can change symbol order. This is ok.
i, j = i+1, j+1
continue
# KiCad changed some flags from a bare flag to a named yes/no flag, which emits one more line here.
NOW_NAMED = ['hide', 'bold', 'italic']
if any(f'{name} yes' in stage1 for name in NOW_NAMED):
j += 1
continue
if any(name in original or name in stage1 for name in NOW_NAMED):
# KiCad changed the position of some flags inside text effects between versions.
i, j = i+1, j+1
continue
if 'generator' in original:
# KiCad changed the generator field from an atom to a str
i, j = i+1, j+1
continue
NEW_FIELDS = [
'generator_version',
'exclude_from_sim',
]
if any(field in stage1 for field in NEW_FIELDS):
# New field, skip only on right (new) side
j += 1
continue
assert original == stage1
i, j = i+1, j+1
|
