WTS/game/scripts/cds.rpy
2024-03-30 18:04:49 +00:00

142 lines
4.3 KiB
Plaintext

python early hide:
import inspect
__register_params = frozenset(inspect.signature(renpy.register_statement).parameters).difference({"name", "parse"})
def register_decorator(cls):
"""
A class decorator which registers a new statement.
The name of the statement will be the class name unless a `name` class attribute is present, which should be a string.
The `parse` method should be a static method that returns an object which will be passed to the other methods as `self`.
Returning an instance of the class is disabled for now.
"""
# security
def raiser(*args, **kwargs):
raise TypeError("Returning an instance of the class is disabled")
cls.__init__ = raiser
name = getattr(cls, "name", cls.__name__)
parse = getattr(cls, "parse", cls) # left in
renpy.register_statement(name,
parse=parse,
**{k:getattr(cls, k) for k in __register_params.intersection(vars(cls))})
return cls
@register_decorator
class dynamic:
block = "possible"
@staticmethod
def parse(l):
rv = {}
def parse_simple(ll):
target = ll.require(ll.name, "variable name")
if target in rv:
ll.error(f"Variable {target} already set in the same dynamic block")
ll.require("=", "equals sign")
expression = ll.simple_expression()
ll.expect_eol()
rv[target] = expression
if l.match(":"):
l.expect_block("dynamic block")
l.expect_eol()
ll = l.subblock_lexer()
while ll.advance():
parse_simple(ll)
else:
parse_simple(l)
return rv
def execute(self):
evaled = {n: eval(e) for n, e in self.items()}
renpy.dynamic(**evaled)
@register_decorator
class chibi:
@staticmethod
def parse(l):
who = l.simple_expression()
action = l.simple_expression()
return (who, action)
def execute(self):
print(f"{self}")
who, action = self
func = eval(f"{who}_chibi.{action}")
# print(f"Execution: {who} {action}")
def lint(self):
who, action = self
try:
chibi = eval(f"{who}_chibi")
except Exception:
renpy.error(f"Character chibi not defined: {who}")
def predict(self):
who, action = self
chibi = eval(f"{who}_chibi")
doll = eval(f"{who}")
layers = (
l[0] for pose in chibi.poses.keys()
for k in doll.states.values() if k[0] and k[2]
for l in k[0].get_layers(k[0]._hash, subpath=posixpath.join("chibi", pose)).values()
)
return layers
@register_decorator
class random:
block = True
predict_all = True
@staticmethod
def parse(l):
l.require(":")
l.expect_eol()
ll = l.subblock_lexer()
blocks = []
while ll.advance():
with ll.catch_error():
weight = 1.0
condition = "True"
if ll.keyword("block"):
ll.expect_block("block")
block = ll.subblock_lexer().renpy_block()
if ll.keyword("weight"):
weight = float(ll.require(ll.float))
if ll.keyword("if"):
ll.expect_block("if block")
condition = ll.require(ll.python_expression)
else:
block = ll.renpy_statement()
blocks.append((block, weight, condition))
return {"blocks": blocks}
def next(self):
blocks = [(block, weight) for block, weight, condition in self["blocks"] if eval(condition)]
total_weight = sum(weight for _, weight in blocks)
n = renpy.random.random() * total_weight
for block, weight in blocks:
if n <= weight:
break
else:
n -= weight
return block