#import streamlit as st
#import pandas as pd
import numpy as np
#import string
import matplotlib.pyplot as plt
#import time
import csv
import math

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

#manually importing Class definitions
class Particle(object): 
    # change __init__ structure to 3d 
    def __init__(self,xPos,yPos,zPos,temp): 
        self.x = xPos
        self.y = yPos
        self.z = zPos
        self.t = temp
        
        self.Sx = 0
        self.Sy = 0
        self.Sz = 0
    
    def changeTemp(self,newTemp):
        self.t = newTemp
    
    def changeX(self,newX):
        self.x = newX/changeFactor
    
    def changeY(self,newY):
        self.y = newY
    
    def changeZ(self,newZ):
        self.z = newZ
        
    def changeVX(self,newXVel):
        self.Sx = newXVel
    
    def changeVY(self,newYVel):
        self.Sy = newYVel
        
    def changeVZ(self,newZVel):
        self.Sz = newZVel
        
class Plate(Particle):
    def __init__(self,xPos,yPos,zPos): 
        self.x = xPos
        self.y = yPos
        self.z = zPos
        self.t = 60
        
        self.Sx = 0
        self.Sy = 0
        self.Sz = 0

    #operates under the assumption of an unmoving plate of constant temperature
    def changeTemp(self,newTemp):
        self.t = 60
        
    def changeVX(self,newXVel):
        self.Sx = 0
    
    def changeVY(self,newYVel):
        self.Sy = 0
        
    def changeVZ(self,newZVel):
        self.Sz = 0

basepS = int(input("What is base side length?"))
frameNum = int(input("How many files (not including F)?"))
h = int(input("How tall is the model (including base)?"))
time = float(input("How many seconds per file?"))

tempArray = np.array([[None]*h]*frameNum) # will store average temp per layer for all files
t=0
num=0
while(num<frameNum):
    #open data and read
    results1=[]
    print("dataX%d.csv"%(num))
    with open("dataX%d.csv"%(num)) as csvfile:
        reader = csv.reader(csvfile, quoting=csv.QUOTE_NONNUMERIC) #change contents to floats
        for row in reader: # each row is a list
            results1.append(row)
    X=np.array(results1)
    results2=[]
    with open("dataY%d.csv"%(num)) as csvfile:
        reader = csv.reader(csvfile, quoting=csv.QUOTE_NONNUMERIC) #change contents to floats
        for row in reader: # each row is a list
            results2.append(row)
    Y=np.array(results2)
    results3=[]
    with open("dataZ%d.csv"%(num)) as csvfile:
        reader = csv.reader(csvfile, quoting=csv.QUOTE_NONNUMERIC) #change contents to floats
        for row in reader: # each row is a list
            results3.append(row)
    Z=np.array(results3)
    results4=[]
    with open("dataT%d.csv"%(num)) as csvfile:
        reader = csv.reader(csvfile, quoting=csv.QUOTE_NONNUMERIC) #change contents to floats
        for row in reader: # each row is a list
            results4.append(row)
    T=np.array(results4)
    #print(results1.shape)
            
    #Temp Verification
    #get average temperatures per layer
    l=0
    j=0
    
    while(l<h and j<len(X)):
        tempTot = 0 
        n=0
        while(j<len(X) and Z[j]<l*math.sqrt(2/3)+.01 and Z[j]>l*math.sqrt(2/3)-.01):
            tempTot+=T[j]
            n+=1
            j+=1
        tempArray[num][l]=tempTot/n
        l+=1
    num+=1
    t+=time
print(tempArray)
i=0
while(i<len(tempArray)):
    #print("T: ",i*time)
    #print("Array: ")
    j=0
    string = "y_{%d}=["%(i+1)
    while(j<len(tempArray[i])-1):
        string += "%f, "%(float(tempArray[i][j]))
        j+=1
    string += "%f"%(float(tempArray[i][j]))
    string+="]"
    print(string)
    i+=1

'''
#reconstruct Particle List (might be unnecessary since I literally just use the data as read later)
particleListR = np.array([[None]*len(results1[0])]*len(results1))
baseNum = 3*basepS**2-3*basepS+1
i = 0
while(i<len(results1)):
    j=0
    while(j<len(results1[0])):
        if(j<baseNum):
            particleListR[i][j]=Plate(results1[i][j],results2[i][j],results3[i][j])
        else:
            particleListR[i][j]=Particle(results1[i][j],results2[i][j],results3[i][j],results4[i][j])
        j+=1
    i+=1
#timeInd = st.slider("When in the sim would you like to look", 0,len(results1), 0)

#plot given time
def plot(timeInd,maxTemp,minTemp):
    fig, ax = plt.subplots(subplot_kw={"projection": "3d"})
    
    x = np.array([None]*len(particleListHistR[0]))
    y = np.array([None]*len(particleListHistR[0]))
    z = np.array([None]*len(particleListHistR[0]))
    temps = np.array([None]*len(particleListHistR[0]))
    i=0
    while(i<len(particleListHistR[timeInd])):
        #print(i)
        x[i] = particleListHistR[timeInd][i].x
        y[i] = particleListHistR[timeInd][i].y
        z[i] = particleListHistR[timeInd][i].z
        temps[i] = particleListHistR[timeInd][i].t
        i+=1
    if(minTemp==0 and maxTemp==0):
        print("Scale Max: ",np.amax(temps)," Min: ", np.amin(temps))
        ax.scatter(x, y, z, c=temps, vmin=np.amin(temps), vmax = np.amax(temps))
    else:
        print("Scale Max: ",maxTemp," Min: ", minTemp)
        ax.scatter(x, y, z, c=temps, vmin=minTemp, vmax = maxTemp)
    
    xStart = np.amax(x)
    xEnd = np.amin(x)
    yStart = np.amax(y)
    yEnd = np.amin(y)
    
    print(xStart," - ", xEnd)
    print(yStart," - ", yEnd)

    #ax.set(xlim=(0, iLength+1), xticks=np.arange(0, iLength+1),
    #       ylim=(0, kLength+1), yticks=np.arange(0, kLength+1))
    #for haxagonal
    ax.set(xlim=(xStart, xEnd), xticks=np.arange(math.floor(xStart), math.ceil(xEnd)),
           ylim=(yStart, yEnd), yticks=np.arange(math.floor(yStart), math.ceil(yEnd)))
    
    plt.show()
'''
'''
xStart = (-1-1/2*(npS-1))/changeFactor
xEnd = (npS+1/2*(npS-1))/changeFactor
yStart = -1/changeFactor
yEnd = 2*npS/changeFactor

#print(xStart," - ", xEnd)
#print(yStart," - ", yEnd)

#ax.set(xlim=(0, iLength+1), xticks=np.arange(0, iLength+1),
#       ylim=(0, kLength+1), yticks=np.arange(0, kLength+1))
#for haxagonal
ax.set(xlim=(xStart, xEnd), xticks=np.arange(xStart, xEnd),
       ylim=(yStart, yEnd), yticks=np.arange(yStart, yEnd))
'''
#st.pyplot(fig)




'''
#UI
Running = True
while (Running):
    print("There are ",len(results1)," frames")
    ind = int(input("Which would you like to look at?"))
    plot(ind,0,0)
    #if(bool(input("Would you like to change the temp range? (y/n)")=="y")):
        #ma = int(input("What should the max be?"))
        #mi = int(input("what should the min be?"))
        #plot(ind,ma,mi)
    if(bool(input("Would you like to keep investigating? (y/n)")=="n")):
        Running = False
'''