from time import sleep
from shutil import rmtree
from CoreLibrary.Param_Model import Param_Model


import matplotlib.pyplot as plt



def clean():
    rmtree("meas_values", ignore_errors=True)


def main(param_model: Param_Model):
    sensor = param_model.devices["mess1"]
    driver = param_model.devices["treib1"]
    driver.set_config({"refresh_rate": 0.2})
    status = driver.get_status()

    plot_data = param_model.get_plot_data()

    for speed in [0.2, 0.5, 1.0]:
        print(speed)
        driver.set_config({"refresh_rate": speed})

        measurement = param_model.new_measurement(name=f"penis_{speed}", writer="CSV")
        measurement.attach_device(device=driver, name="treib1")
        measurement.write_log("ich mag ramen")
        param_model.start_measuring()
        sleep(2)
        param_model.stop_measuring()
    print("main exited")

    plt.plot(plot_data.data["time"], plot_data.data["mess1"])
    plt.show()


if __name__ == '__main__':
    print("HEADLESS RUN")
    pd = Param_Model()
    clean()
    main(pd)
    pd.exit()