computation_2.py

import json
import matplotlib.pyplot as plt
import numpy as np

from scipy.optimize import curve_fit

resisting_current_proportional = []
resisting_current_constant = []
x = np.zeros((2, 12))
h = {
    "h": 6,
    "5": 5,
    "4": 4,
    "3": 3,
    "2": 2,
    # "1": 1,
    "l": 0,
}
bike = {"10": 11, "8": 15, "6": 19, "4": 25, "3": 28, "2": 32, "1": 36}
for ht in h.keys():
    if ht == "h":
        for bk in bike.keys():
            with open(f"./calibration_files/resisting_current_load_h{ht}_b{bk}.json", "r") as f:
                data = json.load(f)
                resisting_current_proportional.append(data["a"])
                resisting_current_constant.append(data["b"])
                x[0, len(resisting_current_proportional) - 1] = h[ht]
                x[1, len(resisting_current_proportional) - 1] = bike[bk]

    else:
        with open(f"./calibration_files/resisting_current_load_h{ht}_b10.json", "r") as f:
            data = json.load(f)
            resisting_current_proportional.append(data["a"])
            resisting_current_constant.append(data["b"])
            x[0, len(resisting_current_proportional) - 1] = h[ht]
            x[1, len(resisting_current_proportional) - 1] = bike["10"]


def find_pig(x, coeff_pig, coeff_ht, coeff_cst):
    return coeff_pig / x[1] + coeff_ht * x[0] + coeff_cst


popt_constant = curve_fit(find_pig, x, resisting_current_constant, p0=[-2.53, -0.023, 0.0])
popt_proportional = curve_fit(find_pig, x, resisting_current_proportional, p0=[-2.53, -0.023, 0.0])
print(popt_constant[0])

plt.plot(x[0, :], resisting_current_constant, "o")
plt.plot(x[0, :], find_pig(x, *popt_constant[0]), "o")
plt.show()

plt.plot(x[1, :], resisting_current_constant, "o")
plt.plot(x[1, :], find_pig(x, *popt_constant[0]), "o")
plt.show()

with open("./ergocycleS2M/parameters/hardware_and_security.json", "r") as f:
    hardware_and_security = json.load(f)

hardware_and_security["resisting_current_cst_gear"] = popt_constant[0][0]
hardware_and_security["resisting_current_cst_hometrainer"] = popt_constant[0][1]
hardware_and_security["resisting_current_cst_constant"] = popt_constant[0][2]
hardware_and_security["resisting_current_prop_gear"] = popt_proportional[0][0]
hardware_and_security["resisting_current_prop_hometrainer"] = popt_proportional[0][1]
hardware_and_security["resisting_current_prop_constant"] = popt_proportional[0][2]

# Writing to .json
json_object = json.dumps(hardware_and_security, indent=4)
with open("./ergocycleS2M/parameters/hardware_and_security.json", "w") as outfile:
    outfile.write(json_object)