#!/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 = "alice_media_convertor.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/y_alice.max(), 'r', label='Alice')
plt.plot(x_bob, y_bob/y_bob.max(), 'b', label='Bob')
plt.title('')
plt.xlabel('Wavelength (nm)')
plt.ylabel('Intensity (Normalized Units)')
plt.grid()

plt.legend(loc="upper right")
plt.xlim(1550.5, 1553.0)
plt.ylim(0, 1.05)

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

plt.show()