import matplotlib.pyplot as plt
import numpy as np
import imageio
import math as mt

images = []
forma = int(input("Ingrese el numero de forma a procesar : "))
n = int(input("Ingrese cantidad de filas: "))
m = int(input("Ingrese cantidad de columnas: "))
a = np.random.random((n, m))
if(forma == 1):
#forma juan
    for j in range(n):
        for k in range(m):
            if j==0: #lado D
                a[j][k] = 100
            elif k==0: #lado A
                a[j][k] = 100*(j/n)
            elif j == n-1: #Lado B
                a[j][k] = 0
            elif k == m-1: #lado C
                a[j][k]=100*(j/n)
            else: # lo que no esta en el borde
                a[j][k]=0
elif(forma == 2):
    for j in range(n):
        for k in range(m):
            if j==0: #lado D
                a[j][k] = 100
            elif k==0: #lado A
                a[j][k] = 100*(j/n)
            elif j == n-1: #Lado B
                a[j][k] = 0
            elif k == m-1: #lado C
                a[j][k]=100*(j/n)
            else: # lo que no esta en el borde
                a[j][k]=0
elif(forma == 3):
    for j in range(n):
        for k in range(m):
            if j==0: #lado D
                a[j][k] = 100*(k/n)**2
            elif k==0: #lado A
                a[j][k] = 0
            elif j == n-1: #Lado B
                a[j][k] = 100*(k/n)**2
            elif k == m-1: #lado C
                a[j][k]=100
            else: # lo que no esta en el borde
                a[j][k]=0
elif(forma == 4):
    for j in range(n):
        for k in range(m):
            if j==0: #lado D
                a[j][k] = 100* mt.sin(mt.pi*(k/n))
            elif k==0: #lado A
                a[j][k] = 0
            elif j == n-1: #Lado B
                a[j][k] = 100* mt.sin(mt.pi*(k/n))
            elif k == m-1: #lado C
                a[j][k]=0
            else: # lo que no esta en el borde
                a[j][k]=0
elif(forma == 5):
    for j in range(n):
        for k in range(m):
            if j==0: #lado D
                a[j][k] = 100
            elif k==0: #lado A
                a[j][k] = 100* j
            elif j == n-1: #Lado B
                a[j][k] = 100* k
            elif k == m-1: #lado C
                a[j][k]=100
            else: # lo que no esta en el borde
                a[j][k]=0
else:
    print ( "Forma no valida")
for i in range(20):
    for j in range(n):
        for k in range(m):
            if j!=0 and k!=0 and j!=n-1 and k!=n-1:
                ecuacion = (a[j][k] + a[j+1][k] + a[j][k+1] + a[j+1][k+1] )/4
                a[j][k]= ecuacion
    plt.imshow(a, cmap=plt.cm.coolwarm)
    plt.savefig("test.png")
    img = imageio.imread("test.png")
    images.append(img)
imageio.mimsave('movie.gif', images)