Differential D2849 Diff 12579 source/tools/entity/scriptlib/__init__.py

Changeset View

Standalone View

source/tools/entity/scriptlib/init.py

This file was added.

				from collections import Counter
				from decimal import Decimal
				from re import split
				from sys import stderr
				from xml.etree import ElementTree


				def warn(s):
				print(f"warn: {s}", file=stderr)


				class SimulTemplateEntity:
				def __init__(self, vfs_root):
				self.vfs_root = vfs_root

				def get_file(self, vfs_path, mod):
				return (self.vfs_root / mod / vfs_path).with_suffix('.xml')

				def get_main_mod(self, vfs_path, mods):
				for mod in mods:
				fp = self.get_file(vfs_path, mod)
				if fp.exists():
				main_mod = mod
				break
				else:
				# default to first mod
				# it should then not exist
				# it will raise an exception when trying to read it
				main_mod = mods[0]
				return main_mod

				def apply_layer(self, base_tag, tag):
				"""
				apply tag layer to base_tag
				"""
				if tag.get('datatype') == 'tokens':
				base_tokens = split(r'\s+', base_tag.text or '')
				tokens = split(r'\s+', tag.text or '')
				final_tokens = base_tokens.copy()
				for token in tokens:
				if token.startswith('-'):
				token_to_remove = token[1:]
				if token_to_remove in final_tokens:
				final_tokens.remove(token_to_remove)
				elif token not in final_tokens:
				final_tokens.append(token)
				base_tag.text = ' '.join(final_tokens)
				elif tag.get('op'):
				op = tag.get('op')
				op1 = Decimal(base_tag.text or '0')
				op2 = Decimal(tag.text or '0')
				if op == 'add':
				base_tag.text = str(op1 + op2)
				elif op == 'mul':
				base_tag.text = str(op1 * op2)
				StanUnsubmitted Not Done Inline Actions Apparently we also support mul_round can you add it? (it's mul + rounding) elif op == 'mul_round ': base_tag.text = str(round(op1 * op2)) Stan: Apparently we also support mul_round can you add it? (it's mul + rounding) ```lang=python…
				else:
				raise ValueError(f"Invalid operator '{op}'")
				else:
				base_tag.text = tag.text
				for child in tag:
				base_child = base_tag.find(child.tag)
				if 'disable' in child.attrib:
				if base_child is not None:
				base_tag.remove(base_child)
				else:
				if 'replace' in child.attrib and base_child is not None:
				base_tag.remove(base_child)
				if base_child is None:
				base_child = ElementTree.Element(child.tag)
				base_tag.append(base_child)
				self.apply_layer(base_child, child)
				if 'replace' in base_child.attrib:
				del base_child.attrib['replace']

				def load_inherited(self, base_path, vfs_path, mods):
				"""
				vfs_path should be relative to base_path in a mod
				"""
				main_mod = self.get_main_mod(base_path / vfs_path, mods)
				fp = self.get_file(base_path / vfs_path, main_mod)
				layer = ElementTree.parse(fp).getroot()
				for el in layer.iter():
				children = [x.tag for x in el]
				duplicates = [x for x, c in Counter(children).items() if c > 1]
				if duplicates:
				for dup in duplicates:
				warn(f"Duplicate child node '{dup}' in tag {el.tag} of {fp}")
				if layer.get('parent'):
				parent = self.load_inherited(base_path, layer.get('parent'), mods)
				self.apply_layer(parent, layer)
				return parent
				else:
				return layer


				def find_files(vfs_root, mods, vfs_path, *ext_list):
				"""
				returns a list of 2-size tuple with:
				- Path relative to the mod base
				- full Path
				"""
				full_exts = ['.' + ext for ext in ext_list]

				def find_recursive(dp, base):
				"""(relative Path, full Path) generator"""
				if dp.is_dir():
				if dp.name != '.svn' and not dp.name.endswith('~'):
				for fp in dp.iterdir():
				yield from find_recursive(fp, base)
				elif dp.suffix in full_exts:
				relative_file_path = dp.relative_to(base)
				yield (relative_file_path, dp.resolve())
				return [(rp, fp) for mod in mods for (rp, fp) in find_recursive(vfs_root / mod / vfs_path, vfs_root / mod)]