import os

import matplotlib.pyplot as plot

from . import _dataset
from .nomogram import *


def grid(start, end, steps):
	step = (end - start) / steps
	i = start
	while i <= end:
		yield i
		i += step


if __name__ == "__main__":
	C = []
	X = list(grid(0, 40, 1000))
	Y = list(grid(0, 50000, 1000))
	for y_scaled in range(0, 1000):
		y = y_scaled * 50
		row = []
		for x_scaled in range(0, 1000):
			x = x_scaled / 25
			try:
				swi = calculate_swi(x, y)
			except (IndexError, RuntimeError):
				swi = -10
			row.append(swi)
		C.append(row)
	plot.pcolormesh(X, Y, C, cmap='viridis', vmin=-10, vmax=10, rasterized=True)
	for _, data in _dataset.INDICES:
		plot.plot([x[0] for x in data], [x[1] for x in data])
	plot.show()