diff --git a/first_exp.py b/first_exp.py new file mode 100644 index 0000000000000000000000000000000000000000..bd0019bb8ff1c3464d213f01d8b29c83b214d655 --- /dev/null +++ b/first_exp.py @@ -0,0 +1,26 @@ +import numpy as np +import matplotlib.pyplot as plt +import pdb + +st = pdb.set_trace + + +import external # imports external.py +import experiment as ex # + + +def my_first_experiment(): + exp = ex.Experiment(lo_freq=5, rx_t=3.125) + tx0_times = np.array([50, 130]) + tx0_amps = np.array([0.5, 0]) + + event_dict = {"tx0": (tx0_times, tx0_amps)} + exp.add_flodict(event_dict) + exp.add_flodict({"rx0_en": (np.array([200, 400]), np.array([1, 0]))}) + + exp.plot_sequence() + plt.show() + +if __name__ == "__main__": + my_first_experiment() + \ No newline at end of file diff --git a/my_first_experiment.py b/my_first_experiment.py deleted file mode 100644 index 30a3356ff0bfb276c5cbde22c0ec3e32eed3af9c..0000000000000000000000000000000000000000 --- a/my_first_experiment.py +++ /dev/null @@ -1,17 +0,0 @@ -import numpy as np -import matplotlib.pyplot as plt -import pdb - -st = pdb.set_trace - - -import external # imports external.py -import experiment as ex # - - -def my_first_experiment(): - exp = ex.Experiment(lo_freq=5, rx_t=3.125) - - -if __name__ == "__main__": - my_first_experiment() \ No newline at end of file diff --git a/second_exp.py b/second_exp.py new file mode 100644 index 0000000000000000000000000000000000000000..a11bad19020a15f42b7a37fa53c90f0974b9644e --- /dev/null +++ b/second_exp.py @@ -0,0 +1,32 @@ +import numpy as np +import matplotlib.pyplot as plt +import pdb + +st = pdb.set_trace + + +import external # imports external.py +import experiment as ex # + + +def my_first_experiment(): + exp = ex.Experiment(lo_freq=5, rx_t=3.125) + + event_dict = { + "tx0": (np.array([50, 130, 200, 360]), np.array([0.5, 0, 0.5j, 0])), + "tx1": (np.array([500, 700]), np.array([0.2, 0])), + "rx0_en": (np.array([400, 800]), np.array([1, 0])), + "rx1_en": (np.array([400, 800]), np.array([1, 0])), + } + exp.add_flodict(event_dict) + exp.plot_sequence() + + rxd, msgs = exp.run() + exp.close_server(only_if_sim=True) + + plt.figure() + plt.plot(np.abs(rxd["rx1"])) + plt.show() +if __name__ == "__main__": + my_first_experiment() + \ No newline at end of file