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


import matplotlib.pyplot as plt



def clean():
    rmtree("meas_values")


def main(param_model: Param_Model):
    sensor = param_model.sensors["mess1"]
    driver = param_model.drivers["treib1"]
    driver.set_config({"speed": 2})
    status = driver.get_status()

    plot_data = param_model.get_plot_data()

    for speed in [1.0, 2.0, 3.0]:
        print(speed)
        driver.set_config({"speed": speed})

        measurement = param_model.new_measurement(name=f"penis_{speed}", writer="CSV")
        measurement.attach_sensor(sensor=sensor, name="mess1")
        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()