WTS/game/scripts/doll/cum.rpy

108 lines
3.3 KiB
Plaintext

init python:
class DollCum(DollMethods):
layer_types = {
"skin": 0,
"cum": 100,
}
layer_modifiers = {
"zorder": None,
}
def __init__(self, obj):
self.char = obj
self._cum = {k: None for k in {"hair", "face", "breasts", "body", "crotch", "pussy", "legs"}}
self._hash = None
def generate_hash(self):
salt = str( [self.char.name, self.char.pose, sorted(list(self._cum.items()))] )
return hash(salt)
def set_cum(self, *args, **kwargs):
if args:
self._cum = {k: args[0] for k in self._cum}
self._cum.update(kwargs)
self.is_stale()
@functools.cache
def get_layers(self, hash):
cum = self._cum
extensions = self.extensions
types = self.layer_types
modifiers = self.layer_modifiers
layers = {}
for part, name in cum.items():
if name is None:
continue
path = os.path.join("characters", self.char.name, self.char.pose, "cum", part, name)
for f in renpy.list_files():
fp, fn = os.path.split(f)
fn, ext = os.path.splitext(fn)
if not fp == path or not ext in extensions:
continue
ltype, *tails = fn.rsplit("_")
if not ltype in types:
print("Invalid layer type for file: {}".format(f))
continue
zorder = types.get(ltype)
if tails:
lmodifier, *tails = tails
if not lmodifier in modifiers:
print("Invalid modifier for file: {}".format(f))
continue
zorder_mod = modifiers.get(lmodifier)
zorder = (zorder + int(zorder_mod))
layers.setdefault(" ".join([part, name, ltype, lmodifier]), [f, zorder])
else:
layers.setdefault(" ".join([part, name, ltype]), [f, zorder])
return layers
@functools.cache
def build_image(self, hash, matrix=None):
if matrix is None:
matrix = self.char.body.hue
processors = {
"skin": lambda file: Transform(file, matrixcolor=matrix),
"default": lambda file: Image(file),
}
layers = self.get_layers(hash)
sprites = []
for identifier, (file, zorder) in layers.items():
cum_type, name, ltype = identifier.rsplit(" ")
processor = processors.get(identifier, processors["default"])
processed_file = processor(file)
sprites.append((identifier, processed_file, zorder))
return sprites
@property
def image(self):
if not renpy.is_skipping() and self.is_stale():
hash = self._hash
sprites = self.build_image(hash)
sprites.sort(key=itemgetter(2))
sprites = [x[1] for x in sprites]
self._image = Fixed(*sprites, fit_first=True)
return self._image