mirror of
https://github.com/MaSzyna-EU07/maszyna.git
synced 2026-10-17 09:49:19 +02:00
88 lines
3.2 KiB
Python
88 lines
3.2 KiB
Python
#!/usr/bin/env python3
|
|
"""Writes //$e descriptions into the switch drive templates (scenery/tory/napedy), so the editor places them with a switch.
|
|
|
|
usage: drive_headers.py <data root> [--write] [--zip <file>]
|
|
without --write it only reports what it would write; --zip packs the written templates with their scenery/ paths
|
|
"""
|
|
|
|
import os
|
|
import re
|
|
import sys
|
|
|
|
from editor_headers import body_lines, decode, load, options, pack, write
|
|
|
|
FOLDER = 'scenery/tory/napedy'
|
|
PATTERN = re.compile(r'^(ele1?|mech[12]?)-rz([lp])-([lp])((?:_[a-z]+)*)\.inc$')
|
|
KINDS = {'ele': 'elektryczny', 'mech': 'mechaniczny'}
|
|
DEFAULT = 'ele'
|
|
HEIGHTS = {'ele': 0.05, 'ele1': 0.05}
|
|
|
|
|
|
def examine(path, file):
|
|
m = PATTERN.match(file)
|
|
if m is None:
|
|
return None, 'not a drive of a single switch'
|
|
raw = load(path)
|
|
text, encoding = decode(raw)
|
|
body = '\n'.join(body_lines(text)).lower()
|
|
count = max([int(n) for n in re.findall(r'\(p(\d+)\)', body)] or [0])
|
|
if count != 7 or 'origin (p2) (p3) (p4)' not in body or 'rotate (p6) (p5) (p7)' not in body:
|
|
return None, 'not origin (p2) (p3) (p4) rotate (p6) (p5) (p7)'
|
|
|
|
model, hand, side, extras = m.groups()
|
|
family = model + extras
|
|
hand = 'left' if hand == 'l' else 'right'
|
|
side = 'left' if side == 'l' else 'right'
|
|
kind = KINDS[model.rstrip('12')]
|
|
words = ' '.join(w.replace('_', '') for w in extras.split('_') if w)
|
|
description = 'Napęd %s %s%s' % (kind, model, (' (' + words + ')') if words else '')
|
|
description += ', rozjazd %s, napęd z %s strony' % ('lewy' if hand == 'left' else 'prawy', 'lewej' if side == 'left' else 'prawej')
|
|
header = [
|
|
'version: 1',
|
|
'name: ' + family,
|
|
'category: switch drive',
|
|
'description: "' + description + '"',
|
|
'switch: {hand: %s, drive: %s, height: %s%s}' % (hand, side, HEIGHTS.get(family, 0), ', default: true' if family == DEFAULT else ''),
|
|
'params:',
|
|
' - {id: 1, role: track, label: Switch}',
|
|
' - {id: 2, role: pos.x}',
|
|
' - {id: 3, role: pos.y}',
|
|
' - {id: 4, role: pos.z}',
|
|
' - {id: 5, role: rot.y}',
|
|
' - {id: 6, role: rot.x}',
|
|
' - {id: 7, role: rot.z}',
|
|
]
|
|
return {'raw': raw, 'text': text, 'encoding': encoding, 'header': header, 'family': family, 'hand': hand, 'side': side}, ''
|
|
|
|
|
|
def main():
|
|
chosen = options(sys.argv, __doc__)
|
|
if chosen is None:
|
|
return 1
|
|
data, writing, target = chosen
|
|
folder = os.path.join(data, FOLDER)
|
|
done = []
|
|
skipped = []
|
|
for file in sorted(os.listdir(folder)):
|
|
if not file.endswith('.inc'):
|
|
continue
|
|
info, reason = examine(os.path.join(folder, file), file)
|
|
if info is None:
|
|
skipped.append((file, reason))
|
|
continue
|
|
done.append((file, info))
|
|
for file, info in done:
|
|
print('%-55s %-30s %-5s drive %s' % (file, info['family'], info['hand'], info['side']))
|
|
for file, reason in skipped:
|
|
print('skipped %s: %s' % (file, reason))
|
|
print('%d described, %d skipped' % (len(done), len(skipped)))
|
|
if writing:
|
|
write(data, FOLDER, 'drives', done)
|
|
if target:
|
|
pack(target, FOLDER, done)
|
|
return 0
|
|
|
|
|
|
if __name__ == '__main__':
|
|
sys.exit(main())
|