# from SigMod import *
# import matplotlib.pyplot as plt
from GolemShot import GolemShot
import numpy as np

def main():
    shot = GolemShot(0)
    time, dz = shot.getDz()
    np.savetxt('plasma_position.txt', np.array((time, dz)).T, fmt='%f')
    return

# time -= shot['plasma_start']
# plt.plot(time, dz, label='dz_mirnov')
# plt.plot(time, np.zeros(shape=len(dz)), label='dz')
# plt.title("vertical displacement")
# plt.legend()
# plt.show()

if __name__ == '__main__':
    main()