import json
import numpy as np
import matplotlib.pyplot as plt

colorList = json.load(open('color/config.json','r'))["color"]

import csv

with open('data/passenger.csv', 'r') as f:
    reader = csv.reader(f)
    header = next(reader)
    data = [row for row in reader]

data = np.array(data).astype(float).T
data[0]=data[0].astype(int)

bar_width = 0.35


plt.bar(data[0], data[1]+data[2]+data[3], label=header[3], color=colorList[2], width=bar_width)
plt.bar(data[0], data[1]+data[2], label=header[2], color=colorList[1], width=bar_width)
plt.bar(data[0], data[1], label=header[1], color=colorList[0], width=bar_width)
# plt.plot(data[0],data[4], label=header[4], color=colorList[3])
plt.legend()
# plt.xlabel('Year')
plt.ylabel('Passenger')
plt.xticks(data[0],rotation=45)
ax2=plt.twinx()
ax2.plot(data[0],data[5], label=header[5], color=colorList[4])
ax2.set_ylabel('Revenue')
plt.legend(loc='upper right')
plt.title('Passenger and Revenue Yearly')
plt.savefig('result/passenger-and-revenue.png',dpi=1024)
plt.show()