import matplotlib.pyplot as plt
import numpy as np

# WMCC өгөгдөл
wmcc_x = np.array([0, 500, 500, 800, 800, 1200]) # Мянган доллараар
wmcc_y = np.array([13.8, 13.8, 15.4, 15.4, 16.2, 16.2]) # Процентоор

# IOS өгөгдөл (Жишээ болгон зохиомол төслүүд авлаа)
ios_x = np.array([0, 300, 300, 600, 600, 900, 900, 1100]) # Нийт хөрөнгө оруулалт
ios_y = np.array([19.0, 19.0, 16.0, 16.0, 14.5, 14.5, 12.0, 12.0]) # Төслүүдийн IRR

# График байгуулах
plt.figure(figsize=(12, 7))

# WMCC муруйг зурах (цэнхэр өнгөөр)
plt.plot(wmcc_x, wmcc_y, color='blue', label='WMCC (Капиталын Жинлэсэн Ахиуц Өртөг)', linewidth=2.5)
plt.fill_between(wmcc_x, wmcc_y, step="pre", alpha=0.1, color='blue') # Будах

# IOS муруйг зурах (улаан өнгөөр, тасархай шугамаар)
plt.plot(ios_x, ios_y, color='red', linestyle='--', label='IOS (Хөрөнгө Оруулалтын Боломж)', linewidth=2)

# Хугарлын цэгүүд дээр босоо шугам татах
plt.axvline(x=500, color='gray', linestyle=':')
plt.axvline(x=800, color='gray', linestyle=':')

# Графикийн гарчиг, тэнхлэгийн нэрс
plt.title('WMCC ба IOS-ийн Харьцуулсан График (5.d)', fontsize=16)
plt.xlabel('Шинээр Татан Төвлөрүүлэх Нийт Капиталын Хэмжээ (мянган $)', fontsize=14)
plt.ylabel('Өртөг / Өгөөж (%)', fontsize=14)

# Тэнхлэгийн утгыг тохируулах
plt.xticks(np.arange(0, 1300, 100))
plt.yticks(np.arange(10, 21, 1))
plt.xlim(0, 1200)
plt.ylim(10, 20)

# Тайлбар, торуулга нэмэх
plt.text(250, 13.5, 'WACC1 = 13.8%', color='blue', fontsize=10, fontweight='bold', ha='center')
plt.text(650, 15.1, 'WACC2 = 15.4%', color='blue', fontsize=10, fontweight='bold', ha='center')
plt.text(1000, 15.9, 'WACC3 = 16.2%', color='blue', fontsize=10, fontweight='bold', ha='center')

plt.text(150, 19.2, 'Төсөл А', color='red', fontsize=10)
plt.text(450, 16.2, 'Төсөл Б', color='red', fontsize=10)
plt.text(750, 14.7, 'Төсөл В', color='red', fontsize=10)

plt.legend(loc='upper right', fontsize=12)
plt.grid(True, linestyle='-', alpha=0.5)

plt.tight_layout()
plt.show()