generate_easing.py

import glob
import os

fnames = glob.glob("easings/*txt")
fnames = list(fnames)
fnames.sort()

ee = []
for i, fname in enumerate(fnames):
    x = []
    y = [0]
    name = os.path.basename(fname).split(".")[0]
    with open(fname, "r") as f:
        for line in f:
            line = line.strip()
            if line == "":
                continue
            nums = line.split()
            if len(nums) != 2:
                continue
            if int(nums[1]) < 0:
                continue
            x.append(int(nums[0]))
            y.append(int(nums[1]))
    ee.append({"x": x, "y": y, "name": name})

for i, ease in enumerate(ee):
    xs = ease["x"]
    ys = ease["y"]
    name = ease["name"]
    print(f"uint8_t ease_{name}(uint16_t v) {{")
    print("\tv=v/8;")
    last_val = 0
    for j, _ in enumerate(xs):
        x = xs[j]
        y = ys[j]
        if y > 255:
            y = 255
        last_val = y
        if j == 0:
            print(f"\tif (v<{x}) {{ return {y}; }}")
        else:
            print(f"\telse if (v<{x}) {{ return  {y}; }}")
    print(f"\t return {last_val};")
    print("}")


import matplotlib.pyplot as plt
import numpy as np

for i, ease in enumerate(ee):
    xs = np.multiply(ease["x"], 8 / 4095)
    ys = ease["y"]
    name = ease["name"]
    plt.plot(xs, ys[1:], alpha=0.4)
    plt.fill_between(xs, ys[1:], alpha=0.2, label=name)
    # plt.gca().add_patch(patch1)
plt.legend(
    loc="upper center", bbox_to_anchor=(0.25, 1.1), ncol=1, fancybox=True, shadow=True
)
plt.xlabel("knob")
plt.ylabel("result")
plt.savefig("easings.png")