#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import matplotlib.pyplot as plt
import numpy as np

DATA_PATH = '/home/seth_poh/Dropbox/telecom_qkd/papers/parallel_traffic/figures/transceiver_spectrums/'

ALICE_SPECTRUM_DATA_FILENAME = "media_convertor_single_photon_spectrum.dat";
#BOB_SPECTRUM_DATA_FILENAME = "bob_media_convertor.dat";
#BOB_SPECTRUM_DATA_FILENAME = "bob_media_convertor_2.dat";

x_alice = []
y_alice = []

x_bob = []
y_bob = []

with open(DATA_PATH + ALICE_SPECTRUM_DATA_FILENAME, 'r') as f_alice:
	lines = f_alice.readlines()
	x_alice = np.array([float(line.split()[0]) for line in lines])
	y_alice = np.array([float(line.split()[1]) for line in lines])

# with open(DATA_PATH + BOB_SPECTRUM_DATA_FILENAME, 'r') as f_bob:
# 	lines = f_bob.readlines()
# 	x_bob = np.array([float(line.split()[0]) for line in lines])
# 	y_bob = np.array([float(line.split()[1]) for line in lines])

plt.plot(x_alice, y_alice, 'r', label='Alice')
#plt.plot(x_bob, y_bob/y_bob.max(), 'b', label='Bob')
plt.title('')
plt.xlabel('Wavelength (nm)')
plt.ylabel('Counts (s$^{-1}$)')
plt.grid()

#plt.legend(loc="upper right")
plt.xlim(1000, 2000)
splt.ylim(0, 2e6)

plt.ticklabel_format(axis="y", style="sci", scilimits=(0,0))

#plt.savefig(DATA_PATH + 'transceiver_spectrums.eps', format='eps')

plt.show()