Modifier_class.py

import os, sys, math, time, copy, pygame, random, colorsys

import visual_functions as visual

class Modifier:

    def __init__(self, value, visual):
        self.value = self.default = value[0]
        self.scaling = {"type":     value[1],
                        "amount":   value[2],
                        "bounds":   value[3]}
        self.name, self.unit, self.dp = visual

    def bump(self, bump_type, bump_magnitude):
        amount = self.scaling["amount"][bump_magnitude]
        identity = (1,-1)[bump_type == "dec"]

        if self.scaling["type"] == "lin":
            self.value += amount * identity
        elif self.scaling["type"] == "geo":
            self.value *= (1+amount) ** identity
        elif self.scaling["type"] == "mod":
            modulo = self.scaling["bounds"][1] - self.scaling["bounds"][0] + 1
            self.value = (self.value + amount * identity) % modulo

        if self.scaling["bounds"][0] != None:
            self.value = max(self.scaling["bounds"][0], self.value)
        elif self.scaling["bounds"][1] != None:
            self.value = min(self.scaling["bounds"][1], self.value)

    def display(self, P_xp, P_yp, marked=False):
        marker = ("", ">> ")[marked]
        num_value = str(round(self.value, self.dp)) + self.unit
        visual.draw_text(marker + "{}: {}".format(self.name, num_value),
                         (P_xp, P_yp), "bottomright")

    def default(self):
        self.value = self.default