17
0
mirror of https://github.com/MaSzyna-EU07/maszyna.git synced 2026-10-20 22:14:07 +02:00
Files
maszyna/tools/editor/turntable_headers.py
maj00r 9474293af7 fix(editor): signals move along their track again, and the editor finds the descriptions of the templates beside the game data
- a press on the mast of a selected signal grabbed the gizmo, whose up axis runs along the mast, and lifted the signal
  instead of moving it along the track; the gizmo gives way over a signal and while one is dragged
- dragged beside its track, off every track, the signal follows the nearest place of the track it stands by instead of
  staying put
- a drop away from any track, and a signal in a file the editor doesn't change, say so instead of doing nothing
- a template without a //$e description of its own takes it from editor/templates/<its path in the scenery folder>; the
  header scripts write there with --sidecar, leaving the templates alone, so a check of the game files by the store
  doesn't take the descriptions away (the descriptions written into the templates on 8.10 and 11.10 were gone, and with
  them every signal and drive the editor knew)
2026-10-18 02:08:19 +02:00

105 lines
4.2 KiB
Python
Executable File

#!/usr/bin/env python3
"""Writes //$e descriptions into the turntable templates of the scenery, so the editor places them as turntables.
usage: turntable_headers.py <data root> [--write] [--zip <file>] [--sidecar]
without --write it only reports what it would write; --zip packs every described turntable template with its scenery/ path
"""
import os
import re
import sys
from editor_headers import LINEMARK, body_lines, decode, load, options, pack, sidecar, write
FOLDER = 'scenery'
ROTATION = re.compile(r'^event\s+\S*tor(\d+)rot\s+animation\s+\S+\s+\S+\s+rotate\s+\S+\s+\S+\s+\S+\s+(\S+)', re.I | re.M)
MODEL = re.compile(r'model\s+0\s+0\s+0\s+0\s+\S*?obrotnica(\d+)m\.t3d\s+\(p6\)', re.I)
def examine(path, relative):
raw = load(path)
if re.search(rb'tor\d+rot\s+animation', raw, re.I) is None:
return None, None
text, encoding = decode(raw)
marks = [line for line in text.splitlines() if line.startswith(LINEMARK)]
if any('category: turntable' in line for line in marks):
return {'raw': raw, 'header': None, 'described': True}, ''
body = '\n'.join(body_lines(text))
lower = body.lower()
if 'origin (p2) (p3) (p4)' not in lower or 'rotate 0 (p5) 0' not in lower:
return None, 'not origin (p2) (p3) (p4) rotate 0 (p5) 0'
model = MODEL.search(body)
if model is None:
return None, 'no obrotnica<length>m.t3d model with surface (p6)'
positions = {}
for number, angle in ROTATION.findall(body):
positions.setdefault(int(number), float(angle))
if not positions:
return None, 'no positions'
count = max(int(n) for n in re.findall(r'\(p(\d+)\)', lower))
if count != 6:
return None, 'uses %d parameters, not 6' % count
length = int(model.group(1))
angles = [positions[n] for n in sorted(positions)]
folder, file = os.path.split(relative)
name = 'obrotnica %d m, %d %s' % (length, len(angles), 'stałe położenia' if len(angles) < 5 else 'stałych położeń')
if folder != 'eng':
name += ' (%s)' % os.path.splitext(file)[0]
description = 'Obrotnica %d m o stałych położeniach mostu: %s' % (length, ' '.join('%g' % a for a in angles))
header = [
'version: 1',
'name: ' + name,
'category: turntable',
'description: "' + description + '"',
'turntable: {length: %d}' % length,
'params:',
' - {id: 1, role: track, label: Name}',
' - {id: 2, role: pos.x}',
' - {id: 3, role: pos.y}',
' - {id: 4, role: pos.z}',
' - {id: 5, role: rot.y}',
' - {id: 6, label: Surface, default: eng/most1-jezdniablach}',
]
return {'raw': raw, 'text': text, 'encoding': encoding, 'header': header, 'described': False, 'length': length, 'angles': angles}, ''
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 root, _, files in os.walk(folder):
for file in sorted(files):
if not file.lower().endswith('.inc'):
continue
relative = os.path.relpath(os.path.join(root, file), folder).replace(os.sep, '/')
info, reason = examine(os.path.join(root, file), relative)
if info is not None:
done.append((relative, info))
elif reason:
skipped.append((relative, reason))
done.sort()
for file, info in done:
if info['described']:
print('%-40s already described' % file)
else:
print('%-40s %d m, positions %s' % (file, info['length'], ' '.join('%g' % a for a in info['angles'])))
for file, reason in skipped:
print('skipped %s: %s' % (file, reason))
fresh = [(file, info) for file, info in done if not info['described']]
print('%d described now, %d already described, %d skipped' % (len(fresh), len(done) - len(fresh), len(skipped)))
if writing and fresh:
write(data, FOLDER, 'turntables', fresh)
if target:
pack(target, FOLDER, done)
if '--sidecar' in sys.argv:
sidecar(data, FOLDER, done)
return 0
if __name__ == '__main__':
sys.exit(main())