# %%
import matplotlib.pyplot as plt
import RigolWFM.wfm as rigol

# %%

hhpion = 'C:/Users/Joschka/Desktop/Coil_Data/New/hhpion.wfm'
hhpioff = 'C:/Users/Joschka/Desktop/Coil_Data/New/hhpioff.wfm'
hhon = 'C:/Users/Joschka/Desktop/Coil_Data/New/hhon.wfm'
hhoff = 'C:/Users/Joschka/Desktop/Coil_Data/New/hhoff.wfm'

scope = 'DS1104Z-S'

# %%
hhpion = rigol.Wfm.from_file(hhpion, scope)
hhpion.plot()
plt.show()
# %%
hhpioff = rigol.Wfm.from_file(hhpioff, scope)
hhpioff.plot()
plt.show()

# %%
hhoff = rigol.Wfm.from_file(hhoff, scope)
hhoff.plot()
plt.show()

# %%
hhon = rigol.Wfm.from_file(hhon, scope)
hhon.plot()
plt.show()


# %%
print(w.channels[0].times)
print(w.channels[0].volts)

# %%

ahhpion = 'C:/Users/Joschka/Desktop/Coil_Data/New/ahhpion.wfm'
ahhpioff = 'C:/Users/Joschka/Desktop/Coil_Data/New/ahhpioff.wfm'
ahhon = 'C:/Users/Joschka/Desktop/Coil_Data/New/ahhon.wfm'
ahhoff = 'C:/Users/Joschka/Desktop/Coil_Data/New/ahhoff.wfm'

scope = 'DS1104Z-S'

# %%
ahhpion = rigol.Wfm.from_file(ahhpion, scope)
ahhpion.plot()
plt.show()
# %%
ahhpioff = rigol.Wfm.from_file(ahhpioff, scope)
ahhpioff.plot()
plt.show()

# %%
ahhoff = rigol.Wfm.from_file(ahhoff, scope)
ahhoff.plot()
plt.show()

# %%
ahhon = rigol.Wfm.from_file(ahhon, scope)
ahhon.plot()
plt.show()