import math
import numpy as np
import matplotlib.pyplot as plt
import csv

plt.style.use('_mpl-gallery-nogrid')

onlyFinal = bool(input("Would you like to look at all the data? (alternative is only final data) (y/n)")=="n")

if(not(onlyFinal)):
	toPng = bool(input("Would you like to simply convert to png? (y/n) ")=="y")
	
	stop = int(input("What is the final numerical index for the surface water files (usually 99 but sometimes 98)"))

	i=1
	results = []
	with open("currentWater1.csv") as csvfile:
		reader = csv.reader(csvfile, quoting=csv.QUOTE_NONNUMERIC) #change contents to floats
		for row in reader: # each row is a list
			results.append(row)
	X, Y = np.meshgrid(np.linspace(0, len(results[0])-1, len(results[0])), np.linspace(0, len(results)-1, len(results)))
	while(i<=stop):
		#import data from corresponding file
		if(i!=1): #by reusing array it saves time that would be used for allocating memory
			name = "currentWater%d.csv" % (i)
			with open(name) as csvfile:
				reader = csv.reader(csvfile, quoting=csv.QUOTE_NONNUMERIC) #change contents to floats
				a=0
				for row in reader: # each row is a list
					results[a]=row
					a+=1
	
		inspecting=True
		#preparing model
		Z = results
		min = np.amin(Z)
		max = np.amax(Z)
		maxI = -1
		maxK = -1
		f=0
		while(f<len(results)):
			k=0
			while(k<len(results[0])):
				if(results[f][k]==max):
					maxI = f
					maxK = k
				k+=1
			f+=1
		plt.close()
		fig, ax = plt.subplots()
		ax.pcolormesh(X, Y, Z, vmin=min, vmax=max)
		while(inspecting):
			if(not(toPng)):
				print("Scale: ",min," - ",max)
				print("Max at i = ",maxI,", k = ",maxK)
				plt.show()
			else:
				name = "water%d%s%f%s%f.png" % (i,"_",np.amin(Z),"-",np.amax(Z))
				plt.savefig(name)
				plt.close()
			pointFinding = False
			if(not(toPng)):
				if(input("Do you want a specific value? (y/n) ")=="y"):
					pointFinding = True
				while(pointFinding):
					k = int(input("At what X?"))
					f = int(input("At what Y?"))
					print("Value at x = ",k," y = ",f,": ", results[f][k])
					if(input("Would you like to find another point? (y/n)")=="n"):
						pointFinding = False
				if(input("Would you like to change the scale? (y/n) ")=="y"):
					min = float(input("What should the minimum be?"))
					max = float(input("What should the maximum be?"))
				elif(input("Would you like to move on to the next? (y/n) ")=="y"):
					inspecting=False
					i+=1
			else:
				fig, ax = plt.subplots()
				ax.pcolormesh(X, Y, Z, vmin=0, vmax=10)
				name = "water%d" % i
				name = "%s_%s" % (name,"0-10.png")
				plt.savefig(name)
				plt.close()
				inspecting = False
				print("I: ",i," of ",stop," complete")
				i+=1
				
	
#import the final data			
results = []
with open("FinalWater.csv") as csvfile:
	reader = csv.reader(csvfile, quoting=csv.QUOTE_NONNUMERIC) #change contents to floats
	for row in reader: # each row is a list
		results.append(row)
Z = results
X, Y = np.meshgrid(np.linspace(0, len(results[0])-1, len(results[0])), np.linspace(0, len(results)-1, len(results)))
min = np.amin(Z)
max = np.amax(Z)
maxI = -1
maxK = -1
i=0
while(i<len(results)):
	k=0
	while(k<len(results[0])):
		if(results[i][k]==max):
			maxI = i
			maxK = k
		k+=1
	i+=1
inspecting=True
while(inspecting):
	print("Scale: ",min," - ",max)
	print("Scale: ",min," - ",max)
	print("Max at i = ",maxI,", k = ",maxK)
	fig, ax = plt.subplots()
	ax.pcolormesh(X, Y, Z, vmin=min, vmax=max)
	plt.show()
	pointFinding = False
	if(input("Do you want a specific value? (y/n) ")=="y"):
		pointFinding = True
	while(pointFinding):
			k = int(input("At what X?"))
			i = int(input("At what Y?"))
			print("Value at x = ",k," y = ",i,": ", results[i][k])
			if(input("Would you like to find another point? (y/n)")=="n"):
				pointFinding = False
		
	if(input("Would you like to change the scale? (y/n) ")=="y"):
		min = float(input("What should the minimum be?"))
		max = float(input("What should the maximum be?"))
	elif(input("Would you like to end the program? (y/n) ")=="y"):
		inspecting=False