def file parser
This commit is contained in:
Stefan Holst committed
1 parent
afb0a64953
commit
153442a10a
1 file changed
+285
@@ -0,0 +1,285 @@
|
|||||||
|
from collections import defaultdict
|
||||||
|
|
||||||
|
from lark import Lark, Transformer, Tree
|
||||||
|
|
||||||
|
from kyupy import readtext
|
||||||
|
|
||||||
|
|
||||||
|
class DefNet:
|
||||||
|
def __init__(self, name):
|
||||||
|
self.name = name
|
||||||
|
self.pins = []
|
||||||
|
|
||||||
|
@property
|
||||||
|
def wires(self):
|
||||||
|
ww = defaultdict(list)
|
||||||
|
[ww[dw.layer].append((int(dw.width), dw.wire_points)) for dw in self.routed if len(dw.wire_points) > 0]
|
||||||
|
return ww
|
||||||
|
|
||||||
|
@property
|
||||||
|
def vias(self):
|
||||||
|
vv = defaultdict(list)
|
||||||
|
[vv[vtype].extend(locs) for dw in self.routed for vtype, locs in dw.vias.items()]
|
||||||
|
return vv
|
||||||
|
|
||||||
|
|
||||||
|
class DefWire:
|
||||||
|
def __init__(self):
|
||||||
|
self.layer = None
|
||||||
|
self.width = None
|
||||||
|
self.points = []
|
||||||
|
|
||||||
|
@property
|
||||||
|
def wire_points(self):
|
||||||
|
start = [self.points[0]]
|
||||||
|
rest = [p for p in self.points[1:] if not isinstance(p[0], str)] # skip over vias
|
||||||
|
return start + rest if len(rest) > 0 else []
|
||||||
|
|
||||||
|
@property
|
||||||
|
def vias(self):
|
||||||
|
vv = defaultdict(list)
|
||||||
|
loc = self.points[0]
|
||||||
|
for p in self.points[1:]:
|
||||||
|
if not isinstance(p[0], str): # new location
|
||||||
|
loc = (loc[0] if p[0] is None else p[0], loc[1] if p[1] is None else p[1]) # if None, keep previous value
|
||||||
|
continue
|
||||||
|
vtype, param = p
|
||||||
|
if isinstance(param, tuple): # expand "DO x BY y STEP xs ys"
|
||||||
|
x_cnt, y_cnt, x_sp, y_sp = param
|
||||||
|
[vv[vtype].append((loc[0] + x*x_sp, loc[1] + y*y_sp, 'N')) for x in range(x_cnt) for y in range(y_cnt)]
|
||||||
|
else:
|
||||||
|
vv[vtype].append((loc[0], loc[1], param or 'N'))
|
||||||
|
return vv
|
||||||
|
|
||||||
|
def __repr__(self):
|
||||||
|
return f'<DefWire {self.layer} {self.width} {self.points}>'
|
||||||
|
|
||||||
|
|
||||||
|
class DefVia:
|
||||||
|
def __init__(self, name):
|
||||||
|
self.name = name
|
||||||
|
self.rowcol = [1, 1]
|
||||||
|
self.cutspacing = [0, 0]
|
||||||
|
|
||||||
|
|
||||||
|
class DefPin:
|
||||||
|
def __init__(self, name):
|
||||||
|
self.name = name
|
||||||
|
self.points = []
|
||||||
|
|
||||||
|
|
||||||
|
class DefFile:
|
||||||
|
def __init__(self):
|
||||||
|
self.rows = []
|
||||||
|
self.tracks = []
|
||||||
|
self.units = []
|
||||||
|
self.vias = {}
|
||||||
|
self.components = {}
|
||||||
|
self.pins = {}
|
||||||
|
self.specialnets = {}
|
||||||
|
self.nets = {}
|
||||||
|
|
||||||
|
|
||||||
|
class DefTransformer(Transformer):
|
||||||
|
def __init__(self): self.def_file = DefFile()
|
||||||
|
def start(self, args): return self.def_file
|
||||||
|
def design(self, args): self.def_file.design = args[0].value
|
||||||
|
def point(self, args): return tuple(int(arg.value) if arg != '*' else None for arg in args)
|
||||||
|
def do_step(self, args): return tuple(map(int, args))
|
||||||
|
def spnet_wires(self, args): return args[0].lower(), args[1:]
|
||||||
|
def net_wires(self, args): return args[0].lower(), args[1:]
|
||||||
|
def sppoints(self, args): return args
|
||||||
|
def points(self, args): return args
|
||||||
|
def net_pin(self, args): return '__pin__', (args[0].value, args[1].value)
|
||||||
|
def net_opt(self, args): return args[0].lower(), args[1].value
|
||||||
|
|
||||||
|
def file_stmt(self, args):
|
||||||
|
value = args[1].value
|
||||||
|
value = value[1:-1] if value[0] == '"' else value
|
||||||
|
setattr(self.def_file, args[0].lower(), value)
|
||||||
|
|
||||||
|
def design_stmt(self, args):
|
||||||
|
stmt = args[0].lower()
|
||||||
|
if stmt == 'units': self.def_file.units.append((args[1].value, args[2].value, int(args[3])))
|
||||||
|
elif stmt == 'diearea': self.def_file.diearea = args[1:]
|
||||||
|
elif stmt == 'row':
|
||||||
|
self.def_file.rows.append((args[1].value, # rowName
|
||||||
|
args[2].value, # siteName
|
||||||
|
(int(args[3]), int(args[4])), # origin x/y
|
||||||
|
args[5].value, # orientation
|
||||||
|
max(args[6][0], args[6][1]), # number of sites
|
||||||
|
max(args[6][2], args[6][3]) # site width
|
||||||
|
))
|
||||||
|
elif stmt == 'tracks':
|
||||||
|
self.def_file.tracks.append((args[1].value, # orientation
|
||||||
|
int(args[2]), # start
|
||||||
|
int(args[3]), # number of tracks
|
||||||
|
int(args[4]), # spacing
|
||||||
|
args[5].value # layer
|
||||||
|
))
|
||||||
|
|
||||||
|
def vias_stmt(self, args):
|
||||||
|
via = DefVia(args[0].value)
|
||||||
|
[setattr(via, opt, val) for opt, val in args[1:]]
|
||||||
|
self.def_file.vias[via.name] = via
|
||||||
|
|
||||||
|
def vias_opt(self, args):
|
||||||
|
opt = args[0].lower()
|
||||||
|
if opt in ['viarule', 'pattern']: val = args[1].value
|
||||||
|
elif opt in ['layers']: val = [arg.value for arg in args[1:]]
|
||||||
|
else: val = [int(arg) for arg in args[1:]]
|
||||||
|
return opt, val
|
||||||
|
|
||||||
|
def comp_stmt(self, args):
|
||||||
|
name = args[0].value
|
||||||
|
kind = args[1].value
|
||||||
|
point = args[2]
|
||||||
|
orientation = args[3].value
|
||||||
|
self.def_file.components[name] = (kind, point, orientation)
|
||||||
|
|
||||||
|
def pins_stmt(self, args):
|
||||||
|
pin = DefPin(args[0].value)
|
||||||
|
[pin.points.append(val) if opt == 'placed' else setattr(pin, opt, val) for opt, val in args[1:]]
|
||||||
|
self.def_file.pins[pin.name] = pin
|
||||||
|
|
||||||
|
def pins_opt(self, args):
|
||||||
|
opt = args[0].lower()
|
||||||
|
if opt in ['net', 'direction', 'use']: val = args[1].value
|
||||||
|
elif opt in ['layer']: val = [args[1].value] + args[2:]
|
||||||
|
elif opt in ['placed']: val = (args[1][0], args[1][1], args[2].value)
|
||||||
|
else: val = []
|
||||||
|
return opt, val
|
||||||
|
|
||||||
|
def spnets_stmt(self, args):
|
||||||
|
dnet = DefNet(args[0].value)
|
||||||
|
for arg in args[1:]:
|
||||||
|
if arg[0] == '__pin__': dnet.pins.append(arg[1])
|
||||||
|
else: setattr(dnet, arg[0], arg[1])
|
||||||
|
self.def_file.specialnets[dnet.name] = dnet
|
||||||
|
|
||||||
|
def nets_stmt(self, args):
|
||||||
|
dnet = DefNet(args[0].value)
|
||||||
|
for arg in args[1:]:
|
||||||
|
if arg[0] == '__pin__': dnet.pins.append(arg[1])
|
||||||
|
else: setattr(dnet, arg[0], arg[1])
|
||||||
|
self.def_file.nets[dnet.name] = dnet
|
||||||
|
|
||||||
|
def spwire(self, args):
|
||||||
|
wire = DefWire()
|
||||||
|
wire.layer = args[0].value
|
||||||
|
wire.width = args[1].value
|
||||||
|
wire.points = args[-1]
|
||||||
|
return wire
|
||||||
|
|
||||||
|
def wire(self, args):
|
||||||
|
wire = DefWire()
|
||||||
|
wire.layer = args[0].value
|
||||||
|
wire.points = args[-1]
|
||||||
|
return wire
|
||||||
|
|
||||||
|
def sppoints_via(self, args):
|
||||||
|
if len(args) == 1: return args[0].value, None
|
||||||
|
else: return args[0].value, args[1]
|
||||||
|
|
||||||
|
def points_via(self, args):
|
||||||
|
if len(args) == 1: return args[0].value, 'N'
|
||||||
|
else: return args[0].value, args[1].value.strip()
|
||||||
|
|
||||||
|
|
||||||
|
GRAMMAR = r"""
|
||||||
|
start: /#[^\n]*/? file_stmt*
|
||||||
|
|
||||||
|
?file_stmt: /VERSION/ ID ";"
|
||||||
|
| /DIVIDERCHAR/ STRING ";"
|
||||||
|
| /BUSBITCHARS/ STRING ";"
|
||||||
|
| design
|
||||||
|
|
||||||
|
design: "DESIGN" ID ";" design_stmt* "END" "DESIGN"
|
||||||
|
|
||||||
|
?design_stmt: /UNITS/ ID ID NUMBER ";"
|
||||||
|
| /DIEAREA/ point+ ";"
|
||||||
|
| /ROW/ ID ID NUMBER NUMBER ID do_step ";"
|
||||||
|
| /TRACKS/ /[XY]/ NUMBER "DO" NUMBER "STEP" NUMBER "LAYER" ID ";"
|
||||||
|
| propdef | vias | nondef | comp | pins | pinprop | spnets | nets
|
||||||
|
|
||||||
|
propdef: "PROPERTYDEFINITIONS" propdef_stmt* "END" "PROPERTYDEFINITIONS"
|
||||||
|
propdef_stmt: /COMPONENTPIN/ ID ID ";"
|
||||||
|
|
||||||
|
vias: "VIAS" NUMBER ";" vias_stmt* "END" "VIAS"
|
||||||
|
vias_stmt: "-" ID vias_opt* ";"
|
||||||
|
vias_opt: "+" /VIARULE/ ID
|
||||||
|
| "+" /CUTSIZE/ NUMBER NUMBER
|
||||||
|
| "+" /LAYERS/ ID ID ID
|
||||||
|
| "+" /CUTSPACING/ NUMBER NUMBER
|
||||||
|
| "+" /ENCLOSURE/ NUMBER NUMBER NUMBER NUMBER
|
||||||
|
| "+" /ROWCOL/ NUMBER NUMBER
|
||||||
|
| "+" /PATTERN/ ID
|
||||||
|
|
||||||
|
nondef: "NONDEFAULTRULES" NUMBER ";" nondef_stmt+ "END" "NONDEFAULTRULES"
|
||||||
|
nondef_stmt: "-" ID ( "+" /HARDSPACING/
|
||||||
|
| "+" /LAYER/ ID "WIDTH" NUMBER "SPACING" NUMBER
|
||||||
|
| "+" /VIA/ ID )* ";"
|
||||||
|
|
||||||
|
comp: "COMPONENTS" NUMBER ";" comp_stmt* "END" "COMPONENTS"
|
||||||
|
comp_stmt: "-" ID ID "+" "PLACED" point ID ";"
|
||||||
|
|
||||||
|
pins: "PINS" NUMBER ";" pins_stmt* "END" "PINS"
|
||||||
|
pins_stmt: "-" ID pins_opt* ";"
|
||||||
|
pins_opt: "+" /NET/ ID
|
||||||
|
| "+" /SPECIAL/
|
||||||
|
| "+" /DIRECTION/ ID
|
||||||
|
| "+" /USE/ ID
|
||||||
|
| "+" /PORT/
|
||||||
|
| "+" /LAYER/ ID point point
|
||||||
|
| "+" /PLACED/ point ID
|
||||||
|
|
||||||
|
pinprop: "PINPROPERTIES" NUMBER ";" pinprop_stmt* "END" "PINPROPERTIES"
|
||||||
|
pinprop_stmt: "-" "PIN" ID "+" "PROPERTY" ID STRING ";"
|
||||||
|
|
||||||
|
spnets: "SPECIALNETS" NUMBER ";" spnets_stmt* "END" "SPECIALNETS"
|
||||||
|
spnets_stmt: "-" ID ( net_pin | net_opt | spnet_wires )* ";"
|
||||||
|
|
||||||
|
spnet_wires: "+" ( /COVER/ | /FIXED/ | /ROUTED/ ) spwire ( "NEW" spwire )*
|
||||||
|
|
||||||
|
spwire: ID NUMBER spwire_opt* sppoints
|
||||||
|
spwire_opt: "+" /SHAPE/ ID
|
||||||
|
| "+" /STYLE/ ID
|
||||||
|
|
||||||
|
sppoints: point ( point | sppoints_via )+
|
||||||
|
sppoints_via: ID do_step?
|
||||||
|
|
||||||
|
nets: "NETS" NUMBER ";" nets_stmt* "END" "NETS"
|
||||||
|
nets_stmt: "-" ID ( net_pin | net_opt | net_wires )* ";"
|
||||||
|
|
||||||
|
net_pin: "(" ID ID ")"
|
||||||
|
net_opt: "+" /USE/ ID
|
||||||
|
| "+" /NONDEFAULTRULE/ ID
|
||||||
|
net_wires: "+" ( /COVER/ | /FIXED/ | /ROUTED/ | /NOSHIELD/ ) wire ( "NEW" wire )*
|
||||||
|
|
||||||
|
wire: ID wire_opt points
|
||||||
|
wire_opt: ( "TAPER" | "TAPERRULE" ID )? ("STYLE" ID)?
|
||||||
|
|
||||||
|
points: point ( point | points_via )+
|
||||||
|
points_via: ID ORIENTATION?
|
||||||
|
|
||||||
|
point: "(" (NUMBER|/\*/) (NUMBER|/\*/) NUMBER? ")"
|
||||||
|
|
||||||
|
do_step: "DO" NUMBER "BY" NUMBER "STEP" (NUMBER|SIGNED_NUMBER) (NUMBER|SIGNED_NUMBER)
|
||||||
|
|
||||||
|
ORIENTATION.2: /F?[NWES]/ WS
|
||||||
|
ID: /[^ \t\f\r\n+][^ \t\f\r\n]*/
|
||||||
|
STRING : "\"" /.*?/s /(?<!\\)(\\\\)*?/ "\""
|
||||||
|
WS: /[ \t\f\r\n]/
|
||||||
|
|
||||||
|
%import common.NUMBER
|
||||||
|
%import common.SIGNED_NUMBER
|
||||||
|
%ignore WS (/#[^\n]*/)?
|
||||||
|
"""
|
||||||
|
|
||||||
|
|
||||||
|
def parse(text):
|
||||||
|
return Lark(GRAMMAR, parser="lalr", transformer=DefTransformer()).parse(text)
|
||||||
|
|
||||||
|
|
||||||
|
def load(file):
|
||||||
|
return parse(readtext(file))
|
||||||
Reference in new issue
Block a user